package classify

import (
	"encoding/json"
	"fmt"
	"os"
	"path/filepath"
)

// Entity 实体定义
type Entity struct {
	ID          string   `json:"id"`
	Name        string   `json:"name"`
	Aliases     []string `json:"aliases"`
	Platform    string   `json:"platform,omitempty"`
	Description string   `json:"description,omitempty"`
	Users       []string `json:"users,omitempty"`
	URL         string   `json:"url,omitempty"`
}

// Entities 实体集合
type Entities struct {
	Platforms []Entity `json:"platforms"`
	Apps      []Entity `json:"apps"`
	Devices   []Entity `json:"devices"`
}

// Relation 实体关系
type Relation struct {
	From        string `json:"from"`
	To          string `json:"to"`
	Type        string `json:"type"`
	Description string `json:"description"`
}

// Relations 关系集合
type Relations struct {
	Relations []Relation `json:"relations"`
}

// ExtractEntities 从实体目录提取实体信息
func ExtractEntities(vaultPath string) (*Entities, *Relations, error) {
	entityDir := filepath.Join(vaultPath, "实体")

	// 检查目录是否存在
	if _, err := os.Stat(entityDir); os.IsNotExist(err) {
		return nil, nil, fmt.Errorf("实体目录不存在: %s", entityDir)
	}

	// 预定义实体（从设计文档）
	entities := &Entities{
		Platforms: []Entity{
			{
				ID:          "elc",
				Name:        "电子秤平台",
				Aliases:     []string{"电子秤后台", "elc平台", "elc.zhiheiot.com"},
				URL:         "https://elc.zhiheiot.com",
				Description: "气站充装作业管理系统",
				Users:       []string{"气站站长", "充装员", "开票员"},
			},
			{
				ID:          "ops",
				Name:        "运营管理平台",
				Aliases:     []string{"运营平台", "运营后台", "rb.zhiheiot.com"},
				URL:         "https://rb.zhiheiot.com",
				Description: "气站运营管理系统",
				Users:       []string{"老板", "管理层", "客服"},
			},
		},
		Apps: []Entity{
			{
				ID:          "safety",
				Name:        "安全用气App",
				Aliases:     []string{"配送App", "电子秤配送App"},
				Platform:    "elc",
				Description: "二维码配送",
			},
			{
				ID:          "delivery",
				Name:        "易配送App",
				Aliases:     []string{"配送App", "运营配送"},
				Platform:    "ops",
				Description: "NFC识别芯片配送",
			},
			{
				ID:          "assistant",
				Name:        "艾信助手App",
				Aliases:     []string{"建档App"},
				Platform:    "both",
				Description: "融合两平台建档",
			},
		},
		Devices: []Entity{
			{
				ID:          "scale",
				Name:        "电子秤",
				Aliases:     []string{"充装电子秤"},
				Platform:    "elc",
				Description: "充装作业设备",
			},
			{
				ID:          "gun",
				Name:        "智能枪",
				Aliases:     []string{"充装枪", "智能充装枪"},
				Platform:    "both",
				Description: "共有硬件",
			},
			{
				ID:          "valve",
				Name:        "智能阀",
				Aliases:     []string{"智能角阀"},
				Platform:    "ops",
				Description: "气瓶阀门",
			},
			{
				ID:          "box",
				Name:        "艾信盒子",
				Aliases:     []string{"4G通信设备"},
				Platform:    "both",
				Description: "共有硬件",
			},
		},
	}

	// 预定义关系
	relations := &Relations{
		Relations: []Relation{
			{From: "gun", To: "scale", Type: "connects", Description: "智能枪连接电子秤"},
			{From: "gun", To: "box", Type: "syncs", Description: "充装数据同步到艾信盒子"},
			{From: "box", To: "elc", Type: "uploads", Description: "上传充装记录"},
			{From: "box", To: "ops", Type: "uploads", Description: "上传充装记录"},
			{From: "valve", To: "delivery", Type: "identified_by", Description: "NFC识别"},
			{From: "safety", To: "elc", Type: "belongs_to", Description: "属于电子秤平台"},
			{From: "delivery", To: "ops", Type: "belongs_to", Description: "属于运营管理平台"},
		},
	}

	return entities, relations, nil
}

// SaveEntities 保存实体和关系到 JSON 文件
func SaveEntities(vaultPath string, entities *Entities, relations *Relations) error {
	// 保存 entities.json
	entitiesPath := filepath.Join(vaultPath, "entities.json")
	entitiesData, err := json.MarshalIndent(entities, "", "  ")
	if err != nil {
		return fmt.Errorf("序列化 entities 失败: %w", err)
	}
	if err := os.WriteFile(entitiesPath, entitiesData, 0644); err != nil {
		return fmt.Errorf("写入 entities.json 失败: %w", err)
	}

	// 保存 relations.json
	relationsPath := filepath.Join(vaultPath, "relations.json")
	relationsData, err := json.MarshalIndent(relations, "", "  ")
	if err != nil {
		return fmt.Errorf("序列化 relations 失败: %w", err)
	}
	if err := os.WriteFile(relationsPath, relationsData, 0644); err != nil {
		return fmt.Errorf("写入 relations.json 失败: %w", err)
	}

	return nil
}