package routers

import (
	"basic.com/valib/logger.git"
	"github.com/gin-gonic/gin"
	"net/http"
	"sync"
	"vamicro/update-service/controller"
)

type (
	router struct {
		methods   []string
		path      string
		handles   gin.HandlersChain
	}
)

var (
	routines []router
)

func init() {
	ver := router{
		methods: []string{http.MethodGet},
		path:    "/version",
		handles: []gin.HandlerFunc{controller.GetAppVersion},
	}

	upload := router{
		methods: []string{http.MethodPost},
		path:    "/upload",
		handles: []gin.HandlerFunc{controller.UploadApps},
	}

	//files := router{
	//	methods: []string{http.MethodPost},
	//	path:    "/files",
	//	handles: []gin.HandlerFunc{controller.UploadFiles},
	//}

	download := router{
		methods: []string{http.MethodGet},
		path:    "/download",
		handles: []gin.HandlerFunc{controller.Download},
	}

	routines = append(routines,
		ver,
		upload,
		//files,
		download,
	)
}

func InitRouter(wg *sync.WaitGroup) *http.Server {
	r := gin.Default()

	for _, v := range routines {
		if len(v.methods) > 0 && len(v.handles) > 0 {
			for _, m := range v.methods {
				r.Handle(m, v.path, v.handles...)
			}
		}
	}

	server := &http.Server{
		Addr:              ":8080",
		Handler:           r,
	}

	wg.Add(1)
	go func() {
		if err := server.ListenAndServe(); nil != err {
			if err != http.ErrServerClosed {
				logger.Fatal("InitRouter failed with error:", err)
			} else {
				logger.Info("InitRouter exited by user")
			}
		}

		wg.Done()
	}()

	return server
}