package ProtoController

import (
	"bytes"
	"dsAutoCode/Dao/ProtoDao"
	"dsAutoCode/Model"
	"dsAutoCode/Service/ProtoService"
	"dsAutoCode/Utils/CommonUtil"
	"github.com/gin-gonic/gin"
	"log"
	"net/http"
	"strings"
	"text/template"
)

/**
功能：生成Dao层的操作代码
作者：黄海
时间：2020-04-23
*/
func CodeDao(c *gin.Context) {
	var operateType string
	var tableName string
	interfaceId := c.Query("interfaceId")
	//真实的表名
	trueTableName := ProtoService.GetTrueTableNameFromInterfaceId(interfaceId)
	//蛇形命名后的表名
	tableName = ProtoDao.GetTrueTableNameFromInterfaceId(interfaceId)
	tableName = strings.Replace(tableName, ".pb.go", "", -1)
	//这个表的主键-->ModelArg:
	_, _, pkFiled, _ := ProtoService.GetParameterByInterfaceId(interfaceId)
	interfaceName := c.Query("interfaceName")

	//获取接口的描述信息
	controllerName := c.Query("controllerName")
	_, _, description := ProtoService.GetControllerInfoByName(controllerName)

	//作者信息
	authorId := c.Query("author_id")
	authorName := ProtoService.GetPersonInfoById(authorId)["person_name"].(string)
	//操作类型
	operateType = ProtoService.GetOperateType(interfaceName)
	//调用模板,
	filesName :=CommonUtil.GetCurrentPath()+ "/Controller/ProtoController/Template/Dao.template"
	t, err := template.ParseFiles(filesName)
	if err != nil {
		log.Fatalln("parse file err:", err)
		return
	}
	buf := new(bytes.Buffer)
	//返回的结构体前缀
	var DeleteIdsArg = false
	var ModelArg = false
	var QueryArg = false

	if operateType == "Page" {
		QueryArg = true
	} else if operateType == "Delete" {
		DeleteIdsArg = true
	} else {
		ModelArg = true
	}

	var content string
	if !DeleteIdsArg{
		var p = map[string]interface{}{
			"tableName":     tableName,
			"operateType":   operateType,
			"CurrentTime":   CommonUtil.GetCurrentTime(),
			"description":   description,
			"authorName":    authorName,
			"modelName":     CommonUtil.StrFirstToLower(tableName),
			"ModelArg":      ModelArg,
			"QueryArg":      QueryArg,
			"trueTableName": trueTableName,
			"PK":            pkFiled,
			"interfaceName": interfaceName,
		}
		if err := t.Execute(buf, p); err != nil {
			log.Fatal("There was an error:", err.Error())
		}
		content = buf.String()
		//删除空格
		content = CommonUtil.DeleteBlankString(content)
	}else{
		content="删除功能请统一使用CommonDao.DeleteIds"
	}

	//返回结果
	c.JSON(http.StatusOK, Model.Res{
		Data:     content,
		Msg:      tableName,
		CodePath: ProtoService.GetGenereateCodePath("Dao", tableName),
	})
}