346 lines
19 KiB
JavaScript
346 lines
19 KiB
JavaScript
// The assistant: a round button in the bottom right corner opens a chat. Each
|
|
// question goes with what the panel shows - the tabs, the active pane, the
|
|
// text of its terminal or its list of files - and the answer streams in.
|
|
// Commands the model suggests come as ```run``` blocks with «Выполнить»: a
|
|
// click sends the command to the active terminal, with the user's rights.
|
|
// Off until the user turns it on in the profile.
|
|
(function () {
|
|
'use strict';
|
|
const PLANE = '<svg width="18" height="18" viewBox="0 0 24 24" fill="currentColor"><path d="M2.4 10.6 20.6 3.3c.8-.3 1.5.4 1.2 1.2l-7.3 18.2c-.3.8-1.5.8-1.8 0l-2.6-6.9-6.9-2.6c-.8-.3-.8-1.5 0-1.8zm7.3 4.4 2 5.2L18 6 9.7 15z"/></svg>';
|
|
const STOP = '<svg width="14" height="14" viewBox="0 0 24 24" fill="currentColor"><rect x="5" y="5" width="14" height="14" rx="2"/></svg>';
|
|
const esc = v => String(v).replace(/[&<>"]/g, c => ({'&': '&', '<': '<', '>': '>', '"': '"'}[c]));
|
|
const fab = document.createElement('button');
|
|
fab.type = 'button';
|
|
fab.className = 'ai-fab';
|
|
fab.hidden = true;
|
|
fab.title = 'ИИ-ассистент';
|
|
fab.setAttribute('aria-label', 'ИИ-ассистент');
|
|
fab.innerHTML = '<svg width="24" height="24" viewBox="0 0 24 24" fill="currentColor"><path d="M10 2.5c.3 0 .5.2.6.5l1 3.7a5 5 0 0 0 3.6 3.6l3.7 1c.6.2.6 1 0 1.2l-3.7 1a5 5 0 0 0-3.6 3.6l-1 3.7c-.2.6-1 .6-1.2 0l-1-3.7a5 5 0 0 0-3.6-3.6l-3.7-1c-.6-.2-.6-1 0-1.2l3.7-1a5 5 0 0 0 3.6-3.6l1-3.7c.1-.3.3-.5.6-.5z"/><path d="M18.5 1.5c.2 0 .3.1.4.3l.4 1.4c.2.6.7 1.1 1.3 1.3l1.4.4c.4.1.4.6 0 .8l-1.4.4c-.6.2-1.1.7-1.3 1.3l-.4 1.4c-.1.4-.7.4-.8 0l-.4-1.4a2 2 0 0 0-1.3-1.3l-1.4-.4c-.4-.2-.4-.7 0-.8l1.4-.4c.6-.2 1.1-.7 1.3-1.3l.4-1.4c.1-.2.2-.3.4-.3z" opacity=".75"/></svg>';
|
|
const box = document.createElement('div');
|
|
box.className = 'ai-box';
|
|
box.innerHTML =
|
|
'<div class="ai-head"><b>Ассистент</b><span class="muted ai-where"></span><span class="spacer"></span>' +
|
|
|
|
'<button type="button" class="btn ghost small ai-clear" title="Новый разговор">Очистить</button>' +
|
|
'<button type="button" class="btn ghost small icon close x ai-close" aria-label="Закрыть"><svg width="12" height="12" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2.6" stroke-linecap="round"><path d="M18 6 6 18M6 6l12 12"/></svg></button></div>' +
|
|
'<div class="ai-list"><p class="muted ai-hello">Спросите о том, что открыто в панели: ассистент видит вкладки и активный терминал или папку. Команды он предложит кнопкой «Выполнить», умеет читать и писать файлы на любой вашей машине. В режиме «Агент» он делает шаги сам, спрашивая перед каждой командой.</p></div>' +
|
|
'<div class="ai-pills">' +
|
|
'<button type="button" class="ai-pill" data-mode="agent" aria-pressed="false" title="Сам выполняет шаги один за другим, спрашивая перед каждой командой и записью файла">' +
|
|
'<svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><path d="M12 8V4H8"/><rect x="4" y="8" width="16" height="12" rx="2"/><path d="M2 14h2M20 14h2M9 13v2M15 13v2"/></svg>Агент</button>' +
|
|
'<button type="button" class="ai-pill" data-mode="think" aria-pressed="false" title="Сначала обдумывает, потом отвечает: дольше, но точнее">' +
|
|
'<svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><path d="M9 18h6M10 22h4"/><path d="M12 2a7 7 0 0 0-4 12.7c.6.5 1 1.3 1 2.3h6c0-1 .4-1.8 1-2.3A7 7 0 0 0 12 2z"/></svg>Думать</button>' +
|
|
'</div>' +
|
|
'<form class="ai-form-in"><textarea rows="1" placeholder="Сообщение…"></textarea>' +
|
|
'<button type="submit" class="ai-send" title="Отправить" aria-label="Отправить">' + PLANE + '</button></form>';
|
|
document.body.append(fab, box);
|
|
const q = s => box.querySelector(s);
|
|
const list = q('.ai-list'), input = q('textarea'), send = q('.ai-send');
|
|
const talk = []; // {role, content}
|
|
let ac = null;
|
|
|
|
// Open and shut with a short slide. On a phone the chat takes the screen
|
|
// and is one step of history: «назад» shuts it.
|
|
const phone = () => matchMedia('(max-width: 640px)').matches;
|
|
const isOpen = () => box.classList.contains('on');
|
|
function show() {
|
|
box.classList.add('on');
|
|
if (phone() && !(history.state && history.state.aiChat)) history.pushState({aiChat: 1}, '', location.href);
|
|
where();
|
|
if (!phone()) input.focus();
|
|
}
|
|
function hide() {
|
|
if (!isOpen()) return;
|
|
if (history.state && history.state.aiChat) { history.back(); return; }
|
|
box.classList.remove('on');
|
|
}
|
|
window.addEventListener('popstate', () => { if (isOpen() && !(history.state && history.state.aiChat)) box.classList.remove('on'); });
|
|
function setOn(on) { fab.hidden = !on; if (!on) hide(); }
|
|
fetch('/api/panel/ai', {cache: 'no-store', credentials: 'same-origin'}).then(r => r.json()).then(d => setOn(!!d.on)).catch(() => {});
|
|
document.addEventListener('lpra:ai', ev => setOn(ev.detail));
|
|
fab.onclick = () => (isOpen() ? hide() : show());
|
|
q('.ai-close').onclick = hide;
|
|
q('.ai-clear').onclick = () => {
|
|
if (ac) ac.abort();
|
|
talk.length = 0;
|
|
list.querySelectorAll('.ai-msg').forEach(m => m.remove());
|
|
q('.ai-hello').hidden = false;
|
|
};
|
|
|
|
// ---------- what the panel shows ----------
|
|
const W = () => window.Workspace;
|
|
function termText(p, lines) {
|
|
const b = p.term && p.term.buffer && p.term.buffer.active;
|
|
if (!b) return '';
|
|
const out = [];
|
|
for (let i = Math.max(0, b.length - lines); i < b.length; i++) {
|
|
const l = b.getLine(i);
|
|
if (l) out.push(l.translateToString(true));
|
|
}
|
|
while (out.length && !out[out.length - 1].trim()) out.pop();
|
|
return out.join('\n');
|
|
}
|
|
function paneLine(p) {
|
|
const kind = {term: 'терминал', ssh: 'SSH-терминал', files: 'файлы', sshfiles: 'файлы по SSH'}[p.kind] || p.kind;
|
|
return kind + ' · ' + (p.tooltip ? p.tooltip() : p.title()) + ' (машина ' + (p.agent ? p.agent.name : '?') + ')';
|
|
}
|
|
function context() {
|
|
const view = document.querySelector('main#app').dataset.view;
|
|
const out = ['Страница: ' + ({machines: 'Машины', terminal: 'WorkSpace', settings: 'Настройки'}[view] || view)];
|
|
out.push('Машины пользователя: ' + machines.map(m => m.name + ' (' + m.user + '@' + m.hostname + (m.online ? '' : ', офлайн') + ')').join('; '));
|
|
const ws = W();
|
|
if (!ws) return out.join('\n');
|
|
const panes = ws.panes(), act = ws.activePane();
|
|
out.push('Панели (' + panes.length + '):');
|
|
panes.forEach(p => out.push((p === act ? '* ' : '- ') + paneLine(p)));
|
|
if (act && act.term) {
|
|
out.push('', 'Активный терминал, последние строки:', '```', termText(act, 120), '```');
|
|
} else if (act && act.entries) {
|
|
out.push('', 'Активная файловая панель: ' + act.cwd + (act.selected.size ? ' · выбрано: ' + [...act.selected].join(', ') : ''));
|
|
out.push(act.entries.slice(0, 200).map(e => (e.is_dir ? e.name + '/' : e.name + ' (' + e.size + ' B)')).join('\n'));
|
|
}
|
|
return out.join('\n');
|
|
}
|
|
function where() {
|
|
const act = W() && W().activePane();
|
|
q('.ai-where').textContent = act ? '· ' + (act.agent ? act.agent.name : '') : '';
|
|
}
|
|
|
|
// ---------- machines ----------
|
|
let machines = [];
|
|
async function loadMachines() {
|
|
try { machines = await fetch('/api/panel/agents/presence', {cache: 'no-store', credentials: 'same-origin'}).then(r => r.json()); } catch (e) { /* keep */ }
|
|
return machines;
|
|
}
|
|
// machineOf: the machine an action names, else the one of the active pane.
|
|
function machineOf(name) {
|
|
const n = (name || '').trim().toLowerCase();
|
|
if (n) return machines.find(m => m.name.toLowerCase() === n || m.id === n || m.hostname.toLowerCase() === n) || null;
|
|
const act = W() && W().activePane();
|
|
if (act && act.agent) return machines.find(m => m.id === act.agent.id) || {id: act.agent.id, name: act.agent.name, online: true};
|
|
return machines.length === 1 ? machines[0] : null;
|
|
}
|
|
|
|
// ---------- actions ----------
|
|
const sleep = ms => new Promise(r => setTimeout(r, ms));
|
|
const lineAt = (b, i) => { const l = b.getLine(i); return l ? l.translateToString(true) : ''; };
|
|
// terminalFor: a terminal of machine m - the active one if it is, else any
|
|
// open one of it, else a new tab.
|
|
async function terminalFor(m) {
|
|
const ws = W(), act = ws.activePane();
|
|
if (act && act.term && !act.ssh && act.agent.id === m.id) return act;
|
|
let t = ws.panes().find(p => p.term && !p.ssh && p.agent.id === m.id);
|
|
if (!t) {
|
|
if (document.querySelector('main#app').dataset.view !== 'terminal') App.go('/terminal');
|
|
ws.open('term', {id: m.id, name: m.name, host: m.hostname, user: m.user});
|
|
t = ws.panes().find(p => p.term && !p.ssh && p.agent.id === m.id);
|
|
for (let i = 0; i < 40 && t && t.state !== 'ok'; i++) await sleep(250);
|
|
await sleep(600);
|
|
}
|
|
return t;
|
|
}
|
|
// runIn sends cmd to terminal t and gives what it printed: until the
|
|
// output has been still for 1.5 s (up to 2 minutes).
|
|
async function runIn(t, cmd, signal) {
|
|
const b = t.term.buffer.active;
|
|
const start = b.baseY + b.cursorY;
|
|
t.send(cmd.replace(/\n/g, '\r') + '\r');
|
|
let last = '', still = 0;
|
|
for (let ms = 0; ms < 120000 && !(signal && signal.aborted); ms += 300) {
|
|
await sleep(300);
|
|
const end = b.baseY + b.cursorY;
|
|
const now = end + ':' + lineAt(b, end);
|
|
if (now === last) { if ((still += 300) >= 1500) break; } else { still = 0; last = now; }
|
|
}
|
|
const lines = [];
|
|
for (let i = start; i <= b.baseY + b.cursorY; i++) lines.push(lineAt(b, i));
|
|
let out = lines.join('\n').replace(/\s+$/, '');
|
|
if (out.length > 8000) out = '…' + out.slice(-8000);
|
|
return out;
|
|
}
|
|
const b64 = text => { const u = new TextEncoder().encode(text); let s = ''; for (let i = 0; i < u.length; i++) s += String.fromCharCode(u[i]); return btoa(s); };
|
|
// perform does one action with the user's rights and tells what came of it.
|
|
async function perform(a, signal) {
|
|
await loadMachines();
|
|
const m = machineOf(a.machine);
|
|
if (!m) return 'Ошибка: нет машины «' + (a.machine || '?') + '». Машины: ' + machines.map(x => x.name).join(', ');
|
|
if (m.online === false) return 'Ошибка: машина ' + m.name + ' офлайн';
|
|
const src = new MachineSource({id: m.id, name: m.name});
|
|
if (a.kind === 'run') {
|
|
const t = await terminalFor(m);
|
|
if (!t) return 'Ошибка: терминал на ' + m.name + ' не открылся';
|
|
return 'Вывод на ' + m.name + ':\n```\n' + await runIn(t, a.body, signal) + '\n```';
|
|
}
|
|
if (a.kind === 'read') {
|
|
const r = await src.op('read', {path: a.path});
|
|
if (!r.ok) return 'Ошибка чтения ' + m.name + ':' + a.path + ': ' + r.error;
|
|
if (r.data.binary) return m.name + ':' + a.path + ' - двоичный файл, ' + r.data.size + ' B';
|
|
let text = new TextDecoder().decode(FileViews.b64ToBytes(r.data.content));
|
|
if (text.length > 20000) text = text.slice(0, 20000) + '\n…';
|
|
return 'Файл ' + m.name + ':' + a.path + (r.data.truncated ? ' (начало)' : '') + ':\n```\n' + text + '\n```';
|
|
}
|
|
if (a.kind === 'ls') {
|
|
const r = await src.op('list', {path: a.path || '~'});
|
|
if (!r.ok) return 'Ошибка: ' + r.error;
|
|
return 'Папка ' + m.name + ':' + r.data.path + ':\n' + (r.data.entries || []).slice(0, 300)
|
|
.map(e => (e.is_dir ? e.name + '/' : e.name + ' ' + e.size + ' B')).join('\n');
|
|
}
|
|
if (a.kind === 'write') {
|
|
const r = await src.op('write', {path: a.path, data: b64(a.body + '\n')});
|
|
return r.ok ? 'Записано: ' + m.name + ':' + a.path : 'Ошибка записи: ' + r.error;
|
|
}
|
|
return 'Ошибка: неизвестное действие';
|
|
}
|
|
// ask first for what changes something: commands and writes.
|
|
function needsYes(a) { return a.kind === 'run' || a.kind === 'write'; }
|
|
function describe(a) {
|
|
const where = a.machine ? a.machine + ': ' : '';
|
|
return a.kind === 'run' ? 'Выполнить на ' + (a.machine || 'активной машине') + ':\n\n' + a.body
|
|
: 'Записать файл ' + where + a.path + ' (' + a.body.length + ' символов)?';
|
|
}
|
|
|
|
// ---------- messages ----------
|
|
// stripThink: the answer without its thinking - what is acted on and kept.
|
|
const stripThink = t => t.replace(/^\s*(?:<think>|::думаю::)[\s\S]*?((?:<\/think>|::ответ::)\s*|$)/, '');
|
|
const LABEL = {run: 'Выполнить', read: 'Прочитать', ls: 'Показать', write: 'Записать'};
|
|
// render: text with ``` blocks; action blocks get their button.
|
|
function render(el, text, done) {
|
|
// Thinking first, folded: ::думаю:: … ::ответ:: (or <think>), open while
|
|
// it is going on.
|
|
let html = '', n = 0;
|
|
const th = /^\s*(?:<think>|::думаю::)([\s\S]*?)(<\/think>|::ответ::|$)/.exec(text);
|
|
if (th) {
|
|
const going = !th[2];
|
|
html += '<details class="ai-think"' + (going ? ' open' : '') + '><summary>' + (going ? 'Думает…' : 'Размышления') + '</summary><div>' +
|
|
esc(th[1].trim()).replace(/\n/g, '<br>') + '</div></details>';
|
|
text = text.slice(th[0].length);
|
|
}
|
|
const parts = text.split(/```/);
|
|
parts.forEach((part, i) => {
|
|
if (i % 2 === 0) { html += '<div class="ai-text">' + esc(part).replace(/\n/g, '<br>') + '</div>'; return; }
|
|
const m = /^([^\n]*)\n?([\s\S]*)$/.exec(part);
|
|
const head = m[1].trim(), kind = head.split(/\s+/)[0], code = m[2].replace(/\n$/, '');
|
|
const closed = i < parts.length - 1 || done;
|
|
const act = LABEL[kind] && closed;
|
|
html += '<div class="ai-code">' + (act && head !== kind ? '<div class="muted ai-cap">' + esc(head) + '</div>' : '') +
|
|
(code || !act ? '<pre>' + esc(code) + '</pre>' : '') +
|
|
(act ? '<button type="button" class="btn small ai-act" data-i="' + (n++) + '">' + LABEL[kind] + '</button>' : '') + '</div>';
|
|
});
|
|
el.innerHTML = html;
|
|
el.dataset.text = text;
|
|
}
|
|
function add(role, text) {
|
|
q('.ai-hello').hidden = true;
|
|
const el = document.createElement('div');
|
|
el.className = 'ai-msg ' + role;
|
|
if (role === 'assistant') render(el, text, true); else el.textContent = text;
|
|
list.appendChild(el);
|
|
list.scrollTop = list.scrollHeight;
|
|
return el;
|
|
}
|
|
// A result goes into the talk as the user's message and shows folded.
|
|
function result(text) {
|
|
const el = add('result', '');
|
|
el.innerHTML = '<details><summary>Результат</summary><pre></pre></details>';
|
|
el.querySelector('pre').textContent = text;
|
|
talk.push({role: 'user', content: text});
|
|
}
|
|
list.addEventListener('click', async ev => {
|
|
const b = ev.target.closest('.ai-act');
|
|
if (!b || b.disabled) return;
|
|
const msg = b.closest('.ai-msg'), a = WSUtil.parseActions(msg.dataset.text || '')[Number(b.dataset.i)];
|
|
if (!a) return;
|
|
b.disabled = true;
|
|
b.textContent = '…';
|
|
const res = await perform(a);
|
|
b.textContent = /^Ошибка/.test(res) ? 'Ошибка' : 'Готово';
|
|
result(res);
|
|
});
|
|
|
|
async function ask(text) {
|
|
if (text != null) { talk.push({role: 'user', content: text}); add('user', text); }
|
|
const el = add('assistant', '');
|
|
el.classList.add('wait');
|
|
ac = new AbortController();
|
|
send.innerHTML = STOP; send.title = 'Остановить';
|
|
let answer = '';
|
|
try {
|
|
await loadMachines();
|
|
const r = await fetch('/api/panel/ai/chat', {
|
|
method: 'POST', headers: {'Content-Type': 'application/json'}, credentials: 'same-origin', signal: ac.signal,
|
|
body: JSON.stringify({messages: talk, context: context(), agent: agentOn(), think: thinkOn()})
|
|
});
|
|
if (r.redirected && new URL(r.url).pathname === '/login') { location.href = '/login'; return; }
|
|
if (!r.body) throw new Error('ошибка ' + r.status);
|
|
const rd = r.body.getReader(), dec = new TextDecoder();
|
|
for (;;) {
|
|
const {done, value} = await rd.read();
|
|
if (done) break;
|
|
answer += dec.decode(value, {stream: true});
|
|
el.classList.remove('wait');
|
|
render(el, answer, false);
|
|
list.scrollTop = list.scrollHeight;
|
|
}
|
|
} catch (e) {
|
|
if (e.name !== 'AbortError') answer += (answer ? '\n' : '') + '[ошибка: ' + (e.message || 'нет связи') + ']';
|
|
} finally {
|
|
el.classList.remove('wait');
|
|
render(el, answer || '[нет ответа]', true);
|
|
talk.push({role: 'assistant', content: stripThink(answer)});
|
|
}
|
|
return {el, answer: stripThink(answer)};
|
|
}
|
|
const pill = k => q('.ai-pill[data-mode="' + k + '"]');
|
|
const agentOn = () => pill('agent').getAttribute('aria-pressed') === 'true';
|
|
const thinkOn = () => pill('think').getAttribute('aria-pressed') === 'true';
|
|
// The modes are kept in this browser.
|
|
box.querySelectorAll('.ai-pill').forEach(b => {
|
|
try { b.setAttribute('aria-pressed', String(localStorage.getItem('lpra.ai.' + b.dataset.mode) === '1')); } catch (e) { /* off */ }
|
|
b.onclick = () => {
|
|
const on = b.getAttribute('aria-pressed') !== 'true';
|
|
b.setAttribute('aria-pressed', String(on));
|
|
try { localStorage.setItem('lpra.ai.' + b.dataset.mode, on ? '1' : ''); } catch (e) { /* not kept */ }
|
|
};
|
|
});
|
|
// talkTurn: one question, and in agent mode the steps after it - the
|
|
// first action of each answer done (asking first for commands and writes)
|
|
// and its result sent back - until an answer asks for nothing.
|
|
async function talkTurn(text) {
|
|
let turn = await ask(text);
|
|
for (let step = 0; agentOn() && ac && !ac.signal.aborted && step < 15; step++) {
|
|
const a = WSUtil.parseActions(turn.answer)[0];
|
|
if (!a) break;
|
|
const b = turn.el.querySelector('.ai-act');
|
|
if (needsYes(a) && !await UI.confirm(describe(a), {ok: a.kind === 'run' ? 'Выполнить' : 'Записать'})) {
|
|
if (b) { b.disabled = true; b.textContent = 'Отклонено'; }
|
|
result('Пользователь отклонил это действие. Предложи другое или закончи.');
|
|
turn = await ask(null);
|
|
continue;
|
|
}
|
|
if (b) { b.disabled = true; b.textContent = '…'; }
|
|
const res = await perform(a, ac.signal);
|
|
if (b) b.textContent = /^Ошибка/.test(res) ? 'Ошибка' : 'Готово';
|
|
result(res);
|
|
if (ac.signal.aborted) break;
|
|
turn = await ask(null);
|
|
}
|
|
ac = null;
|
|
send.innerHTML = PLANE; send.title = 'Отправить';
|
|
}
|
|
q('.ai-form-in').onsubmit = ev => {
|
|
ev.preventDefault();
|
|
if (ac) { ac.abort(); return; }
|
|
const text = input.value.trim();
|
|
if (!text) return;
|
|
input.value = '';
|
|
grow();
|
|
where();
|
|
talkTurn(text);
|
|
};
|
|
// The field grows with the text, up to a few lines.
|
|
function grow() { input.style.height = '40px'; input.style.height = Math.min(140, input.scrollHeight + 2) + 'px'; }
|
|
input.addEventListener('input', grow);
|
|
input.addEventListener('keydown', ev => {
|
|
if (ev.key === 'Enter' && !ev.shiftKey && !ev.isComposing) { ev.preventDefault(); q('.ai-form-in').requestSubmit(); }
|
|
});
|
|
})();
|