# coding: utf-8
# -------------------------------------------------------------------
# aaPanel
# -------------------------------------------------------------------
# Copyright (c) 2015-2099 aaPanel(www.aapanel.com) All rights reserved.
# -------------------------------------------------------------------
# Author: zhwwen <zhw@aapanel.com>
# -------------------------------------------------------------------
#
# ------------------------------
# OLS管理模块
# ------------------------------
import public,re,json,os

class ols:
    _main_conf_path = '/usr/local/lsws/conf/httpd_config.conf'
    _detail_conf_path = '/www/server/panel/vhost/openlitespeed/detail/{}.conf'

    def get_value(self,get):
        """
        :param get:
        get.sitename
        :return:
        """
        keys = ["enableGzipCompress","gzipCompressLevel","rollingSize","maxConnections","maxSSLConnections","connTimeout","maxKeepAliveReq"]
        conf = public.readFile(self._main_conf_path)
        if not conf:
            return {}
        data = {}
        for k in keys:
            rep = r'{}\s+(\w+)'.format(k)
            tmp = re.search(rep,conf)
            if tmp:
                data[k] = tmp.groups(1)
            else:
                data[k] = ''
        return data

    def set_value(self,get):
        """
        :param get:
        get.array
        :return:
        """
        conf = public.readFile(self._main_conf_path)
        data = json.loads(get.array)
        for k in data:
            rep = r'{}\s+(\w+)'.format(k)
            tmp = re.search(rep,conf)
            if tmp:
                conf = re.sub('{}.*'.format(k),'{} {}'.format(k,data[k]),conf)
            else:
                conf = conf + '\n{} {}'.format(k,data[k])
        public.writeFile(self._main_conf_path,conf)
        public.serviceReload()
        return public.success_v2("Setup Successfully!")

    # 获取站点静态文件缓存配置
    def get_static_cache(self,get):
        data = {}
        sitename = public.M('sites').where("id=?", (get.id,)).getField('name')
        conf = public.readFile(self._detail_conf_path.format(sitename))
        rep = r'expiresByType\s+(.*)'
        tmp = re.search(rep,conf)

        if tmp:
            tmp = tmp.groups(1)[0].split(',')
            for i in tmp:
                if 'image' in i:
                    rep = r'image\/\*=A(\d+)'
                    data['image'] = re.search(rep,conf).groups(1)[0]
                    print(data['image'])
                if 'css' in i:
                    rep = r'css=A(\d+)'
                    data['css'] = re.search(rep, conf).groups(1)[0]
                if 'javascript' in i:
                    rep = r'javascript=A(\d+)'
                    data['javascript'] = re.search(rep, conf).groups(1)[0]
                if 'font' in i:
                    rep = r'font.*=A(\d+)'
                    data['font'] = re.search(rep, conf).groups(1)[0]
        return data

    def set_static_cache(self,get):
        """
        :param get:
        get.values
        get.id
        :return:
        """
        try:
            sitename = public.M('sites').where("id=?", (get.id,)).getField('name')
            conf = public.readFile(self._detail_conf_path.format(sitename))
            print(self._detail_conf_path.format(sitename))
            values = json.loads(get.values)
            for k in values:
                rep = r'{}/[\*\w+=]=A\d+'.format(k)
                old_cache = re.search(rep,conf)
                if not old_cache:
                    continue
                old_cache = old_cache.group()
                new_cache = re.sub(r'\d+',values[k],old_cache)
                conf = conf.replace(old_cache,new_cache)
            public.writeFile(self._detail_conf_path.format(sitename),conf)
            public.serviceReload()
            return public.return_msg_gettext(True, public.lang("Setup Successfully!"))
        except Exception as e:
            return e

    def get_private_cache_conf(self,id):
        sitepath = public.M('sites').where("id=?", (id,)).getField('path')
        # 获取动态缓存配置
        file = "{}/.htaccess".format(sitepath)
        conf = ''
        if os.path.exists(file):
            conf = public.readFile(file)
        return conf,file

    def get_private_cache(self,get):
        """
        :param get:
        get.id
        :return:
        """
        conf = self.get_private_cache_conf(get.id)[0]
        rep = '#*BTLSCACHE(.|\n)+BTLSCACHE_END.*'
        tmp = re.search(rep,conf)
        data = {}
        if tmp:
            # 获取排除文件
            tmp = tmp.group()
            rep = r'#\s*excluding.*\n.*\((.*)\)'
            tmp = re.search(rep,tmp)
            if tmp:
                data['exclude_file'] = [i+'.php' for i in tmp.groups(1)[0].split('|')]
            print(data)
            # 获取缓存时间
            rep = r'max-age=(\d+)'
            tmp = re.search(rep,conf)
            if tmp:
                data['maxage'] = tmp.groups(1)[0]
        return data

    def get_private_cache_status(self,get):
        conf = self.get_private_cache_conf(get.id)[0]
        if 'BTLSCACHE_BEGIN' not in conf:
            return False
        return True

    def switch_private_cache(self,get):
        """
        get.id
        :param get:
        :return:
        """
        conf = self.get_private_cache_conf(get.id)[0]
        file = self.get_private_cache_conf(get.id)[1]
        if 'BTLSCACHE_BEGIN' not in conf:
            confstr = r"""#######################BTLSCACHE_BEGIN#######################

<IfModule LiteSpeed>
RewriteEngine on
CacheLookup on
# for those not met above condition, enable private cache.
RewriteCond %{REQUEST_METHOD} ^HEAD|GET$
## select which pages to serve from private cache
RewriteCond %{HTTP_COOKIE} !page_contain_cachetoken=yes
# with other condition
RewriteCond %{QUERY_STRING} !s=[a-fA-F0-9]{32}
# excluding certain URLs
RewriteCond %{REQUEST_URI} !/(login|register|usercp|private|profile|cron|image)\.php$
# private cache for however long set in cache policy for php pages only
RewriteRule (.*\.php)?$ - [E=Cache-Control:max-age=120]
RewriteRule (.*\.php)?$ - [E=Cache-Control:private]
</IfModule>

#######################BTLSCACHE_END#######################
"""
            with open(file, "r+") as f:
                f.seek(0)
                f.write(confstr)
                f.write(conf)
                f.close()
            public.serviceReload()
            return public.return_msg_gettext(True, public.lang("Setup successfully!"))
        else:
            bt_conf_rep = '#.*BTLSCACHE_BEGIN(.|\n)+BTLSCACHE_END#*\n'
            conf = re.sub(bt_conf_rep,'',conf)
            print(conf)
            public.writeFile(file,conf)
            public.serviceReload()
            return public.return_msg_gettext(True, public.lang("Setup successfully!"))

    def set_private_cache(self,get):
        """
        get.exclude_file
        get.max_age
        :param get:
        :return:
        """
        conf = self.get_private_cache_conf(get.id)[0]
        file_name = self.get_private_cache_conf(get.id)[1]
        bt_conf_rep = '#.*BTLSCACHE_BEGIN(.|\n)+BTLSCACHE_END#*\n'
        bt_conf = re.search(bt_conf_rep, conf)
        if bt_conf:
            bt_conf = bt_conf.group()
            exclude_files = []
            for file in get.exclude_file.split('\n'):
                exclude_files.append(file.split('.')[0])
            exclude_file = "|".join(exclude_files)
            old_exc_rep = r'RewriteCond\s+\%\{REQUEST_URI\}\s+\!/\(.*\)\\\.php\$'
            new_exc = 'RewriteCond %{REQUEST_URI} !/(' + exclude_file + r')\.php$'
            bt_conf = re.sub(old_exc_rep,new_exc,bt_conf)
            old_max_age_rep = r'max-age=\d+'
            new_max_age = 'max-age={}'.format(int(get.max_age))
            bt_conf = re.sub(old_max_age_rep,new_max_age,bt_conf)
            conf = re.sub(bt_conf_rep,bt_conf,conf)
            public.writeFile(file_name,conf)
        public.serviceReload()
        return public.return_msg_gettext(True, public.lang("Setup successfully!"))

    def _get_site_domain(self):
        site = []
        site_list = public.M("sites").field("id,name").select()
        for i in site_list:
            domain_list = public.M("domain").where("pid=?", (i["id"],)).field("name").select()
            l = []
            for domain in domain_list:
                l.append(domain["name"])
            # site[i["name"]] = l
            site.append({'sitename':i["name"],'domainlist':l})
        return site

    def _get_need_create_site(self):
        siteinfo = self._get_site_domain()
        data = []
        for s in siteinfo:
            path = '/www/server/panel/vhost/openlitespeed/{}.conf'.format(s['sitename'])
            if not os.path.exists(path):
                data.append(s)
        return data

    def _get_siteconf_info(self):
        siteinfo = self._get_need_create_site()
        phpv_reg = r'enable-php-(\w+)\.conf'
        rundir_reg = r'root\s+(.*);'
        for s in siteinfo:
            path = '/www/server/panel/vhost/nginx/{}.conf'.format(s['sitename'])
            conf = public.readFile(path)
            ap_path = '/www/server/panel/vhost/apache/{}.conf'.format(s['sitename'])
            ap_conf = public.readFile(ap_path)
            tmp = re.search(r'ServerName\s+SSL\.(.*)',ap_conf)
            s['phpv'] = re.search(phpv_reg,conf).groups(1)[0]
            s['rundir']  = re.search(rundir_reg.format(s),conf).groups(1)[0]
            s['ssl_domain'] =tmp.groups(1)[0] if tmp else None
            s['port'] = re.search(r'listen\s+(\d+);',conf).groups(1)[0]
        return siteinfo

    def _make_args(self):
        # 获取没有ols配置的域名

        # 获取php版本和执行目录和是否开启ssl
        siteinfo = self._get_siteconf_info()
        return siteinfo

    def init_ols(self,get):
        # 构造需要传入的参数
        import panelSite
        ps = panelSite.panelSite()
        siteinfo = self._make_args()
        for s in siteinfo:
            # 创建配置文件
            get.port = s['port']
            get.webname = {'domain':s['sitename'],'domainlist':s['domainlist']}
            ps.openlitespeed_add_site(get,siteinfo)
            # 处理证书
            if s['ssl_domain']:
                get.first_domain = s['ssl_domain']
                ps.set_ols_ssl(get,s['ssl_domain'])
