1 files changed, 133 insertions, 0 deletions
diff --git a/clashup b/clashup
new file mode 100755
index 000000000000..b7010f2a6094
--- /dev/null
+++ b/clashup
@@ -0,0 +1,133 @@
+#!/usr/bin/env python3
+
+import requests
+import yaml
+import json
+import os
+import shutil
+import logging
+import time
+import argparse
+import subprocess
+import hashlib
+
+logging.basicConfig(format='%(asctime)s %(levelname)s: %(message)s', level=logging.INFO)
+
+class ClashUp:
+
+    EXAMPLE_CONF = '''{
+    "http_port": 7890,
+    "socks5_port": 7891,
+    "redir_port": 7892,
+    "allow_lan": true,
+    "external_controller": "127.0.0.1:9090",
+    "subscribe_url": "",
+    "is_subscribe_banned": false,
+    "custom_rules": []
+}
+'''
+    def __init__(self):
+        self.conf_file_path = os.path.expanduser('~/.config/clash/clashup.json')
+        self.clash_conf_path = os.path.expanduser('~/.config/clash/config.yaml')
+        self.clash_conf_path_old = os.path.expanduser('~/.config/clash/config.yaml.old')
+        self.cache_file_path = os.path.expanduser('~/.cache/clashup')
+
+    def load_conf(self):
+        if not os.path.isfile(self.conf_file_path):
+            with open(self.conf_file_path, 'w') as f:
+                f.write(self.EXAMPLE_CONF)
+            raise OSError('plz edit ~/.config/clash/clashup.json')
+        with open(self.conf_file_path) as f:
+            raw_config_text = f.read()
+            raw_config = json.loads(raw_config_text)
+        if not raw_config.get('subscribe_url'):
+            raise ValueError('subscribe_url can not be empty')
+        self.config = raw_config
+        hash_item = hashlib.sha256(raw_config_text.encode())
+        self.config_hash = hash_item.hexdigest()
+
+    def download(self, use_proxy=False):
+        if use_proxy:
+            proxy = {
+                'http': 'http://127.0.0.1:{}'.format(self.config['http_port']),
+                'https': 'http://127.0.0.1:{}'.format(self.config['http_port'])
+            }
+            res = requests.get(self.config['subscribe_url'], timeout=5, proxies=proxy)
+        else:
+            res = requests.get(self.config['subscribe_url'], timeout=5)
+        res.raise_for_status()
+        raw_clash_conf = yaml.safe_load(res.text)
+        return raw_clash_conf
+    
+    def _load_conf(self, config, local_config_key, config_key):
+        if local_config_key in self.config:
+            config[config_key] = self.config[local_config_key]
+
+    def parse_config(self, config):
+        self._load_conf(config, 'http_port', 'port')
+        self._load_conf(config, 'socks5_port', 'socks-port')
+        self._load_conf(config, 'redir_port', 'redir-port')
+        self._load_conf(config, 'allow_lan', 'allow-lan')
+        self._load_conf(config, 'external_controller', 'external-controller')
+        for rule in self.config.get('custom_rules', []):
+            config['Rule'].append(rule)
+        return config
+
+    def save(self, config):
+        if os.path.isfile(self.clash_conf_path):
+            shutil.move(self.clash_conf_path, self.clash_conf_path_old)
+        with open(self.clash_conf_path, 'w') as f:
+            f.write(yaml.safe_dump(config))
+
+    def update(self, use_proxy):
+        logging.info('Update Start')
+        try:
+            raw_clash_conf = self.download(use_proxy)
+            parsed_clash_conf = self.parse_config(raw_clash_conf)
+            self.save(parsed_clash_conf)
+            logging.info('Update Finish')
+        except requests.exceptions.RequestException:
+            logging.warning('Update Fail')
+
+    def update_time_cache(self):
+        if not os.path.isfile(self.cache_file_path):
+            self._write_cache()
+            return True
+        else:
+            with open(self.cache_file_path, 'r') as f:
+                cache_text = f.read().split('-')
+            last_time = float(cache_text[1])
+            if cache_text[0] != self.config_hash or time.time() - last_time > 86400:
+                self._write_cache()
+                return True
+            else:
+                return False
+
+    def _write_cache(self):
+        with open(self.cache_file_path, 'w') as f:
+            f.write('{}-{}'.format(self.config_hash, time.time()))
+
+    def run(self):
+        parser = argparse.ArgumentParser()
+        parser.add_argument('--pre', action='store_true')
+        parser.add_argument('--post', action='store_true')
+        args = parser.parse_args()
+        self.load_conf()
+        if args.pre:
+            if self.config['is_subscribe_banned']:
+                logging.info('Subscribe is banned, pass this run')
+            else:
+                self.update(False)
+        elif args.post:
+            if self.config['is_subscribe_banned']:
+                if self.update_time_cache():
+                    self.update(True)
+                    subprocess.run(['systemctl', '--user', 'restart', 'clash'])
+                else:
+                    logging.info('config file updated in 24h, pass this run')
+            else:
+                logging.info('pass this run')
+
+
+if __name__ == '__main__':
+    ClashUp().run()