add mudbjson mgr

apiconfig.py

@@ -3,6 +3,9 @@ TRANSFER_MUL = 1.0
 NODE_ID = 1
 API_INTERFACE = 'sspanelv2' #mudbjson, sspanelv2, sspanelv3, sspanelv3ssr, muapiv2(not support)
 
+#mudb
+MUDB_FILE = 'mudb.json'
+
 # Mysql
 MYSQL_HOST = '127.0.0.1'
 MYSQL_PORT = 3306

db_transfer.py

@@ -231,7 +231,7 @@ class MuJsonTransfer(DbTransfer):
 		import json
 		rows = None
 
-		config_path = "mudb.json"
+		config_path = get_config().MUDB_FILE
 		with open(config_path, 'r+') as f:
 			rows = shell.parse_json_in_str(f.read().decode('utf8'))
 			for row in rows:
@@ -249,7 +249,7 @@ class MuJsonTransfer(DbTransfer):
 	def pull_db_all_user(self):
 		rows = None
 
-		config_path = "mudb.json"
+		config_path = get_config().MUDB_FILE
 		with open(config_path, 'r+') as f:
 			rows = shell.parse_json_in_str(f.read().decode('utf8'))
 			for row in rows:

mujson_mgr.py

@@ -0,0 +1,218 @@
+#!/usr/bin/python
+# -*- coding: UTF-8 -*-
+
+import traceback
+from shadowsocks import common, shell
+from configloader import load_config, get_config
+import getopt
+import sys
+
+class MuJsonLoader(object):
+	def __init__(self):
+		self.json = None
+
+	def load(self, path):
+		with open(path, 'r+') as f:
+			self.json = shell.parse_json_in_str(f.read().decode('utf8'))
+
+	def save(self, path):
+		import json
+		if self.json:
+			output = json.dumps(self.json, sort_keys=True, indent=4, separators=(',', ': '))
+			with open(path, 'w') as f:
+				f.write(output)
+
+class MuMgr(object):
+	def __init__(self):
+		self.config_path = get_config().MUDB_FILE
+		self.data = MuJsonLoader()
+
+	def userinfo(self, user):
+		ret = ""
+		for key in user.keys():
+			ret += '\n'
+			if key in ['transfer_enable', 'u', 'd'] :
+				val = user[key]
+				if val / 1024 < 4:
+					ret += "    %s : %s" % (key, val)
+				elif val / 1024**2 < 4:
+					val /= float(1024)
+					ret += "    %s : %s  K bytes" % (key, val)
+				elif val / 1024**3 < 4:
+					val /= float(1024**2)
+					ret += "    %s : %s  M bytes" % (key, val)
+				else:
+					val /= float(1024**3)
+					ret += "    %s : %s  G bytes" % (key, val)
+			else:
+				ret += "    %s : %s" % (key, user[key])
+		return ret
+
+	def add(self, user):
+		up = {'enable': True, 'u': 0, 'd': 0, 'passwd': "m", 'method': "aes-128-cfb",
+		'protocol': "auth_sha1_v2_compatible",
+		'obfs': "tls1.2_ticket_auth_compatible",
+		'transfer_enable': 1125899906842624}
+		up.update(user)
+
+		self.data.load(self.config_path)
+		for row in self.data.json:
+			match = False
+			if 'user' in user and row['user'] == user['user']:
+				match = True
+			if 'port' in user and row['port'] == user['port']:
+				match = True
+			if match:
+				print("user [%s] port [%s] already exist" % (row['user'], row['port']))
+				return
+		self.data.json.append(up)
+		print("### add user info %s" % self.userinfo(up))
+		self.data.save(self.config_path)
+
+	def edit(self, user):
+		self.data.load(self.config_path)
+		for row in self.data.json:
+			match = True
+			if 'user' in user and row['user'] != user['user']:
+				match = False
+			if 'port' in user and row['port'] != user['port']:
+				match = False
+			if match:
+				print("edit user [%s]" % (row['user'],))
+				row.update(user)
+				print("### new user info %s" % self.userinfo(row))
+				break
+		self.data.save(self.config_path)
+
+	def delete(self, user):
+		self.data.load(self.config_path)
+		index = 0
+		for row in self.data.json:
+			match = True
+			if 'user' in user and row['user'] != user['user']:
+				match = False
+			if 'port' in user and row['port'] != user['port']:
+				match = False
+			if match:
+				print("delete user [%s]" % row['user'])
+				del self.data.json[index]
+				break
+			index += 1
+		self.data.save(self.config_path)
+
+	def clear_ud(self, user):
+		up = {'u': 0, 'd': 0}
+		self.data.load(self.config_path)
+		for row in self.data.json:
+			match = True
+			if 'user' in user and row['user'] != user['user']:
+				match = False
+			if 'port' in user and row['port'] != user['port']:
+				match = False
+			if match:
+				print("clear user %s" % self.userinfo(row['user']))
+				row.update(up)
+		self.data.save(self.config_path)
+
+	def list_user(self, user):
+		self.data.load(self.config_path)
+		for row in self.data.json:
+			match = True
+			if 'user' in user and row['user'] != user['user']:
+				match = False
+			if 'port' in user and row['port'] != user['port']:
+				match = False
+			if match:
+				print("### user [%s] info %s" % (row['user'], self.userinfo(row)))
+
+def print_server_help():
+    print('''usage: python mujson_manage.py -a|-d|-e|-c|-l [OPTION]...
+
+Actions:
+  -a ADD                 add/edit a user
+  -d DELETE              delete a user
+  -e EDIT                edit a user
+  -c CLEAR               set u/d to zero
+  -l LIST                display a user infomation or all users infomation
+
+Options:
+  -u USER                the user name
+  -p PORT                server port
+  -k PASSWORD            password
+  -m METHOD              encryption method, default: aes-128-cfb
+  -O PROTOCOL            protocol plugin, default: auth_sha1_v2_compatible
+  -o OBFS                obfs plugin, default: tls1.2_ticket_auth_compatible
+  -t TRANSFER            max transfer for G bytes, default: 1048576, can be float point number
+
+General options:
+  -h, --help             show this help message and exit
+''')
+
+def main():
+	shortopts = 'hadeclu:p:k:O:o:m:t'
+	longopts = ['help']
+	action = None
+	user = {}
+	try:
+		optlist, args = getopt.getopt(sys.argv[1:], shortopts, longopts)
+		for key, value in optlist:
+			if key == '-a':
+				action = 1
+			elif key == '-d':
+				action = 2
+			elif key == '-e':
+				action = 3
+			elif key == '-l':
+				action = 4
+			elif key == '-c':
+				action = 0
+			elif key == '-u':
+				user['user'] = value
+			elif key == '-p':
+				user['port'] = int(value)
+			elif key == '-k':
+				user['passwd'] = value
+			elif key == '-o':
+				user['obfs'] = value
+			elif key == '-O':
+				user['protocal'] = value
+			elif key == '-m':
+				user['method'] = value
+			elif key == '-t':
+				val = float(value)
+				try:
+					val = int(value)
+				except:
+					pass
+				user['transfer_enable'] = val * (1024 ** 3)
+			elif key in ('-h', '--help'):
+				print_server_help()
+				sys.exit(0)
+	except getopt.GetoptError as e:
+		print(e)
+		sys.exit(2)
+
+	manage = MuMgr()
+	if action == 0:
+		manage.clear_ud(user)
+	elif action == 1:
+		if 'user' in user and 'port' in user:
+			manage.add(user)
+		else:
+			print("You have to set the user name and port with -u/-p")
+	elif action == 2:
+		if 'user' in user or 'port' in user:
+			manage.delete(user)
+		else:
+			print("You have to set the user name or port with -u/-p")
+	elif action == 3:
+		if 'user' in user or 'port' in user:
+			manage.edit(user)
+		else:
+			print("You have to set the user name or port with -u/-p")
+	elif action == 4:
+		manage.list_user(user)
+
+if __name__ == '__main__':
+	main()
+