schema

var (

	// CookStates contains all the states for the Cook machine.
	CookStates = ssC
	// CookGroups contains all the state groups for the Cook machine.
	CookGroups = sgC
)

var CookSchema = SchemaMerge(

	llm.LLMAgentSchema,
	am.Schema{

		ssC.ErrIngredients: {},
		ssC.ErrCooking:     {},

		ssC.DBStarting: {
			Require: S{ssC.Start},
			Remove:  S{ssC.DBReady},
		},
		ssC.DBReady: {
			Require: S{ssC.Start},
			Remove:  S{ssC.DBStarting},
		},

		ssC.IngredientsReady: {},
		ssC.RecipeReady:      {Require: S{ssC.IngredientsReady}},
		ssC.StepCompleted:    {Multi: true},

		ssC.StoryJoke:               {},
		ssC.StoryWakingUp:           {Tags: S{base.TagManual}},
		ssC.StoryIngredientsPicking: {Tags: S{base.TagPrompt}},
		ssC.StoryRecipePicking:      {Tags: S{base.TagPrompt}},
		ssC.StoryCookingStarted:     {Tags: S{base.TagPrompt}},
		ssC.StoryMealReady: {
			Tags:   S{base.TagPrompt, base.TagManual},
			Remove: S{ssC.InputPending},
		},
		ssC.StoryMemoryWipe: {Tags: S{base.TagPrompt}},
		ssC.StoryStartAgain: {Tags: S{base.TagPrompt}},

		ssC.CheckStories: {
			Multi:   true,
			Require: S{ssC.Start},
		},
		ssC.StoryChanged: {
			Multi: true,
			After: S{ssC.CheckStories},
		},

		ssC.RestoreCharacter: {
			Auto:    true,
			Require: S{ssC.DBReady},
			Remove:  sgC.Character,
		},
		ssC.GenCharacter: {
			Require: S{ssC.DBReady},
			Remove:  sgC.Character,
			Tags:    S{base.TagPrompt, base.TagTrigger},
		},
		ssC.CharacterReady: {Remove: sgC.Character},

		ssC.RestoreJokes: {
			Auto:    true,
			Require: S{ssC.DBReady, ssC.CharacterReady},
			Remove:  sgC.Jokes,
		},
		ssC.GenJokes: {
			Require: S{ssC.CharacterReady, ssC.DBReady},
			Remove:  sgC.Jokes,
			Tags:    S{base.TagPrompt},
		},
		ssC.JokesReady: {Remove: sgC.Jokes},

		ssC.RestoreResources: {
			Auto:    true,
			Require: S{ssC.CharacterReady, ssC.DBReady},
			Remove:  sgC.Resources,
		},
		ssC.GenResources: {
			Require: S{ssC.CharacterReady, ssC.DBReady},
			Remove:  sgC.Resources,
			Tags:    S{base.TagPrompt, base.TagTrigger},
		},
		ssC.ResourcesReady: {Remove: sgC.Resources},

		ssC.GenSteps: {
			Auto:    true,
			Require: S{ssC.StoryCookingStarted},
			Remove:  S{ssC.StepsReady},
			Tags:    S{base.TagPrompt},
		},
		ssC.StepsReady: {
			Require: S{ssC.RecipeReady},
			Remove:  S{ssC.GenSteps},
		},

		ssC.GenStepComments: {
			Auto:    true,
			Require: S{ssC.StoryCookingStarted},
			Remove:  S{ssC.StepCommentsReady},
			Tags:    S{base.TagPrompt},
		},
		ssC.StepCommentsReady: {Remove: S{ssC.GenStepComments}},

		ssC.Orienting: {
			Multi: true,
			Tags:  S{base.TagPrompt},
		},
		ssC.OrientingMove: {},

		ssC.Start: StateAdd(saLLM[ssLLM.Start], State{
			Add: S{ssC.CheckStories, ssC.DBStarting},
		}),
		ssC.Ready: StateAdd(saLLM[ssLLM.Ready], State{
			Auto:    true,
			Require: S{ssC.CharacterReady, ssC.ResourcesReady},
		}),
		ssC.Interrupted: StateAdd(saLLM[ssLLM.Interrupted], State{

			Remove: sgC.Interruptable,
		}),
		ssC.Prompt: StateAdd(saLLM[ssLLM.Prompt], State{}),
	})

var Exception = am.StateException

var LLMResources = ParamsGenResources{
	Phrases: map[string]string{
		"NoCookingNoJokes":      "No jokes without cooking.",
		"IngredientsPicking":    "Tell me what cooking ingredients do you have at hand, we need at least %d to continue.",
		"IngredientsPickingEnd": "OK, I have the ingredients I need.",
		"RecipePicking":         "Ok, let me check my books for what we could make...",
		"ResumeNeeded":          "You have to Resume for me to do anything.",
		"WokenUp":               "OK I'm ready to start",
		"CookingStarted":        "You chose %s as the recipe which we will cook, got it. I will plan all the cooking steps nicely for us. Just press those buttons on the right, once I'm done.",
		"CharacterReady": sp(`
			Welcome to Cook - your AI-powered cooking assistant! It will help you pick a meal from the ingredients you have, and then you can cook it together (wink wink).
	`),
		ssC.StoryMealReady: "We made it, the meal is ready! You can enjoy it now. I hope you had fun cooking with us.",
		"ReqLimitReached":  "You have reached the limit of %d requests per session. Please come back later.",
	},
}

var MatchIngredients = regexp.MustCompile("^Ingredient")

var MatchSteps = regexp.MustCompile(`^Step`)

var MemMealReady = "StepMealReady"

var SAdd = am.SAdd

var SchemaMerge = am.SchemaMerge

var StateAdd = am.StateAdd

var StateSet = am.StateSet

var StoryCookingStarted = &Story{
	Story: shared.Story[Story]{
		StoryInfo: shared.StoryInfo{
			State: ssC.StoryCookingStarted,
			Title: "Cooking Started",
			Desc:  "The main story, the bot translates the recipe into actionable steps, then comments while the user completes them.",
		},
	},
	Cook: shared.StoryActor{
		Trigger: amhelp.Cond{
			Is: S{ssC.Ready, ssC.RecipeReady},
		},
	},
	Memory: shared.StoryActor{
		Trigger: amhelp.Cond{
			Not: S{MemMealReady},
		},
	},
}

var StoryIngredientsPicking = &Story{
	Story: shared.Story[Story]{
		StoryInfo: shared.StoryInfo{
			State: ssC.StoryIngredientsPicking,
			Title: "Ingredients Picking",
			Desc:  "The bot asks the user what ingredients they have at hand.",
		},
	},
	Cook: shared.StoryActor{
		Trigger: amhelp.Cond{
			Is:  S{ssC.Ready},
			Not: S{ssC.IngredientsReady, ssC.StoryWakingUp},
		},
	},
}

var StoryJoke = &Story{
	Story: shared.Story[Story]{
		StoryInfo: shared.StoryInfo{
			State: ssC.StoryJoke,
			Title: "Joke",
			Desc:  "In this story the bot tells a joke when asked to, based on the character.",
		},

		CanActivate: func(s *Story) bool {
			mem := s.Memory.Mach
			stepStates := mem.StateNamesMatch(MatchSteps)
			stepsNow := mem.Time(stepStates).Sum(nil) + s.Epoch
			freq := 1.5

			return s.Tick == 0 || float64(stepsNow)*freq >= float64(s.Tick)
		},
	},
}

var StoryMealReady = &Story{
	Story: shared.Story[Story]{
		StoryInfo: shared.StoryInfo{
			State: ssC.StoryMealReady,
			Title: "Meal Ready",
			Desc:  "This story is the end of the flow, the recipe should have been materialized by now.",
		},
	},
	Memory: shared.StoryActor{
		Trigger: amhelp.Cond{
			Is: S{MemMealReady},
		},
	},
}

var StoryMemoryWipe = &Story{
	Story: shared.Story[Story]{
		StoryInfo: shared.StoryInfo{
			State: ssC.StoryMemoryWipe,
			Title: "Memory Wipe",
			Desc:  "The bot will clean both short term and long term memory.",
		},
	},
}

var StoryRecipePicking = &Story{
	Story: shared.Story[Story]{
		StoryInfo: shared.StoryInfo{
			State: ssC.StoryRecipePicking,
			Title: "Recipe Picking",
			Desc:  "The bot offers some recipes, based on the ingredients.",
		},
	},
	Cook: shared.StoryActor{
		Trigger: amhelp.Cond{
			Is:  S{ssC.Ready, ssC.IngredientsReady},
			Not: S{ssC.RecipeReady},
		},
	},
}

var StoryStartAgain = &Story{
	Story: shared.Story[Story]{
		StoryInfo: shared.StoryInfo{
			State: ssC.StoryStartAgain,
			Title: "Start Again",
			Desc:  "The session will re-start, keeping the bot's memory.",
		},
	},
}

var StoryWakingUp = &Story{
	Story: shared.Story[Story]{
		StoryInfo: shared.StoryInfo{
			State: ssC.StoryWakingUp,
			Title: "Waking Up",
			Desc:  "The waking up story is the bot starting on either cold or warm boot.",
		},
	},
	Cook: shared.StoryActor{
		Trigger: amhelp.Cond{
			Not: S{ssC.Ready},
		},
	},
}