import type { ItemModule } from '@sammo-ts/logic/items/types.js';
import type { UniqueItemPool } from './uniqueLottery.js';

const LEGACY_UNIQUE_ITEM_KEYS: Readonly<Record<ItemModule['slot'], readonly string[]>> = {
    horse: [
        'che_명마_07_백마',
        'che_명마_07_기주마',
        'che_명마_07_오환마',
        'che_명마_07_백상',
        'che_명마_08_양주마',
        'che_명마_08_흉노마',
        'che_명마_09_과하마',
        'che_명마_09_의남백마',
        'che_명마_10_대완마',
        'che_명마_10_옥추마',
        'che_명마_11_서량마',
        'che_명마_11_화종마',
        'che_명마_12_사륜거',
        'che_명마_12_옥란백용구',
        'che_명마_13_절영',
        'che_명마_13_적로',
        'che_명마_14_적란마',
        'che_명마_14_조황비전',
        'che_명마_15_한혈마',
        'che_명마_15_적토마',
    ],
    weapon: [
        'che_무기_07_동추',
        'che_무기_07_철편',
        'che_무기_07_철쇄',
        'che_무기_07_맥궁',
        'che_무기_08_유성추',
        'che_무기_08_철질여골',
        'che_무기_09_쌍철극',
        'che_무기_09_동호비궁',
        'che_무기_10_삼첨도',
        'che_무기_10_대부',
        'che_무기_11_고정도',
        'che_무기_11_이광궁',
        'che_무기_12_철척사모',
        'che_무기_12_칠성검',
        'che_무기_13_사모',
        'che_무기_13_양유기궁',
        'che_무기_14_언월도',
        'che_무기_14_방천화극',
        'che_무기_15_청홍검',
        'che_무기_15_의천검',
    ],
    book: [
        'che_서적_07_위료자',
        'che_서적_07_사마법',
        'che_서적_07_한서',
        'che_서적_07_논어',
        'che_서적_08_전론',
        'che_서적_08_사기',
        'che_서적_09_장자',
        'che_서적_09_역경',
        'che_서적_10_시경',
        'che_서적_10_구국론',
        'che_서적_11_상군서',
        'che_서적_11_춘추전',
        'che_서적_12_산해경',
        'che_서적_12_맹덕신서',
        'che_서적_13_관자',
        'che_서적_13_병법24편',
        'che_서적_14_한비자',
        'che_서적_14_오자병법',
        'che_서적_15_노자',
        'che_서적_15_손자병법',
    ],
    item: [
        'che_의술_정력견혈산',
        'che_의술_청낭서',
        'che_의술_태평청령',
        'che_의술_상한잡병론',
        'che_보물_도기',
        'che_조달_주판',
        'che_내정_납금박산로',
        'che_전략_평만지장도',
        'che_숙련_동작',
        'che_명성_구석',
        'che_척사_오악진형도',
        'che_격노_구정신단경',
        'che_징병_낙주',
        'che_저격_매화수전',
        'che_저격_비도',
        'che_위압_조목삭',
        'che_공성_묵자',
        'che_집중_전국책',
        'che_환술_논어집해',
        'che_진압_박혁론',
        'che_부적_태현청생부',
        'che_저지_삼황내문',
        'che_행동_서촉지형도',
        'che_간파_노군입산부',
        'che_불굴_상편',
        'che_약탈_옥벽',
        'che_농성_주서음부',
        'che_농성_위공자병법',
        'che_계략_육도',
        'che_계략_삼략',
        'che_상성보정_과실주',
        'che_능력치_지력_이강주',
        'che_능력치_무력_두강주',
        'che_능력치_통솔_보령압주',
        'che_훈련_철벽서',
        'che_훈련_단결도',
        'che_사기_춘화첩',
        'che_사기_초선화',
        'che_회피_태평요술',
        'che_필살_둔갑천서',
    ],
};

export const buildLegacyDefaultUniqueItemPool = (itemRegistry: Map<string, ItemModule>): UniqueItemPool => {
    const pool: UniqueItemPool = { horse: {}, weapon: {}, book: {}, item: {} };
    for (const slot of ['horse', 'weapon', 'book', 'item'] as const) {
        const count = slot === 'item' ? 1 : 2;
        for (const itemKey of LEGACY_UNIQUE_ITEM_KEYS[slot]) {
            const item = itemRegistry.get(itemKey);
            if (item?.slot === slot && !item.buyable) {
                pool[slot]![itemKey] = count;
            }
        }
    }
    return pool;
};