package main import ( "context" "database/sql" "flag" "log" "log/slog" "net/http/cookiejar" "os" "time" "gitea.local.lab/Lbenedar/foundry_helper_service/internal/foundry" "gitea.local.lab/Lbenedar/foundry_helper_service/internal/foundry/requests" "gitea.local.lab/Lbenedar/foundry_helper_service/internal/foundry/transport" _ "github.com/mattn/go-sqlite3" ) type service_mode string const ( API_MODE = "api" DISCORD_BOT_MODE = "discord" TG_BOT_MODE = "tg" ) type config struct { mode service_mode port string env string db struct { dsn string maxOpenConns int maxIdleConns int maxIdleTime string } } type application struct { cfg config foundryApp *foundry.FoundryApi slogger *slog.Logger logger *log.Logger } var version = "0.1.0" func openDB(cfg config) (*sql.DB, error) { db, err := sql.Open("sqlite3", cfg.db.dsn) if err != nil { return nil, err } db.SetMaxIdleConns(cfg.db.maxIdleConns) db.SetMaxOpenConns(cfg.db.maxOpenConns) duration, err := time.ParseDuration(cfg.db.maxIdleTime) if err != nil { return nil, err } db.SetConnMaxIdleTime(duration) ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) defer cancel() err = db.PingContext(ctx) if err != nil { return nil, err } return db, nil } // TODO: change default values func (app *application) parseFlags() *requests.FoundryHttpRequest { flag.StringVar(&app.cfg.port, "port", ":9090", "Service port") flag.StringVar(&app.cfg.env, "env", "development", "Enivornment (development|staging|production)") flag.StringVar(&app.cfg.db.dsn, "db-dsn", "file:db/default.db?cache=shared", "PostgreSQL DSN") flag.IntVar(&app.cfg.db.maxOpenConns, "db-max-open-conns", 25, "PostgreSQL max open connections") flag.IntVar(&app.cfg.db.maxIdleConns, "db-max-idle-conns", 25, "PostgreSQL max idle connections") flag.StringVar(&app.cfg.db.maxIdleTime, "db-max-idle-time", "15m", "PostgreSQL max connection idle time") var mode string flag.StringVar(&mode, "service_mode", "api", "Type of service mode(api|discord|tg)") foundryHttpData := requests.FoundryHttpRequest{SessionID: nil} flag.StringVar(&foundryHttpData.Host, "foundry_host", "127.0.0.1", "Address to connect to Foundry") flag.StringVar(&foundryHttpData.Password, "foundry_pass", "", "Password to connect to Foundry") var logLevelStr string flag.StringVar(&logLevelStr, "log_level", "info", "Password to connect to Foundry(debug|info|warn|error)") flag.Parse() app.cfg.mode = service_mode(mode) logLevel, ok := StringToLogLevel[logLevelStr] if !ok { logLevel = slog.LevelInfo } app.slogger = slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{Level: logLevel})) if !ok { app.slogger.Warn("Wrong log level. Started with log_level=info", "received", logLevelStr) } return &foundryHttpData } // TODO: Handle DB connection error func main() { app := application{foundryApp: &foundry.FoundryApi{}} foundryHttpData := app.parseFlags() jar, err := cookiejar.New(nil) if err != nil { app.slogger.Error("Error", "text", err.Error()) return } foundryHttpData.Jar = jar app.slogger.Info("", "dsn", app.cfg.db.dsn) db, err := openDB(app.cfg) if err != nil { app.slogger.Error("Error when opening database connection", "err", err) return } defer db.Close() app.slogger.Info("Database connection pool established") app.foundryApp.SetTransport(transport.NewFoundryTransport(db, app.slogger, foundryHttpData)) go app.foundryApp.StartListenFoundry() app.serve() }