// Package notes 存放站长为地点加的别名与备注。
// 备注按 rond 的地点 ID（src_pk）而非数据库自增 id 存储，
// 因此重新上传备份、地点换了新 id 之后内容依然挂在原地。
package notes

import (
	"context"
	"database/sql"
	"strings"
	"time"
)

type Note struct {
	SrcPK     int
	Alias     string
	Content   string
	RondNote  string
	UpdatedAt time.Time
}

type Store struct{ DB *sql.DB }

// All 返回某个数据集的全站备注（站点只有一个站长账号，读的时候不按 user_id 过滤，
// 这样匿名访客也能看到别名；写的时候才需要登录用户的 id）。
// 后写覆盖先写，保证同一地点只保留一条。必须按 dataset_id 过滤：
// src_pk 只在单个数据集内唯一，跨数据集同编号是完全不同的地点。
func (s *Store) All(ctx context.Context, datasetID int64) (map[int]Note, error) {
	out := map[int]Note{}
	rows, err := s.DB.QueryContext(ctx,
		`SELECT src_pk, alias, note, rond_note, updated_at FROM place_notes
		  WHERE dataset_id=$1 ORDER BY updated_at DESC`, datasetID)
	if err != nil {
		return out, err
	}
	defer rows.Close()
	for rows.Next() {
		var n Note
		if err := rows.Scan(&n.SrcPK, &n.Alias, &n.Content, &n.RondNote, &n.UpdatedAt); err != nil {
			return out, err
		}
		out[n.SrcPK] = n
	}
	return out, rows.Err()
}

// SetNote 只改备注、保留别名（批量「清空备注」用）；两者都为空则删除记录。
func (s *Store) SetNote(ctx context.Context, userID, datasetID int64, srcPK int, content string) error {
	var alias string
	err := s.DB.QueryRowContext(ctx,
		`SELECT alias FROM place_notes WHERE user_id=$1 AND dataset_id=$2 AND src_pk=$3`,
		userID, datasetID, srcPK).Scan(&alias)
	if err != nil && err != sql.ErrNoRows {
		return err
	}
	return s.Save(ctx, userID, datasetID, srcPK, alias, content)
}

// Delete 删除某地点的整条备注与别名。
func (s *Store) Delete(ctx context.Context, userID, datasetID int64, srcPK int) error {
	_, err := s.DB.ExecContext(ctx,
		`DELETE FROM place_notes WHERE user_id=$1 AND dataset_id=$2 AND src_pk=$3`, userID, datasetID, srcPK)
	return err
}

// Save 写入别名与备注；两者都为空时清空手写内容（rond 回填的那一块要留着）。
func (s *Store) Save(ctx context.Context, userID, datasetID int64, srcPK int, alias, content string) error {
	alias = strings.TrimSpace(alias)
	content = strings.TrimSpace(content)
	if alias == "" && content == "" {
		// 没有 rond 备注时直接删行；有则只清空手写部分，别把回填结果一起删掉
		_, err := s.DB.ExecContext(ctx,
			`DELETE FROM place_notes WHERE user_id=$1 AND dataset_id=$2 AND src_pk=$3 AND rond_note=''`,
			userID, datasetID, srcPK)
		if err != nil {
			return err
		}
		_, err = s.DB.ExecContext(ctx,
			`UPDATE place_notes SET alias='', note='', updated_at=now()
			  WHERE user_id=$1 AND dataset_id=$2 AND src_pk=$3`, userID, datasetID, srcPK)
		return err
	}
	_, err := s.DB.ExecContext(ctx, `INSERT INTO place_notes (user_id, dataset_id, src_pk, alias, note, updated_at)
		VALUES ($1,$2,$3,$4,$5,now())
		ON CONFLICT (user_id, dataset_id, src_pk)
		DO UPDATE SET alias=EXCLUDED.alias, note=EXCLUDED.note, updated_at=now()`,
		userID, datasetID, srcPK, alias, content)
	return err
}

// BackfillRond 把 rond 备份里按次到访记录的备注（ZVISIT.ZREMARK_）汇总进 rond_note：
// 同一地点多条备注全部保留，去重后按时间升序、每条一行并带上日期。
// 只重写 rond_note 一列，站长手写的 note 一概不动。返回写了多少行。
func (s *Store) BackfillRond(ctx context.Context, userID, datasetID int64) (int, error) {
	// DISTINCT 在「日期 + 内容」这一层做，同一天重复写的同一句只留一条
	const agg = `
	WITH raw AS (
		SELECT p.src_pk, to_char(v.arrival, 'YYYY-MM-DD') || ' ' || btrim(v.remark) AS line
		FROM visits v JOIN places p ON p.id = v.place_id
		WHERE v.dataset_id = $1 AND btrim(COALESCE(v.remark, '')) <> ''
	), uniq AS (
		SELECT DISTINCT src_pk, line, left(line, 10) AS day FROM raw
	), agg AS (
		SELECT src_pk, string_agg(line, E'\n' ORDER BY day, line) AS txt FROM uniq GROUP BY src_pk
	)`
	res, err := s.DB.ExecContext(ctx, agg+`
		INSERT INTO place_notes (user_id, dataset_id, src_pk, alias, note, rond_note, updated_at)
		SELECT $2, $1, src_pk, '', '', txt, now() FROM agg
		ON CONFLICT (user_id, dataset_id, src_pk) DO UPDATE SET rond_note = EXCLUDED.rond_note`,
		datasetID, userID)
	if err != nil {
		return 0, err
	}
	affected, _ := res.RowsAffected()

	// 备注被删掉的地点要清掉旧值，否则会一直留着过期的 rond 内容。
	// 限定在本数据集的地点里：datasetID 不合法时子查询为空，一行都不会动，
	// 不会把别的数据集的回填结果误清掉。
	if _, err := s.DB.ExecContext(ctx, `
		UPDATE place_notes n SET rond_note = ''
		WHERE n.user_id = $2 AND n.dataset_id = $1 AND n.rond_note <> ''
		  AND n.src_pk IN (SELECT src_pk FROM places WHERE dataset_id = $1)
		  AND NOT EXISTS (
			SELECT 1 FROM visits v JOIN places p ON p.id = v.place_id
			WHERE v.dataset_id = $1 AND p.src_pk = n.src_pk AND btrim(COALESCE(v.remark, '')) <> ''
		  )`, datasetID, userID); err != nil {
		return 0, err
	}
	return int(affected), nil
}