lib.typ

#import "@preview/hydra:0.5.2": hydra
#import "@preview/t4t:0.4.1": get
#import "@preview/headcount:0.1.0": dependent-numbering
#import "@preview/subpar:0.2.0"

#let defaults = (
  meta: (
    project-group: "No group name provided",
    participants: (),
    supervisors: (),
    field-of-study: "Computer Science",
    project-type: "Semester Project"
  ),
  en: (
    title: "Untitled",
    theme: "",
    abstract: [],
    department: "Department of Computer Science",
    department-url: "https://www.cs.aau.dk",
  ),
  dk: (
    title: "Uden titel",
    theme: "",
    abstract: [],
    department: "Institut for Datalogi",
    department-url: "https://www.dat.aau.dk",
  ),
)


#let _is-chapter-page(chapter-label) = {
  let current = counter(page).get()
  return query(chapter-label).any(m =>
    counter(page).at(m.location()) == current
  )
}

// only applies when pages are not roman numbered, thus no label argument
#let _custom-header(name: none) = context {
  if not _is-chapter-page(<chapter>) {
    if calc.even(here().page()) [
      #counter(page).display("1 of 1", both: true)
      #h(1fr)
      #name #hydra(1)
    ] else [
      #hydra(2)
      #h(1fr)
      #counter(page).display("1 of 1", both: true)
    ]
  }
}

#let _custom-footer(chapter-label) = context {
  if _is-chapter-page(chapter-label) {
    align(center, counter(page).display(page.numbering))
  }
}

#let clear-double-page() = {
  set page(header: none, footer: none)
  pagebreak(weak: true, to: "odd")
}


#let _set-chapter-style(numbering: none, name: none, body) = {
  set heading(numbering: numbering, outlined: true)
  set page(header: _custom-header(name: name))
  counter(heading).update(0) // Reset the chapter counter (appendices start at A)
  // numbering of figures and equations
  set figure(numbering: dependent-numbering(numbering)) if numbering != none
  set math.equation(numbering: dependent-numbering("(" + numbering + ")")) if numbering != none
  // set subpar.grid(numbering: depnedent-numbering(numbering)) if numbering != none
  
  // references show chapter / appendix
  show heading.where(level: 1): set heading(supplement: name)
  
  show heading.where(level: 1): it => {
    clear-double-page()

    counter(figure.where(kind: image)).update(0)
    counter(figure.where(kind: table)).update(0)
    counter(figure.where(kind: raw)).update(0)
    counter(math.equation).update(0)
    
    set par(first-line-indent: 0pt, justify: false)
    show: block
    v(3cm)
    if name != none { // set chapter/appendix whatever if exists
      text(size: 18pt)[#it.supplement #counter(heading).display()]
      v(.5cm)
    }
    text(size: 24pt)[#it.body <chapter>] // allow us to query this label to make header work properly
    v(.75cm)
  }
  body // actually show what comes afterwards
}

#let _custom-heading(it, p-top, p-bottom) = {
  if it.numbering == none {
    block(pad(top: p-top, bottom: p-bottom, it.body))
  } else {
    pad(top: p-top, bottom: p-bottom, grid(
      columns: (30pt, 1fr),
      counter(heading).display(it.numbering),
      it.body
    ))
  }
}

#let mainmatter(body) = {
  clear-double-page()
  set page(
    numbering: "1",
    header: _custom-header(),
    footer: _custom-footer(<chapter>)
  )
  counter(page).update(1)
  _set-chapter-style(numbering: none, name: none, body)
}

#let chapters(body) = {
  _set-chapter-style(numbering: "1.1", name: "Chapter", body)
}

#let backmatter(body) = {
  _set-chapter-style(numbering: none, name: none, body)
}

#let appendix(body) = {
  _set-chapter-style(numbering: "A.1", name: "Appendix", body)
}

#let _today = datetime.today()
#let _summer = datetime(year: _today.year(), month: 7, day: 1)
#let _is-spring-semester = _today < _summer
#let _semester-dk = if _is-spring-semester {
  "Forårssemesteret"
} else {
  "Efterårssemesteret"
}
#let _semester-en = if _is-spring-semester {
  "Spring Semester"
} else {
  "Fall Semester"
}

// English Abstract page.
#let _titlepage-en(meta, en) = {
  page(
    grid(
      columns: (1fr, 1fr),
      rows: (3fr, 7fr, 30pt),
      column-gutter: 10pt,
      image("/AAUgraphics/aau_logo_en.svg", width: 90%),
      align(right + horizon)[
        #strong(en.department)\
        Aalborg University\
        #link(en.department-url)
      ],

      grid(
        gutter: 16pt,
        [*Title:*\ #en.title],
        [*Theme:*\ #en.theme],
        [*Project Period:*\ #_semester-en #_today.year()],
        [*Project Group:*\ #meta.project-group],
        [*Participants:*\ #meta.participants.join("\n")],
        if type(meta.supervisors) == array [
          *Supervisors:*\ #meta.supervisors.join("\n")
        ] else [
          *Supervisor:*\ #meta.supervisors
        ],
        [*Copies:* 1],
        [*Number of Pages:* #context counter(page).final().first()],
        [*Date of Completion:*\ #datetime.today().display("[day]/[month]-[year]")],
      ),
      [*Abstract:*\ #box(width: 100%, stroke: .5pt, inset: 4pt, en.abstract)],

      grid.cell(colspan: 2, text(size: 10pt)[
        _The content of this report is freely available, but publication (with reference) may only be pursued due to agreement with the author._
      ])
    )
  )
}

// Danish Abstract page.
#let _titelside-dk(meta, dk) = {
  set text(lang: "da")
  page(
    grid(
      columns: (1fr, 1fr),
      rows: (3fr, 7fr, 30pt),
      column-gutter: 10pt,
      image("/AAUgraphics/aau_logo_da.svg", width: 90%),
      align(right + horizon)[
        #strong(dk.department)\
        Aalborg Universitet\
        #link(dk.department-url)
      ],

      grid(
        gutter: 16pt,
        [*Titel:*\ #dk.title],
        [*Tema:*\ #dk.theme],
        [*Projektperiode:*\ #_semester-dk #_today.year()],
        [*Projektgruppe:*\ #meta.project-group],
        [*Deltagere:*\ #meta.participants.join("\n")],
        if type(meta.supervisors) == array [
          *Vejledere:*\ #meta.supervisors.join("\n")
        ] else [
          *Vejleder:*\ #meta.supervisors
        ],
        [*Opsalgstal:* 1],
        [*Sidetal:* #context counter(page).final().first()],
        [*Afleveringsdato:*\ #datetime.today().display("[day]/[month]-[year]")],
      ),
      [*Resumé:*\ #box(width: 100%, stroke: .5pt, inset: 4pt, dk.abstract)],

      grid.cell(colspan: 2, text(size: 10pt)[
        _Rapportens indhold er frit tilgængeligt, men offentliggørelse (med kildeangivelse) må kun ske efter aftale med forfatterne._
      ])
    )
  )
  set text(lang: "en")
}

#let project(
  meta: (:),
  en: (:),
  dk: (:),
  body,
) = {
  let dk-is-set = dk != (:)

  meta = get.dict-merge(defaults.meta, meta)
  en = get.dict-merge(defaults.en, en)
  dk = get.dict-merge(defaults.dk, dk)

  // Set the document's basic properties.
  set document(author: meta.participants, title: en.title)
  set page(numbering: none)

  let aau-blue = rgb(33, 26, 82)
  let body-font = ("Palatino Linotype", "New Computer Modern") // primary and fallback

  // Set document preferences, font family, heading format etc.
  set text(font: body-font, lang: "en")
  set par(
    first-line-indent: 1em,
    spacing: 0.65em,
    justify: true,
  )

  set figure(numbering: dependent-numbering("1.1"))
  set math.equation(numbering: dependent-numbering("(1.1)"))

  // spacing around enumerations and lists
  show enum: set par(spacing: .5cm) // spacing around whole list
  show list: set par(spacing: .5cm)
  set enum(indent: .5cm, spacing: .5cm) // indentation and spacing between elements
  set list(indent: .5cm, spacing: .5cm)

  show terms: set par(first-line-indent: 0pt, spacing: 1em)
  show quote: set pad(x: 2em)

  show link: it => if type(it.dest) == str { // external link
    underline(it)
  } else { it }
  
  show bibliography: set par(spacing: 1em)

  // Chapter styling (preface, outline etc)
  show heading.where(level: 1): it => {
    clear-double-page()
    set par(first-line-indent: 0pt, justify: false)
    show: block
    v(3cm)
    text(size: 24pt)[#it.body <titlepages-chapter>] // label allows for header/footer to work properly
    v(.75cm)
  }
  // style heading with spacing before and after + spacing between number and name
  show heading.where(level: 2): it => _custom-heading(it, 10pt, 5pt)
  show heading.where(level: 3): it => _custom-heading(it, 5pt, 3pt)
  set heading(outlined: false) // changed in frontmatter

  // can be customized with figure.where(kind: table) etc...
  show figure: f => pad(top: 5pt, bottom: 10pt, f)
  // bold the figure title and number
  show figure.caption: c => context {
    set text(10pt)
    strong[#c.supplement #c.counter.display(c.numbering)#c.separator]
    c.body
  }

  show outline.entry.where(level: 1): it => {
    v(12pt, weak: true)
    strong(it)
  }

  // Front/cover page.
  page(
    background: image("/AAUgraphics/aau_waves.svg", width: 100%, height: 100%),
    grid(
      columns: (100%), // needed to not set uneven margins
      rows: (50%, 20%, 30%),
      align(center + bottom,
        box(fill: aau-blue, inset: 18pt, radius: 1pt, clip: false, {
          set text(fill: white, 12pt)
          align(center)[
            #text(2em, weight: 700, en.title)\
            #v(5pt)
            #en.theme\
            #v(10pt)
            #meta.participants.join(", ", last: " & ")\
            #text(10pt)[Computer Science, #meta.project-group, #datetime.today().year()]
            #v(10pt)
            #meta.project-type
          ]
        }
      )),
      none,
      align(center, image("/AAUgraphics/aau_logo_circle_en.svg", width: 25%))
    )
  )

  counter(page).update(1)
  set page(numbering: "I", footer: none, margin: (inside: 2.8cm, outside: 4.1cm))

  page(align(bottom)[
    #set text(size: 10pt)
    #set par(first-line-indent: 0em)

    Copyright \u{00A9} Aalborg University #datetime.today().year()\
    #v(0.2cm)
    This report is typeset using the Typst system.
  ])

  _titlepage-en(meta, en)

  if dk-is-set {
    clear-double-page()
    _titelside-dk(meta, dk)
  }

  set page(footer: _custom-footer(<titlepages-chapter>))

  body
}