# -*- coding: utf-8 -*-
"""系统信息"""
import logging
import uuid
import time
import re
import os
import json
import hashlib

import datetime

from website import errors
from website import settings
from website.handler import APIHandler, WebHandler, authenticated, operation_log, permission
from website.util import sysinfo, rsa


class VersionHandler(APIHandler):
    @authenticated
    # @permission([100014, 100016])
    def post(self):
        self.finish()

class IdentifycodeHandler(APIHandler):
    @authenticated
    # @permission([100014, 100015])
    # @operation_log("资产管理中心", "系统激活", "查询", "查询本地识别码", "查询本地识别码")
    def post(self):
        code = sysinfo.get_identify_code()

        self.finish({"result": code})

class LicenseUploadHandler(APIHandler):
    @authenticated
    # @permission([100014, 100015])
    # @operation_log("资产管理中心", "系统激活", "导入", "上传license文件", "上传license文件")
    def post(self):
        file_metas = self.request.files.get('file', None)
        if not file_metas:
            raise errors.HTTPAPIError(errors.ERROR_BAD_REQUEST, "请上传文件")

        file = file_metas[0]
        filename = file.filename
        punctuation = """!"#$%&'()*+,/:;<=>?@[\]^`{|}~ """
        regex = re.compile('[%s]' % re.escape(punctuation))
        filename = regex.sub("", filename.replace('..', ''))
        file_size = len(file.body)

        if file_size > 10 * 1024 * 1024:
            raise errors.HTTPAPIError(errors.ERROR_METHOD_NOT_ALLOWED, 'Exceed 10M size limit')

        md5_str = hashlib.md5(file.body).hexdigest()

        filepath = settings.rsa_license_file

        try:
            body = file['body']
            public_key = rsa.load_pub_key_string(open(settings.rsa_public_file).read().strip('\n').encode('utf-8'))
            plaintext = rsa.decrypt(public_key, body)
            plaintext_json = json.loads(self.tostr(plaintext))

            syscode = plaintext_json["syscode"]
            expireat = plaintext_json["expireat"]

            current_syscode = sysinfo.get_identify_code()
            if syscode != current_syscode:
                raise errors.HTTPAPIError(errors.ERROR_BAD_REQUEST, "license激活失败，请重新激活")

            row = self.db_app.get("select id from license where syscode=%s", syscode)
            if row:
                self.db_app.update(
                    "update license set expireat=%s where syscode=%s", str(expireat), syscode
                )
            else:
                self.db_app.insert(
                    "insert into license(syscode, expireat) values(%s, %s)",
                    syscode, expireat
                )
            self.r_app.set("system:license", json.dumps({"syscode":syscode, "expireat":expireat}))

            with open(filepath, 'wb') as f:
                f.write(file['body'])

            logging.info(plaintext_json)
        except Exception as e:
            logging.info(e)
            raise errors.HTTPAPIError(errors.ERROR_BAD_REQUEST, "license激活失败，请重新激活")

        self.finish()

class ActivateInfoHandler(APIHandler):
    @authenticated
    # @permission([100014, 100015])
    # @operation_log("资产管理中心", "系统激活", "查询", "查询系统激活信息", "查询系统激活信息")
    def post(self):
        license_str = ""
        activate_at = ""
        expire_at = ""
        date_remain = 0

        row = self.db_app.get(
            "select create_time, expireat from license limit 1"
        )
        if row:
            license_str = open(settings.rsa_license_file, 'r').read()
            activate_at = str(row["create_time"])
            expire_at = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(int(row["expireat"])))
            now = datetime.datetime.now()
            delta = (datetime.datetime.fromtimestamp(int(row["expireat"])).date() - now.date()).days
            date_remain = delta if delta > 0 else 0

        data = {
            "system": settings.system_info[settings.system_type]["name"],
            "license": license_str,
            "activate_at": activate_at,
            "expire_at": expire_at,
            "date_remain": date_remain
        }

        self.finish(data)


class InfoHandler(APIHandler):
    def post(self):

        self.finish()