#import "@preview/ctheorems:1.1.3": *
#import "@preview/zebraw:0.5.5": zebraw, zebraw-init
#import "@preview/subpar:0.2.2"

/*
 * Примечания:
 * - Код разбит по модулям, каждый из которых объявляется и инициализируется
 *   в переменной modules. Методы именуются в змеином стиле, при этом
 *   приватные начинаются с подчёркивания.
 *
 *   В силу отсутствия адекватной реализации приватных методов в Typst
 *   приватность --- лишь условное соглашение: такие методы не предназначены
 *   для вызова извне, хотя фактически такая возможность имеется. Не делайте
 *   так!
 *
 * - Чтобы не засорять кодовую базу, строковые константы вынесены в
 *   переменную strings.
 */


#let strings = (
  title: (
    minobrnauki: "МИНОБРНАУКИ РОССИИ\nФедеральное государственное бюджетное образовательное учреждение\nвысшего образования\n",
    sgu: "«САРАТОВСКИЙ НАЦИОНАЛЬНЫЙ ИССЛЕДОВАТЕЛЬСКИЙ \n ГОСУДАРСТВЕННЫЙ УНИВЕРСИТЕТ \n ИМЕНИ Н. Г. ЧЕРНЫШЕВСКОГО»",
    city: "Саратов",
    worktypes: (
      referat: [РЕФЕРАТ],
      coursework: [КУРСОВАЯ РАБОТА],
      diploma: [Выпускная квалификационная работа],
      autoref: [работа],
      nir: [работа],
      pract: [Отчёт о практике],
    ),
    from_course: "курса",
    from_group: "группы",
    from_speciality: "направления",
    scientific_advisor: "Научный руководитель",
    department_chair: "Заведующий кафедрой",
    reviewed: "Проверено"
  ),
  student: (
    male: "студента",
    female: "студентки",
    plural: "студентов",
  ),
  caps_headings: (
    [ОПРЕДЕЛЕНИЯ, ОБОЗНАЧЕНИЯ И СОКРАЩЕНИЯ],
    [СОДЕРЖАНИЕ],
    [ВВЕДЕНИЕ],
    [ЗАКЛЮЧЕНИЕ],
    [СПИСОК ИСПОЛЬЗОВАННЫХ ИСТОЧНИКОВ],
  ),
  specialities: (
    spec_kb: [10.05.01 --- Компьютерная безопасность],
    bac_ivt: [09.03.01 --- Информатика и вычислительная техника],
    bac_moais: [02.03.03 --- Математическое обеспечение и администрирование информационных систем],
    bac_fiit: [02.03.02 --- Фундаментальная информатика и информационные технологии],
    bac_pi: [09.03.04 --- Программная инженерия],
    mag_ivt: [09.04.01 --- Информатика и вычислительная техника],
    mag_moais: [02.04.03 --- Математическое обеспечение и администрирование информационных систем],
  ),
  error: (
    no_sex: [*6.21 КоАП РФ*],
    undefined_spec: [*НЕИЗВЕСТНАЯ СПЕЦИАЛЬНОСТЬ*],
  ),
)

// Переменная отвечающая за размер отступа красной строки
#let indent = 1.25cm
#let toc_number_title_gap = 1em
// Level-2 numbers should start under the title text of their level-1 parent,
// not under the parent number.
#let toc_level_indent = 1.5em
#let font_size = 14pt
#let main_font = "Tempora"
#let figure_font_size = font_size - 2pt
#let line_spacing = 1.75em / 2
#let styled = [#set text(red)].func()
#let space = [ ].func()
#let sequence = [].func()

#let mono_small_font_size = font_size - 2pt
// Typst raw internally renders at about 80% of surrounding text size.
// To get 12pt code, as in LaTeX \small SCVerbatim, set the surrounding
// size to 15pt before passing raw to zebraw.
#let code_block_raw_base_size = 15pt
// Моноширинный шрифт для inline-code, блоков кода и URL.
// В LaTeX-шаблоне это T2A/cmtt (в PDF: SFTT1200/SFTT1440 из cm-super).
// Для Typst используем локально сохранённый OpenType-эквивалент:
// fonts/CMUTypewriterText-Regular.otf.
#let mono_font = ("CMU Typewriter Text", "Courier New")
// Номера строк в LaTeX fancyvrb набираются маленьким Computer Modern Roman,
// а не моноширинным шрифтом.
#let line_number_font = ("CMU Serif", "Times New Roman")
#let line_number_font_size = 6pt
#let line_number_baseline = 1pt
// Extra left inset for code lines: increases the visual gap between line
// numbers and code while the whole block is shifted left by the same amount.
#let code_numbering_gap_extra = 0.75em
#let code_numbering_outdent = 1.5em + code_numbering_gap_extra

#let code-inline(code) = text(font: mono_font, size: font_size, code.text)

#let code-block-raw(code) = {
  show raw: set text(font: mono_font)
  set text(size: code_block_raw_base_size)
  // TODO: нужно выяснить почему нужно переоборачивать raw, чтобы размер был
  // правильным
  // Сам zebraw строит breakable-блок и сохраняет непрерывную нумерацию строк
  // при разрыве страницы. Внешний block нужен только для расширения ширины
  // после выноса номеров строк в левое поле.
  pad(
    left: -code_numbering_outdent,
    zebraw(inset: (left: code_numbering_gap_extra), raw(block: true, code.text)),
  )
}

#let code-block(..args, code) = {
  show raw: set text(font: mono_font)
  set text(size: code_block_raw_base_size)
  pad(
    left: -code_numbering_outdent,
    zebraw(inset: (left: code_numbering_gap_extra), ..args, code),
  )
}

#let modules = (
  /*
   * Модуль информации об авторе Позволяет получать более полную и
   * отформатированную информацию об авторе: студент/студентки/студентов
   * (get_author_sex), номер курса, группы и так далее. Конечно, некоторые из
   * этих вещей в открытом виде лежат в словаре author, но некоторые данные
   * нуждаются в постобработке или вовсе получаются косвенным путём.
   * Необходимость в постобработке может возникнуть и позднее, поэтому
   * настоятельно рекомендуется пользоваться именно этими методами, а не
   * получать код напрямую.
   */
  author_info: (
    /*
     * Определяет студенческо-половую сущность автора.
     * Принимает:
     *  - author - словарь про автора
     * Возвращает:
     *  - В зависимости от пола: "студента", "студентки", "студентов"
     */
    get_author_sex: author => {
      return strings.student.at(
        author.at("sex"),
        default: strings.error.no_sex,
      )
    },
    /*
     * Определяет курс автора (по номеру группы)
     * Принимает:
     *  - author - словарь про автора
     * Возвращает:
     *  - Номер курса в строковом формате
     */
    get_author_course: author => {
      return author.group.at(0)
    },
    /*
     * Определяет группу автора
     * Принимает:
     *  - author - словарь про автора
     * Возвращает:
     *  - Номер группы в строковом формате
     * Примечание: на текущий момент функция ничего не делает, но, возможно,
     * это не навсегда и в будущем здесь что-то может появиться. Чтобы потом не
     * мучиться с рефакторингом, лучше получать номер группы через неё.
     */
    get_author_group: author => {
      return author.group
    },
    /*
     * Определяет код и название направления автора
     * Принимает:
     *  - author - словарь про автора
     * Возвращает:
     *  - author.speciality, если таковой указан
     *  - В противном случае, если студент учится на КНиИТе, для студента
     *    работает аттракцион невиданной технологичности: код и название
     *    специальности определяются автоматически
     */
    get_speciality: author => {
      if author.at("speciality", default: none) != none {
        return author.speciality
      } else if author.faculty == [КНиИТ] {
        let specialities = (
          strings.specialities.bac_fiit,
          strings.specialities.bac_ivt,
          strings.specialities.spec_kb,
          strings.specialities.bac_moais,
          strings.specialities.bac_pi,
          [], // x6x
          strings.specialities.mag_ivt,
          strings.specialities.mag_moais,
          [], // x8x
        )
        if author.group.at(1) == "7" {
          // x7x
          if author.group.at(2) == "1" {
            // x71
            return specialities.at(6) // ивт
          } else if author.group.at(2) == "3" {
            // x73
            return specialities.at(7) // моаис
          }
          return []
        }
        let id = int(author.group.at(1)) - 1

        if id < 0 or id > 8 {
          return []
        }
        return specialities.at(id)
      }
      return strings.error.undefined_spec
    },
  ),
  /*
   * Модуль титульного листа Здесь происходит генерация титульного листа и
   * определяются все необходимые для этого методы. Поскольку его создание ---
   * задача не такая уж и тривиальная, здесь есть много приватных методов и ваш
   * покорный слуга ещё раз напоминает о нежелательности их вызова извне. Если
   * такая необходимость всё же возникает, лучше переименовать метод и убрать
   * подчёркивание: впоследствии это можно будет хотя бы как-то отладить.
   *
   * Изначально задумывалось, что единственный доступный для вызова извне метод
   * --- make, а в остальных не должно возникнуть необходимости.
   */
  title: (
    /*
     * Отвечает за вывод названия министерства и
     * университета на титульном листе
     */
    _default_header: (chair: none) => {
      set align(center)
      // set text(font: "Tempora")
      text(strings.title.minobrnauki)
      v(0.2em)
      text(weight: "bold", strings.title.sgu)
      if chair != none and chair != "" {
        v(0em)
        text("Кафедра " + chair)
      }
      set align(left)
    },
    /*
     * Отвечает за вывод тела титульного листа: заголовок (название работы,
     * если не определено иное), тип работы, информация об авторе
     *
     * Информация о проверяющем преподавателе генерируется не здесь по
     * историческим причинам.
     */
    _default_body: data => {
      set align(center)
      v(3cm)
      text(weight: "bold", upper(data.title))
      par(data.worktype)
      v(1.5cm)
      set align(left)
      text(data.group + "\n")
      text(data.speciality + "\n")
      text(data.faculty + "\n" + data.author)
    },
    /*
     * Отвечает за вывод города и года на титульном листе
     */
    _default_footer: () => {
      v(1fr)
      set align(center)
      text(strings.title.city + " " + str(datetime.today().year()))
    },
    /*
     * Подпись для титульного листа
     */
    _signature: (comment, post, name) => {
      text(comment)
      v(-0.5em)
      grid(
        columns: (1fr,) * 3,
        align: (left, center, right),
        row-gutter: 5pt,
        post, block(inset: (y: 13pt), line(length: 3cm, stroke: .4pt)), name
      )
    },
    /*
     * Строка "студента такой-то группы" для титульного листа
     */
    _get_author_string: (self, author) => {
      let sex = (self.author_info.get_author_sex)(author)
      let course = (self.author_info.get_author_course)(author)
      let group = (self.author_info.get_author_group)(author)
      if course.len() > 0 {
        course = (self.utils.strglue)(course, strings.title.from_course)
      }
      if group.len() > 0 {
        group = (self.utils.strglue)(group, strings.title.from_group)
      }
      let result = (self.utils.strglue)(sex, course, group)
      return result
    },
    /*
     * Получает заголовок титульного листа.
     * Принимает:
     *  - info - информация о документе
     * Возвращает:
     *  - В зависимости от типа:
     *    - Тему работы, если это не автореферат и не отчёт по НИРу
     *    - В противном случае названия соответствующих типов работ
     */
    _get_title_string: info => {
      if info.type == "autoref" {
        return [АВТОРЕФЕРАТ]
      }
      if info.type == "nir" {
        return [ОТЧЁТ О НАУЧНО-ИССЛЕДОВАТЕЛЬСКОЙ РАБОТЕ]
      }
      return info.at("title", default: [Тема работы])
    },
    /*
     * Генерирует строки текста для вывода на титульном листе
     * Принимает:
     *  - info - информация о документе
     * Возвращает:
     *  - Словарь:
     *     title: Заголовок
     *     worktype: Тип работы
     *     group: студент(а|ки|ов) s курса sex группы
     *     specialty: направления 69.14.88 --- Специальность
     *     faculty: факультета XXX
     *     author: автор(ы)? работы
     */
    _get_strings: (self, info) => {
      let author = info.at("author", default: (:))
      let title_string = (self.title._get_title_string)(info)
      let worktype = strings.title.worktypes.at(info.type, default: [])
      let group_string = (self.title._get_author_string)(self, author)
      let speciality_string = (self.utils.strglue)(
        strings.title.from_speciality,
        (self.author_info.get_speciality)(author),
      )
      let faculty_string = "факультета " + author.faculty
      return (
        title: title_string,
        worktype: worktype,
        group: group_string,
        speciality: speciality_string,
        faculty: faculty_string,
        author: author.name,
      )
    },
    /*
     * Генерирует титульный лист
     * Принимает:
     *  - info - информация о документе
     */
    make: (self, info) => {
      let strs = (self.title._get_strings)(self, info)
      (self.title._default_header)(chair: info.at("chair", default: none))
      (self.title._default_body)(strs)
      v(1fr)
      if info.type == "referat" {
        (self.title._signature)(info.inspector.degree, info.inspector.name)
      } else if info.type == "coursework" {
        (self.title._signature)(strings.title.scientific_advisor, info.scientific_advisor.degree, info.scientific_advisor.name)
        (self.title._signature)(strings.title.department_chair, info.department_chair.degree, info.department_chair.name)
      }
      (self.title._default_footer)()
    },
  ),
  /*
   * Модуль генерации документа Здесь содержатся методы, влияющие на вид всего
   * документа в целом. Главный из них --- make --- вызывается из точки входа в
   * стилевой файл и отвечает за всё оформление выходного документа.
   */
  document: (
    apply_heading_styles: it => {
      set text(size: font_size)
      if it.depth == 1 {
        pagebreak(weak: true)
      }
      if strings.caps_headings.contains(it.body) {
        align(center, it.body)
      } else {
        pad(left: indent, it)
      }
      v(line_spacing / 2)
    },
    /*
     * Генерирует страницу содержания
     * Принимает:
     *  - info - информация о документе
     */
    make_toc: (info: ()) => {
      show outline.entry: it => {
        let heading-text = it
          .at("element", default: (:))
          .at("body", default: "")

        let is-outlined = it.element.outlined
        let is-annex = state("annex", false).at(it.element.location())
        let is-caps = heading-text in strings.caps_headings

        // TODO: Хотелось бы запретить создавать ненумерованные заголовки. По
        // какой-то причине это не срабаотывает для самого заголовка содержания. Не
        // понятно почему оно сюда вообще попадает.
        // assert(is-caps or (not is-outlined) or it.element.numbering != none)

        if is-annex or is-caps {
          let prefix = none
          if is-annex {
            prefix = [ПРИЛОЖЕНИЕ #it.prefix() #heading-text]
          } else if is-caps {
            prefix = heading-text + sym.space
          }
          return link(
            it.element.location(),
            it.indented(
              none,
              prefix
                + sym.space
                + box(width: 1fr, it.fill)
                + sym.space
                + sym.wj
                + it.page(),
            ),
          )
        }

        link(
          it.element.location(),
          pad(
            left: toc_level_indent * (it.level - 1),
            grid(
              columns: (auto, toc_number_title_gap, 1fr),
              column-gutter: 0pt,
              align: top,
              it.prefix(),
              [],
              heading-text
                + sym.space
                + box(width: 1fr, it.fill)
                + sym.space
                + sym.wj
                + it.page(),
            ),
          ),
        )
      }
      outline(title: [СОДЕРЖАНИЕ])
    },
    /*
     * Генерирует весь документ
     * Принимает:
     *  - info - информация о документе
     *  - doc  - содержимое документа
     */
    make: (self, info: (), settings, doc) => {
      set page(
        paper: "a4",
        margin: (
          top: 2cm,
          bottom: 2cm,
          left: 2.5cm,
          right: 1.5cm,
        ),
      )
      set text(size: font_size, lang: "ru", font: main_font)

      if settings.title_page.at("enabled", default: true) {
        (self.title.make)(self, info)
      }

      show: thmrules.with(qed-symbol: $square$)
      show heading: self.document.apply_heading_styles
      show link: it => {
        if type(it.dest) == str and it.dest.starts-with("http") {
          text(font: mono_font, size: mono_small_font_size, it.body)
        } else {
          it
        }
      }
      show raw.where(block: false): code-inline
      show raw.where(block: true): code-block-raw

      set par(
        // Выравнивание по ширине
        justify: true,
        // отвечает за красные строки там, где их нет, но они должны быть
        first-line-indent: (amount: indent, all: true),
        leading: line_spacing,
        spacing: line_spacing,
      )

      // Вывод содержания
      if settings.contents_page.enabled {
        (self.document.make_toc)(info: info)
      }

      // Оформление элементов содержимого документа
      set ref(supplement: "")
      set heading(numbering: "1.1", hanging-indent: -indent)
      set page(
        footer: context [
          #h(1fr)
          #counter(page).display("1")
        ],
      )
      set page(numbering: "1")
      set math.equation(numbering: "(1)")

      set figure.caption(separator: [ -- ])
      show figure.where(kind: image): set figure(supplement: [Рисунок])
      show figure.caption: set text(size: figure_font_size)

      show figure.where(kind: table): set figure.caption(position: top)
      show figure.where(kind: table): set text(size: figure_font_size)
      show figure.where(kind: table): set block(breakable: true)
      show figure.caption.where(kind: table): set align(left)

      set terms(separator: [ --- ])

      set quote(block: true)

      set bibliography(title: "СПИСОК ИСПОЛЬЗОВАННЫХ ИСТОЧНИКОВ")
      set bibliography(style: "gost-7-1-2003.csl")

      // Примерный размер "---". Определено на глаз, как сделать более точным
      // не разбирался.
      let shift-back-list = 1.5em
      set list(indent: indent - shift-back-list, marker: [#strong([---])])
      let shift-back-enum = 1.25em
      set enum(indent: indent - shift-back-enum)

      // Вывод самого документа
      doc
    },
  ),
  /*
   * Помощник Модуль-помощник не содержит особой функциональности и не имеет
   * конкретного назначения, но содержащиеся в нём методы могут быть полезны
   * где угодно и не привязаны к конкретной части документа
   */
  utils: (
    /*
     * Склеивает строки, игнорируя пустые
     * Принимает:
     *  - divider - разделитель (по умолчанию пробел)
     *  - соединяемые строки в неограниченном количестве
     * Примечание: метод похож по своей сути на join, но отличается от него
     * наличием проверки на пустые строки: рядом с ними разделитель не
     * ставится, что исключает присутствие двух разделителей подряд.
     */
    strglue: (divider: " ", ..strings) => {
      let result = ""
      for string in strings.pos() {
        if (
          (result.len() > 0) and ((type(string) != str) or (string.len() > 0))
        ) {
          result = result + " "
        }
        // TODO: string было бы неплохо обернуть в str(), но баг в системе
        // типов Typst не позволяет этого сделать
        result = result + string
      }
      return result
    },
  ),
)

#let defabbr = {
  heading(
    numbering: none,
    outlined: true,
    [ОПРЕДЕЛЕНИЯ, ОБОЗНАЧЕНИЯ И СОКРАЩЕНИЯ],
  )
}

#let intro = {
  heading(numbering: none, outlined: true, [ВВЕДЕНИЕ])
}


#let conclusion = {
  heading(numbering: none, outlined: true, [ЗАКЛЮЧЕНИЕ])
}

#let annex-letters = "АБВГДЕЖЗИКЛМНПРСТУФХЦШЩЭЮЯABCDEFGHJKLMNPQRSTUVWXYZ"
#let annex-numbering(..nums) = {
  let number = nums.pos().first() - 1
  annex-letters.clusters().at(number, default: str(number))
}

// thx https://github.com/typst-g7-32/modern-g7-32
#let annexes-start(it) = {
  set heading(
    numbering: annex-numbering,
    outlined: true,
  )
  set align(center)

  show heading: it => {
    if it.depth == 1 {
      pagebreak(weak: true)
    }
    set text(size: font_size)
    let letter = counter(heading).display(annex-numbering)
    [ПРИЛОЖЕНИЕ #letter \ #it.body #v(line_spacing / 2)]
  }

  state("annex").update(true)
  counter(heading).update(0)

  it
}

#let thm-format = thmplain.with(
  inset: 0em,
  padding: (y: font_size / 2),
  titlefmt: strong,
  bodyfmt: emph,
  namefmt: x => [(#x)],
  base: none,
  separator: [#strong(".")#h(0.2em)],
)
#let definition = thm-format("definition", "Определение")
#let theorem = thm-format("theorem", "Теорема")
#let proof = thmproof("proof", "Доказательство")

// TODO: по какой-то причине эта функция всегда вставляет пустую строку в конце
#let source-file(file) = {
  raw(read(file), block: true)
}

/*
 * Точка входа, просто вызывает modules.document.make
 */
#let conf(
  title: none,
  info: (),
  type: "referat",
  settings: (),
  doc,
) = {
  info.title = title
  info.type = type
  settings.title_page = settings.at("title_page", default: (:))
  (modules.document.make)(
    modules,
    info: info,
    settings,
    [
      #show: zebraw-init.with(
        background-color: auto,
        hanging-indent: true,
        numbering-font-args: (
          font: line_number_font,
          size: line_number_font_size,
          baseline: line_number_baseline,
        ),
        lang-font-args: (font: mono_font),
      )
      #doc
    ],
  )
}