import RawAnswers, { BasicRawAnswers, HostRawAnswers } from './raw-answers';

export type RecordValue<T> = T extends Record<any, infer R> ? R : any;

export type ParsedHostAnswers = ({
    present:
        | 'not_mentioned'
        | 'not_before_making_a_choice'
        | 'mentioned_in_policy'
        | 'mentioned_in_popup';
    legal_basis_type: 'consent' | 'legitimate_interes' | 'not_mentioned';
    was_processing_necessary: 'yes' | 'no' | 'not_sure';
} & (
    | {
          consent_problems:
              | 'claims_consent_but_sends_before_consent'
              | 'claims_consent_but_there_was_no_easy_refuse';
      }
    | { consent_problems: 'none'; outside_eu: 'yes' | 'no' | 'not_sure' }
)) & {
    legitimate_interest_activity_specified: 'no' | 'precise' | 'vague';
    outside_eu: 'yes' | 'no' | 'not_sure';
    legitimate_interest_description?: string;
};

export type ParsedAnswers = BasicRawAnswers & { hosts: Record<string, ParsedHostAnswers> };

function parseHostAnswers(
    raw_answers: Record<keyof HostRawAnswers, string>
): Record<string, ParsedHostAnswers> {
    const result: Record<string, Record<string, string>> = {};
    for (const [key, value] of Object.entries(raw_answers)) {
        const [masked_host, attr] = key.split('|');
        const host = masked_host.replace(/_/g, '.');
        if (!result[host]) {
            result[host] = {} as ParsedHostAnswers;
        }
        result[host][attr] = value;
    }
    return result as Record<string, ParsedHostAnswers>;
}

export function parseAnswers({
    zaimek,
    is_incognito_different,
    policy_readable,
    popup_type,
    cookie_wall,
    passive_consent_description,
    mentions_passive_consent,
    rejection_is_hard,
    administrator_identity_available_before_choice,
    popup_action,
    ...rest
}: RawAnswers): ParsedAnswers {
    return {
        zaimek,
        is_incognito_different,
        policy_readable,
        popup_type,
        cookie_wall,
        passive_consent_description,
        mentions_passive_consent,
        rejection_is_hard,
        administrator_identity_available_before_choice,
        popup_action,
        hosts: parseHostAnswers(rest),
    } as ParsedAnswers;
}