flang: выборочный импорт «только» — чтобы не дублировать чужой код

Импорт вносил все имена модуля, и достаточно было одного совпадения, чтобы
связывание отказало с FLANG_DUPLICATE_NAME. Цена выяснилась сразу: лексеру
ядра пришлось скопировать к себе четыре тотальные функции stdlib из-за одной
одноимённой — «Обрезать пробелы», которая в библиотеке обычная, а в лексере
обязана быть тотальной.

`использует «Списки» из "…" только «Сумма», «Длина»` берёт названное и
оставляет остальное снаружи. Просьба не расширяет видимое: имя, не попавшее в
«экспортирует», не придёт и по «только» — иначе экспорт перестал бы что-либо
значить.

Проверено: 870 тестов flang, 0 падений.

Author: the-homeless-god

flang/src/lexer.mjs

@@ -86,6 +86,7 @@ export const KEYWORDS = {
   and: ["и", "and"],
   with: ["с", "with"],
   from: ["из", "from"],
+  only: ["только", "only"],
   to: ["в", "к", "to", "into", "in", "onto"],
   by: ["по", "by"],
   at: ["у", "at"],

flang/src/link.mjs

@@ -63,6 +63,18 @@ export async function linkProgram(entryFile, source, parse, options = {}) {
   const sources = new Map()
   const withImports = new Set()
 
+  /**
+   * Пересечение «что модуль отдаёт» и «что у него просят».
+   *
+   * Просьба не расширяет видимое: `только «Внутри»` для неэкспортированного
+   * имени ничего не даёт — иначе `экспортирует` перестал бы что-либо значить.
+   */
+  const narrow = (exported, requested) => {
+    if (requested === undefined || requested === null) return exported
+    if (exported === null) return requested
+    return new Set([...requested].filter((name) => exported.has(name)))
+  }
+
   const merge = (program, file, visible) => {
     for (const type of program.types ?? []) {
       if (visible !== null && !visible.has(type.name)) continue
@@ -102,6 +114,10 @@ export async function linkProgram(entryFile, source, parse, options = {}) {
     }
   }
 
+  /* Что именно берём из каждого файла: `null` — всё видимое, иначе набор имён
+     из `только`. Пересечение с `экспортирует` считается в merge. */
+  const wanted = new Map()
+
   const load = async (file, text, isEntry) => {
     if (loaded.has(file)) return null
     /* Цикл импортов — ошибка, а не молчаливая остановка: иначе часть объявлений
@@ -155,6 +171,12 @@ export async function linkProgram(entryFile, source, parse, options = {}) {
         )
         continue
       }
+      if (Array.isArray(entry.only)) {
+        const already = wanted.get(target)
+        wanted.set(target, already === undefined ? new Set(entry.only) : new Set([...already, ...entry.only]))
+      } else {
+        wanted.set(target, null)
+      }
       const child = await load(target, imported, false)
       /* Имя модуля в `использует` обязано совпадать с его заголовком: иначе
          читатель видит одно имя, а получает объявления из другого файла. */
@@ -170,7 +192,7 @@ export async function linkProgram(entryFile, source, parse, options = {}) {
 
     loading.pop()
     loaded.add(file)
-    merge(program, file, isEntry ? null : exportsOf(program))
+    merge(program, file, isEntry ? null : narrow(exportsOf(program), wanted.get(file)))
     return program
   }
 

flang/src/parser.mjs

@@ -568,7 +568,18 @@ class Parser {
       const category = this.expectName("ожидалось имя категории")
       this.expectKw("from", "после имени категории ожидалось 'из'")
       const from = this.expectName("ожидался путь")
-      value.imports.push({ category, from })
+      /* `только «А», «Б»` — выборочный импорт. Без него модуль вносит все свои
+         имена, и достаточно одного совпадения, чтобы связывание отказало:
+         именно так лексеру ядра пришлось дублировать четыре функции stdlib
+         из-за одной одноимённой. Выбор имён — способ сказать «мне нужна вот
+         эта функция», не переименовывая чужой модуль. */
+      const entry = { category, from }
+      if (this.eatKw("only")) {
+        const names = [this.expectName("ожидалось имя, которое нужно импортировать")]
+        while (this.eatPunct(",")) names.push(this.expectName("ожидалось имя, которое нужно импортировать"))
+        entry.only = names
+      }
+      value.imports.push(entry)
       this.endLine()
       return
     }

flang/test/link.test.mjs

@@ -297,6 +297,96 @@ test("у входного файла видно всё, даже если он 
   assert.deepEqual(names.sort(), ["Внутри", "Наружу"])
 })
 
+/* ─────────────────────────── выборочный импорт ──────────────────────────── */
+
+test("«только» берёт названные имена и не тащит остальные", async () => {
+  const dir = await sandbox({
+    "решение.flang": `модуль «Решение»
+  использует «Списки» из ${JSON.stringify(join(stdlib, "lists.flang"))} только «Сумма», «Длина»
+
+тотальная функция «Среднее»
+  принимает элементы: список числа
+  возвращает число
+  «Сумма» от элементы делить на «Длина» от элементы
+`,
+  })
+  const result = await link(join(dir, "решение.flang"), await readEntry(dir, "решение.flang"))
+  assert.deepEqual(result.functions.map((fn) => fn.name).sort(), ["Длина", "Среднее", "Сумма"])
+
+  const { stdout } = await run("node", [
+    cli,
+    "run",
+    join(dir, "решение.flang"),
+    "--function",
+    "Среднее",
+    "--args",
+    JSON.stringify({ элементы: [2, 4, 6] }),
+  ])
+  assert.equal(JSON.parse(stdout).result, 4)
+})
+
+test("«только» снимает конфликт: одноимённую функцию можно не брать", async () => {
+  const dir = await sandbox({
+    "чужой.flang": `модуль «Чужой»
+
+тотальная функция «Нужная»
+  возвращает число
+  1
+
+тотальная функция «Общая»
+  возвращает число
+  2
+`,
+    "главный.flang": `модуль «Главный»
+  использует «Чужой» из "чужой.flang" только «Нужная»
+
+тотальная функция «Общая»
+  возвращает число
+  3
+
+тотальная функция «Позвать»
+  возвращает число
+  «Нужная» плюс «Общая»
+`,
+  })
+  const result = await link(join(dir, "главный.flang"), await readEntry(dir, "главный.flang"))
+  assert.deepEqual(
+    result.diagnostics,
+    [],
+    `конфликта быть не должно: одноимённая функция не импортировалась, получено ${JSON.stringify(result.diagnostics)}`,
+  )
+  const { stdout } = await run("node", [cli, "run", join(dir, "главный.flang"), "--function", "Позвать", "--args", "{}"])
+  assert.equal(JSON.parse(stdout).result, 4)
+})
+
+test("«только» не расширяет видимое сверх «экспортирует»", async () => {
+  const dir = await sandbox({
+    "модуль.flang": `модуль «Модуль»
+  экспортирует «Наружу»
+
+тотальная функция «Наружу»
+  возвращает число
+  1
+
+тотальная функция «Внутри»
+  возвращает число
+  2
+`,
+    "главный.flang": `модуль «Главный»
+  использует «Модуль» из "модуль.flang" только «Внутри»
+
+тотальная функция «Ноль»
+  возвращает число
+  0
+`,
+  })
+  const result = await link(join(dir, "главный.flang"), await readEntry(dir, "главный.flang"))
+  assert.ok(
+    !result.functions.some((fn) => fn.name === "Внутри"),
+    "неэкспортированное имя не должно приходить по просьбе",
+  )
+})
+
 /* ──────────────────── файл без импортов не меняется вовсе ───────────────── */
 
 test("без «использует» разбор остаётся прежним: связывание ничего не подменяет", async () => {