package models

import (
	"database/sql"
	"gitee.com/ha666/gen_sqlx_code/utils"
	"strings"
)

type IndexInfo struct {
	NonUnique      int
	IndexName      string
	ColumnName     string
	Comment        string
	IndexComment   string
	ColumnNameCase string
}

func GetIndexInfos(database_name, table_name string) (index_infos []IndexInfo, err error) {
	rows, err := mainDB.Query("SELECT NON_UNIQUE,INDEX_NAME,COLUMN_NAME,`COMMENT`,INDEX_COMMENT FROM INFORMATION_SCHEMA.STATISTICS WHERE table_schema = ? AND table_name = ?;", database_name, table_name)
	defer rows.Close()
	if err != nil {
		return index_infos, err
	}
	return _IndexInfoRowsToArray(rows)
}

func _IndexInfoRowsToArray(rows *sql.Rows) (models []IndexInfo, err error) {
	for rows.Next() {
		model := IndexInfo{}
		err = rows.Scan(&model.NonUnique, &model.IndexName, &model.ColumnName, &model.Comment, &model.IndexComment)
		if err != nil {
			return models, err
		}
		if len(model.Comment) > 0 {
			model.Comment = strings.Replace(model.Comment, "\r", "", -1)
			model.Comment = strings.Replace(model.Comment, "\n", "", -1)
		}
		if len(model.IndexComment) > 0 {
			model.IndexComment = strings.Replace(model.IndexComment, "\r", "", -1)
			model.IndexComment = strings.Replace(model.IndexComment, "\n", "", -1)
		}
		model.ColumnNameCase = model.ColumnName
		utils.ToBigHump(&model.ColumnNameCase)
		models = append(models, model)
	}
	return models, err
}