package controllers

import (
	"github.com/crawlab-team/crawlab/core/ds"
	"github.com/crawlab-team/crawlab/core/errors"
	"github.com/crawlab-team/crawlab/core/models/models/v2"
	"github.com/crawlab-team/crawlab/core/models/service"
	"github.com/gin-gonic/gin"
	"go.mongodb.org/mongo-driver/bson/primitive"
)

func PostDatabase(c *gin.Context) {
	// data source
	var payload struct {
		Name        string   `json:"name"`
		Type        string   `json:"type"`
		Description string   `json:"description"`
		Host        string   `json:"host"`
		Port        int      `json:"port"`
		Url         string   `json:"url"`
		Hosts       []string `json:"hosts"`
		Database    string   `json:"database"`
		Username    string   `json:"username"`
		Password    string   `json:"-,omitempty"`
		ConnectType string   `json:"connect_type"`
		Status      string   `json:"status"`
		Error       string   `json:"error"`
	}
	if err := c.ShouldBindJSON(&payload); err != nil {
		HandleErrorBadRequest(c, err)
		return
	}

	u := GetUserFromContextV2(c)

	// add data source to db
	dataSource := models.DatabaseV2{
		Name:        payload.Name,
		Type:        payload.Type,
		Description: payload.Description,
		Host:        payload.Host,
		Port:        payload.Port,
		URI:         payload.Url,
		Database:    payload.Database,
		Username:    payload.Username,
		Password:    payload.Password,
		ConnectType: payload.ConnectType,
		Status:      payload.Status,
		Error:       payload.Error,
	}
	dataSource.SetCreated(u.Id)
	dataSource.SetUpdated(u.Id)
	id, err := service.NewModelServiceV2[models.DatabaseV2]().InsertOne(dataSource)
	if err != nil {
		HandleErrorInternalServerError(c, err)
		return
	}
	dataSource.Id = id

	// check data source status
	go func() {
		_ = ds.GetDataSourceServiceV2().CheckStatus(id)
	}()

	HandleSuccessWithData(c, dataSource)
}

func PutDatabaseById(c *gin.Context) {
	id, err := primitive.ObjectIDFromHex(c.Param("id"))
	if err != nil {
		HandleErrorInternalServerError(c, err)
		return
	}

	// data source
	var dataSource models.DatabaseV2
	if err := c.ShouldBindJSON(&dataSource); err != nil {
		HandleErrorBadRequest(c, err)
		return
	}

	err = service.NewModelServiceV2[models.DatabaseV2]().ReplaceById(id, dataSource)
	if err != nil {
		HandleErrorInternalServerError(c, err)
		return
	}

	// check data source status
	go func() {
		_ = ds.GetDataSourceServiceV2().CheckStatus(id)
	}()
}

func PostDatabaseChangePassword(c *gin.Context) {
	id, err := primitive.ObjectIDFromHex(c.Param("id"))
	if err != nil {
		HandleErrorBadRequest(c, err)
		return
	}
	var payload struct {
		Password string `json:"password"`
	}
	if err := c.ShouldBindJSON(&payload); err != nil {
		HandleErrorBadRequest(c, err)
		return
	}
	if payload.Password == "" {
		HandleErrorBadRequest(c, errors.ErrorDataSourceMissingRequiredFields)
		return
	}
	u := GetUserFromContextV2(c)
	if err := ds.GetDataSourceServiceV2().ChangePassword(id, payload.Password, u.Id); err != nil {
		HandleErrorInternalServerError(c, err)
		return
	}
	HandleSuccess(c)
}