require "metanorma/cli/compiler"
require "metanorma/cli/generator"
require "metanorma/cli/collection"
require "metanorma/cli/git_template"
require "metanorma/cli/thor_with_config"
require "metanorma/cli/commands/config"
require "metanorma/cli/commands/template_repo"
require "metanorma/cli/commands/site"
require "mnconvert"

module Metanorma
  module Cli
    class Command < ThorWithConfig
      class_option :progress, aliases: "-s", type: :boolean, default: false,
                              desc: "Show progress for long running tasks (like download)"

      desc "new NAME", "Create new Metanorma document"
      option :type, aliases: "-t", required: true, desc: "Document type"
      option :doctype, aliases: "-d", required: true, desc: "Metanorma doctype"
      option :overwrite, aliases: "-y", type: :boolean,
                         desc: "Overwrite existing document"
      option :template, aliases: "-l",
                        desc: "Git hosted remote or local FS template skeleton"

      def new(name)
        create_new_document(name, options)
      end

      desc "compile FILENAME", "Compile to a metanorma document"
      option :type, aliases: "-t", desc: "Type of standard to generate"
      option :extensions, aliases: "-x", type: :string,
                          desc: "Type of extension to generate per type"
      option :format, aliases: "-f", default: :asciidoc,
                      desc: "Format of source file: eg. asciidoc"
      option :require, aliases: "-r", desc: "Require LIBRARY prior to execution"
      option :wrapper, aliases: "-w", type: :boolean,
                       desc: "Create wrapper folder for HTML output"
      option :asciimath, aliases: "-a", type: :boolean,
                         desc: "Keep Asciimath in XML output instead of converting it to MathM"
      option :datauriimage, aliases: "-d", type: :boolean,
                            desc: "Encode HTML output images as data URIs"
      option :relaton, aliases: "-R",
                       desc: "Export Relaton XML for document to nominated filename"
      option :extract, aliases: "-e",
                       desc: "Export sourcecode fragments from this document to nominated directory"
      option :version, aliases: "-v",
                       desc: "Print version of code (accompanied with -t)"
      option :output_dir, aliases: "-o",
                          desc: "Directory to save compiled files"
      option :strict, aliases: "-S", type: :boolean,
                      desc: "Strict compilation: abort if there are any errors"
      option :agree_to_terms, type: :boolean, desc: "Agree / Disagree with all third-party licensing terms "\
                                                    "presented (WARNING: do know what you are agreeing with!)"
      option :install_fonts, type: :boolean, default: true,
                             desc: "Install required fonts"
      option :continue_without_fonts, type: :boolean,
                                      desc: "Continue processing even when fonts are missing"

      def compile(file_name = nil)
        if file_name && !options[:version]
          documents = select_wildcard_documents(file_name) || [file_name]
          documents.each { |document| compile_document(document, options.dup) }

        elsif options[:version]
          invoke(:version, [], type: options[:type], format: options[:format])

        elsif options.keys.size >= 2
          UI.say("Need to specify a file to process")

        else
          invoke :help
        end
      end

      desc "collection FILENAME", "Render HTML pages from XML/YAML colection"
      option :format, aliases: "-x", type: :string, desc: "Formats to generate"
      option :output_folder, aliases: "-w",
                             desc: "Directory to save compiled files"
      option :coverpage, aliases: "-c", desc: "Liquid template"
      option :agree_to_terms, type: :boolean, desc: "Agree / Disagree with all third-party licensing terms "\
                                                    "presented (WARNING: do know what you are agreeing with!)"
      option :install_fonts, type: :boolean, default: true,
                             desc: "Install required fonts"
      option :continue_without_fonts, type: :boolean,
                                      desc: "Continue processing even when fonts are missing"
      option :strict, aliases: "-S", type: :boolean, \
                      desc: "Strict compilation: abort if there are any errors"

      def collection(filename = nil)
        if filename
          coll_options = options.dup
          coll_options[:compile] = filter_compile_options(coll_options)
          Metanorma::Cli::Collection.render(filename, coll_options)
        else
          UI.say("Need to specify a file to process")
        end
      rescue ArgumentError => e
        UI.say e.message
      end

      desc "convert FILENAME", "Convert STS XML to Metanorma adoc"
      option :output_format,
             type: :string,
             desc: "Output format: xml|adoc|iso|niso"
      option :output_file, type: :string, desc: "Output file"
      option :imagesdir,
             type: :string,
             desc: "For STS input only: folder with images"
      option :input_format,
             type: :string,
             desc: "Input format: metanorma|sts|rfc"
      option :debug, type: :boolean, desc: "Enable debug output"
      option :sts_type,
             type: :string,
             desc: "For STS input only: type of standard to generate"
      option :check_type, type: :string, desc: "Check against XSD/DTD"
      option :xsl_file, type: :string, desc: "Path to XSL file"
      option :split_bibdata,
             type: :boolean,
             desc: "For STS input only: create MN Adoc and Relaton XML"
      option :semantic,
             type: :boolean,
             desc: "For STS input only: generate semantic XML"

      def convert(inputfile)
        MnConvert.convert(inputfile, options)
      rescue Error => e
        UI.say e.message
      end

      desc "version", "Version of the code"
      option :type, aliases: "-t", required: false,
                    desc: "Type of standard to generate"
      option :format, aliases: "-f", default: :asciidoc,
                      desc: "Format of source file: eg. asciidoc"

      def version
        Metanorma::Cli.load_flavors
        backend_version(options[:type]) || supported_backends
        options[:type] or dependencies_versions
      rescue NameError => e
        UI.say(e)
      end

      desc "list-extensions", "List supported extensions"
      def list_extensions(type = nil)
        single_type_extensions(type) || all_type_extensions
      rescue LoadError
        UI.say("Couldn't load #{type}, please provide a valid type!")
      end

      desc "list-doctypes", "List supported doctypes"
      def list_doctypes(type = nil)
        processors = backend_processors

        if type && processors[type.to_sym]
          processors = { type.to_sym => processors[type.to_sym] }
        end

        print_doctypes_table(processors)
      end

      desc "template-repo", "Manage metanorma templates repository"
      subcommand :template_repo, Metanorma::Cli::Commands::TemplateRepo

      desc "site", "Manage site for metanorma collections"
      subcommand :site, Metanorma::Cli::Commands::Site

      desc "config", "Manage configuration file"
      subcommand :config, Metanorma::Cli::Commands::Config

      def self.exit_on_failure?
        true
      end

      private

      def single_type_extensions(type)
        type or return false
        format_keys = find_backend(type).output_formats.keys
        UI.say("Supported extensions: #{join_keys(format_keys)}.")
        true
      end

      def all_type_extensions
        Metanorma::Cli.load_flavors
        message = "Supported extensions per type: \n"
        Metanorma::Registry.instance.processors.each do |type_sym, processor|
          format_keys = processor.output_formats.keys
          message += "  #{type_sym}: #{join_keys(format_keys)}.\n"
        end
        UI.say(message)
      end

      def backend_version(type)
        type and UI.say(find_backend(type).version)
      end

      def backend_processors
        @backend_processors ||= begin
          Metanorma::Cli.load_flavors
          Metanorma::Registry.instance.processors
        end
      end

      def find_backend(type)
        load_flavours(type)
        Metanorma::Registry.instance.find_processor(type&.to_sym)
      end

      def supported_backends
        UI.say("Metanorma #{Metanorma::VERSION}")
        UI.say("Metanorma::Cli #{VERSION}")
        Metanorma::Cli.load_flavors
        Metanorma::Registry.instance.processors.map do |_type, processor|
          UI.say(processor.version)
        end
      end

      DEPENDENCY_GEMS =
        %w(html2doc isodoc metanorma-utils mn2pdf mn-requirements isodoc-i18n
           metanorma-plugin-datastruct metanorma-plugin-glossarist
           metanorma-plugin-lutaml relaton-cli pubid glossarist fontist
           plurimath lutaml expressir xmi lutaml-model emf2svg unitsml
           vectory ogc-gml oscal).freeze

      def dependencies_versions
        versions = Gem.loaded_specs
        DEPENDENCY_GEMS.sort.each do |k|
          UI.say("#{k} #{versions[k].version}")
        end
      end

      def join_keys(keys)
        [keys[0..-2].join(", "), keys.last].join(" and ")
      end

      def create_new_document(name, options)
        Metanorma::Cli::Generator.run(
          name,
          type: options[:type],
          doctype: options[:doctype],
          template: options[:template],
          overwrite: options[:overwrite],
        )
      end

      def load_flavours(type)
        Metanorma::Cli.load_flavors
        unless Metanorma::Registry.instance.find_processor(type&.to_sym)
          require "metanorma-#{type}"
        end
      end

      def print_doctypes_table(processors)
        table_data = processors.map do |type_sym, processor|
          [type_sym.to_s,
           processor.input_format,
           join_keys(processor.output_formats.keys)]
        end
        UI.table(["Type", "Input", "Supported output format"], table_data)
      end

      def select_wildcard_documents(filename)
        if filename.include?("*")
          Dir.glob(Pathname.new(filename))
        end
      end

      def compile_document(filename, options)
        Metanorma::Cli.load_flavors
        errors = Metanorma::Cli::Compiler.compile(filename, options)
        errors.each { |error| Util.log(error, :error) }
        exit(1) if errors.any?
      end
    end
  end
end