81 lines
3 KiB
JavaScript
81 lines
3 KiB
JavaScript
|
// 铸渊专线订阅服务 · 流量计数 + subscription-userinfo 头
|
|||
|
|
// 部署：BS-GZ-006 · 2026-08-23 · 配额 1000GiB/自然月重置
|
|||
|
|
// 计数源：iptables 计数规则（relay 端口 8443-8446 进出字节）
|
|||
|
|
const http = require('http');
|
|||
|
|
const fs = require('fs');
|
|||
|
|
const path = require('path');
|
|||
|
|
const { execFile } = require('child_process');
|
|||
|
|
|
|||
|
|
const PORT = 3944;
|
|||
|
|
const DATA_DIR = '/var/www/guanghulab-front-door/zy-sub';
|
|||
|
|
const STATE_FILE = '/opt/zy-sub-service/state.json';
|
|||
|
|
const TOTAL_QUOTA = 1000 * 1024 * 1024 * 1024;
|
|||
|
|
|
|||
|
|
let state = { month: '', upload: 0, download: 0, lastIn: 0, lastOut: 0 };
|
|||
|
|
try { Object.assign(state, JSON.parse(fs.readFileSync(STATE_FILE, 'utf8'))); } catch {}
|
|||
|
|
|
|||
|
|
function monthKey(d = new Date()) {
|
|||
|
|
return d.getFullYear() + '-' + String(d.getMonth() + 1).padStart(2, '0');
|
|||
|
|
}
|
|||
|
|
function save() {
|
|||
|
|
try { fs.writeFileSync(STATE_FILE, JSON.stringify(state)); }
|
|||
|
|
catch (e) { console.error('save fail:', e.message); }
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function readCounter(chain, tag, cb) {
|
|||
|
|
execFile('iptables-save', ['-c'], (err, out) => {
|
|||
|
|
let bytes = 0;
|
|||
|
|
if (!err) {
|
|||
|
|
for (const line of out.split('\n')) {
|
|||
|
|
if (line.includes('zy-sub-acct-' + tag) && line.includes('-A ' + chain + ' ')) {
|
|||
|
|
const m = line.match(/^\[(\d+):(\d+)\]/);
|
|||
|
|
if (m) bytes += parseInt(m[2], 10);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
cb(bytes);
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function poll() {
|
|||
|
|
readCounter('INPUT', 'in', (inB) => {
|
|||
|
|
readCounter('OUTPUT', 'out', (outB) => {
|
|||
|
|
const mk = monthKey();
|
|||
|
|
if (state.month !== mk) { state.month = mk; state.upload = 0; state.download = 0; }
|
|||
|
|
if (inB < state.lastIn) state.lastIn = inB; // 计数器重启归零时对齐
|
|||
|
|
if (outB < state.lastOut) state.lastOut = outB;
|
|||
|
|
state.upload += inB - state.lastIn;
|
|||
|
|
state.download += outB - state.lastOut;
|
|||
|
|
state.lastIn = inB; state.lastOut = outB;
|
|||
|
|
save();
|
|||
|
|
});
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
setInterval(poll, 60 * 1000);
|
|||
|
|
poll();
|
|||
|
|
|
|||
|
|
const used = () => state.upload + state.download;
|
|||
|
|
const userinfo = () =>
|
|||
|
|
`upload=${state.upload}; download=${state.download}; total=${TOTAL_QUOTA}; expire=0`;
|
|||
|
|
|
|||
|
|
http.createServer((req, res) => {
|
|||
|
|
const url = req.url.split('?')[0];
|
|||
|
|
const m = url.match(/^\/zy-sub\/([0-9a-f]{16,64})\/(clash\.yaml|shadowrocket\.txt|status)$/);
|
|||
|
|
if (!m) { res.writeHead(404); return res.end('not found'); }
|
|||
|
|
if (m[2] === 'status') {
|
|||
|
|
res.writeHead(200, { 'Content-Type': 'application/json' });
|
|||
|
|
return res.end(JSON.stringify({
|
|||
|
|
month: state.month, upload: state.upload, download: state.download,
|
|||
|
|
total: TOTAL_QUOTA, used: used(), remaining: Math.max(0, TOTAL_QUOTA - used())
|
|||
|
|
}, null, 2));
|
|||
|
|
}
|
|||
|
|
fs.readFile(path.join(DATA_DIR, m[1], m[2]), (err, data) => {
|
|||
|
|
if (err) { res.writeHead(404); return res.end('not found'); }
|
|||
|
|
res.writeHead(200, {
|
|||
|
|
'Content-Type': m[2].endsWith('.yaml') ? 'text/yaml; charset=utf-8' : 'text/plain; charset=utf-8',
|
|||
|
|
'subscription-userinfo': userinfo(),
|
|||
|
|
'profile-update-interval': '24'
|
|||
|
|
});
|
|||
|
|
res.end(data);
|
|||
|
|
});
|
|||
|
|
}).listen(PORT, '127.0.0.1', () => console.log('zy-sub-service on', PORT));
|