package model

import (
	"cc/db"
	"errors"
	"math/rand"
)

// Project 项目
type Project struct {
	ID              int    `db:"id" json:"id"`                             // 项目ID
	Name            string `db:"name" json:"name"`                         // 项目名称
	Code            int    `db:"code" json:"code"`                         // 项目编码
	User            string `db:"user" json:"user"`                         // 项目用户
	Description     string `db:"description" json:"description"`           // 项目描述
	UpdateFrequency int    `db:"update_frequency" json:"update_frequency"` // 更新频率
	LocaltionID     int    `db:"localtion_id" json:"localtion_id"`         // 项目位置ID
	WeatherTypeID   int    `db:"weather_type_id" json:"weather_type_id"`   // 天气类型ID
	Dev             bool   `db:"dev" json:"dev"`                           // 是否为开发模式
}

// GetAll 获取所有项目
func (pro *Project) GetAll() (rv []Project, rer error) {
	sql := "SELECT * FROM projects"
	rer = db.DB.Select(&rv, sql)
	return
}

// GetProjectByID 通过项目ID获取项目
func (pro *Project) GetProjectByID() (rer error) {
	sql := "SELECT * FROM projects WHERE id =?"
	return db.DB.Get(pro, sql, pro.ID)
}

// GetProjectByCode 通过项目代码获取项目
func (pro *Project) GetProjectByCode() (rer error) {
	sql := "SELECT * FROM projects WHERE code =?"
	return db.DB.Get(pro, sql, pro.Code)
}

// CreateProject 创建项目
func (pro *Project) CreateProject() (rer error) {
	// 判断是否存在相同的项目名称和天气类型
	var count int
	err := db.DB.Get(&count, "SELECT COUNT(*) FROM projects WHERE name =? AND weather_type_id =?", pro.Name, pro.WeatherTypeID)
	if err != nil {
		return err
	}
	if count > 0 {
		return errors.New("已存在相同配置")
	}
	// 判断项目代码，当前项目代码不为0时，必须是4位数字
	if pro.Code != 0 {
		if pro.Code < 1000 || pro.Code > 9999 {
			return errors.New("项目代码必须是4位数字")
		}
		err := db.DB.Get(&count, "SELECT COUNT(*) FROM projects WHERE code =?", pro.Code)
		if err != nil {
			return err
		}
		if count > 0 {
			return errors.New("项目代码已存在,请重新输入,或者将项目代码设置为0,系统将随机生成一个4位数字的项目代码")
		}
	} else {
		// 随机生成一个4位数字的项目代码
		for {
			pro.Code = rand.Intn(9000) + 1000
			err := db.DB.Get(&count, "SELECT COUNT(*) FROM projects WHERE code =?", pro.Code)
			if err != nil {
				return err
			}
			if count == 0 {
				break
			}
		}
	}
	// 更新频率为0时，设置为60分钟
	if pro.UpdateFrequency == 0 {
		pro.UpdateFrequency = 60
	}
	// 插入数据
	sql := "INSERT INTO projects (name, code, user, description, update_frequency, localtion_id, weather_type_id, dev) VALUES (?,?,?,?,?,?,?,?)"
	_, rer = db.DB.Exec(sql, pro.Name, pro.Code, pro.User, pro.Description, pro.UpdateFrequency, pro.LocaltionID, pro.WeatherTypeID, pro.Dev)
	return
}

// UpdateProject 更新项目
func (pro *Project) UpdateProject() (rer error) {
	// 更新频率为0时，设置为60分钟
	if pro.UpdateFrequency == 0 {
		pro.UpdateFrequency = 60
	}
	// 更新数据
	sql := "UPDATE projects SET name=?, user=?, description=?, update_frequency=?, localtion_id=?, weather_type_id=?, dev=? WHERE id=?"
	_, rer = db.DB.Exec(sql, pro.Name, pro.User, pro.Description, pro.UpdateFrequency, pro.LocaltionID, pro.WeatherTypeID, pro.Dev, pro.ID)
	return
}

// DeleteProject 删除项目
func (pro *Project) DeleteProject() (rer error) {
	sql := "DELETE FROM projects WHERE id=?"
	_, rer = db.DB.Exec(sql, pro.ID)
	return
}