'''Wizard Kit: Settings - UFD'''
# pylint: disable=C0326,E0611
# vim: sts=2 sw=2 ts=2

from collections import OrderedDict
from settings.main import KIT_NAME_FULL,KIT_NAME_SHORT

# General
DOCSTRING = '''WizardKit: Build UFD

Usage:
  build-ufd [options] --ufd-device PATH --linux PATH
            [--linux-minimal PATH]
            [--main-kit PATH]
            [--winpe PATH]
            [--extra-dir PATH]
  build-ufd (-h | --help)

Options:
  -e PATH, --extra-dir PATH
  -g PATH, --linux-dgpu PATH
  -k PATH, --main-kit PATH
  -l PATH, --linux PATH
  -m PATH, --linux-minimal PATH
  -u PATH, --ufd-device PATH
  -w PATH, --winpe PATH

  -d --debug            Enable debug mode
  -h --help             Show this page
  -v --verbose          Enable verbose mode
  -M --use-mbr          Use real MBR instead of GPT w/ Protective MBR
  -F --force            Bypass all confirmation messages. USE WITH EXTREME CAUTION!
  -U --update           Don't format device, just update
'''
ISO_LABEL = '{}_LINUX'.format(KIT_NAME_SHORT)
UFD_LABEL = '{}_UFD'.format(KIT_NAME_SHORT)
UFD_SOURCES = OrderedDict({
  'Linux':            {'Arg': '--linux',          'Type': 'ISO'},
  'Linux (dGPU)':     {'Arg': '--linux-dgpu',     'Type': 'ISO'},
  'Linux (Minimal)':  {'Arg': '--linux-minimal',  'Type': 'ISO'},
  'WinPE':            {'Arg': '--winpe',          'Type': 'ISO'},
  'Main Kit':         {'Arg': '--main-kit',       'Type': 'KIT'},
  'Extra Dir':        {'Arg': '--extra-dir',      'Type': 'DIR'},
  })

# Definitions: Boot entries
BOOT_ENTRIES = {
  # Path to check:      Comment to remove
  '/arch_minimal':      'UFD-MINIMAL',
  '/dgpu':              'UFD-DGPU',
  '/sources/boot.wim':  'UFD-WINPE',
  }
BOOT_FILES = {
  # Directory:            extension
  '/arch/boot/syslinux':  'cfg',
  '/EFI/boot':            'conf',
  }

# Definitions: Sources and Destinations
## NOTES: Paths are relative to the root of the ISO/UFD
##        Sources use rsync's trailing slash syntax
ITEMS = {
  'Extra Dir': (
    ('/',                         '/'),
    ),
  'Linux': (
    ('/arch',                     '/'),
    ('/isolinux',                 '/'),
    ('/EFI/boot',                 '/EFI/'),
    ('/EFI/memtest86',            '/EFI/'),
    ),
  'Linux (dGPU)': (
    ('/arch/boot/archiso.img',    '/dgpu/'),
    ('/arch/boot/vmlinuz',        '/dgpu/'),
    ('/arch/pkglist.x86_64.txt',  '/dgpu/'),
    ('/arch/x86_64',              '/dgpu/'),
    ),
  'Linux (Minimal)': (
    ('/arch/boot/archiso.img',    '/arch_minimal/'),
    ('/arch/boot/vmlinuz',        '/arch_minimal/'),
    ('/arch/pkglist.x86_64.txt',  '/arch_minimal/'),
    ('/arch/x86_64',              '/arch_minimal/'),
    ),
  'Main Kit': (
    ('/',                         '/{}/'.format(KIT_NAME_FULL)),
    ),
  'WinPE': (
    ('/bootmgr',                  '/'),
    ('/bootmgr.efi',              '/'),
    ('/en_us',                    '/'),
    ('/Boot/',                    '/boot/'),
    ('/EFI/Boot/',                '/EFI/Microsoft/'),
    ('/EFI/Microsoft/',           '/EFI/Microsoft/'),
    ('/Boot/BCD',                 '/sources/'),
    ('/Boot/boot.sdi',            '/sources/'),
    ('/bootmgr',                  '/sources/'),
    ('/sources/boot.wim',         '/sources/'),
    ),
  }

if __name__ == '__main__':
  print("This file is not meant to be called directly.")