1
0
Fork 1
mirror of https://github.com/hay-kot/homebox.git synced 2024-12-30 10:36:40 +00:00
homebox/backend/app/api/main.go

149 lines
3.6 KiB
Go
Raw Normal View History

2022-08-30 02:30:36 +00:00
package main
import (
"context"
2022-09-14 05:15:01 +00:00
"os"
"path/filepath"
2022-08-30 02:30:36 +00:00
"time"
atlas "ariga.io/atlas/sql/migrate"
"entgo.io/ent/dialect/sql/schema"
"github.com/hay-kot/homebox/backend/app/api/docs"
"github.com/hay-kot/homebox/backend/ent"
"github.com/hay-kot/homebox/backend/internal/config"
"github.com/hay-kot/homebox/backend/internal/migrations"
"github.com/hay-kot/homebox/backend/internal/repo"
"github.com/hay-kot/homebox/backend/internal/services"
"github.com/hay-kot/homebox/backend/pkgs/server"
2022-08-30 02:30:36 +00:00
_ "github.com/mattn/go-sqlite3"
2022-09-03 18:38:35 +00:00
"github.com/rs/zerolog/log"
2022-08-30 02:30:36 +00:00
)
2022-09-14 04:06:07 +00:00
var (
Version = "0.1.0"
Commit = "HEAD"
BuildTime = "now"
)
2022-08-30 02:30:36 +00:00
// @title Go API Templates
// @version 1.0
// @description This is a simple Rest API Server Template that implements some basic User and Authentication patterns to help you get started and bootstrap your next project!.
// @contact.name Don't
// @license.name MIT
// @BasePath /api
// @securityDefinitions.apikey Bearer
// @in header
// @name Authorization
// @description "Type 'Bearer TOKEN' to correctly set the API Key"
func main() {
2022-09-14 05:15:01 +00:00
path := ""
if len(os.Args) > 1 {
path = os.Args[1]
}
2022-09-03 18:38:35 +00:00
2022-09-14 05:15:01 +00:00
cfg, err := config.NewConfig(path)
2022-08-30 02:30:36 +00:00
if err != nil {
panic(err)
}
docs.SwaggerInfo.Host = cfg.Swagger.Host
if err := run(cfg); err != nil {
panic(err)
}
}
func run(cfg *config.Config) error {
2022-09-04 03:27:02 +00:00
app := new(cfg)
app.setupLogger()
2022-08-30 02:30:36 +00:00
// =========================================================================
// Initialize Database & Repos
err := os.MkdirAll(cfg.Storage.Data, 0755)
if err != nil {
log.Fatal().Err(err).Msg("failed to create data directory")
}
c, err := ent.Open("sqlite3", cfg.Storage.SqliteUrl)
2022-08-30 02:30:36 +00:00
if err != nil {
2022-09-03 18:38:35 +00:00
log.Fatal().
Err(err).
Str("driver", "sqlite").
Str("url", cfg.Storage.SqliteUrl).
2022-09-03 18:38:35 +00:00
Msg("failed opening connection to sqlite")
2022-08-30 02:30:36 +00:00
}
defer func(c *ent.Client) {
err := c.Close()
if err != nil {
log.Fatal().Err(err).Msg("failed to close database connection")
}
2022-08-30 02:30:36 +00:00
}(c)
temp := filepath.Join(os.TempDir(), "migrations")
err = migrations.Write(temp)
if err != nil {
return err
}
dir, err := atlas.NewLocalDir(temp)
if err != nil {
return err
}
options := []schema.MigrateOption{
schema.WithDir(dir),
schema.WithDropColumn(true),
schema.WithDropIndex(true),
}
err = c.Schema.Create(context.Background(), options...)
if err != nil {
2022-09-03 18:38:35 +00:00
log.Fatal().
Err(err).
Str("driver", "sqlite").
Str("url", cfg.Storage.SqliteUrl).
2022-09-03 18:38:35 +00:00
Msg("failed creating schema resources")
2022-08-30 02:30:36 +00:00
}
err = os.RemoveAll(temp)
if err != nil {
log.Fatal().Err(err).Msg("failed to remove temporary directory for database migrations")
return err
}
2022-08-30 02:30:36 +00:00
app.db = c
app.repos = repo.New(c, cfg.Storage.Data)
app.services = services.NewServices(app.repos)
2022-08-30 02:30:36 +00:00
// =========================================================================
// Start Server
app.server = server.NewServer(
server.WithHost(app.conf.Web.Host),
server.WithPort(app.conf.Web.Port),
)
2022-08-30 02:30:36 +00:00
routes := app.newRouter(app.repos)
2022-09-04 03:27:02 +00:00
if app.conf.Mode != config.ModeDevelopment {
app.logRoutes(routes)
}
2022-08-30 02:30:36 +00:00
2022-09-03 18:38:35 +00:00
log.Info().Msgf("Starting HTTP Server on %s:%s", app.server.Host, app.server.Port)
2022-08-30 02:30:36 +00:00
// =========================================================================
// Start Reoccurring Tasks
2022-09-04 03:27:02 +00:00
go app.startBgTask(time.Duration(24)*time.Hour, func() {
2022-09-03 09:24:28 +00:00
_, err := app.repos.AuthTokens.PurgeExpiredTokens(context.Background())
if err != nil {
2022-09-03 18:38:35 +00:00
log.Error().
Err(err).
Msg("failed to purge expired tokens")
2022-09-03 09:24:28 +00:00
}
2022-08-30 02:30:36 +00:00
})
return app.server.Start(routes)
}