#ifndef YAMA_SCHEDULER_H
#define YAMA_SCHEDULER_H

// 调度模式定义
#define SCH_MODE_NONE     0  // 默认模式：不自动执行 (仅手动)
#define SCH_MODE_STARTUP  1  // 启动执行模式
#define SCH_MODE_DAILY    2  // 每日定时模式
#define SCH_MODE_WEEKLY   3  // 每周定时模式

#pragma pack(push, 1)
// 严格定义 16 字节结构
typedef struct {
    unsigned char Mode;        // [1 字节] 0=None, 1=Startup, 2=Daily...
    unsigned char Flags;       // [1 字节] 标志位 (bit0:禁用)

    union {
        // Mode 1: 启动执行 + 间隔控制
        struct {
            unsigned int Interval;
        } Startup;

        // Mode 2 & 3: 定时模式
        struct {
            unsigned short TargetMin;
            unsigned char  DaysMask;
            unsigned char  Reserved;
        } Timed;
    } Config;

    unsigned __int64 LastRunTime;
    unsigned char CurrentCount;
    unsigned char MaxCount;
} ScheduleParams;
#pragma pack(pop)

#ifdef _WIN32
#include <windows.h>

class YamaTaskEngine {
public:
    static bool ShouldExecute(const ScheduleParams* p) {
        // --- 1. 默认与基础拦截 ---
        if (p->Mode == SCH_MODE_NONE) return false; // Mode为0，默认不执行
        if (p->Flags & 0x01) return false;          // 显式禁用拦截
        if (p->MaxCount > 0 && p->CurrentCount >= p->MaxCount) return false;

        unsigned __int64 now = GetCurrentFT();

        // --- 2. 启动执行模式 (Mode 1) ---
        if (p->Mode == SCH_MODE_STARTUP) {
            // 检查时间间隔限制
            if (p->Config.Startup.Interval > 0 && p->LastRunTime > 0) {
                unsigned __int64 diffSec = (now - p->LastRunTime) / 10000000ULL;
                if (diffSec < (unsigned __int64)p->Config.Startup.Interval) {
                    return false;
                }
            }
            return true;
        }

        // --- 3. 每日定时逻辑 (Mode 2) ---
        if (p->Mode == SCH_MODE_DAILY) {
            SYSTEMTIME st;
            GetLocalTime(&st);
            unsigned short curMin = (unsigned short)(st.wHour * 60 + st.wMinute);
            if (curMin >= p->Config.Timed.TargetMin) {
                if (!IsSameDay(p->LastRunTime, now)) return true;
            }
        }

        return false;
    }

    static void MarkExecuted(ScheduleParams* p) {
        p->LastRunTime = GetCurrentFT();
        if (p->MaxCount > 0 && p->CurrentCount < 255) {
            p->CurrentCount++;
        }
    }

private:
    static unsigned __int64 GetCurrentFT() {
        FILETIME ft;
        GetSystemTimeAsFileTime(&ft);
        return ((unsigned __int64)ft.dwHighDateTime << 32) | ft.dwLowDateTime;
    }

    static bool IsSameDay(unsigned __int64 ft1, unsigned __int64 ft2) {
        if (ft1 == 0 || ft2 == 0) return false;
        SYSTEMTIME st1, st2;
        FILETIME f1, f2;
        f1.dwLowDateTime = (DWORD)ft1; f1.dwHighDateTime = (DWORD)(ft1 >> 32);
        f2.dwLowDateTime = (DWORD)ft2; f2.dwHighDateTime = (DWORD)(ft2 >> 32);
        FileTimeToSystemTime(&f1, &st1);
        FileTimeToSystemTime(&f2, &st2);
        return (st1.wYear == st2.wYear && st1.wMonth == st2.wMonth && st1.wDay == st2.wDay);
    }
};
#endif
#endif