secubox-openwrt/package/secubox/luci-app-mitmproxy/htdocs/luci-static/resources/mitmproxy/api.js

'use strict';
'require rpc';

var callMitmproxy = rpc.declare({
	object: 'luci.mitmproxy',
	method: 'get_status'
});

var callGetConfig = rpc.declare({
	object: 'luci.mitmproxy',
	method: 'get_config'
});

var callGetStats = rpc.declare({
	object: 'luci.mitmproxy',
	method: 'get_stats'
});

var callGetRequests = rpc.declare({
	object: 'luci.mitmproxy',
	method: 'get_requests',
	params: ['limit']
});

var callGetTopHosts = rpc.declare({
	object: 'luci.mitmproxy',
	method: 'get_top_hosts',
	params: ['limit']
});

var callGetCaInfo = rpc.declare({
	object: 'luci.mitmproxy',
	method: 'get_ca_info'
});

var callServiceStart = rpc.declare({
	object: 'luci.mitmproxy',
	method: 'service_start'
});

var callServiceStop = rpc.declare({
	object: 'luci.mitmproxy',
	method: 'service_stop'
});

var callServiceRestart = rpc.declare({
	object: 'luci.mitmproxy',
	method: 'service_restart'
});

var callSetConfig = rpc.declare({
	object: 'luci.mitmproxy',
	method: 'set_config',
	params: ['key', 'value']
});

var callClearData = rpc.declare({
	object: 'luci.mitmproxy',
	method: 'clear_data'
});

return {
	getStatus: function() {
		return callMitmproxy().catch(function() {
			return { running: false, enabled: false };
		});
	},

	getConfig: function() {
		return callGetConfig().catch(function() {
			return {};
		});
	},

	getStats: function() {
		return callGetStats().catch(function() {
			return { total_requests: 0, unique_hosts: 0, flow_file_size: 0 };
		});
	},

	getRequests: function(limit) {
		return callGetRequests(limit || 50).catch(function() {
			return { requests: [] };
		});
	},

	getTopHosts: function(limit) {
		return callGetTopHosts(limit || 20).catch(function() {
			return { hosts: [] };
		});
	},

	getCaInfo: function() {
		return callGetCaInfo().catch(function() {
			return { installed: false };
		});
	},

	serviceStart: function() {
		return callServiceStart();
	},

	serviceStop: function() {
		return callServiceStop();
	},

	serviceRestart: function() {
		return callServiceRestart();
	},

	setConfig: function(key, value) {
		return callSetConfig(key, value);
	},

	clearData: function() {
		return callClearData();
	},

	getAllData: function() {
		return Promise.all([
			this.getStatus(),
			this.getConfig(),
			this.getStats(),
			this.getTopHosts(10),
			this.getCaInfo()
		]).then(function(results) {
			return {
				status: results[0],
				config: results[1],
				stats: results[2],
				topHosts: results[3],
				caInfo: results[4]
			};
		});
	},

	formatBytes: function(bytes) {
		if (!bytes || bytes === 0) return '0 B';
		var k = 1024;
		var sizes = ['B', 'KB', 'MB', 'GB', 'TB'];
		var i = Math.floor(Math.log(bytes) / Math.log(k));
		return parseFloat((bytes / Math.pow(k, i)).toFixed(2)) + ' ' + sizes[i];
	},

	formatNumber: function(num) {
		if (!num) return '0';
		if (num >= 1000000) return (num / 1000000).toFixed(1) + 'M';
		if (num >= 1000) return (num / 1000).toFixed(1) + 'K';
		return num.toString();
	}
};