//
//  KBSkinService.m
//  keyBoard
//

#import "KBSkinService.h"

#import "KBSkinManager.h"
#import "KBConfig.h"
#import "KBKeyboardPermissionManager.h"
#import "KBNetworkManager.h"
#import "KBHUD.h"

#if __has_include(<SSZipArchive/SSZipArchive.h>)
#import <SSZipArchive/SSZipArchive.h>
#endif

@implementation KBSkinService

#pragma mark - Icon short-name mapping (local default)

/// 本地维护的一份“逻辑按键标识 -> 图标短文件名”映射表。
/// - 若后端 skinJSON 未提供 key_icons，则远程 Zip 模式会回退使用本表；仅依赖 zip_url + 命名规范即可。
+ (NSDictionary<NSString *, NSString *> *)kb_defaultIconShortNames {
    static NSDictionary<NSString *, NSString *> *map;
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        map = @{
            // 字母键（大小写共用）
            @"letter_q_lower": @"key_q", @"letter_q_upper": @"key_q",
            @"letter_w_lower": @"key_w", @"letter_w_upper": @"key_w",
            @"letter_e_lower": @"key_e", @"letter_e_upper": @"key_e",
            @"letter_r_lower": @"key_r", @"letter_r_upper": @"key_r",
            @"letter_t_lower": @"key_t", @"letter_t_upper": @"key_t",
            @"letter_y_lower": @"key_y", @"letter_y_upper": @"key_y",
            @"letter_u_lower": @"key_u", @"letter_u_upper": @"key_u",
            @"letter_i_lower": @"key_i", @"letter_i_upper": @"key_i",
            @"letter_o_lower": @"key_o", @"letter_o_upper": @"key_o",
            @"letter_p_lower": @"key_p", @"letter_p_upper": @"key_p",

            @"letter_a_lower": @"key_a", @"letter_a_upper": @"key_a",
            @"letter_s_lower": @"key_s", @"letter_s_upper": @"key_s",
            @"letter_d_lower": @"key_d", @"letter_d_upper": @"key_d",
            @"letter_f_lower": @"key_f", @"letter_f_upper": @"key_f",
            @"letter_g_lower": @"key_g", @"letter_g_upper": @"key_g",
            @"letter_h_lower": @"key_h", @"letter_h_upper": @"key_h",
            @"letter_j_lower": @"key_j", @"letter_j_upper": @"key_j",
            @"letter_k_lower": @"key_k", @"letter_k_upper": @"key_k",
            @"letter_l_lower": @"key_l", @"letter_l_upper": @"key_l",

            @"letter_z_lower": @"key_z", @"letter_z_upper": @"key_z",
            @"letter_x_lower": @"key_x", @"letter_x_upper": @"key_x",
            @"letter_c_lower": @"key_c", @"letter_c_upper": @"key_c",
            @"letter_v_lower": @"key_v", @"letter_v_upper": @"key_v",
            @"letter_b_lower": @"key_b", @"letter_b_upper": @"key_b",
            @"letter_n_lower": @"key_n", @"letter_n_upper": @"key_n",
            @"letter_m_lower": @"key_m", @"letter_m_upper": @"key_m",

            // 功能键
            @"space":     @"key_space", // 空格键
            @"backspace": @"key_del",   // 删除键
            @"shift":     @"key_up",    // Shift（上箭头）
            @"mode_123":  @"key_123",   // 字母面板左下角 "123"
            @"ai":        @"key_ai",    // 自定义 AI 键
            @"return":    @"key_send"   // 发送/换行键
        };
    });
    return map;
}

+ (instancetype)shared {
    static KBSkinService *s; static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{ s = [KBSkinService new]; });
    return s;
}

- (void)applySkinWithJSON:(NSDictionary *)skinJSON
         fromViewController:(UIViewController *)presenting
                 completion:(KBSkinApplyCompletion)completion {
    if (skinJSON.count == 0) {
        if (completion) completion(NO);
        return;
    }

    // 1. 点击应用皮肤时，检查键盘启用 & 完全访问状态，并尽量给出友好提示。
    KBKeyboardPermissionManager *perm = [KBKeyboardPermissionManager shared];
    BOOL enabled = [perm isKeyboardEnabled];
    KBFARecord fa = [perm lastKnownFullAccess];
    BOOL hasFullAccess = (fa == KBFARecordGranted);

    if (!enabled || !hasFullAccess) {
        // 引导页（内部有自己的展示策略，避免过度打扰）
        [perm presentPermissionIfNeededFrom:presenting];

        // 简单提示：皮肤可以应用，但未开启完全访问时扩展无法读取 App Group 中的图片。
        [KBHUD showInfo:KBLocalized(@"皮肤已应用，键盘需开启“允许完全访问”后才能显示图片")];
    }

#if KB_SKIN_ICON_USE_REMOTE
    [self kb_applySkinUsingRemoteIcons:skinJSON completion:completion];
#else
    [self kb_applySkinUsingLocalIcons:skinJSON completion:completion];
#endif
}

#pragma mark - Internal helpers

/// 本地模式：key_icons 的 value 为 Assets 名称（主 App bundle 内），导出到 App Group 后再写入主题。
- (void)kb_applySkinUsingLocalIcons:(NSDictionary *)skin completion:(KBSkinApplyCompletion)completion {
    NSString *skinId = skin[@"id"] ?: @"local";
    NSString *name = skin[@"name"] ?: skinId;

    NSDictionary *iconNames = [skin[@"key_icons"] isKindOfClass:NSDictionary.class] ? skin[@"key_icons"] : @{};
    NSMutableDictionary<NSString *, NSString *> *iconPathMap = [NSMutableDictionary dictionary];

    NSURL *containerURL = [[NSFileManager defaultManager] containerURLForSecurityApplicationGroupIdentifier:AppGroup];
    if (containerURL && iconNames.count > 0) {
        NSString *skinsRoot = [containerURL.path stringByAppendingPathComponent:@"Skins"];
        NSString *skinRoot  = [skinsRoot stringByAppendingPathComponent:skinId];
        NSString *iconsDir  = [skinRoot stringByAppendingPathComponent:@"icons"];
        [[NSFileManager defaultManager] createDirectoryAtPath:iconsDir
                                  withIntermediateDirectories:YES
                                                   attributes:nil
                                                        error:NULL];

        [iconNames enumerateKeysAndObjectsUsingBlock:^(NSString *identifier, NSString *imageName, BOOL *stop) {
            if (![imageName isKindOfClass:NSString.class] || imageName.length == 0) return;
            UIImage *img = [UIImage imageNamed:imageName];
            if (!img) return;
            NSData *data = UIImagePNGRepresentation(img);
            if (data.length == 0) return;
            NSString *fileName = [NSString stringWithFormat:@"%@.png", identifier];
            NSString *fullPath = [iconsDir stringByAppendingPathComponent:fileName];
            if ([data writeToFile:fullPath atomically:YES]) {
                NSString *relative = [NSString stringWithFormat:@"Skins/%@/icons/%@", skinId, fileName];
                iconPathMap[identifier] = relative;
            }
        }];
    }

    NSMutableDictionary *themeJSON = [skin mutableCopy];
    if (iconPathMap.count > 0) {
        themeJSON[@"key_icons"] = iconPathMap.copy; // value 改为 App Group 相对路径
    }

    // 先应用颜色 / hidden_keys / key_icons（此时为相对路径，扩展从 App Group 读取）
    BOOL themeOK = [[KBSkinManager shared] applyThemeFromJSON:themeJSON];

    // 再处理背景图片（background_image），与原逻辑一致
    NSString *bgURL = skin[@"background_image"] ?: @"";
    if (bgURL.length == 0) {
        if (completion) completion(themeOK);
        if (themeOK) {
            [KBHUD showInfo:KBLocalized(@"已应用皮肤（无背景图）")];
        } else {
            [KBHUD showInfo:KBLocalized(@"应用皮肤失败")];
        }
        return;
    }

    [[KBNetworkManager shared] GET:bgURL parameters:nil headers:nil completion:^(id jsonOrData, NSURLResponse *response, NSError *error) {
        NSData *data = ([jsonOrData isKindOfClass:NSData.class] ? (NSData *)jsonOrData : nil);
        // 压缩尺寸，避免 Keychain 过大
        if (data && data.length > 0) {
            UIImage *img = [UIImage imageWithData:data];
            if (img) {
                CGFloat maxW = 1500.0;
                if (img.size.width > maxW) {
                    CGFloat scale = maxW / img.size.width;
                    CGSize newSize = CGSizeMake(maxW, floor(img.size.height * scale));
                    UIGraphicsBeginImageContextWithOptions(newSize, YES, 1.0);
                    [img drawInRect:CGRectMake(0, 0, newSize.width, newSize.height)];
                    UIImage *resized = UIGraphicsGetImageFromCurrentImageContext();
                    UIGraphicsEndImageContext();
                    img = resized ?: img;
                }
                data = UIImageJPEGRepresentation(img, 0.85) ?: data;
            }
        }
        dispatch_async(dispatch_get_main_queue(), ^{
            NSData *payload = data;
            if (payload.length == 0) {
                // 兜底：生成一张简单渐变图片
                CGSize size = CGSizeMake(1200, 600);
                UIGraphicsBeginImageContextWithOptions(size, YES, 1.0);
                CGContextRef ctx = UIGraphicsGetCurrentContext();
                UIColor *c1 = [UIColor colorWithRed:0.76 green:0.91 blue:0.86 alpha:1];
                UIColor *c2 = [UIColor colorWithRed:0.93 green:0.97 blue:0.91 alpha:1];
                if ([skin[@"id"] hasPrefix:@"dark"]) {
                    c1 = [UIColor colorWithRed:0.1 green:0.12 blue:0.16 alpha:1];
                    c2 = [UIColor colorWithRed:0.22 green:0.24 blue:0.28 alpha:1];
                }
                CGColorSpaceRef space = CGColorSpaceCreateDeviceRGB();
                NSArray *colors = @[(__bridge id)c1.CGColor, (__bridge id)c2.CGColor];
                CGFloat locs[] = {0,1};
                CGGradientRef grad = CGGradientCreateWithColors(space, (__bridge CFArrayRef)colors, locs);
                CGContextDrawLinearGradient(ctx, grad, CGPointZero, CGPointMake(size.width, size.height), 0);
                CGGradientRelease(grad); CGColorSpaceRelease(space);
                UIImage *img = UIGraphicsGetImageFromCurrentImageContext();
                UIGraphicsEndImageContext();
                payload = UIImageJPEGRepresentation(img, 0.9);
            }
            BOOL ok = (payload.length > 0)
                ? [[KBSkinManager shared] applyImageSkinWithData:payload skinId:skinId name:name]
                : themeOK;
            if (completion) completion(ok);
            [KBHUD showInfo:(ok ? KBLocalized(@"已应用，切到键盘查看") : KBLocalized(@"应用皮肤失败"))];
        });
    }];
}

/// 远程 Zip 模式：skinJSON 提供 zip_url，一套皮肤一个压缩包。
/// - Zip 解压路径：AppGroup/Skins/<skinId>/...
///   * 图标：icons/<shortName>.png，例如 icons/key_a.png
///   * 背景：background.png（可选）
/// - skinJSON.key_icons 的 value 填写 Zip 内的图标“短文件名”（不含路径，可不含扩展名），例如 "key_a"
///   应用时会被转换为相对 App Group 根目录的路径：Skins/<skinId>/icons/<shortName>.png
- (void)kb_applySkinUsingRemoteIcons:(NSDictionary *)skin completion:(KBSkinApplyCompletion)completion {
    NSString *skinId = skin[@"id"] ?: @"remote";
    NSString *name = skin[@"name"] ?: skinId;
    NSString *zipURL = skin[@"zip_url"] ?: @""; // 新协议：远程 Zip 包地址

    // key_icons 可选：
    // - 若后端提供 key_icons，则优先使用服务端映射；
    // - 若未提供，则回退到本地默认映射（kb_defaultIconShortNames），这样后端只需返回 id/name/zip_url。
    NSDictionary *iconShortNames = nil;
    if ([skin[@"key_icons"] isKindOfClass:NSDictionary.class]) {
        iconShortNames = skin[@"key_icons"];
    } else {
        iconShortNames = [self.class kb_defaultIconShortNames];
    }

    NSURL *containerURL = [[NSFileManager defaultManager] containerURLForSecurityApplicationGroupIdentifier:AppGroup];
    if (!containerURL) {
        if (completion) completion(NO);
        [KBHUD showInfo:KBLocalized(@"无法访问共享容器，应用皮肤失败")];
        return;
    }

    NSString *skinsRoot = [containerURL.path stringByAppendingPathComponent:@"Skins"];
    NSString *skinRoot  = [skinsRoot stringByAppendingPathComponent:skinId];
    NSString *iconsDir  = [skinRoot stringByAppendingPathComponent:@"icons"];
    [[NSFileManager defaultManager] createDirectoryAtPath:iconsDir
                              withIntermediateDirectories:YES
                                               attributes:nil
                                                    error:NULL];

    NSFileManager *fm = [NSFileManager defaultManager];
    BOOL isDir = NO;
    BOOL hasIconsDir = [fm fileExistsAtPath:iconsDir isDirectory:&isDir] && isDir;
    NSArray *contents = hasIconsDir ? [fm contentsOfDirectoryAtPath:iconsDir error:NULL] : nil;
    BOOL hasCachedAssets = (contents.count > 0);

    NSString *bgPath = [skinRoot stringByAppendingPathComponent:@"background.png"];

    dispatch_group_t group = dispatch_group_create();
    __block BOOL zipOK = YES;

#if __has_include(<SSZipArchive/SSZipArchive.h>)
    // 若本地尚未缓存该皮肤资源且提供了 zip_url，则下载并解压 Zip 包。
    if (!hasCachedAssets && zipURL.length > 0) {
        dispatch_group_enter(group);
        [[KBNetworkManager shared] GET:zipURL parameters:nil headers:nil completion:^(id jsonOrData, NSURLResponse *response, NSError *error) {
            NSData *data = ([jsonOrData isKindOfClass:NSData.class] ? (NSData *)jsonOrData : nil);
            if (error || data.length == 0) {
                zipOK = NO;
                dispatch_group_leave(group);
                return;
            }

            // 将 Zip 写入临时路径再解压
            [[NSFileManager defaultManager] createDirectoryAtPath:skinRoot
                                      withIntermediateDirectories:YES
                                                       attributes:nil
                                                            error:NULL];
            NSString *zipPath = [skinRoot stringByAppendingPathComponent:@"skin.zip"];
            if (![data writeToFile:zipPath atomically:YES]) {
                zipOK = NO;
                dispatch_group_leave(group);
                return;
            }

            NSError *unzipError = nil;
            BOOL ok = [SSZipArchive unzipFileAtPath:zipPath
                                       toDestination:skinRoot
                                           overwrite:YES
                                            password:nil
                                               error:&unzipError];
            [fm removeItemAtPath:zipPath error:nil];
            if (!ok || unzipError) {
                zipOK = NO;
            }
            dispatch_group_leave(group);
        }];
    }
#else
    zipOK = NO;
#endif

    dispatch_group_notify(group, dispatch_get_main_queue(), ^{
        // 构造 key_icons -> App Group 相对路径 映射
        NSMutableDictionary<NSString *, NSString *> *iconPathMap = [NSMutableDictionary dictionary];
        [iconShortNames enumerateKeysAndObjectsUsingBlock:^(NSString *identifier, NSString *shortName, BOOL *stop) {
            if (![shortName isKindOfClass:NSString.class] || shortName.length == 0) return;
            NSString *fileName = shortName;
            // 若未带扩展名，默认按 .png 处理
            if (fileName.pathExtension.length == 0) {
                fileName = [fileName stringByAppendingPathExtension:@"png"];
            }
            NSString *relative = [NSString stringWithFormat:@"Skins/%@/icons/%@", skinId, fileName];
            iconPathMap[identifier] = relative;
        }];

        NSMutableDictionary *themeJSON = [skin mutableCopy];
        themeJSON[@"id"] = skinId;
        if (iconPathMap.count > 0) {
            themeJSON[@"key_icons"] = iconPathMap.copy;
        }

        BOOL themeOK = [[KBSkinManager shared] applyThemeFromJSON:themeJSON];

        // 背景图优先从 Zip 解压出的 background.png 读取
        NSData *bgData = [NSData dataWithContentsOfFile:bgPath];
        BOOL ok = themeOK;
        if (bgData.length > 0) {
            ok = [[KBSkinManager shared] applyImageSkinWithData:bgData skinId:skinId name:name];
        }

        if (!zipOK && !hasCachedAssets) {
            ok = NO;
        }

        if (completion) completion(ok);
        [KBHUD showInfo:(ok ? KBLocalized(@"已应用，切到键盘查看") : KBLocalized(@"应用皮肤失败"))];
    });
}

@end