# Copyright (c) 2017 VisualDL Authors. All Rights Reserve.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# =======================================================================

import os
import mimetypes
from flask import (Response, send_from_directory)


class Template(object):
    extname = ['.html', '.js', '.css']
    mimetypes = ['text/html', 'application/javascript', 'text/css']

    defaults = {
        'PUBLIC_PATH': '/app',
        'API_TOKEN_KEY': '',
        'TELEMETRY_ID': '',
        'THEME': ''
    }

    __files = {}

    def __init__(self, path, **context):
        if not os.path.exists(path):
            raise Exception("template file does not exist.")
        self.path = path
        self.__add_mime_types()
        for root, _dirs, files in os.walk(path):
            for file in files:
                if any(file.endswith(name) for name in self.extname):
                    file_path = os.path.join(root, file)
                    rel_path = os.path.relpath(file_path, path).replace(os.path.sep, '/')
                    with open(file_path, "r", encoding="UTF-8") as f:
                        content = f.read()
                        envs = self.defaults.copy()
                        envs.update(context)
                        for key, value in envs.items():
                            content = content.replace("{{" + key + "}}", value)
                    self.__files[rel_path] = content, mimetypes.guess_type(file)[0]

    def render(self, file):
        if file in self.__files:
            return Response(response=self.__files[file][0], mimetype=self.__files[file][1])
        return send_from_directory(self.path, file)

    def __add_mime_types(self):
        for i, ext in enumerate(self.extname):
            mimetypes.add_type(self.mimetypes[i] or 'application/octet-stream', ext)