namespace LMS.Repository.MJPackage
{
    public class MJSubmitImageModel
    {
        /// <summary>
        /// bot 类型，mj(默认)或niji
        /// MID_JOURNEY | 枚举值: NIJI_JOURNEY
        /// </summary>
        public string? BotType { get; set; }

        /// <summary>
        /// 提示词。
        /// </summary>
        public string Prompt { get; set; }

        /// <summary>
        /// 垫图base64数组。
        /// </summary>
        public List<string>? Base64Array { get; set; }

        /// <summary>
        /// 账号过滤
        /// </summary>
        public AccountFilter? AccountFilter { get; set; }

        /// <summary>
        /// 自定义参数。
        /// </summary>
        public string? State { get; set; }

        /// <summary>
        /// 回调地址, 为空时使用全局notifyHook。
        /// </summary>
        public string? NotifyHook { get; set; }
    }

    public class AccountFilter
    {
        /// <summary>
        /// 过滤指定实例的账号
        /// </summary>
        public string? InstanceId { get; set; }

        /// <summary>
        /// 账号模式 RELAX | FAST | TURBO
        /// </summary>
        public List<GenerationSpeedMode>? Modes { get; set; } = new List<GenerationSpeedMode>();

        /// <summary>
        /// 账号是否 remix（Midjourney Remix）
        /// </summary>
        public bool? Remix { get; set; }

        /// <summary>
        /// 账号是否 remix（Nijiourney Remix）
        /// </summary>
        public bool? NijiRemix { get; set; }

        /// <summary>
        /// 账号过滤时，remix 自动提交视为账号的 remix 为 false
        /// </summary>
        public bool? RemixAutoConsidered { get; set; }
    }

    /// <summary>
    /// 生成速度模式枚举.
    /// </summary>
    public enum GenerationSpeedMode
    {
        RELAX,
        FAST,
        TURBO
    }
}