#import "./title-pages/main.typ" as tp
#import "./shared.typ": universities
#import "./helpers.typ": *

#import "./style.typ"
#import "./utils.typ"

/// Coursework template for NURE
/// - university (str): University code, default "ХНУРЕ"
/// - subject (str): Subject short name
/// - title (str): Work title
/// - authors (array): List of author dictionaries
/// - mentors (array): List of mentor dictionaries
/// - task-list (dict): Task metadata
/// - calendar-plan (dict): Calendar plan table and approval date
/// - abstract (dict): Keywords and abstract text
/// - bib-path (str): Path to bibliography file
/// - appendices (content): Appendix content
#let coursework(
  doc,
  university: "ХНУРЕ",
  subject: none,
  title: none,
  authors: (),
  mentors: (),
  task-list: (),
  calendar-plan: (),
  abstract: (),
  bib-path: none,
  appendices: (),
) = {
  set document(title: title, author: authors.map(c => c.name))

  show: style.dstu.with(skip: 1)

  tp.cw.nure(
    university,
    subject,
    title,
    authors,
    mentors,
    task-list,
    calendar-plan,
    abstract,
  )

  doc

  // Bibliography with DSTU formatting
  {
    show regex("^\\d+\\."): it => [#it#h(0.5cm)]
    show block: it => [#it.body#parbreak()]
    bibliography(bib-path, title: [Перелік джерел посилання], style: "csl/dstu-3008-2015.csl", full: true)
  }

  style.appendices(appendices)
}

/// Practice and Laboratory works template
/// - layout (str): "default", "minimal", or "complex"
/// - university (str): University code
/// - edu-program (str): Education program code
/// - subject (str): Subject code
/// - type (str): Work type (ЛБ, ПЗ, КР, РФ, ІДЗ)
/// - number (int): Work number
/// - title (str): Work title
/// - authors (array): List of authors
/// - mentors (array): List of mentors
#let pz-lb(
  doc,
  layout: "default",
  university: "ХНУРЕ",
  edu-program: none,
  subject: none,
  type: none,
  number: none,
  title: none,
  authors: (),
  mentors: (),
  skip-heading: false,
) = {
  assert(authors.len() > 0, message: "At least one author required")

  let edu-program = if edu-program != none { edu-program } else { authors.first().edu-program }
  let uni = universities.at(university)

  set document(title: title, author: authors.map(c => c.name))

  show: style.dstu.with(skip: 1)

  // Select layout variant
  let layouts = (
    "complex": () => tp.pz-lb.complex(uni, edu-program, subject, type, number, title, authors, mentors),
    "ХНУРЕ": () => tp.pz-lb.nure(uni, edu-program, subject, type, number, title, authors, mentors),
    "default": () => tp.pz-lb.nure(uni, edu-program, subject, type, number, title, authors, mentors),
  )

  (layouts.at(university, default: layouts.default))()

  if not skip-heading {
    pagebreak(weak: true)

    // Set heading counter based on title/number
    if title == none {
      if number == none { context counter(heading).update(1) } else {
        context counter(heading).update(number)
      }
    } else {
      if number != none {
        context counter(heading).update(number - 1)
      }

      heading(eval(title, mode: "markup"))
    }
  }

  doc
}