<PackageReference Include="Microsoft.Agents.AI.OpenAI" Version="1.1.0" />

public enum EmployeeRole
{
    Employee,
    Manager,
    HRAdmin
}

public static class UserRoleHelper
{
    public static bool IsSkillVisibleTo(AgentSkill skill, EmployeeRole role)
    {
        var name = skill.Frontmatter.Name;
        // 管理技能仅经理和HR可见
        if (name.StartsWith("manager-") && role == EmployeeRole.Employee) 
            return false;
        // HR管理技能仅HR可见
        if (name.StartsWith("hr-admin-") && role != EmployeeRole.HRAdmin) 
            return false;
        return true;
    }
}

public sealed class SimulatedRemoteApiSkillsSource : AgentSkillsSource
{
    private readonly string _apiEndpoint;
    public SimulatedRemoteApiSkillsSource(string apiEndpoint)
    {
        _apiEndpoint = apiEndpoint;
    }
    public override async Task<IList<AgentSkill>> GetSkillsAsync(CancellationToken cancellationToken = default)
    {
        Console.WriteLine($"📡 [RemoteApiSource] 从 {_apiEndpoint} 拉取技能列表...");
        await Task.Delay(500, cancellationToken); // 模拟网络延迟
        var entries = GetMockGlobalSkills();
        var skills = new List<AgentSkill>();
        foreach (var entry in entries)
        {
            try
            {
                var skill = new AgentInlineSkill(entry.Name, entry.Description, entry.Instructions);
                skills.Add(skill);
            }
            catch (ArgumentException ex)
            {
                Console.WriteLine($"⚠️ [RemoteApiSource] 跳过非法技能 '{entry.Name}': {ex.Message[..Math.Min(60, ex.Message.Length)]}");
            }
        }
        Console.WriteLine($"✅ [RemoteApiSource] 已成功加载 {skills.Count} 个远程技能");
        return skills;
    }
    private static IList<SkillApiEntry> GetMockGlobalSkills()
    {
        return new List<SkillApiEntry>
        {
            new("expense-report",  "（全局v1）企业费用报销政策",    "全局版报销规则：..."),
            new("hr-onboarding",   "（全局）新员工入职流程",         "入职材料清单：..."),
            new("leave-policy",    "（全局）请假制度和申请流程",      "年假/病假/事假规则：..."),
            new("manager-review",  "（全局）绩效评估指南",           "季度评估流程：..."),
            new("hr-admin-audit",  "（全局）HR 审计和合规",          "合规审查清单：..."),
        };
    }
}

public sealed record SkillApiEntry(
  string Name, 
  string Description, 
  string Instructions, 
  string[]? Tags = null);

public sealed class CachingSkillsSource : AgentSkillsSource
{
    private readonly AgentSkillsSource _innerSource;
    private readonly TimeSpan _ttl;
    private IList<AgentSkill>? _cache;
    private DateTime _cacheExpiresAt = DateTime.MinValue;
    private readonly SemaphoreSlim _lock = new(1, 1);
    
    public CachingSkillsSource(AgentSkillsSource innerSource, TimeSpan ttl)
    {
        _innerSource = innerSource;
        _ttl = ttl;
    }
    
    public override async Task<IList<AgentSkill>> GetSkillsAsync(CancellationToken cancellationToken = default)
    {
        if (_cache != null && DateTime.UtcNow < _cacheExpiresAt)
        {
            Console.WriteLine($"⚡ [CachingSource] 命中缓存（过期时间: {_cacheExpiresAt.ToLocalTime():HH:mm:ss}）");
            return _cache;
        }
        await _lock.WaitAsync(cancellationToken);
        try
        {
            // 双重检查锁（避免并发刷新）
            if (_cache != null && DateTime.UtcNow < _cacheExpiresAt)
                return _cache;
            Console.WriteLine("🔄 [CachingSource] 缓存未命中，从内层 Source 刷新...");
            _cache = await _innerSource.GetSkillsAsync(cancellationToken);
            _cacheExpiresAt = DateTime.UtcNow.Add(_ttl);
            Console.WriteLine($"✅ [CachingSource] 缓存已更新，{_cache.Count} 个技能，有效至 {_cacheExpiresAt.ToLocalTime():HH:mm:ss}");
            return _cache;
        }
        finally
        {
            _lock.Release();
        }
    }
    
    public void InvalidateCache() { _cache = null; _cacheExpiresAt = DateTime.MinValue; }
    public bool IsCacheValid => _cache != null && DateTime.UtcNow < _cacheExpiresAt;
}

public class SimulatedLocalApiSkillsFactory
{
    public static async Task<IList<AgentInlineSkill>> GetSkillsAsync(CancellationToken cancellationToken = default)
    {
        Console.WriteLine("🏠 [LocalApiFactory] 正在从本地获取技能列表...");

        var localSkills = new List<AgentInlineSkill>
        {
            // 覆盖全局 expense-report，使用 Contoso 定制规则
            new AgentInlineSkill("expense-report", "（Contoso定制v2）企业费用报销政策",
                """
                # Contoso 定制报销规则（2025版）
                - 差旅费上限提升至 8000 元/次
                - 新增"远程协作设备补贴"类目，≤3000元免审批
                - 年末报销截止日期：12月25日
                """),
            // 新增：Contoso 特有的技能
            new AgentInlineSkill("contoso-benefits", "Contoso 员工福利计划详情", "福利：弹性办公、年度体检、学习补贴..."),
        };
        Console.WriteLine($"✅ [LocalApiFactory] 已成功加载 {localSkills.Count} 个本地技能");
        return localSkills;
    }
}

// ── Source 1：本地定制技能（覆盖全局版，注册顺序靠前 → 优先）──
var localCustomSkills = await SimulatedLocalApiSkillsFactory.GetSkillsAsync();
// ── Source 2：模拟全局技能库（Remote API，通用政策）──
var globalSource = new SimulatedRemoteApiSkillsSource("https://global-skills.contoso.com/api");
// ── Source 3：带缓存的远程 Source（生产环境推荐）──
var cachedGlobalSource = new CachingSkillsSource(globalSource, TimeSpan.FromMinutes(60));

AgentSkillsProvider BuildProviderForRole(EmployeeRole role)
{
    Console.WriteLine($"\n🔨 开始构建 {role} 角色的 Provider...");
    return new AgentSkillsProviderBuilder()
        // 本地定制优先（先注册 → first-wins）
        .UseSkills(localCustomSkills)
        // 全局技能库（带缓存）
        .UseSource(cachedGlobalSource)
        // 角色感知过滤
        .UseFilter(s => UserRoleHelper.IsSkillVisibleTo(s, role))
        // 自定义 Prompt：企业内部语气
        .UsePromptTemplate("""
            你是 Contoso 集团的企业服务助手。
            ## 你掌握的企业知识库
            {skills}
            ## 工作原则
            遇到政策性问题，**先加载该技能的详细指引**，再作答。请确保所有建议符合 Contoso 最新官方规定。
            {resource_instructions}
            {script_instructions}
            """)
        .Build();
}

// 验证三个角色看到的技能集合
foreach (var role in Enum.GetValues<EmployeeRole>())
{
    var provider = BuildProviderForRole(role);
    var srcField = typeof(AgentSkillsProvider).GetField("_source", BindingFlags.Instance | BindingFlags.NonPublic);
    var src = (AgentSkillsSource?)srcField?.GetValue(provider);
    var skills = src is null ? 
        new List<AgentSkill>() : (await src.GetSkillsAsync()).ToList();
    Console.WriteLine("\n━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━");
    Console.WriteLine($"👤 [{role}] 可见技能（{skills.Count} 个）：");
    foreach (var s in skills)
    {
        string origin = s.Frontmatter.Description.StartsWith("（Contoso") ? "（本地定制）" :
                        s.Frontmatter.Description.StartsWith("（全局") ? "（全局库）" : "（其他）";
        Console.WriteLine($"    • {s.Frontmatter.Name} {origin}");
    }
}