リポジトリ / emssr / independent_common_compiler

// Common scope の VS Code 拡張。
//
// web 版と同じ server.py を使う。拡張が足すのは三つだけになる。
//
//   1. server.py の起動と停止
//   2. notebook を webview で開く
//   3. 編集中の .cmn を hub へ送り、開いている全パネルを更新する
//
// 計算は拡張の中に無い。平文は server の /state へ渡り、各ウィンドウの
// kernel がそれぞれ実行する。lane の作業領域は交わらない（20 式 1.2）ので、
// 何枚開いても state が混ざらない。

const vscode = require('vscode');
const cp = require('child_process');
const fs = require('fs');
const path = require('path');
const http = require('http');

let server = null;      // 起動した server.py の process
let out = null;         // その標準出力を写す channel

function cfg(key) {
  return vscode.workspace.getConfiguration('commonScope').get(key);
}

// ===== server.py の場所 =====
//
// 設定が空なら、開いている folder の下から common_scope_notebook/server.py を探す。

function findRoot() {
  const set = cfg('root');
  if (set) return set;
  const rel = path.join('continuous_republic', 'common_apps', 'common_scope_notebook');
  for (const f of vscode.workspace.workspaceFolders || []) {
    for (const c of [path.join(f.uri.fsPath, rel), f.uri.fsPath]) {
      if (fs.existsSync(path.join(c, 'server.py'))) return c;
    }
  }
  return null;
}

function url(pathname) {
  return `http://127.0.0.1:${cfg('port')}${pathname || '/'}`;
}

function alive() {
  return new Promise(ok => {
    const req = http.get(url('/api/state'), r => { r.resume(); ok(r.statusCode === 200); });
    req.on('error', () => ok(false));
    req.setTimeout(400, () => { req.destroy(); ok(false); });
  });
}

async function ensureServer() {
  if (await alive()) return true;
  const root = findRoot();
  if (!root) {
    vscode.window.showErrorMessage(
      'common_scope_notebook が見つかりません。設定 commonScope.root にフォルダを書いてください。');
    return false;
  }
  out = out || vscode.window.createOutputChannel('Common scope');
  server = cp.spawn(cfg('python'),
    [path.join(root, 'server.py'), '--port', String(cfg('port')), '--no-browser'],
    { cwd: root });
  server.stdout.on('data', d => out.append(d.toString()));
  server.stderr.on('data', d => out.append(d.toString()));
  server.on('exit', c => { out.appendLine(`server 終了 (${c})`); server = null; });

  for (let k = 0; k < 40; k++) {
    await new Promise(r => setTimeout(r, 250));
    if (await alive()) return true;
  }
  out.show(true);
  vscode.window.showErrorMessage('server.py が起動しません。出力を確認してください。');
  return false;
}

// ===== hub へ送る =====

function post(body) {
  return new Promise((ok, no) => {
    const data = Buffer.from(JSON.stringify(body), 'utf8');
    const req = http.request(url('/state'),
      { method: 'POST', headers: { 'Content-Length': data.length } },
      r => { r.resume(); ok(r.statusCode); });
    req.on('error', no);
    req.end(data);
  });
}

// 同じフォルダに .dic があれば採用語彙として一緒に送る
function dicBeside(file) {
  const dir = path.dirname(file);
  const own = path.join(dir, path.basename(file, '.cmn') + '.dic');
  if (fs.existsSync(own)) return fs.readFileSync(own, 'utf8');
  for (const f of fs.readdirSync(dir)) {
    if (f.endsWith('.dic')) return fs.readFileSync(path.join(dir, f), 'utf8');
  }
  return null;
}

async function send() {
  const ed = vscode.window.activeTextEditor;
  if (!ed || !ed.document.fileName.endsWith('.cmn')) {
    vscode.window.showWarningMessage('.cmn を開いてから実行してください。');
    return;
  }
  if (!(await ensureServer())) return;
  await post({
    notebook: path.basename(ed.document.fileName),
    spec: { source: ed.document.getText(), dic: dicBeside(ed.document.fileName),
            seeds: null, words: null, budget: null },
    view: { inputs: null, cut: null, out: 0 },
  });
}

// ===== webview =====

function panel(title, pathname) {
  const p = vscode.window.createWebviewPanel('commonScope', title,
    vscode.ViewColumn.Beside, { enableScripts: true, retainContextWhenHidden: true });
  const src = url(pathname);
  p.webview.html = `<!DOCTYPE html><html><head><meta charset="utf-8">
<style>html,body,iframe{margin:0;padding:0;height:100%;width:100%;border:0}</style>
</head><body><iframe src="${src}"></iframe></body></html>`;
  return p;
}

const PANES = {
  '層と机': 'chamber', '箱の DAG': 'dag', 'ツイ数平面': 'tsui',
  'lane64': 'lane', '接合秩': 'joint', '音節と箱': 'text',
};

function activate(ctx) {
  ctx.subscriptions.push(
    vscode.commands.registerCommand('commonScope.open', async () => {
      if (await ensureServer()) panel('Common scope notebook', '/');
    }),
    vscode.commands.registerCommand('commonScope.send', send),
    vscode.commands.registerCommand('commonScope.pane', async () => {
      const pick = await vscode.window.showQuickPick(Object.keys(PANES),
        { placeHolder: '開くパネル' });
      if (!pick) return;
      if (await ensureServer()) panel(pick, '/?pane=' + PANES[pick]);
    }),
    vscode.commands.registerCommand('commonScope.stop', () => {
      if (server) { server.kill(); server = null; }
    }));
}

function deactivate() {
  if (server) server.kill();
}

module.exports = { activate, deactivate };