model

type Agent struct {
	Idx                int
	TeamName           string
	OriginalName       string
	GameName           string
	Profile            *Profile
	ProfileDescription *string
	Role               Role
	Connection         *websocket.Conn
	HasError           bool
}

type BroadcastPacket struct {
	Id     string `json:"id"`
	Idx    int    `json:"idx"`
	Day    int    `json:"day"`
	IsDay  bool   `json:"is_day"`
	Agents []struct {
		Idx     int     `json:"idx"`
		Team    string  `json:"team"`
		Name    string  `json:"name"`
		Profile *string `json:"profile,omitempty"`
		Avatar  *string `json:"avatar,omitempty"`
		Role    string  `json:"role"`
		IsAlive bool    `json:"is_alive"`
	} `json:"agents"`
	Event     string  `json:"event"`
	Message   *string `json:"message,omitempty"`
	FromIdx   *int    `json:"from_idx,omitempty"`
	ToIdx     *int    `json:"to_idx,omitempty"`
	BubbleIdx *int    `json:"bubble_idx,omitempty"`
}

type Config struct {
	Server              ServerConfig              `yaml:"server"`
	Game                GameConfig                `yaml:"game"`
	Logic               LogicConfig               `yaml:"logic"`
	Matching            MatchingConfig            `yaml:"matching"`
	CustomProfile       CustomProfileConfig       `yaml:"custom_profile"`
	JSONLogger          JSONLoggerConfig          `yaml:"json_logger"`
	GameLogger          GameLoggerConfig          `yaml:"game_logger"`
	RealtimeBroadcaster RealtimeBroadcasterConfig `yaml:"realtime_broadcaster"`
	TTSBroadcaster      TTSBroadcasterConfig      `yaml:"tts_broadcaster"`
}

type Connection struct {
	TeamName     string
	OriginalName string
	Conn         *websocket.Conn
	Header       *http.Header
}

type CustomProfileConfig struct {
	Enable          bool                 `yaml:"enable"`
	ProfileEncoding map[string]string    `yaml:"profile_encoding"`
	Profiles        []Profile            `yaml:"profiles"`
	DynamicProfile  DynamicProfileConfig `yaml:"dynamic_profile"`
}

type DynamicProfileConfig struct {
	Enable    bool     `yaml:"enable"`
	Prompt    string   `yaml:"prompt"`
	Attempts  int      `yaml:"attempts"`
	Model     string   `yaml:"model"`
	MaxTokens int      `yaml:"max_tokens"`
	Avatars   []string `yaml:"avatars"`
}

type GameConfig struct {
	AgentCount     int        `yaml:"agent_count"`
	MaxDay         int        `yaml:"max_day"`
	VoteVisibility bool       `yaml:"vote_visibility"`
	Talk           TalkConfig `yaml:"talk"`
	Whisper        TalkConfig `yaml:"whisper"`
	Vote           struct {
		MaxCount      int  `yaml:"max_count"`
		AllowSelfVote bool `yaml:"allow_self_vote"`
	} `yaml:"vote"`
	AttackVote struct {
		MaxCount      int  `yaml:"max_count"`
		AllowSelfVote bool `yaml:"allow_self_vote"`
		AllowNoTarget bool `yaml:"allow_no_target"`
	} `yaml:"attack_vote"`
}

type GameLoggerConfig struct {
	Enable    bool   `yaml:"enable"`
	OutputDir string `yaml:"output_dir"`
	Filename  string `yaml:"filename"`
}

type GameStatus struct {
	Day             int
	MediumResult    *Judge
	DivineResult    *Judge
	ExecutedAgent   *Agent
	AttackedAgent   *Agent
	Guard           *Guard
	Votes           []Vote
	AttackVotes     []Vote
	Talks           []Talk
	Whispers        []Talk
	StatusMap       map[Agent]Status
	RemainCountMap  *map[Agent]int
	RemainLengthMap *map[Agent]int
	RemainSkipMap   *map[Agent]int
}

type Guard struct {
	Day    int   `json:"day"`
	Agent  Agent `json:"agent"`
	Target Agent `json:"target"`
}

type Info struct {
	GameID         string           `json:"game_id"`
	Day            int              `json:"day"`
	Agent          *Agent           `json:"agent"`
	Profile        *string          `json:"profile,omitempty"`
	MediumResult   *Judge           `json:"medium_result,omitempty"`
	DivineResult   *Judge           `json:"divine_result,omitempty"`
	ExecutedAgent  *Agent           `json:"executed_agent,omitempty"`
	AttackedAgent  *Agent           `json:"attacked_agent,omitempty"`
	VoteList       []Vote           `json:"vote_list,omitempty"`
	AttackVoteList []Vote           `json:"attack_vote_list,omitempty"`
	TalkList       []Talk           `json:"-"`
	WhisperList    []Talk           `json:"-"`
	StatusMap      map[Agent]Status `json:"status_map"`
	RoleMap        map[Agent]Role   `json:"role_map"`
	RemainCount    *int             `json:"remain_count,omitempty"`
	RemainLength   *int             `json:"remain_length,omitempty"`
	RemainSkip     *int             `json:"remain_skip,omitempty"`
}

type JSONLoggerConfig struct {
	Enable    bool   `yaml:"enable"`
	OutputDir string `yaml:"output_dir"`
	Filename  string `yaml:"filename"`
}

type Judge struct {
	Day    int     `json:"day"`
	Agent  Agent   `json:"agent"`
	Target Agent   `json:"target"`
	Result Species `json:"result"`
}

type LogicConfig struct {
	DayPhases   []Phase                `yaml:"day_phases"`
	NightPhases []Phase                `yaml:"night_phases"`
	Roles       map[int]map[string]int `yaml:"roles"`
}

type MatchWeight struct {
	RoleIdxs map[Role][]int `json:"role_idxs"`
	Weight   float64        `json:"weight"`
}

type MatchingConfig struct {
	SelfMatch    bool   `yaml:"self_match"`
	IsOptimize   bool   `yaml:"is_optimize"`
	TeamCount    int    `yaml:"team_count"`
	GameCount    int    `yaml:"game_count"`
	OutputPath   string `yaml:"output_path"`
	InfiniteLoop bool   `yaml:"infinite_loop"`
}

type Packet struct {
	Request        *Request `json:"request"`
	Info           *Info    `json:"info,omitempty"`
	Setting        *Setting `json:"setting,omitempty"`
	TalkHistory    *[]Talk  `json:"talk_history,omitempty"`
	WhisperHistory *[]Talk  `json:"whisper_history,omitempty"`
}

type Phase struct {
	Name      string   `yaml:"name"`
	Actions   []string `yaml:"actions"`
	OnlyDay   *int     `yaml:"only_day,omitempty"`
	ExceptDay *int     `yaml:"except_day,omitempty"`
}

type Profile struct {
	Name      string            `yaml:"name"`
	AvatarURL string            `yaml:"avatar_url"`
	VoiceID   int               `yaml:"voice_id"`
	Arguments map[string]string `yaml:",inline"`
}

type RealtimeBroadcasterConfig struct {
	Enable    bool          `yaml:"enable"`
	Delay     time.Duration `yaml:"delay"`
	OutputDir string        `yaml:"output_dir"`
	Filename  string        `yaml:"filename"`
}

type Request struct {
	Type            string
	RequireResponse bool
}

type Role struct {
	Name    string
	Team    Team
	Species Species
}

type ServerConfig struct {
	WebSocket struct {
		Host string `yaml:"host"`
		Port int    `yaml:"port"`
	} `yaml:"web_socket"`
	Authentication struct {
		Enable bool `yaml:"enable"`
	} `yaml:"authentication"`
	Timeout struct {
		Action     time.Duration `yaml:"action"`
		Response   time.Duration `yaml:"response"`
		Acceptable time.Duration `yaml:"acceptable"`
	} `yaml:"timeout"`
	MaxContinueErrorRatio float64 `yaml:"max_continue_error_ratio"`
}

type Setting struct {
	AgentCount     int          `json:"agent_count"`
	MaxDay         *int         `json:"max_day,omitempty"`
	RoleNumMap     map[Role]int `json:"role_num_map"`
	VoteVisibility bool         `json:"vote_visibility"`
	Talk           struct {
		TalkSetting `json:",inline"`
	} `json:"talk"`
	Whisper struct {
		TalkSetting `json:",inline"`
	} `json:"whisper"`
	Vote struct {
		MaxCount      int  `json:"max_count"`
		AllowSelfVote bool `json:"allow_self_vote"`
	} `json:"vote"`
	AttackVote struct {
		MaxCount      int  `json:"max_count"`
		AllowSelfVote bool `json:"allow_self_vote"`
		AllowNoTarget bool `json:"allow_no_target"`
	} `json:"attack_vote"`
	Timeout struct {
		Action   int `json:"action"`
		Response int `json:"response"`
	} `json:"timeout"`
}

type Species string

type Status string

type TTSBroadcasterConfig struct {
	Enable         bool          `yaml:"enable"`
	Async          bool          `yaml:"async"`
	TargetDuration time.Duration `yaml:"target_duration"`
	SegmentDir     string        `yaml:"segment_dir"`
	TempDir        string        `yaml:"temp_dir"`
	Host           string        `yaml:"host"`
	Timeout        time.Duration `yaml:"timeout"`
	FfmpegPath     string        `yaml:"ffmpeg_path"`
	FfprobePath    string        `yaml:"ffprobe_path"`
	ConvertArgs    []string      `yaml:"convert_args"`
	DurationArgs   []string      `yaml:"duration_args"`
	PreConvertArgs []string      `yaml:"pre_convert_args"`
	SplitArgs      []string      `yaml:"split_args"`
}

type Talk struct {
	Idx   int    `json:"idx"`
	Day   int    `json:"day"`
	Turn  int    `json:"turn"`
	Agent Agent  `json:"agent"`
	Text  string `json:"text"`
}

type TalkConfig struct {
	MaxCount struct {
		PerAgent int `yaml:"per_agent"`
		PerDay   int `yaml:"per_day"`
	} `yaml:"max_count"`
	MaxLength struct {
		CountInWord   bool `yaml:"count_in_word"`
		CountSpaces   bool `yaml:"count_spaces"`
		PerTalk       int  `yaml:"per_talk"`
		MentionLength int  `yaml:"mention_length"`
		PerAgent      int  `yaml:"per_agent"`
		BaseLength    int  `yaml:"base_length"`
	} `yaml:"max_length"`
	MaxSkip int `yaml:"max_skip"`
}

type TalkSetting struct {
	MaxCount struct {
		PerAgent int `json:"per_agent"`
		PerDay   int `json:"per_day"`
	} `json:"max_count"`
	MaxLength struct {
		CountInWord   *bool `json:"count_in_word,omitempty"`
		CountSpaces   *bool `json:"count_spaces,omitempty"`
		PerTalk       *int  `json:"per_talk,omitempty"`
		MentionLength *int  `json:"mention_length,omitempty"`
		PerAgent      *int  `json:"per_agent,omitempty"`
		BaseLength    *int  `json:"base_length,omitempty"`
	} `json:"max_length"`
	MaxSkip int `json:"max_skip"`
}

type Team string

type Vote struct {
	Day    int   `json:"day"`
	Agent  Agent `json:"agent"`
	Target Agent `json:"target"`
}