package session

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

// 任务会话标记：定时任务（scheduler）每次触发创建的会话在此留痕。
//
// 用途有二：
//  1. 出处可查——standalone 任务每次一个新会话，时间戳命名与手工临时对话
//     无法区分，没有标记就既不能在 UI 归组、也不敢做自动清理；
//  2. 会话清扫的依据——只有带标记的会话才按保留期回收（手工会话、
//     existing 模式的复用线程、项目绑定会话一律不动）。
//
// 与 wsbind 同一套 sidecar 命名约定：<会话ID>.job.json。

// JobMarker 是任务会话的出处标记。
type JobMarker struct {
	JobID   string    `json:"job_id"`
	JobName string    `json:"job_name"`
	At      time.Time `json:"at"` // 会话创建时刻（清扫按此判保留期）
}

// JobMarkerPath 标记 sidecar 路径。
func JobMarkerPath(id string) string {
	return filepath.Join(Dir(), id+".job.json")
}

// SaveJobMarker 写入任务会话标记（best-effort 调用方不判错也行）。
func SaveJobMarker(id, jobID, jobName string) error {
	return SaveJobMarkerAt(id, jobID, jobName, time.Now())
}

// SaveJobMarkerAt 同 SaveJobMarker，时间由调用方指定（测试注入用）。
func SaveJobMarkerAt(id, jobID, jobName string, at time.Time) error {
	if id == "" || jobID == "" {
		return nil
	}
	b, err := json.Marshal(JobMarker{JobID: jobID, JobName: jobName, At: at})
	if err != nil {
		return err
	}
	if err := os.MkdirAll(Dir(), 0o755); err != nil {
		return err
	}
	return os.WriteFile(JobMarkerPath(id), b, 0o644)
}

// LoadJobMarker 读取任务会话标记；未标记返回 false。
func LoadJobMarker(id string) (JobMarker, bool) {
	b, err := os.ReadFile(JobMarkerPath(id))
	if err != nil {
		return JobMarker{}, false
	}
	var m JobMarker
	if json.Unmarshal(b, &m) != nil {
		return JobMarker{}, false
	}
	return m, true
}

// RemoveJobMarker 删除标记（随会话清扫一起走）。
func RemoveJobMarker(id string) error {
	err := os.Remove(JobMarkerPath(id))
	if os.IsNotExist(err) {
		return nil
	}
	return err
}