package controllers

import (
	"net/http"
	"os"

	"forge.redroom.link/yves/sqldb"
	beego "github.com/beego/beego/v2/server/web"
	"github.com/rs/zerolog/log"
)

// Operations about schema
type SchemaController struct {
	beego.Controller
}

// @Title GetTablesList
// @Description Get database tables list
// @Success 200 {string} success !
// @Failure 500 query error
// @router / [get]
func (s *SchemaController) GetTablesList() {
	db := sqldb.Open(os.Getenv("driverdb"), os.Getenv("paramsdb"))
	data, err := db.ListTables()
	if err != nil {
		log.Error().Msg(err.Error())
		s.Data["json"] = map[string]string{"error": err.Error()}
		s.Ctx.Output.SetStatus(http.StatusInternalServerError)
	}
	s.Data["json"] = data
	s.ServeJSON()

	db.Close()
}

// @Title GetSchema
// @Description Get table schema
// @Param	table			path 	string	true		"Name of the table"
// @Success 200 success !
// @Failure 204 no table
// @router /:table [get]
func (s *SchemaController) GetSchema() {
	table := s.GetString(":table")
	db := sqldb.Open(os.Getenv("driverdb"), os.Getenv("paramsdb"))
	data, err := db.Table(table).GetSchema()
	if err != nil {
		log.Error().Msg(err.Error())
		s.Data["json"] = map[string]string{"error": err.Error()}
		s.Ctx.Output.SetStatus(http.StatusNoContent)
	}
	s.Data["json"] = data
	s.ServeJSON()

	db.Close()
}