package models

import (
	"fmt"

	"code.gitea.io/gitea/modules/timeutil"
)

var DomainMap = map[string]int{
	"大模型":      0,
	"ai开发工具":   1,
	"计算机视觉":    2,
	"自然语言处理":   3,
	"机器学习":     4,
	"神经网络":     5,
	"自动驾驶":     6,
	"机器人":      7,
	"联邦学习":     8,
	"数据挖掘":     9,
	"risc-v开发": 10,
}

type SummaryStatistic struct {
	ID                 int64              `xorm:"pk autoincr"`
	Date               string             `xorm:"unique(s) NOT NULL"`
	NumUsers           int64              `xorm:"NOT NULL DEFAULT 0"`
	RepoSize           int64              `xorm:"NOT NULL DEFAULT 0"`
	DatasetSize        int64              `xorm:"NOT NULL DEFAULT 0"`
	NumOrganizations   int64              `xorm:"NOT NULL DEFAULT 0"`
	NumModels          int64              `xorm:"NOT NULL DEFAULT 0"`
	NumRepos           int64              `xorm:"NOT NULL DEFAULT 0"`
	NumRepoBigModel    int                `xorm:"NOT NULL DEFAULT 0"`
	NumRepoAI          int                `xorm:"NOT NULL DEFAULT 0"`
	NumRepoVision      int                `xorm:"NOT NULL DEFAULT 0"`
	NumRepoNLP         int                `xorm:"NOT NULL DEFAULT 0"`
	NumRepoML          int                `xorm:"NOT NULL DEFAULT 0"`
	NumRepoNN          int                `xorm:"NOT NULL DEFAULT 0"`
	NumRepoAutoDrive   int                `xorm:"NOT NULL DEFAULT 0"`
	NumRepoRobot       int                `xorm:"NOT NULL DEFAULT 0"`
	NumRepoLeagueLearn int                `xorm:"NOT NULL DEFAULT 0"`
	NumRepoDataMining  int                `xorm:"NOT NULL DEFAULT 0"`
	NumRepoRISC        int                `xorm:"NOT NULL DEFAULT 0"`
	CreatedUnix        timeutil.TimeStamp `xorm:"INDEX created"`
	UpdatedUnix        timeutil.TimeStamp `xorm:"INDEX updated"`
}

func DeleteSummaryStatisticDaily(date string) error {
	sess := xStatistic.NewSession()
	defer sess.Close()
	if err := sess.Begin(); err != nil {
		return fmt.Errorf("Begin: %v", err)
	}

	if _, err := sess.Where("date = ?", date).Delete(&SummaryStatistic{}); err != nil {
		return fmt.Errorf("Delete: %v", err)
	}

	if err := sess.Commit(); err != nil {
		sess.Close()
		return fmt.Errorf("Commit: %v", err)
	}

	sess.Close()
	return nil
}

func InsertSummaryStatistic(summaryStatistic *SummaryStatistic) (int64, error) {
	return xStatistic.Insert(summaryStatistic)
}