package application

import (
	"context"
	"database/sql"
	"fmt"
	"net/http"

	"github.com/gin-gonic/gin"
)

type Ctrl interface {
	RegisterRoutes(engine *gin.Engine)
}

type Application struct {
	srv  *http.Server
	pool *sql.DB
}

func (app *Application) Start() error {
	return app.srv.ListenAndServe()
}

func (app *Application) Stop() {
	err := app.pool.Close()
	if err != nil {
		fmt.Println("关闭数据库链接遭遇错误")
		fmt.Println(err)
	}
	err = app.srv.Shutdown(context.Background())
	if err != nil {
		fmt.Println("优雅关闭服务遭遇错误")
		fmt.Println(err)
	}
}

func NewApplication(cs []Ctrl, p *sql.DB) *Application {
	route := gin.Default()
	for _, ctrl := range cs {
		ctrl.RegisterRoutes(route)
	}
	srv := &http.Server{
		Addr:    ":8443",
		Handler: route,
	}
	return &Application{srv, p}
}