import configparser
import os.path


# [danmaku]
# exec
DANMAKU_FACTORY_EXEC = "DanmakuFactory"
# speed
DANMAKU_SPEED = 12
# font
DEFAULT_FONT_NAME = "Sarasa Term SC"
# resolution
VIDEO_RESOLUTION = "1280x720"
# [ffmpeg]
# exec
FFMPEG_EXEC = "ffmpeg"
# hevc
FFMPEG_USE_HEVC = False
# nvidia_gpu
FFMPEG_USE_NVIDIA_GPU = False
# intel_gpu
FFMPEG_USE_INTEL_GPU = False
# bitrate
VIDEO_BITRATE = "2.5M"
# crf
VIDEO_CRF = 28
# [video]
# title
VIDEO_TITLE = "【永恒de草薙直播录播】直播于 {}"
# [clip]
# each_sec
VIDEO_CLIP_EACH_SEC = 6000
# overflow_sec
VIDEO_CLIP_OVERFLOW_SEC = 5
# [recorder]
# bili_dir
BILILIVE_RECORDER_DIRECTORY = "./"
# xigua_dir
XIGUALIVE_RECORDER_DIRECTORY = "./"
# output_dir
VIDEO_OUTPUT_DIR = "./"


def load_config():
    if not os.path.exists("config.ini"):
        write_config()
        return False
    config = configparser.ConfigParser()
    config.read("config.ini", encoding="utf-8")
    if config.has_section("danmaku"):
        section = config['danmaku']
        global DANMAKU_FACTORY_EXEC, DANMAKU_SPEED, DEFAULT_FONT_NAME, VIDEO_RESOLUTION
        DANMAKU_FACTORY_EXEC = section.get('exec', DANMAKU_FACTORY_EXEC)
        DANMAKU_SPEED = section.getfloat('speed', DANMAKU_SPEED)
        DEFAULT_FONT_NAME = section.get('font', DEFAULT_FONT_NAME)
        VIDEO_RESOLUTION = section.get('resolution', VIDEO_RESOLUTION)
    if config.has_section("video"):
        section = config['video']
        global VIDEO_TITLE
        VIDEO_TITLE = section.get('title', VIDEO_TITLE)
    if config.has_section("clip"):
        section = config['clip']
        global VIDEO_CLIP_EACH_SEC, VIDEO_CLIP_OVERFLOW_SEC
        VIDEO_CLIP_EACH_SEC = section.getfloat('each_sec', VIDEO_CLIP_EACH_SEC)
        VIDEO_CLIP_OVERFLOW_SEC = section.getfloat('overflow_sec', VIDEO_CLIP_OVERFLOW_SEC)
    if config.has_section("ffmpeg"):
        section = config['ffmpeg']
        global FFMPEG_EXEC, FFMPEG_USE_HEVC, FFMPEG_USE_NVIDIA_GPU, FFMPEG_USE_INTEL_GPU, VIDEO_BITRATE, VIDEO_CRF
        FFMPEG_EXEC = section.get('exec', FFMPEG_EXEC)
        FFMPEG_USE_HEVC = section.getboolean('hevc', FFMPEG_USE_HEVC)
        FFMPEG_USE_NVIDIA_GPU = section.getboolean('nvidia_gpu', FFMPEG_USE_NVIDIA_GPU)
        FFMPEG_USE_INTEL_GPU = section.getboolean('intel_gpu', FFMPEG_USE_INTEL_GPU)
        VIDEO_BITRATE = section.get('bitrate', VIDEO_BITRATE)
        VIDEO_CRF = section.getfloat('crf', VIDEO_CRF)
    if config.has_section("recorder"):
        global BILILIVE_RECORDER_DIRECTORY, XIGUALIVE_RECORDER_DIRECTORY, VIDEO_OUTPUT_DIR
        section = config['recorder']
        BILILIVE_RECORDER_DIRECTORY = section.get('bili_dir', BILILIVE_RECORDER_DIRECTORY)
        XIGUALIVE_RECORDER_DIRECTORY = section.get('xigua_dir', XIGUALIVE_RECORDER_DIRECTORY)
        VIDEO_OUTPUT_DIR = section.get('output_dir', VIDEO_OUTPUT_DIR)
    return True


def get_config():
    config = {
        'danmaku': {
            'exec': DANMAKU_FACTORY_EXEC,
            'speed': DANMAKU_SPEED,
            'font': DEFAULT_FONT_NAME,
            'resolution': VIDEO_RESOLUTION,
        },
        'video': {
            'title': VIDEO_TITLE,
        },
        'clip': {
            'each_sec': VIDEO_CLIP_EACH_SEC,
            'overflow_sec': VIDEO_CLIP_OVERFLOW_SEC,
        },
        'ffmpeg': {
            'exec': FFMPEG_EXEC,
            'hevc': FFMPEG_USE_HEVC,
            'nvidia_gpu': FFMPEG_USE_NVIDIA_GPU,
            'intel_gpu': FFMPEG_USE_INTEL_GPU,
            'bitrate': VIDEO_BITRATE,
            'crf': VIDEO_CRF,
        },
        'recorder': {
            'bili_dir': BILILIVE_RECORDER_DIRECTORY,
            'xigua_dir': XIGUALIVE_RECORDER_DIRECTORY,
            'output_dir': VIDEO_OUTPUT_DIR,
        },
    }
    return config


def write_config():
    config = configparser.ConfigParser()
    _config = get_config()
    for _i in _config:
        config[_i] = _config[_i]
    with open("config.ini", "w", encoding="utf-8") as f:
        config.write(f)
    return True


load_config()