Первая версия

3 files changed, 139 insertions, 0 deletions

diff --git a/cmd/app/importer/importer.go b/cmd/app/importer/importer.go
new file mode 100644
index 0000000..0cc9bb7
--- /dev/null
+++ b/cmd/app/importer/importer.go
@@ -0,0 +1,47 @@
+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(&quotes); 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/app/main.go
new file mode 100644
index 0000000..dcd3eb0
--- /dev/null
+++ b/cmd/app/main.go
@@ -0,0 +1,43 @@
+package main
+
+import (
+	"log"
+	"os"
+
+	"github.com/urfave/cli/v2"
+	"sh.org.ru/cmd/app/importer"
+	"sh.org.ru/cmd/app/serve"
+)
+
+func main() {
+	app := &cli.App{
+		Name: "app",
+		Commands: []*cli.Command{
+			{
+				Name:   "serve",
+				Action: serve.Run,
+				Flags: []cli.Flag{
+					&cli.StringFlag{
+						Name:  "config",
+						Value: "./config/dev.yaml",
+						Usage: "config",
+					},
+				},
+			},
+			{
+				Name:   "import",
+				Action: importer.Run,
+				Flags: []cli.Flag{
+					&cli.StringFlag{
+						Name:  "config",
+						Value: "./config/dev.yaml",
+						Usage: "config",
+					},
+				},
+			},
+		},
+	}
+	if err := app.Run(os.Args); err != nil {
+		log.Fatal(err)
+	}
+}
diff --git a/cmd/app/serve/serve.go b/cmd/app/serve/serve.go
new file mode 100644
index 0000000..dcc296b
--- /dev/null
+++ b/cmd/app/serve/serve.go
@@ -0,0 +1,49 @@
+package serve
+
+import (
+	"github.com/labstack/echo/v4"
+	"github.com/labstack/echo/v4/middleware"
+	"github.com/ssoda/captcha"
+	"github.com/uptrace/bun/extra/bundebug"
+	"github.com/urfave/cli/v2"
+	"sh.org.ru/pkg/config"
+	"sh.org.ru/pkg/db"
+	"sh.org.ru/pkg/handler"
+	"sh.org.ru/static"
+)
+
+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)
+	db.AddQueryHook(bundebug.NewQueryHook(bundebug.WithVerbose(cfg.Debug)))
+
+	h := handler.Handler{DB: db}
+
+	e := echo.New()
+
+	e.Use(middleware.Recover())
+	e.Use(middleware.Logger())
+
+	e.GET("/", h.Index)
+	e.GET("/random", h.Random)
+	e.GET("/quote/:id", h.Quote)
+	e.GET("/add", h.AddQuote)
+	e.POST("/add", h.AddQuotePost)
+	e.GET("/add/success", h.AddQuoteSuccess)
+	e.GET("/captcha/*", echo.WrapHandler(captcha.Server(400, 65)))
+
+	func(g *echo.Group) {
+		g.GET("/", h.Admin)
+		g.POST("/action", h.AdminAction)
+	}(e.Group("/admin", middleware.BasicAuth(func(u, p string, ctx echo.Context) (bool, error) {
+		return cfg.Admins[u] == p, nil
+	})))
+
+	e.StaticFS("/", static.FS)
+
+	return e.Start(cfg.Listen)
+}