#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import translator
import logging
import os
import json

COMMOND_GEN_EXCEL = 'gen_excel'
COMMOND_GEN_XML = 'gen_xml'
COMMOND_TRANSLATE = 'translate'
COMMOND_STATUS = 'status'
COMMOND_HELP ='help'
COMMOND_EXIT = 'exit'

API_CONFIG_FILE_PAHT = 'config.json'

def run_commond():
    '''用于在命令行窗口中执行的代码块'''
    config_logging()
    show_msg()    
    
    while True:
        msg = input('>>> ')
        cmd = get_commond(msg)

        if cmd == COMMOND_EXIT:
            break
        if cmd == COMMOND_HELP:
            on_help()
        elif cmd == COMMOND_GEN_EXCEL:
            on_translate(msg)
        elif cmd == COMMOND_GEN_XML:
            on_generate(msg)
        elif cmd == COMMOND_STATUS:
            on_status(msg)
        elif cmd == COMMOND_TRANSLATE:
            on_auto(msg)
        else:
            print('Invalid commond')        
    print('程序已退出')
    input()

def show_msg():
    '''输出程序的提示信息'''
    print('\n欢迎使用 Android 字符串翻译小工具\n\n'
        + '用法示例：\n'
        + '    1. 获取帮助请使用命令：' + COMMOND_HELP + '\n'
        + '    2. 根据当前路径下的 strings.xml 生成用于翻译的 Excel，并且指定翻译的语言: \n' 
        + '       ' + COMMOND_GEN_EXCEL +  ' zh-rCN,zh-rTW,jp,fr\n' 
        + '    3. 根据当前路径下的 Translate.xls 生成各种需要翻译的 strings.xml：' + COMMOND_GEN_XML + '\n'
        + '    4. 检查翻译进度：' + COMMOND_STATUS + '\n'
        + '    5. 自动翻译，以10条记录为一个单位：' + COMMOND_TRANSLATE + '\n'
        + '    6. 结束程序使用命令：' + COMMOND_EXIT + '\n\n'
        + '项目地址：https://github.com/Shouheng88/Android-translator\n'
        + '作者：WngShhng\n')

def on_help():
    '''输入帮助'''
    show_msg()

def config_logging():
    '''配置日志'''
    LOG_FORMAT = "%(asctime)s - %(levelname)s - %(message)s"
    DATE_FORMAT = "%m/%d/%Y %H:%M:%S %p"
    logging.basicConfig(filename='translator.log', filemode='a', level=logging.DEBUG, format=LOG_FORMAT, datefmt=DATE_FORMAT)
    logging.FileHandler(filename='translator.log', encoding='utf-8')

def get_commond(msg):
    '''获取输入的命令'''
    msg = msg.strip()
    logging.debug('inputed mssage ' + msg)
    arr = msg.split(' ')
    if len(arr) == 0:
        logging.debug('no commond')
        return 'invalid commond'
    return arr[0].strip()

def get_params(cmd, msg):
    '''获取输入的命令的参数'''
    msg = msg.strip()
    logging.debug(msg)
    
    index = msg.find(cmd)
    if index == -1:
        return []
    
    start = len(cmd)+index
    return msg[start:].strip().split(' ')

def on_translate(msg):
    '''翻译'''
    logging.debug('========== ' + COMMOND_GEN_EXCEL)
    if not check_excel_exists():
        return
    if not check_strings_exists():
        return
    params = get_params(COMMOND_GEN_EXCEL, msg)
    logging.debug(params)
    if len(params) == 0:
        print('Invliad params ')
        return
    languages = params[0].split(',')
    translator.create_translate_resources(languages)

def check_excel_exists():
    '''创建翻译的 Excel 之前检验该文件是否已经存在，防止覆盖导致用户数据丢失'''
    if os.path.exists(translator.EXCEL_FILE_NAME):
        msg = input('文件 ' + translator.EXCEL_FILE_NAME + ' 已存在，继续操作将会覆盖之前的文件，是否继续操作？(Y/N)\n')
        cmd = msg.strip().lower()
        if cmd == 'y' or cmd == 'yes':
            return True
        else:
            return False
    return True

def check_strings_exists():
    '''创建翻译的 Excel 之前检验该 strings.xml 是否存在，如果不存在则无法创建Excel'''
    if not os.path.exists(translator.STRINGS_XML_FILE_NAME):
        print('文件 ' + translator.STRINGS_XML_FILE_NAME + ' 不存在，添加该文件到当前目录之后再尝试\n')
        return False
    return True

def on_generate(msg):
    '''根据翻译的结果生成资源'''
    logging.debug('========== ' + COMMOND_GEN_XML)
    if not check_excel_for_generate():
        return
    translator.create_translated_resources()

def check_excel_for_generate():
    '''在生成各国的翻译的字符串资源之前检验 Excel 文件是否存在'''
    if not os.path.exists(translator.EXCEL_FILE_NAME):
        print('文件 ' + translator.EXCEL_FILE_NAME + ' 不存在，添加该文件到当前目录之后再尝试\n')
        return False
    return True

def on_status(msg):
    '''检查当前的翻译状态'''
    logging.debug('========== ' + COMMOND_STATUS)
    if not check_excel_for_generate():
        return
    result = translator.get_translate_status()
    for k,v in result.items():
        print(k + (' : %.2f'%(v)))

def on_auto(msg):
    '''自动翻译'''
    logging.debug('========== ' + COMMOND_TRANSLATE)
    if not check_excel_for_generate():
        return
    if not get_api_config():
        return
    translator.auto_translate()

def get_api_config():
    '''从文件中读取第三方翻译api的配置信息'''
    logging.debug('========= get api config')
    if not os.path.exists(API_CONFIG_FILE_PAHT):
        print('文件 ' + API_CONFIG_FILE_PAHT + ' 不存在\n')
        return False
    with open(API_CONFIG_FILE_PAHT, 'r') as f:
        ret = json.load(f)
        translator.config_baidu_api(ret['app_id'], ret['app_secret'], ret['mappings'])
    logging.debug(translator.config.APP_ID)
    logging.debug(translator.config.APP_SECRET)
    logging.debug(translator.config.MAPPING)
    return True

# 运行程序
run_commond()