540 lines
16 KiB
Plaintext
540 lines
16 KiB
Plaintext
/* BARE_OS_BIN_API 1.0.0 — bump when staged /bin script semantics change (see developer guide). */
|
|
/** Shared helpers for drive-resident /bin scripts (prepended before each command). */
|
|
function bareStdin(ctx) {
|
|
return typeof ctx.shellStdin === 'string' ? ctx.shellStdin : ''
|
|
}
|
|
|
|
/** @param {number} mode @param {'file' | 'directory' | 'symlink'} type */
|
|
function bareFormatModeString(mode, type) {
|
|
const typeChar = type === 'directory' ? 'd' : type === 'symlink' ? 'l' : '-'
|
|
const perm = mode & 0o777
|
|
const r = (bit) => (perm & bit ? 'r' : '-')
|
|
const w = (bit) => (perm & bit ? 'w' : '-')
|
|
const x = (bit) => (perm & bit ? 'x' : '-')
|
|
return (
|
|
typeChar +
|
|
r(0o400) +
|
|
w(0o200) +
|
|
x(0o100) +
|
|
r(0o040) +
|
|
w(0o020) +
|
|
x(0o010) +
|
|
r(0o004) +
|
|
w(0o002) +
|
|
x(0o001)
|
|
)
|
|
}
|
|
|
|
/** @param {number} mtimeMs @param {number} [nowMs] */
|
|
function bareFormatLsMtime(mtimeMs, nowMs) {
|
|
const now = nowMs != null ? nowMs : Date.now()
|
|
const d = new Date(mtimeMs)
|
|
const months = [
|
|
'Jan',
|
|
'Feb',
|
|
'Mar',
|
|
'Apr',
|
|
'May',
|
|
'Jun',
|
|
'Jul',
|
|
'Aug',
|
|
'Sep',
|
|
'Oct',
|
|
'Nov',
|
|
'Dec'
|
|
]
|
|
const mon = months[d.getMonth()]
|
|
const day = String(d.getDate()).padStart(2, ' ')
|
|
const sixMo = 180 * 24 * 3600 * 1000
|
|
if (Math.abs(now - mtimeMs) > sixMo) {
|
|
const yr = String(d.getFullYear()).padStart(4, ' ')
|
|
return mon + ' ' + day + ' ' + yr
|
|
}
|
|
const hh = String(d.getHours()).padStart(2, '0')
|
|
const mm = String(d.getMinutes()).padStart(2, '0')
|
|
return mon + ' ' + day + ' ' + hh + ':' + mm
|
|
}
|
|
|
|
/** @param {number} size */
|
|
function barePosixBlocks(size) {
|
|
return Math.ceil(Number(size) / 512) || 0
|
|
}
|
|
|
|
/**
|
|
* Raw stdout for NUL/binary when **`process.stdout.write`** is missing.
|
|
* If **`ctx.bareOsBinWrite(Uint8Array|string)`** is set (tests / host), use it.
|
|
* @param {Record<string, unknown>} ctx
|
|
* @param {string | Uint8Array} chunk
|
|
* @returns {boolean}
|
|
*/
|
|
function bareOsEmitRaw(ctx, chunk) {
|
|
if (typeof ctx.bareOsBinWrite === 'function') {
|
|
const b4 = ctx.b4a
|
|
const u8 =
|
|
typeof chunk === 'string'
|
|
? b4 && typeof b4.from === 'function'
|
|
? b4.from(chunk)
|
|
: new TextEncoder().encode(chunk)
|
|
: chunk
|
|
ctx.bareOsBinWrite(u8 instanceof Uint8Array ? u8 : new Uint8Array(u8))
|
|
return true
|
|
}
|
|
const w = globalThis.process?.stdout?.write
|
|
if (typeof w === 'function') {
|
|
w.call(globalThis.process.stdout, chunk)
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
/** Session env map (`vfs.env`, then `ctx.env`). Never throws. */
|
|
function bareOsEnv(ctx) {
|
|
const v = ctx && ctx.vfs && ctx.vfs.env
|
|
if (v && typeof v === 'object') return v
|
|
const e = ctx && ctx.env
|
|
if (e && typeof e === 'object') return e
|
|
return {}
|
|
}
|
|
|
|
/**
|
|
* Strict POSIX-ish decimal integer (no octal, no exponent, no empty).
|
|
* @param {unknown} s
|
|
* @returns {number}
|
|
*/
|
|
function bareOsParseDecInt(s) {
|
|
const t = String(s == null ? '' : s).trim()
|
|
if (!/^[+-]?(?:0|[1-9][0-9]*)$/.test(t)) return NaN
|
|
const n = Number.parseInt(t, 10)
|
|
return Number.isSafeInteger(n) ? n : NaN
|
|
}
|
|
|
|
/** @param {unknown} s */
|
|
function bareOsParseNonNegInt(s) {
|
|
const n = bareOsParseDecInt(s)
|
|
return n >= 0 ? n : NaN
|
|
}
|
|
|
|
/**
|
|
* @param {Record<string, unknown>} ctx
|
|
* @param {string} name
|
|
* @param {number} fallback
|
|
* @param {number} [min]
|
|
* @param {number} [max]
|
|
*/
|
|
function bareOsEnvInt(ctx, name, fallback, min, max) {
|
|
const raw = bareOsEnv(ctx)[name]
|
|
if (raw == null || raw === '') return fallback
|
|
const n = Number.parseInt(String(raw), 10)
|
|
if (!Number.isFinite(n)) return fallback
|
|
let v = n
|
|
if (min != null && v < min) v = min
|
|
if (max != null && v > max) v = max
|
|
return v
|
|
}
|
|
|
|
/**
|
|
* @param {Record<string, unknown>} ctx
|
|
* @param {string} msg
|
|
* @param {number} [code]
|
|
*/
|
|
function bareOsFail(ctx, msg, code) {
|
|
if (msg) ctx.console.error(msg)
|
|
ctx.exitCode = code == null ? 1 : code
|
|
}
|
|
|
|
/** @param {unknown} e */
|
|
function bareOsIsNotFoundErr(e) {
|
|
const code = e && typeof e === 'object' ? e.code : ''
|
|
if (code === 'ENOENT') return true
|
|
const msg = String((e && e.message) || e || '')
|
|
return /ENOENT|No such file|not found/i.test(msg)
|
|
}
|
|
|
|
/**
|
|
* @param {Record<string, unknown>} ctx
|
|
* @param {unknown} buf
|
|
* @returns {Uint8Array}
|
|
*/
|
|
function bareOsToU8(ctx, buf) {
|
|
if (!buf) return new Uint8Array(0)
|
|
if (buf instanceof Uint8Array) return buf
|
|
if (ctx && ctx.b4a && typeof ctx.b4a.from === 'function') return ctx.b4a.from(buf)
|
|
return new Uint8Array(buf)
|
|
}
|
|
|
|
/** @param {string} dir @param {string} name */
|
|
function bareOsJoinPath(dir, name) {
|
|
const d = String(dir || '').replace(/\/+$/, '')
|
|
const n = String(name || '').replace(/^\/+/, '')
|
|
if (!d || d === '/') return '/' + n
|
|
return d + '/' + n
|
|
}
|
|
|
|
/** @param {string} p */
|
|
function bareOsBaseName(p) {
|
|
const t = String(p || '').replace(/\/+$/, '')
|
|
if (!t || t === '/') return t === '/' ? '/' : ''
|
|
const i = t.lastIndexOf('/')
|
|
return i < 0 ? t : t.slice(i + 1) || t
|
|
}
|
|
|
|
/** @param {string} p */
|
|
function bareOsParentDir(p) {
|
|
const t = String(p || '').replace(/\/+$/, '') || '/'
|
|
if (t === '/') return '/'
|
|
const i = t.lastIndexOf('/')
|
|
return i <= 0 ? '/' : t.slice(0, i) || '/'
|
|
}
|
|
|
|
/** @param {string} p */
|
|
function bareOsNormPath(p) {
|
|
return String(p || '').replace(/\/+$/, '') || '/'
|
|
}
|
|
|
|
/**
|
|
* @param {Record<string, unknown>} ctx
|
|
* @param {string} p
|
|
*/
|
|
function bareOsResolvePath(ctx, p) {
|
|
if (ctx && ctx.vfs && typeof ctx.vfs.resolveLogical === 'function') {
|
|
try {
|
|
return String(ctx.vfs.resolveLogical(p) || p)
|
|
} catch {
|
|
/* fall through */
|
|
}
|
|
}
|
|
return String(p || '')
|
|
}
|
|
|
|
/**
|
|
* True when dest is src or lives under src (self-copy / self-move).
|
|
* @param {Record<string, unknown>} ctx
|
|
* @param {string} src
|
|
* @param {string} dest
|
|
*/
|
|
function bareOsDestInsideSrc(ctx, src, dest) {
|
|
const s = bareOsNormPath(bareOsResolvePath(ctx, src))
|
|
const d = bareOsNormPath(bareOsResolvePath(ctx, dest))
|
|
if (s === d) return true
|
|
if (s === '/') return d !== '/'
|
|
return d === s || d.startsWith(s + '/')
|
|
}
|
|
|
|
const BARE_OS_B64_ALPH =
|
|
'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/'
|
|
|
|
/** @param {Uint8Array} u8 */
|
|
function bareOsB64Encode(u8) {
|
|
let out = ''
|
|
let i = 0
|
|
for (; i + 2 < u8.length; i += 3) {
|
|
const n = (u8[i] << 16) | (u8[i + 1] << 8) | u8[i + 2]
|
|
out +=
|
|
BARE_OS_B64_ALPH[(n >> 18) & 63] +
|
|
BARE_OS_B64_ALPH[(n >> 12) & 63] +
|
|
BARE_OS_B64_ALPH[(n >> 6) & 63] +
|
|
BARE_OS_B64_ALPH[n & 63]
|
|
}
|
|
const rest = u8.length - i
|
|
if (rest === 1) {
|
|
const n = u8[i] << 16
|
|
out += BARE_OS_B64_ALPH[(n >> 18) & 63] + BARE_OS_B64_ALPH[(n >> 12) & 63] + '=='
|
|
} else if (rest === 2) {
|
|
const n = (u8[i] << 16) | (u8[i + 1] << 8)
|
|
out +=
|
|
BARE_OS_B64_ALPH[(n >> 18) & 63] +
|
|
BARE_OS_B64_ALPH[(n >> 12) & 63] +
|
|
BARE_OS_B64_ALPH[(n >> 6) & 63] +
|
|
'='
|
|
}
|
|
return out
|
|
}
|
|
|
|
/**
|
|
* RFC 4648 Base64 decode (also accepts URL-safe alphabet). Rejects junk.
|
|
* @param {string} s
|
|
* @returns {Uint8Array}
|
|
*/
|
|
function bareOsB64Decode(s) {
|
|
const t = String(s).replace(/\s+/g, '')
|
|
if (!t) return new Uint8Array(0)
|
|
if (t.length % 4 === 1) throw new Error('invalid base64 length')
|
|
let pad = 0
|
|
if (t.endsWith('==')) pad = 2
|
|
else if (t.endsWith('=')) pad = 1
|
|
const body = pad ? t.slice(0, t.length - pad) : t
|
|
const bytes = []
|
|
let buf = 0
|
|
let bits = 0
|
|
for (let i = 0; i < body.length; i++) {
|
|
const c = body[i]
|
|
let v = BARE_OS_B64_ALPH.indexOf(c)
|
|
if (v < 0) {
|
|
if (c === '-') v = 62
|
|
else if (c === '_') v = 63
|
|
else throw new Error('invalid base64 character')
|
|
}
|
|
buf = (buf << 6) | v
|
|
bits += 6
|
|
if (bits >= 8) {
|
|
bits -= 8
|
|
bytes.push((buf >> bits) & 255)
|
|
}
|
|
}
|
|
if (pad) {
|
|
const want = Math.floor((body.length * 6) / 8)
|
|
if (bytes.length > want) bytes.length = want
|
|
}
|
|
return new Uint8Array(bytes)
|
|
}
|
|
|
|
/**
|
|
* @param {string} s
|
|
* @returns {Uint8Array}
|
|
*/
|
|
function bareOsHexDecode(s) {
|
|
const t = String(s).replace(/\s+/g, '')
|
|
if (t.length % 2 !== 0) throw new Error('odd hex length')
|
|
const out = new Uint8Array(t.length / 2)
|
|
for (let i = 0; i < out.length; i++) {
|
|
const pair = t.slice(i * 2, i * 2 + 2)
|
|
if (!/^[0-9a-fA-F]{2}$/.test(pair)) throw new Error('invalid hex')
|
|
out[i] = Number.parseInt(pair, 16)
|
|
}
|
|
return out
|
|
}
|
|
|
|
/** @param {Uint8Array} u8 */
|
|
function bareOsHexEncode(u8) {
|
|
let s = ''
|
|
for (let i = 0; i < u8.length; i++) s += u8[i].toString(16).padStart(2, '0')
|
|
return s
|
|
}
|
|
|
|
/**
|
|
* Interactive Discord bot setup: ~/.discord/.env + hello-world plugin.
|
|
*/
|
|
|
|
var INIT_DISCORD_HELLO_JSON =
|
|
'{\n' +
|
|
' "name": "hello",\n' +
|
|
' "description": "Greet someone from Bare OS",\n' +
|
|
' "version": "1.0.0",\n' +
|
|
' "options": [\n' +
|
|
' {\n' +
|
|
' "name": "who",\n' +
|
|
' "description": "Name to greet",\n' +
|
|
' "required": false\n' +
|
|
' }\n' +
|
|
' ],\n' +
|
|
' "run": "echo Hello, ${who:-world} — from $USER on $HOSTNAME",\n' +
|
|
' "title": "hello"\n' +
|
|
'}\n'
|
|
|
|
function initDiscordHasFlag(argv, names) {
|
|
for (let i = 1; i < argv.length; i++) {
|
|
const a = String(argv[i] || '')
|
|
for (let n = 0; n < names.length; n++) {
|
|
if (a === names[n]) return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
function initDiscordArg(argv, names) {
|
|
for (let i = 1; i < argv.length; i++) {
|
|
const a = String(argv[i] || '')
|
|
for (let n = 0; n < names.length; n++) {
|
|
const name = names[n]
|
|
if (a === name) return argv[i + 1] != null ? String(argv[i + 1]) : ''
|
|
if (a.indexOf(name + '=') === 0) return a.slice(name.length + 1)
|
|
}
|
|
}
|
|
return ''
|
|
}
|
|
|
|
function initDiscordParseEnv(text) {
|
|
const out = {}
|
|
const lines = String(text || '').split(/\r?\n/)
|
|
for (let i = 0; i < lines.length; i++) {
|
|
let s = lines[i].replace(/^\uFEFF/, '').trim()
|
|
if (!s || s.charAt(0) === '#') continue
|
|
if (s.indexOf('export ') === 0) s = s.slice(7).trim()
|
|
const eq = s.indexOf('=')
|
|
if (eq < 1) continue
|
|
let val = s.slice(eq + 1).trim()
|
|
if (
|
|
(val.charAt(0) === '"' && val.charAt(val.length - 1) === '"') ||
|
|
(val.charAt(0) === "'" && val.charAt(val.length - 1) === "'")
|
|
) {
|
|
val = val.slice(1, -1)
|
|
}
|
|
out[s.slice(0, eq).trim()] = val
|
|
}
|
|
return out
|
|
}
|
|
|
|
function initDiscordMaskToken(tok) {
|
|
const t = String(tok || '')
|
|
if (t.length < 8) return '(set)'
|
|
return t.slice(0, 4) + '…' + t.slice(-4)
|
|
}
|
|
|
|
function initDiscordUpsertEnv(text, key, value) {
|
|
const line = key + '=' + value
|
|
const re = new RegExp('^\\s*(?:export\\s+)?' + key.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\$&') + '=.*$', 'm')
|
|
if (re.test(text)) return text.replace(re, line)
|
|
const body = text && !/\n$/.test(text) ? text + '\n' : text || ''
|
|
return body + line + '\n'
|
|
}
|
|
|
|
async function initDiscordReadText(ctx, path) {
|
|
try {
|
|
const buf = await ctx.vfs.readFile(path)
|
|
if (!buf) return ''
|
|
if (ctx.b4a && typeof ctx.b4a.toString === 'function') return ctx.b4a.toString(buf)
|
|
return String(buf)
|
|
} catch {
|
|
return ''
|
|
}
|
|
}
|
|
|
|
async function initDiscordWriteText(ctx, path, text) {
|
|
const buf = ctx.b4a && typeof ctx.b4a.from === 'function' ? ctx.b4a.from(text) : text
|
|
await ctx.vfs.writeFile(path, buf)
|
|
}
|
|
|
|
async function initDiscordMkdir(ctx, path) {
|
|
if (!ctx.vfs || typeof ctx.vfs.mkdir !== 'function') return
|
|
try {
|
|
await ctx.vfs.mkdir(path, { recursive: true })
|
|
} catch {
|
|
/* exists */
|
|
}
|
|
}
|
|
|
|
async function initDiscordAsk(ctx, prompt, fallback) {
|
|
if (typeof ctx.readLine !== 'function') return fallback || ''
|
|
const hint = fallback ? ' [' + fallback + ']' : ''
|
|
const raw = await ctx.readLine(prompt + hint + ': ')
|
|
const s = String(raw == null ? '' : raw).trim()
|
|
return s || fallback || ''
|
|
}
|
|
|
|
async function run(ctx, argv) {
|
|
const argv0 = argv[0] || 'init-discord'
|
|
if (initDiscordHasFlag(argv, ['-h', '--help'])) {
|
|
ctx.console.log(
|
|
'usage: ' +
|
|
argv0 +
|
|
' [--token TOKEN] [--guild ID] [--whitelist ID,ID] [--yes] [--force] [--plugin-only]\n\n' +
|
|
'Create ~/.discord, write ~/.discord/.env, and install a hello-world plugin.\n' +
|
|
'Interactive by default (asks for token, guild id, whitelist). Use flags for scripts.\n\n' +
|
|
' --token TOKEN Bot token (Developer Portal → Bot). Never logged.\n' +
|
|
' --guild ID Guild id for instant slash registration.\n' +
|
|
' --whitelist ID,ID Discord user ids allowed to use the bot.\n' +
|
|
' --yes Non-interactive: require --token (or an existing .env).\n' +
|
|
' --force Overwrite an existing token / hello.json.\n' +
|
|
' --plugin-only Only ensure the hello plugin; leave .env alone.\n\n' +
|
|
'Then: discord-bot --check --env ~/.discord/.env\n' +
|
|
' systemctl daemon-reload && systemctl start bare-os-discord'
|
|
)
|
|
ctx.exitCode = 0
|
|
return
|
|
}
|
|
if (!ctx.vfs || typeof ctx.vfs.writeFile !== 'function') {
|
|
ctx.console.error(argv0 + ': vfs.writeFile unavailable')
|
|
ctx.exitCode = 1
|
|
return
|
|
}
|
|
|
|
const yes = initDiscordHasFlag(argv, ['--yes', '-y'])
|
|
const force = initDiscordHasFlag(argv, ['--force'])
|
|
const pluginOnly = initDiscordHasFlag(argv, ['--plugin-only'])
|
|
let tokenFlag = initDiscordArg(argv, ['--token'])
|
|
let guildFlag = initDiscordArg(argv, ['--guild', '--guild-id'])
|
|
let whiteFlag = initDiscordArg(argv, ['--whitelist', '--id-whitelist'])
|
|
|
|
await initDiscordMkdir(ctx, '~/.discord')
|
|
await initDiscordMkdir(ctx, '~/.discord/plugins')
|
|
|
|
const envPath = '~/.discord/.env'
|
|
const prevText = await initDiscordReadText(ctx, envPath)
|
|
const prev = initDiscordParseEnv(prevText)
|
|
const prevToken = prev.DISCORD_TOKEN || prev.BOT_TOKEN || ''
|
|
const prevGuild = prev.DISCORD_GUILD_ID || ''
|
|
const prevWhite = prev.DISCORD_ID_WHITELIST || ''
|
|
|
|
if (!pluginOnly) {
|
|
let token = tokenFlag
|
|
let guild = guildFlag
|
|
let white = whiteFlag
|
|
if (!yes) {
|
|
ctx.console.log('Bare OS Discord setup')
|
|
ctx.console.log('Token from Developer Portal → Bot → Reset Token (not the OAuth2 client secret).')
|
|
if (prevToken && !token) {
|
|
ctx.console.log('Existing token: ' + initDiscordMaskToken(prevToken) + ' (Enter keeps it)')
|
|
}
|
|
token = await initDiscordAsk(ctx, 'DISCORD_TOKEN', token || '')
|
|
if (!token && prevToken) token = prevToken
|
|
guild = await initDiscordAsk(
|
|
ctx,
|
|
'DISCORD_GUILD_ID (optional, instant slash commands)',
|
|
guild || prevGuild
|
|
)
|
|
white = await initDiscordAsk(
|
|
ctx,
|
|
'DISCORD_ID_WHITELIST (your user id, comma-separated)',
|
|
white || prevWhite
|
|
)
|
|
} else if (!token) {
|
|
token = prevToken
|
|
}
|
|
if (!token) {
|
|
ctx.console.error(argv0 + ': DISCORD_TOKEN is required (flag --token or interactive prompt)')
|
|
ctx.exitCode = 1
|
|
return
|
|
}
|
|
if (prevToken && token !== prevToken && !force && yes) {
|
|
ctx.console.error(argv0 + ': ~/.discord/.env already has a token (pass --force to replace)')
|
|
ctx.exitCode = 1
|
|
return
|
|
}
|
|
let next = prevText || '# Bare OS Discord bot\n'
|
|
next = initDiscordUpsertEnv(next, 'DISCORD_TOKEN', token)
|
|
if (guild) next = initDiscordUpsertEnv(next, 'DISCORD_GUILD_ID', guild)
|
|
if (white) next = initDiscordUpsertEnv(next, 'DISCORD_ID_WHITELIST', white)
|
|
await initDiscordWriteText(ctx, envPath, next)
|
|
if (ctx.env) {
|
|
ctx.env.DISCORD_TOKEN = token
|
|
if (guild) ctx.env.DISCORD_GUILD_ID = guild
|
|
if (white) ctx.env.DISCORD_ID_WHITELIST = white
|
|
}
|
|
if (ctx.vfs.env) {
|
|
ctx.vfs.env.DISCORD_TOKEN = token
|
|
if (guild) ctx.vfs.env.DISCORD_GUILD_ID = guild
|
|
if (white) ctx.vfs.env.DISCORD_ID_WHITELIST = white
|
|
}
|
|
ctx.console.log('Wrote ' + envPath + ' (token ' + initDiscordMaskToken(token) + ')')
|
|
if (guild) ctx.console.log('Guild id ' + guild)
|
|
if (white) ctx.console.log('Whitelist ' + white)
|
|
}
|
|
|
|
const pluginPath = '~/.discord/plugins/hello.json'
|
|
const existingPlugin = await initDiscordReadText(ctx, pluginPath)
|
|
if (existingPlugin && !force) {
|
|
ctx.console.log('Kept existing ' + pluginPath)
|
|
} else {
|
|
await initDiscordWriteText(ctx, pluginPath, INIT_DISCORD_HELLO_JSON)
|
|
ctx.console.log('Installed example plugin ' + pluginPath + ' (/hello)')
|
|
}
|
|
|
|
ctx.console.log('')
|
|
ctx.console.log('Next:')
|
|
ctx.console.log(' discord-bot --check --env ~/.discord/.env')
|
|
ctx.console.log(' discord-bot --env ~/.discord/.env')
|
|
ctx.console.log(' systemctl daemon-reload && systemctl start bare-os-discord')
|
|
ctx.console.log(' /plugins reload (after the bot is online)')
|
|
ctx.exitCode = 0
|
|
}
|