module IsoDoc
  module Iso
    class WordConvert < IsoDoc::WordConvert
      def figure_cleanup(xml)
        super
        xml.xpath("//div[@class = 'figure']//table[@class = 'dl']").each do |t|
          t["class"] = "figdl"
          d = t.add_previous_sibling("<div class='figdl' " \
                                     "style='page-break-after:avoid;'/>")
          t.parent = d.first
        end
      end

      def word_annex_cleanup1(docxml, lvl)
        docxml.xpath("//h#{lvl}[ancestor::*[@class = 'Section3']]").each do |h2|
          h2.name = "p"
          h2["class"] = "a#{lvl}"
        end
      end

      def word_annex_cleanup(docxml)
        (2..6).each { |i| word_annex_cleanup1(docxml, i) }
      end

      def word_annex_cleanup_h1(docxml)
        docxml.xpath("//h1[@class = 'Annex']").each do |h|
          h.name = "p"
          h["class"] = "ANNEX"
        end
        %w(BiblioTitle ForewordTitle IntroTitle).each do |s|
          docxml.xpath("//*[@class = '#{s}']").each do |h|
            h.name = "p"
          end
        end
      end

      def style_cleanup(docxml)
        word_annex_cleanup_h1(docxml)
        figure_style(docxml)
        new_styles(docxml)
        index_cleanup(docxml)
      end

      def index_cleanup(docxml)
        docxml.xpath("//div[@class = 'index']").each do |i|
          i.xpath(".//p | .//li").each do |p|
            p["style"] ||= ""
            p["style"] += "margin-bottom:0px;"
          end
        end
      end

      def figure_style(docxml)
        docxml.xpath("//div[@class = 'figure']").each do |f|
          f["style"] ||= ""
          f["style"] += "text-align:center;"
        end
      end

      def quote_style(docxml)
        docxml.xpath("//div[@class = 'Quote' or @class = 'Note' or " \
                     "@class = 'Example' or @class = 'Admonition']").each do |d|
                       quote_style1(d)
                     end
      end

      def quote_style1(div)
        div.xpath(".//li").each do |p|
          p["style"] ||= ""
          p["style"] += "font-size:#{default_fonts({})[:smallerfontsize]};"
        end
      end

      def sourcecode_style
        "Code"
      end

      STYLESMAP = {
        example: "Example",
        note: "Note",
        Sourcecode: "Code",
        tabletitle: "Tabletitle",
        Biblio: "MsoNormal",
        figure: "MsoNormal",
        SourceTitle: "FigureTitle",
      }.freeze

      def new_styles(docxml)
        self.class::STYLESMAP.each do |k, v|
          docxml.xpath("//*[@class = '#{k}']").each { |s| s["class"] = v }
        end
        docxml.xpath("//div[@class = 'Section3']//p[@class = 'Tabletitle']")
          .each { |t| t["class"] = "AnnexTableTitle" }
        docxml.xpath("//*[@class = 'zzHelp']/p[not(@class)]").each do |p|
          p["class"] = "zzHelp"
        end
        quote_style(docxml)
      end

      def authority_hdr_cleanup(docxml)
        { "boilerplate-license": "zzWarningHdr",
          "boilerplate-copyright": "zzCopyrightHdr" }.each do |k, v|
            docxml.xpath("//div[@class = '#{k}']").each do |d|
              d.xpath(".//h1").each do |p|
                p.name = "p"
                p["class"] = v
              end
            end
          end
      end

      def authority_cleanup(docxml)
        insert = docxml.at("//div[@id = 'boilerplate-license-destination']")
        auth = docxml.at("//div[@class = 'boilerplate-license']")&.remove
        auth&.xpath(".//p[not(@class)]")&.each { |p| p["class"] = "zzWarning" }
        auth and insert and insert.children = auth
        insert = docxml.at("//div[@id = 'boilerplate-copyright-destination']")
        auth = docxml.at("//div[@class = 'boilerplate-copyright']")&.remove
        auth&.xpath(".//p[not(@class)]")&.each do |p|
          p["class"] = "zzCopyright"
        end
        auth&.xpath(".//p[@id = 'boilerplate-message']")&.each do |p|
          p["class"] = "zzCopyright1"
        end
        auth&.xpath(".//p[@id = 'boilerplate-address']")&.each do |p|
          p["class"] = "zzAddress"
        end
        auth&.xpath(".//p[@id = 'boilerplate-place']")&.each do |p|
          p["class"] = "zzCopyright1"
        end
        auth and insert and insert.children = auth
        coverpage_note_cleanup(docxml)
      end

      def word_cleanup(docxml)
        authority_hdr_cleanup(docxml)
        super
        style_cleanup(docxml)
        docxml
      end

      # supply missing annex title
      def make_WordToC(docxml, level)
        toc = ""
        xpath = (1..level).each.map { |i| "//h#{i}" }.join (" | ")
        docxml.xpath(xpath).each do |h|
          x = ""
          x = @anchor[h.parent["id"]][:xref] if h["class"] == "ANNEX"
          toc += word_toc_entry(h.name[1].to_i, x + header_strip(h))
        end
        toc.sub(/(<p class="MsoToc1">)/,
                %{\\1#{word_toc_preface(level)}}) + WORD_TOC_SUFFIX1
      end
    end
  end
end