module IsoDoc::Function
  module References

    # This is highly specific to ISO, but it's not a bad precedent for
    # references anyway; keeping here instead of in IsoDoc::Iso for now
    def docid_l10n(x)
      return x if x.nil?
      x.gsub(/All Parts/i, @all_parts_lbl.downcase)
    end

    # TODO generate formatted ref if not present
    def nonstd_bibitem(list, b, ordinal, bibliography)
      list.p **attr_code(iso_bibitem_entry_attrs(b, bibliography)) do |r|
        id = bibitem_ref_code(b)
        identifier = render_identifier(id)
        if bibliography then ref_entry_code(r, ordinal, identifier, id)
        else
          r << "#{identifier}, "
        end
        reference_format(b, r)
      end
    end

    def std_bibitem_entry(list, b, ordinal, biblio)
      list.p **attr_code(iso_bibitem_entry_attrs(b, biblio)) do |ref|
        prefix_bracketed_ref(ref, ordinal) if biblio
        id = bibitem_ref_code(b)
        identifier = render_identifier(id)
        ref << identifier
        date_note_process(b, ref)
        ref << ", "
        reference_format(b, ref)
      end
    end

    # if t is just a number, only use that ([1] Non-Standard)
    # else, use both ordinal, as prefix, and t
    def ref_entry_code(r, ordinal, t, id)
      if id["type"] == "metanorma"
        prefix_bracketed_ref(r, t)
      else
        prefix_bracketed_ref(r, ordinal)
        if !t.empty? && !%w(DOI ISSN ISBN).include?(id["type"])
          r << "#{t}, "
        end
      end
    end

    def bibitem_ref_code(b)
      id = b.at(ns("./docidentifier[@type = 'metanorma']"))
      id ||= b.at(ns("./docidentifier[not(@type = 'DOI' or @type = 'metanorma' "\
                   "or @type = 'ISSN' or @type = 'ISBN')]"))
      id ||= b.at(ns("./docidentifier")) 
      return id if id
      id = Nokogiri::XML::Node.new("docidentifier", b.document)
      id << "(NO ID)"
      id
    end

    def render_identifier(id)
      docid_prefix(id["type"], id.text.sub(/^\[/, "").sub(/\]$/, ""))
    end

    def docid_prefix(prefix, docid)
      docid = "#{prefix} #{docid}" if prefix && !omit_docid_prefix(prefix)
      docid_l10n(docid)
    end

    def omit_docid_prefix(prefix)
      return true if prefix.nil? || prefix.empty?
      return %w(ISO IEC ITU metanorma).include? prefix
    end

    def date_note_process(b, ref)
      date_note = b.at(ns("./note[text()][contains(.,'ISO DATE:')]"))
      return if date_note.nil?
      date_note.content = date_note.content.gsub(/ISO DATE: /, "")
      date_note.children.first.replace("<p>#{date_note.content}</p>")
      footnote_parse(date_note, ref)
    end

    def iso_bibitem_entry_attrs(b, biblio)
      { id: b["id"], class: biblio ? "Biblio" : "NormRef" }
    end

    def iso_title(b)
      title = b.at(ns("./title[@language = '#{@lang}' and @type = 'main']")) ||
        b.at(ns("./title[@language = '#{@lang}']")) ||
        b.at(ns("./title[@type = 'main']")) ||
        b.at(ns("./title"))
      title
    end

    # reference not to be rendered because it is deemed implicit
    # in the standards environment
    def implicit_reference(b)
      false
    end

    def prefix_bracketed_ref(ref, text)
      ref << "[#{text}]"
      insert_tab(ref, 1)
    end

    def reference_format(b, r)
      if ftitle = b.at(ns("./formattedref"))
        ftitle&.children&.each { |n| parse(n, r) }
      else
        title = iso_title(b)
        r.i do |i|
          title&.children&.each { |n| parse(n, i) }
        end
      end
    end

    ISO_PUBLISHER_XPATH =
      "./contributor[xmlns:role/@type = 'publisher']/"\
      "organization[abbreviation = 'ISO' or xmlns:abbreviation = 'IEC' or "\
      "xmlns:name = 'International Organization for Standardization' or "\
      "xmlns:name = 'International Electrotechnical Commission']".freeze

    def is_standard(b)
      ret = false
      b.xpath(ns("./docidentifier")).each do |id|
        next if id["type"].nil? ||
          %w(metanorma DOI ISSN ISBN).include?(id["type"])
        ret = true
      end
      ret
    end

    def biblio_list(f, div, biblio)
      i = 0
      f.xpath(ns("./bibitem | ./note")).each do |b|
        next if implicit_reference(b)
        i += 1 if b.name == "bibitem"
        if b.name == "note" then note_parse(b, div)
        elsif(is_standard(b)) then std_bibitem_entry(div, b, i, biblio)
        else
          nonstd_bibitem(div, b, i, biblio)
        end
      end
    end

    def norm_ref(isoxml, out, num)
      q = "//bibliography/references[title = 'Normative References' or "\
        "title = 'Normative references']"
      f = isoxml.at(ns(q)) or return num
      out.div do |div|
        num = num + 1
        clause_name(num, @normref_lbl, div, nil)
        f.elements.reject do |e|
          %w(reference title bibitem note).include? e.name
        end.each { |e| parse(e, div) }
        biblio_list(f, div, false)
      end
      num
    end

    BIBLIOGRAPHY_XPATH = "//bibliography/clause[title = 'Bibliography'] | "\
      "//bibliography/references[title = 'Bibliography']".freeze

    def bibliography(isoxml, out)
      f = isoxml.at(ns(BIBLIOGRAPHY_XPATH)) || return
      page_break(out)
      out.div do |div|
        div.h1 @bibliography_lbl, **{ class: "Section3" }
        f.elements.reject do |e|
          %w(reference title bibitem note).include? e.name
        end.each { |e| parse(e, div) }
        biblio_list(f, div, true)
      end
    end

    def bibliography_parse(node, out)
      title = node&.at(ns("./title"))&.text || ""
      out.div do |div|
        anchor(node['id'], :label, false) and
          clause_parse_title(node, div, node.at(ns("./title")), out) or
          div.h2 title, **{ class: "Section3" }
        node.elements.reject do |e|
          %w(reference title bibitem note).include? e.name
        end.each { |e| parse(e, div) }
        biblio_list(node, div, true)
      end
    end

    def format_ref(ref, prefix, isopub, date, allparts)
      ref = docid_prefix(prefix, ref)
      return "[#{ref}]" if /^\d+$/.match(ref) && !prefix &&
        !/^\[.*\]$/.match(ref)
        ref
    end

    def reference_names(ref)
      isopub = ref.at(ns(ISO_PUBLISHER_XPATH))
      docid = bibitem_ref_code(ref)
      prefix = docid["type"]
      date = ref.at(ns("./date[@type = 'published']"))
      allparts = ref.at(ns("./extent[@type='part'][referenceFrom='all']"))
      reference = format_ref(docid_l10n(docid.text), prefix, isopub, date,
                             allparts)
      @anchors[ref["id"]] = { xref: reference }
    end

    # def ref_names(ref)
    #  linkend = ref.text
    # linkend.gsub!(/[\[\]]/, "") unless /^\[\d+\]$/.match linkend
    # @anchors[ref["id"]] = { xref: linkend }
    # end
  end
end