diff --git a/Dockerfile b/Dockerfile index 977c118..fda1d32 100644 --- a/Dockerfile +++ b/Dockerfile @@ -6,7 +6,7 @@ RUN apk update --no-cache && apk add --no-cache tzdata COPY go.mod go.sum ./ RUN go mod download && go mod verify COPY . . -RUN go build -o shorg ./cmd/app +RUN go build -o shorg ./cmd/shorg # Runtime container FROM alpine:3.20 @@ -20,4 +20,4 @@ ENV TZ=Europe/Moscow EXPOSE 8000 -CMD ["/app/shorg"] \ No newline at end of file +ENTRYPOINT ["/app/shorg"] \ No newline at end of file diff --git a/cmd/app/importer/importer.go b/cmd/app/importer/importer.go deleted file mode 100644 index 0cc9bb7..0000000 --- a/cmd/app/importer/importer.go +++ /dev/null @@ -1,47 +0,0 @@ -package importer - -import ( - "encoding/json" - "os" - - "github.com/urfave/cli/v2" - "sh.org.ru/pkg/config" - "sh.org.ru/pkg/db" - "sh.org.ru/pkg/model" -) - -func Run(c *cli.Context) error { - configFile := c.String("config") - cfg, err := config.New(configFile) - if err != nil { - return err - } - db := db.New(cfg.DB) - - file := c.Args().First() - - quotes := []string{} - - fp, err := os.Open(file) - if err != nil { - return err - } - defer fp.Close() - - if err := json.NewDecoder(fp).Decode("es); err != nil { - return err - } - - for _, text := range quotes { - q := &model.Quote{ - Quote: text, - Approved: true, - Archive: true, - } - if _, err := db.NewInsert().Model(q).Exec(c.Context); err != nil { - return err - } - } - - return nil -} diff --git a/cmd/app/main.go b/cmd/shorg/main.go similarity index 61% rename from cmd/app/main.go rename to cmd/shorg/main.go index dcd3eb0..df36817 100644 --- a/cmd/app/main.go +++ b/cmd/shorg/main.go @@ -5,8 +5,9 @@ import ( "os" "github.com/urfave/cli/v2" - "sh.org.ru/cmd/app/importer" - "sh.org.ru/cmd/app/serve" + + "sh.org.ru/cmd/shorg/migrator" + "sh.org.ru/cmd/shorg/serve" ) func main() { @@ -24,17 +25,7 @@ func main() { }, }, }, - { - Name: "import", - Action: importer.Run, - Flags: []cli.Flag{ - &cli.StringFlag{ - Name: "config", - Value: "./config/dev.yaml", - Usage: "config", - }, - }, - }, + migrator.Migrator(), }, } if err := app.Run(os.Args); err != nil { diff --git a/cmd/bun/main.go b/cmd/shorg/migrator/migrator.go similarity index 65% rename from cmd/bun/main.go rename to cmd/shorg/migrator/migrator.go index 62c9441..01dd819 100644 --- a/cmd/bun/main.go +++ b/cmd/shorg/migrator/migrator.go @@ -1,32 +1,21 @@ -package main +package migrator import ( + "encoding/json" "fmt" "log" "os" "strings" "github.com/uptrace/bun/migrate" - "sh.org.ru/cmd/bun/migrations" + "github.com/urfave/cli/v2" + "sh.org.ru/migrations" "sh.org.ru/pkg/config" "sh.org.ru/pkg/db" - - "github.com/urfave/cli/v2" + "sh.org.ru/pkg/model" ) -func main() { - app := &cli.App{ - Name: "bun", - Commands: []*cli.Command{ - newDBCommand(migrations.Migrations), - }, - } - if err := app.Run(os.Args); err != nil { - log.Fatal(err) - } -} - -func newDBCommand(migrations *migrate.Migrations) *cli.Command { +func Migrator() *cli.Command { return &cli.Command{ Name: "db", Usage: "manage database migrations", @@ -37,12 +26,12 @@ func newDBCommand(migrations *migrate.Migrations) *cli.Command { Flags: []cli.Flag{ &cli.StringFlag{ Name: "config", - Value: "./config/dev.yaml", + Value: "./config/config.yaml", Usage: "config", }, }, Action: func(c *cli.Context) error { - migrator, err := newMigrator(c, migrations) + migrator, err := newMigrator(c, migrations.Migrations) if err != nil { return err } @@ -56,12 +45,12 @@ func newDBCommand(migrations *migrate.Migrations) *cli.Command { Flags: []cli.Flag{ &cli.StringFlag{ Name: "config", - Value: "./config/dev.yaml", + Value: "./config/config.yaml", Usage: "config", }, }, Action: func(c *cli.Context) error { - migrator, err := newMigrator(c, migrations) + migrator, err := newMigrator(c, migrations.Migrations) if err != nil { return err } @@ -86,12 +75,12 @@ func newDBCommand(migrations *migrate.Migrations) *cli.Command { Flags: []cli.Flag{ &cli.StringFlag{ Name: "config", - Value: "./config/dev.yaml", + Value: "./config/config.yaml", Usage: "config", }, }, Action: func(c *cli.Context) error { - migrator, err := newMigrator(c, migrations) + migrator, err := newMigrator(c, migrations.Migrations) if err != nil { return err } @@ -116,12 +105,12 @@ func newDBCommand(migrations *migrate.Migrations) *cli.Command { Flags: []cli.Flag{ &cli.StringFlag{ Name: "config", - Value: "./config/dev.yaml", + Value: "./config/config.yaml", Usage: "config", }, }, Action: func(c *cli.Context) error { - migrator, err := newMigrator(c, migrations) + migrator, err := newMigrator(c, migrations.Migrations) if err != nil { return err } @@ -135,12 +124,12 @@ func newDBCommand(migrations *migrate.Migrations) *cli.Command { Flags: []cli.Flag{ &cli.StringFlag{ Name: "config", - Value: "./config/dev.yaml", + Value: "./config/config.yaml", Usage: "config", }, }, Action: func(c *cli.Context) error { - migrator, err := newMigrator(c, migrations) + migrator, err := newMigrator(c, migrations.Migrations) if err != nil { return err } @@ -154,12 +143,12 @@ func newDBCommand(migrations *migrate.Migrations) *cli.Command { Flags: []cli.Flag{ &cli.StringFlag{ Name: "config", - Value: "./config/dev.yaml", + Value: "./config/config.yaml", Usage: "config", }, }, Action: func(c *cli.Context) error { - migrator, err := newMigrator(c, migrations) + migrator, err := newMigrator(c, migrations.Migrations) if err != nil { return err } @@ -171,6 +160,49 @@ func newDBCommand(migrations *migrate.Migrations) *cli.Command { } fmt.Printf("created migration %s (%s)\n", mf.Name, mf.Path) + return nil + }, + }, + { + Name: "import", + Usage: "import json with quotes", + Flags: []cli.Flag{ + &cli.StringFlag{ + Name: "config", + Value: "./config/config.yaml", + Usage: "config", + }, + }, + Action: func(c *cli.Context) error { + configFile := c.String("config") + cfg, err := config.New(configFile) + if err != nil { + return err + } + db := db.New(cfg.DB) + file := c.Args().First() + quotes := []string{} + fp, err := os.Open(file) + if err != nil { + return err + } + defer fp.Close() + + if err := json.NewDecoder(fp).Decode("es); err != nil { + return err + } + + for _, text := range quotes { + q := &model.Quote{ + Quote: text, + Approved: true, + Archive: true, + } + if _, err := db.NewInsert().Model(q).Exec(c.Context); err != nil { + return err + } + } + return nil }, }, diff --git a/cmd/app/serve/serve.go b/cmd/shorg/serve/serve.go similarity index 89% rename from cmd/app/serve/serve.go rename to cmd/shorg/serve/serve.go index 0609083..13354b7 100644 --- a/cmd/app/serve/serve.go +++ b/cmd/shorg/serve/serve.go @@ -27,8 +27,11 @@ func Run(c *cli.Context) error { e.HTTPErrorHandler = handler.ErrorHandler - e.Use(middleware.Recover()) - e.Use(middleware.Logger()) + e.Use( + middleware.Recover(), + middleware.Logger(), + middleware.RemoveTrailingSlash(), + ) e.GET("/", h.Index) e.GET("/quote/:id", h.Quote) @@ -41,6 +44,7 @@ func Run(c *cli.Context) error { func(g *echo.Group) { g.GET("/", h.Admin) g.POST("/action", h.AdminAction) + g.GET("/export", h.AdminExport) }(e.Group("/admin", middleware.BasicAuth(func(u, p string, ctx echo.Context) (bool, error) { return cfg.Admins[u] == p, nil }))) diff --git a/cmd/bun/migrations/20241005143542_1_init.go b/migrations/20241005143542_1_init.go similarity index 100% rename from cmd/bun/migrations/20241005143542_1_init.go rename to migrations/20241005143542_1_init.go diff --git a/cmd/bun/migrations/main.go b/migrations/main.go similarity index 100% rename from cmd/bun/migrations/main.go rename to migrations/main.go diff --git a/pkg/handler/admin.go b/pkg/handler/admin.go index 8531011..75fb650 100644 --- a/pkg/handler/admin.go +++ b/pkg/handler/admin.go @@ -55,3 +55,21 @@ func (h *Handler) AdminAction(c echo.Context) error { return c.Redirect(http.StatusFound, "/admin/") } + +func (h *Handler) AdminExport(c echo.Context) error { + quotes := []model.Quote{} + err := h.DB.NewSelect(). + Model((*model.Quote)(nil)). + Order("id ASC"). + Scan(c.Request().Context(), "es) + if err != nil { + return err + } + + quotesString := make([]string, 0, len(quotes)) + for _, q := range quotes { + quotesString = append(quotesString, q.Quote) + } + + return c.JSON(http.StatusOK, quotesString) +} diff --git a/pkg/tpl/add.templ b/pkg/tpl/add.templ index 0611d1b..14b57eb 100644 --- a/pkg/tpl/add.templ +++ b/pkg/tpl/add.templ @@ -14,27 +14,10 @@ templ AddQuotePage(form *AddQuoteForm, err string) {
- } } diff --git a/pkg/tpl/add_templ.go b/pkg/tpl/add_templ.go index 04fb615..d47582b 100644 --- a/pkg/tpl/add_templ.go +++ b/pkg/tpl/add_templ.go @@ -92,33 +92,20 @@ func AddQuotePage(form *AddQuoteForm, err string) templ.Component { if templ_7745c5c3_Err != nil { return templ_7745c5c3_Err } - _, templ_7745c5c3_Err = templ_7745c5c3_Buffer.WriteString("\">") + _, templ_7745c5c3_Err = templ_7745c5c3_Buffer.WriteString("\"> ") if templ_7745c5c3_Err != nil { return templ_7745c5c3_Err } diff --git a/pkg/tpl/quote.templ b/pkg/tpl/quote.templ index 58c0d4f..34429d1 100644 --- a/pkg/tpl/quote.templ +++ b/pkg/tpl/quote.templ @@ -8,13 +8,22 @@ import ( templ Quote(quote *model.Quote) {