import configparser
import os.path


# [danmaku]
# use_danmu2ass
DANMAKU_USE_DANMU2ASS = False
# use_danmakufactory
DANMAKU_USE_DANMAKUFACTORY = True
# exec
DANMAKU_EXEC = "DanmakuFactory"
# speed
DANMAKU_SPEED = 12
# font
DANMAKU_FONT_NAME = "Sarasa Term SC"
# font_size
DANMAKU_FONT_SIZE = 40
# resolution
VIDEO_RESOLUTION = "1280x720"
# [ffmpeg]
# exec
FFMPEG_EXEC = "ffmpeg"
# [handbrake]
# exec
HANDBRAKE_EXEC = "HandBrakeCli"
# preset_file
HANDBRAKE_PRESET_FILE = "handbrake.json"
# preset
HANDBRAKE_PRESET = "NvEnc"
# encopt
HANDBRAKE_ENCOPT = ""
# [video]
# enabled
VIDEO_ENABLED = False
# title
VIDEO_TITLE = "【永恒de草薙直播录播】直播于 {}"
# desc
VIDEO_DESC = "弹幕来源：B站直播\r\n\r\n" + \
             "原主播：永恒de草薙\r\n往期节目单查询：https://comment.sc.jerryyan.top\r\n\r\n" + \
             "好多人这时候就开始问了，在哪直播呀，哎。对吧…咱来啦啊。在哔哩哔哩啊，" \
             "无论你是用网页百度搜索哔哩哔哩官网或者是用手机的APP下载一个哔哩哔哩，" \
             "啊，都能找到原主播。大概每天晚七点半左右吧…一般都是往左。然后的话呢搜索永恒de草薙就能找到他。" \
             "那么今天的话呢今天的录播也就发完了…他是本期的主播永恒。咱明天同一时间不见不散…拜拜！！"
# tid
VIDEO_TID = 17
# tags
VIDEO_TAGS = "永恒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_EXEC, DANMAKU_SPEED, DANMAKU_FONT_NAME, VIDEO_RESOLUTION, DANMAKU_FONT_SIZE, \
            DANMAKU_USE_DANMU2ASS, DANMAKU_USE_DANMAKUFACTORY
        DANMAKU_USE_DANMU2ASS = section.getboolean('use_danmu2ass', DANMAKU_USE_DANMU2ASS)
        DANMAKU_USE_DANMAKUFACTORY = section.getboolean('use_danmakufactory', DANMAKU_USE_DANMAKUFACTORY)
        DANMAKU_EXEC = section.get('exec', DANMAKU_EXEC)
        DANMAKU_SPEED = section.getfloat('speed', DANMAKU_SPEED)
        DANMAKU_FONT_NAME = section.get('font', DANMAKU_FONT_NAME)
        DANMAKU_FONT_SIZE = section.getint('font_size', DANMAKU_FONT_SIZE)
        VIDEO_RESOLUTION = section.get('resolution', VIDEO_RESOLUTION)
    if config.has_section("video"):
        section = config['video']
        global VIDEO_ENABLED, VIDEO_TITLE, VIDEO_DESC, VIDEO_TID, VIDEO_TAGS
        VIDEO_ENABLED = section.getboolean('enabled', VIDEO_ENABLED)
        VIDEO_TITLE = section.get('title', VIDEO_TITLE)
        VIDEO_DESC = section.get('desc', VIDEO_DESC)
        VIDEO_TID = section.getint('tid', VIDEO_TID)
        VIDEO_TAGS = section.get('tags', VIDEO_TAGS)
    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_EXEC = section.get('exec', FFMPEG_EXEC)
    if config.has_section("handbrake"):
        section = config['handbrake']
        global HANDBRAKE_EXEC, HANDBRAKE_PRESET_FILE, HANDBRAKE_PRESET, HANDBRAKE_ENCOPT
        HANDBRAKE_EXEC = section.get('exec', HANDBRAKE_EXEC)
        HANDBRAKE_PRESET_FILE = section.get('preset_file', HANDBRAKE_PRESET_FILE)
        HANDBRAKE_PRESET = section.get('preset', HANDBRAKE_PRESET)
        HANDBRAKE_ENCOPT = section.get('encopt', HANDBRAKE_ENCOPT)
    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_EXEC,
            'use_danmu2ass': DANMAKU_USE_DANMU2ASS,
            'use_danmakufactory': DANMAKU_USE_DANMAKUFACTORY,
            'speed': DANMAKU_SPEED,
            'font': DANMAKU_FONT_NAME,
            'font_size': DANMAKU_FONT_SIZE,
            'resolution': VIDEO_RESOLUTION,
        },
        'video': {
            'enabled': VIDEO_ENABLED,
            'title': VIDEO_TITLE,
            'desc': VIDEO_DESC,
            'tid': VIDEO_TID,
            'tags': VIDEO_TAGS,
        },
        'clip': {
            'each_sec': VIDEO_CLIP_EACH_SEC,
            'overflow_sec': VIDEO_CLIP_OVERFLOW_SEC,
        },
        'ffmpeg': {
            'exec': FFMPEG_EXEC,
        },
        'handbrake': {
            'exec': HANDBRAKE_EXEC,
            'preset_file': HANDBRAKE_PRESET_FILE,
            'preset': HANDBRAKE_PRESET,
            'encopt': HANDBRAKE_ENCOPT,
        },
        '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()