scen.go

package main

import (
	"log"
	"os"

	"github.com/BurntSushi/toml"
)

type RoleplayScenario struct {
	ID          string
	Action      string `toml:"action"`
	ActionDesc  string `toml:"action_desc"`
	MsgRequest  string `toml:"msg_request"`
	MsgSuccess  string `toml:"msg_success"`
	MsgForced   string `toml:"msg_forced"`
	MsgDenied   string `toml:"msg_denied"`
	IsDeniable  bool   `toml:"is_deniable"`
	IsForceable bool   `toml:"is_forceable"`
	SortLevel   int    `toml:"sort_level"`
}

type BySortLevelThenID []RoleplayScenario

func (s BySortLevelThenID) Len() int { return len(s) }

func (s BySortLevelThenID) Swap(i, j int) { s[i], s[j] = s[j], s[i] }

func (s BySortLevelThenID) Less(i, j int) bool {
	if s[i].SortLevel != s[j].SortLevel {
		return s[i].SortLevel < s[j].SortLevel
	}
	return s[i].ID < s[j].ID
}

func rpScenFileReader(filePath string) ([]RoleplayScenario) {
	var r map[string]toml.Primitive

	meta, err := toml.DecodeFile(filePath, &r)
	if err != nil {
		log.Printf("failed to parse TOML file %s: %s", filePath, err)
		return []RoleplayScenario{}
	}

	actions := make([]RoleplayScenario, 0, len(r))

	for id, p := range r {
		var a RoleplayScenario
		a.ID = id
		a.IsDeniable = true
		a.IsForceable = true

		err := meta.PrimitiveDecode(p, &a)
		if err != nil {
			log.Printf("failed to decode action %s: %s", id, err)
			return []RoleplayScenario{}
		}

		actions = append(actions, a)
		rpScensRes[a.ID+"_y"] = a.MsgSuccess
		rpScensRes[a.ID+"_n"] = a.MsgDenied
		rpScensRes[a.ID+"_f"] = a.MsgForced
	}

	return actions
}

func rpScenReadDir(actionsDir string) {
	items, _ := os.ReadDir(actionsDir)

	for _, item := range items {
		if item.IsDir() {
			rpScenReadDir(actionsDir+"/"+item.Name())
		} else {
			rpScens = append(rpScens, rpScenFileReader(actionsDir+"/"+item.Name())...)
		}
	}
}