package cmd

import (
	"context"
	"fmt"
	"log/slog"
	"net"
	"net/http"

	"github.com/labstack/echo-contrib/session"
	"github.com/labstack/echo/v4"
	echomiddleware "github.com/labstack/echo/v4/middleware"
	_ "github.com/mattn/go-sqlite3"
	"github.com/michaeljs1990/sqlitestore"
	"github.com/spf13/cobra"
	"github.com/spf13/viper"
	"gitrepo.ru/neonxp/gorum/contextlib"
	"gitrepo.ru/neonxp/gorum/db"
	"gitrepo.ru/neonxp/gorum/middleware"
	"gitrepo.ru/neonxp/gorum/repository"
	"gitrepo.ru/neonxp/gorum/routes"
	"gitrepo.ru/neonxp/gorum/utils"
	"gitrepo.ru/neonxp/gorum/views"
	"gitrepo.ru/neonxp/gorum/views/assets"
)

var (
	theme         string
	listen        string
	sessionSecret string
	serverCmd     = &cobra.Command{
		Use:   "serve",
		Short: "Run server",
		Long:  `Run forum server`,
		RunE: func(cmd *cobra.Command, args []string) error {
			return serve(cmd.Context())
		},
	}
)

func init() {
	serverCmd.PersistentFlags().StringVar(&theme, "theme", "default", "theme to use")
	serverCmd.PersistentFlags().StringVar(&listen, "listen", ":8000", "bind address to listen")
	serverCmd.PersistentFlags().StringVar(&sessionSecret, "session_secret", "s3cr3t", "sessions secret")
	viper.BindPFlag("theme", serverCmd.Flags().Lookup("theme"))
	viper.BindPFlag("listen", serverCmd.Flags().Lookup("listen"))
	viper.BindPFlag("session_secret", serverCmd.Flags().Lookup("session_secret"))
}

func serve(ctx context.Context) error {

	slog.Debug(
		"params",
		slog.String("listen", listen),
		slog.String("theme", theme),
		slog.String("session_secret", sessionSecret),
	)
	ctx = context.WithValue(ctx, contextlib.ThemeKey, theme)

	orm, err := db.GetDB(dbFile)
	if err != nil {
		return err
	}

	defer orm.Close()

	userRepo := repository.NewUser(orm)
	nodeRepo := repository.NewNode(orm)

	r := routes.NewRouter(userRepo, nodeRepo)

	e := echo.New()

	e.HideBanner = true

	e.HTTPErrorHandler = func(err error, c echo.Context) {
		_ = utils.Render(c, views.ErrorPage(err))
	}

	sessionStore, err := sqlitestore.NewSqliteStoreFromConnection(orm.DB, "sessions", "", 0, []byte(sessionSecret))
	if err != nil {
		return fmt.Errorf("failed init session store: %w", err)
	}

	e.Use(
		echomiddleware.Recover(),
		echomiddleware.Gzip(),
		// echomiddleware.CSRFWithConfig(echomiddleware.CSRFConfig{
		// 	Skipper:      echomiddleware.DefaultSkipper,
		// 	TokenLength:  32,
		// 	TokenLookup:  "form:" + echo.HeaderXCSRFToken,
		// 	ContextKey:   "csrf",
		// 	CookieName:   "_csrf",
		// 	CookieMaxAge: 86400,
		// }),
		session.Middleware(sessionStore),
		middleware.UserMiddleware(),
	)

	e.GET("/register", r.Register)
	e.POST("/register", r.Register)
	e.GET("/login", r.Login)
	e.POST("/login", r.Login)
	e.POST("/logout", r.Logout)

	e.GET("/", r.Node)
	e.GET("/p/:id", r.Node)
	e.GET("/p/:id/new", r.NewPost)
	e.POST("/p/:id/new", r.NewPost)
	e.GET("/t/:id", r.Node)
	e.GET("/t/:id/new", r.NewTopic)
	e.POST("/t/:id/new", r.NewTopic)

	e.StaticFS("/assets", assets.FS)

	slog.InfoContext(ctx, "started gorum", slog.String("bind", listen))

	server := http.Server{
		Addr:     listen,
		Handler:  e,
		ErrorLog: slog.NewLogLogger(slog.Default().Handler(), slog.LevelError),
		ConnContext: func(cctx context.Context, c net.Conn) context.Context {
			return ctx
		},
	}
	if err := server.ListenAndServe(); err != http.ErrServerClosed {
		return fmt.Errorf("server failed: %w", err)
	}

	return nil
}