package extraction

import (
	"encoding/json"
	"errors"
	"strings"
)

// Dify orchestrates the model steps. These rules also apply to its final output
// before the existing story writers can save or merge anything.
func WorkflowPrompts() (string, string)         { return eventExtractionPrompt, dailyExtractionPrompt }
func WorkflowContent(in GenerationInput) string { return generationContent(in) }

func ValidateWorkflowResult(in GenerationInput, result GenerationResult, modelRef string) (GenerationResult, error) {
	result.Summary = strings.TrimSpace(result.Summary)
	result.DecisionReason = strings.TrimSpace(result.DecisionReason)
	result.OpenTopic = strings.TrimSpace(result.OpenTopic)
	result.Keywords = normalizedKeywords(result.Keywords)
	if result.Stories == nil || result.DailyEvents == nil {
		return GenerationResult{}, errors.New("workflow event arrays required")
	}
	if err := normalizeEvents(&result, in); err != nil {
		return GenerationResult{}, err
	}
	normalizeDecisionContent(&result)
	if !validGeneratedContent(result) {
		return GenerationResult{}, errors.New("workflow content failed validation")
	}
	actual := make([]DailyEvent, 0, len(result.DailyEvents))
	for _, d := range result.DailyEvents {
		if d.TimeScope != "today_actual" || strings.TrimSpace(d.Evidence) == "" || !strings.Contains(in.Content, d.Evidence) {
			return GenerationResult{}, errors.New("daily event lacks original evidence")
		}
		future := false
		for _, phrase := range []string{"明天", "明早", "前几天", "打算", "计划", "尚未", "还没动手", "想着", "还想", "一会儿", "以后再"} {
			if strings.Contains(d.Content, phrase) {
				future = true
				break
			}
		}
		if !future {
			actual = append(actual, d)
		}
	}
	result.DailyEvents = actual
	result.ModelRef = modelRef
	result.PromptVersion = "content-events-dify-v1"
	result.AllowStory = result.Meaningful && result.StoryWorthy
	return result, nil
}

func ParseWorkflowResult(in GenerationInput, raw json.RawMessage, modelRef string) (GenerationResult, error) {
	var result GenerationResult
	if json.Unmarshal(raw, &result) != nil {
		return result, errors.New("invalid workflow extraction JSON")
	}
	return ValidateWorkflowResult(in, result, modelRef)
}