diff --git a/instance.ts b/instance.ts index cac7593..459022a 100644 --- a/instance.ts +++ b/instance.ts @@ -6,6 +6,26 @@ interface ChatMessage { message: string; } +function datetime(): string { + const now = new Date(); + const p = (n: number) => String(n).padStart(2, '0'); + return `${now.getFullYear()}${p(now.getMonth() + 1)}${p(now.getDate())} ${p(now.getHours())}${p(now.getMinutes())}${p(now.getSeconds())}`; +} + +// Message body only. `.+?` (not `\S+`) so a manual save named 'my map' still parses. +const SAVE_LINE_RE = /Saving to (.+?) \((?:non-)?blocking\)\./; + +// Factorio's autosaves are literally named '_autosave1'..'_autosaveN'. +const AUTOSAVE_NAME_RE = /^_autosave(\d+)$/; + +function parse_save(line: string): {name: string, slot?: number} | null { + const m = SAVE_LINE_RE.exec(line); + if (!m) return null; + + const slot = AUTOSAVE_NAME_RE.exec(m[1]); + return {name: m[1], slot: slot ? Number(slot[1]) : undefined}; +} + export default async function(context: InstancePluginContext) { const {instance, host, plugin, logger} = context; const queue: ChatMessage[] = []; @@ -38,32 +58,38 @@ export default async function(context: InstancePluginContext) { }); instance.hooks.output.attach(plugin.name, async (output) => { - if (output.type !== 'action') { - return; + if (output.type === 'action') { + const dt = datetime(); + const action = output.action; + let message = output.message.trim(); + + if (action === 'CHAT' || action === 'SHOUT') { + message = message.replace(/\[special-item=.*?\]/g, '').replace(/<@/g, '<@\u200c>'); + const message_index = message.indexOf(':'); + const message_username = message_index === -1 ? message : message.slice(0, message_index); + const message_content = message_index === -1 ? '' : message.slice(message_index + 1).trim(); + + queue.push({datetime: dt, message: `**\`${message_username}\`**: ${message_content}`}); + } else if (action === 'JOIN' || action === 'LEAVE' || action === 'KICK' || action === 'BAN' || action === 'UNBANNED') { + const message_index = message.indexOf(' '); + const message_username = message_index === -1 ? message : message.slice(0, message_index); + const message_content = message_index === -1 ? '' : message.slice(message_index + 1).trim(); + const player_count = instance.playersOnline.size + (action === 'JOIN' ? 0 : -1); + + queue.push({datetime: dt, message: `**\`${message_username}\`** ${message_content} (${player_count})`}); + } + + await flush(); + } else if (output.type === 'info') { + let message = output.message.trim(); + const save_slot = parse_save(message); + + if (!save_slot) { + return; + } + + const dt = datetime(); + queue.push({datetime: dt, message: `**AUTO SAVE SLOT ${save_slot}**`}); } - - const now = new Date(); - const p = (n: number) => String(n).padStart(2, '0'); - const dt = `${now.getFullYear()}${p(now.getMonth() + 1)}${p(now.getDate())} ${p(now.getHours())}${p(now.getMinutes())}${p(now.getSeconds())}`; - const action = output.action; - let message = output.message.trim(); - - if (action === 'CHAT' || action === 'SHOUT') { - message = message.replace(/\[special-item=.*?\]/g, '').replace(/<@/g, '<@\u200c>'); - const message_index = message.indexOf(':'); - const message_username = message_index === -1 ? message : message.slice(0, message_index); - const message_content = message_index === -1 ? '' : message.slice(message_index + 1).trim(); - - queue.push({datetime: dt, message: `**\`${message_username}\`**: ${message_content}`}); - } else { - const message_index = message.indexOf(' '); - const message_username = message_index === -1 ? message : message.slice(0, message_index); - const message_content = message_index === -1 ? '' : message.slice(message_index + 1).trim(); - const player_count = instance.playersOnline.size + (action === 'JOIN' ? 0 : -1); - - queue.push({datetime: dt, message: `**\`${message_username}\`** ${message_content} (${player_count})`}); - } - - await flush(); }); }