package SyncDao

import (
	"dsAutoCode/Utils/CommonUtil"
	"dsAutoCode/Utils/DbUtil"
	"dsAutoCode/Utils/FileUtil"
	"dsAutoCode/models"
	"encoding/json"
	"fmt"
	"github.com/holdno/snowFlakeByGo"
	"go/ast"
	"go/parser"
	"go/token"
	"io/ioutil"
	"log"
	"os"
	"path/filepath"
	"reflect"
	"strings"
	"unicode"
)

var db = DbUtil.Engine

/**
功能：清空表格
作者：黄海
时间：2020-04-21
*/
func ProtoTableClear() {
	//禁用外键约束
	sql := "SET foreign_key_checks=0"
	db.Exec(sql)
	//1、清空
	sql = "truncate table t_proto_files"
	db.Exec(sql)
	sql = "truncate table t_proto_interface"
	db.Exec(sql)
	sql = "truncate table t_proto_struct"
	db.Exec(sql)
	sql = "truncate table t_proto_struct_parameter"
	db.Exec(sql)
	sql = "truncate table t_swagger_controller"
	db.Exec(sql)
	sql = "truncate table t_swagger_controller_parameter"
	db.Exec(sql)
	//禁用外键约束
	sql = "SET foreign_key_checks=1"
	db.Exec(sql)
}

/**
功能：将proto转化为数据库中
作者：黄海
时间：2020-04-22
*/
func ProtoTableToDb() {
	var IdWorker *snowFlakeByGo.Worker
	IdWorker, _ = snowFlakeByGo.NewWorker(0)
	//声明事务
	session := db.NewSession()
	defer session.Close()
	//启动事务
	err := session.Begin()
	if err != nil {
		log.Fatalln(err)
	}
	//遍历所有proto文件
	pathname := CommonUtil.GetCurrentParentPath()+"/dsBaseWeb/Business"
	files, _ := FileUtil.WalkDir(pathname, ".pb.go")
	for m := 0; m < len(files); m++ {
		//临时变量
		_map := make(map[string]int64, 1)

		//1、读取go源文件
		b, _ := ioutil.ReadFile(files[m])
		var protoFile models.TProtoFiles
		protoFile.FileId = CommonUtil.ConvertInt64ToString(IdWorker.GetId())
		protoFile.FilePath, protoFile.FileName = filepath.Split(files[m])
		//通过事务插入数据
		_, err = session.Insert(protoFile)
		if err != nil {
			session.Rollback()
			log.Fatalln(err)
		}
		//2、解析.go文件， Go!
		src := string(b)
		fset := token.NewFileSet()
		f, err := parser.ParseFile(fset, "", src, 0)

		if err != nil {
			panic(err)
		}
		//解析.go文件
		for i := 0; i < len(f.Decls); i++ {
			if reflect.TypeOf(f.Decls[i]).String() == "*ast.GenDecl" {
				typeDecl := f.Decls[i].(*ast.GenDecl)
				for j := 0; j < len(typeDecl.Specs); j++ {
					if reflect.TypeOf(typeDecl.Specs[j]).String() == "*ast.TypeSpec" {
						structName := typeDecl.Specs[j].(*ast.TypeSpec).Name
						//未实现的为代码自动生成，不参与运算
						if strings.Index(structName.Name, "Unimplemented") == 0 {
							continue
						}
						// 首字母小写的不参加运算.小写的不是我们生成的结构体，我们就是这么要求的~
						if !unicode.IsUpper([]rune(structName.Name)[0]) {
							continue
						}
						//如果是struct,表示是输入输出参数列表
						if reflect.TypeOf(typeDecl.Specs[j].(*ast.TypeSpec).Type).String() == "*ast.StructType" {
							var protoStruct models.TProtoStruct
							protoStruct.StructId = CommonUtil.ConvertInt64ToString(IdWorker.GetId())
							protoStruct.FileId = protoFile.FileId
							protoStruct.StructName = structName.Name
							//通过事务插入数据
							_, err = session.Insert(protoStruct)
							if err != nil {
								session.Rollback()
								log.Fatalln(err)
							}
							_map[structName.Name] = CommonUtil.ConvertStringToInt64(protoStruct.StructId)
							//属性
							if reflect.TypeOf(typeDecl.Specs[j].(*ast.TypeSpec).Type).String() == "*ast.StructType" {
								structDecl := typeDecl.Specs[j].(*ast.TypeSpec).Type.(*ast.StructType)
								fields := structDecl.Fields.List
								for _, v := range fields {
									//支持字符串数组
									if reflect.TypeOf(v.Type).String() == "*ast.ArrayType" {
										fieldName := v.Names[0].Name
										typeExpr := "string"
										var protoStructParameter models.TProtoStructParameter
										protoStructParameter.StructId = protoStruct.StructId
										protoStructParameter.StructParameterId = CommonUtil.ConvertInt64ToString(IdWorker.GetId())
										protoStructParameter.StructParameterName = fieldName
										protoStructParameter.StructParameterType = typeExpr
										protoStructParameter.IsRepeate = 1
										//通过事务插入数据
										_, err = session.Insert(protoStructParameter)
										if err != nil {
											session.Rollback()
											log.Fatalln(err)
										}
									}
									//普通简单类型
									if reflect.TypeOf(v.Type).String() == "*ast.Ident" {
										fieldName := v.Names[0].Name
										typeExpr := v.Type
										var protoStructParameter models.TProtoStructParameter
										protoStructParameter.StructId = protoStruct.StructId
										protoStructParameter.StructParameterId = CommonUtil.ConvertInt64ToString(IdWorker.GetId())
										protoStructParameter.StructParameterName = fieldName
										protoStructParameter.StructParameterType = fmt.Sprintf("%s", typeExpr)
										protoStructParameter.IsRepeate = 0
										//通过事务插入数据
										_, err = session.Insert(protoStructParameter)
										if err != nil {
											session.Rollback()
											log.Fatalln(err)
										}
									}
								}
							}
						}

						//如果是inerface，表示是接口列表
						if reflect.TypeOf(typeDecl.Specs[j].(*ast.TypeSpec).Type).String() == "*ast.InterfaceType" {
							//这个是自动生成的，需要处理
							if strings.HasSuffix(structName.Name, "Server") {
								//解析下面定义的接口有哪些？
								service := typeDecl.Specs[j].(*ast.TypeSpec).Type.(*ast.InterfaceType)
								for k := 0; k < len(service.Methods.List); k++ {
									//这个方法的入参是什么？回参是统一的
									serviceName := service.Methods.List[k].Names[0].Name
									argName := service.Methods.List[k].Type.(*ast.FuncType).Params.List[1].Type.(*ast.StarExpr).X.(*ast.Ident).Name
									//美丽的对齐字符串
									var protoInterface models.TProtoInterface
									//根据参数名需要知道argName--->struct_id是多少？
									protoInterface.StructId = CommonUtil.ConvertInt64ToString(_map[argName])
									protoInterface.FileId = protoFile.FileId
									protoInterface.InterfaceId = CommonUtil.ConvertInt64ToString(IdWorker.GetId())
									protoInterface.InterfaceName = serviceName
									//通过事务插入数据
									_, err = session.Insert(protoInterface)
									if err != nil {
										session.Rollback()
										log.Fatalln(err)
									}
								}
							}
						}
					}
				}
			}
		}
	}
	//事务
	err = session.Commit()
	if err != nil {
		log.Fatalln(err)
	}
}

/**
功能：清空表格
作者：黄海
时间：2020-04-21
*/
func SwaggerTableClear() {
	//禁用外键约束
	sql := "SET foreign_key_checks=0"
	db.Exec(sql)
	//1、清空
	sql = "truncate table t_swagger_controller"
	db.Exec(sql)
	sql = "truncate table t_swagger_controller_parameter"
	db.Exec(sql)
	//禁用外键约束
	sql = "SET foreign_key_checks=1"
	db.Exec(sql)
}



/**
功能：读取swagger的生成json
作者：黄海
时间：2020-03-16
*/
func readSwaggerJson() interface{} {
	//读取doc目录下的swagger.json
	parentDir:=CommonUtil.GetParentPath(CommonUtil.GetCurrentDirectory())
	swaggerFile := parentDir+"/dsBaseWeb/docs/swagger.json"
	f, err := os.Open(swaggerFile)
	if err != nil {
		log.Fatal("读取swagger.json失败！")
		return nil
	}
	content, _ := ioutil.ReadAll(f)
	var jsonBody interface{}
	err = json.Unmarshal(content, &jsonBody)
	if err != nil {
		fmt.Println("ERROR: ", err.Error())
		return nil
	}
	return jsonBody
}

/**
功能：将swagger配置文件中的json数据整理入库
作者：黄海
时间：2020-04-22
*/
func SwaggerTableToDb() {
	//读取json文件
	jsonBody := readSwaggerJson()
	var IdWorker *snowFlakeByGo.Worker
	IdWorker, _ = snowFlakeByGo.NewWorker(0)

	m := jsonBody.(map[string]interface{})
	n := m["paths"].(map[string]interface{})
	for k, v := range n {
		t := v.(map[string]interface{})
		httpType := "post"
		//非空检查器
		if t["get"] != nil {
			httpType = "get"
		}
		r1 := t[httpType].(map[string]interface{})
		//描述
		description := r1["description"].(string)

		var sc models.TSwaggerController
		sc.ControllerId = CommonUtil.ConvertInt64ToString(IdWorker.GetId())
		t1 := strings.Split(k, "/")
		sc.ControllerName = t1[len(t1)-1]

		if r1["x-interfacename"] != nil {
			var a = r1["x-interfacename"].([]interface{})
			sc.InterfaceName = a[0].(string)
		} else {
			sc.InterfaceName = t1[len(t1)-1]
		}
		sc.Description = description
		sc.HttpType = httpType
		_, err := db.Insert(sc)
		if err != nil {
			log.Fatalln(err)
		}
		//参数
		if r1["parameters"]!=nil{
			parameters := r1["parameters"].([]interface{})
			//遍历这些参数
			for i := 0; i < len(parameters); i++ {
				parameter := parameters[i].(map[string]interface{})
				var swaggerParameter models.TSwaggerControllerParameter
				swaggerParameter.ControllerId = sc.ControllerId
				swaggerParameter.ControllerParameterGetType = parameter["in"].(string)
				swaggerParameter.ControllerParameterId = CommonUtil.ConvertInt64ToString(IdWorker.GetId())
				swaggerParameter.ControllerParameterName = parameter["name"].(string)
				var isRequird bool
				if _, ok := parameter["required"]; ok {
					isRequird = parameter["required"].(bool)
				} else {
					isRequird = false
				}
				if isRequird {
					swaggerParameter.IsRequired = 1
				} else {
					swaggerParameter.IsRequired = 0
				}
				//描述
				swaggerParameter.ControllerParameterDescription = parameter["description"].(string)
				swaggerParameter.ControllerParameterType = parameter["type"].(string)
				_, err := db.Insert(swaggerParameter)
				if err != nil {
					log.Fatalln(err)
				}
			}
		}
	}
}