cli.js

#!/usr/bin/env node

const path = require('path')
const os = require('os')
const fs = require('fs-extra')
const generate = require('./lib/generate')
const register = require('./lib/register')
const info = require('./lib/info')
const ref = require('./lib/ref')
const conf = require('./lib/conf')
const license = require('./lib/license')

const args = process.argv.slice(2)
const HOME = os.homedir()
const CONFIG_DIR = path.join(HOME, '.cf-warp')
if (!fs.existsSync(CONFIG_DIR)) {
	fs.mkdirSync(CONFIG_DIR)
}

const resovle = file => path.join(CONFIG_DIR, file)
const exists = file => fs.exists(resovle(file))
const read = file => fs.readFile(resovle(file), 'utf-8')
const write = (file, content) => fs.writeFile(resovle(file), content, 'utf-8')
function sleep(ms) {
	return new Promise(resolve => {
		setTimeout(resolve, ms)
	})
}
function printInfo(data) {
	console.log(
		`Your Warp credentials are located at "${resovle(
			'data.json'
		)}", and WireGuard connection file is "${resovle('cf-warp.conf')}".`
	)
	console.log(
		`You currently have ${data.account.quota / 1000000000}GB Warp+ quota.`
	)
	console.log('To get your current Warp+ quota, simply run "cf-warp".')
	console.log(
		'To increase your Warp+ quota by 10 GB, simply run "cf-warp 10".'
	)
}
async function init() {
	console.log('Initializing Warp credentials...\n')
	const keys = await generate()
	const data = await register(keys)
	await ref(data) // enable Warp+
	const combined = Object.assign({}, keys, data, await info(data))
	await write('data.json', JSON.stringify(combined, null, 2))
	await write('cf-warp.conf', conf(combined))
	printInfo(combined)
}
;(async () => {
	const [dE, wE] = await Promise.all([
		exists('data.json'),
		exists('cf-warp.conf')
	])
	if (!dE) {
		// regenerate if data.json doesn't exists
		return init()
	}
	let data
	try {
		data = JSON.parse(await read('data.json'))
	} catch (e) {
		console.log(
			'"data.json" is corrupted, all the credentials will be reset...\n'
		)
		return init()
	}
	if (!wE) {
		console.log(
			'"cf-warp.conf" missing but "data.json" exists, regenerating a "cf-warp.conf"...\n'
		)
		await write('cf-warp.conf', conf(data))
	}

	if (args[0] === '--license') {
		// License
		const licenseStr = args[1]
		if (!licenseStr) {
			console.log('Please provide a valid license string from your app.')
			console.log('Example: cf-warp --license 1a2b3d4e-1a2b3d4e-1a2b3d4e')
			return
		}
		console.log(
			`Trying to register your device under the license "${licenseStr}"`
		)
		await license({ ...data, license: licenseStr })
		console.log(
			'Done. You may now manage your device on other devices under the same license.'
		)
		console.log()
	} else {
		// referrer
		const n = parseInt(args[0])
		if (!isNaN(n)) {
			console.log(`Prepare faking Warp+ referrer for ${n} times.`)
			for (let i = 1; i <= n; i++) {
				await sleep(20000)
				await ref(data)
				console.log(`#${i} fake referrer finished`)
			}
			console.log()
		}
	}
	const newData = await info(data)
	printInfo(newData)
	await write(
		'data.json',
		JSON.stringify(Object.assign({}, data, newData), null, 2)
	)
})()