# frozen_string_literal: true

require 'liquid'
require 'tomlrb'
require 'tags/layout'

Liquid::Template.error_mode = :strict
Liquid::Template.register_tag('layout', Layout)

module Archival
  class Builder
    attr_reader :page_templates

    def initialize(config, *_args)
      @config = config
      refresh_config
    end

    def refresh_config
      @file_system = Liquid::LocalFileSystem.new(
        @config.root, '%s.liquid'
      )
      @variables = {}
      @object_types = {}
      @page_templates = {}

      Liquid::Template.file_system = @file_system

      objects_definition_file = File.join(@config.root,
                                          'objects.toml')
      if File.file? objects_definition_file
        @object_types = read_toml(objects_definition_file)
      end

      update_pages
      update_objects
    end

    def full_rebuild
      Layout.reset_cache
      refresh_config
    end

    def update_pages
      do_update_pages(File.join(@config.root, @config.pages_dir))
    end

    def do_update_pages(dir, prefix = nil)
      add_prefix = lambda { |entry|
        prefix ? File.join(prefix, entry) : entry
      }
      Dir.foreach(dir) do |entry|
        if File.directory? entry
          unless [
            '.', '..'
          ].include?(entry)
            update_pages(File.join(dir, entry),
                         add_prefix(entry))
          end
        elsif File.file? File.join(dir, entry)
          if entry.end_with?('.liquid') && !(entry.start_with? '_')
            page_name = File.basename(entry,
                                      '.liquid')
            template_file = File.join(
              @config.pages_dir,
              add_prefix.call(page_name)
            )
            content = @file_system.read_template_file(template_file)
            content += dev_mode_content if @config.dev_mode
            @page_templates[add_prefix.call(page_name)] =
              Liquid::Template.parse(content)
          end
        end
      end
    end

    def update_objects
      do_update_objects(File.join(@config.root,
                                  @config.objects_dir))
    end

    def do_update_objects(dir)
      objects = {}
      @object_types.each do |name, _definition|
        objects[name] = []
        obj_dir = File.join(dir, name)
        if File.directory? obj_dir
          Dir.foreach(obj_dir) do |file|
            if file.end_with? '.toml'
              object = read_toml(File.join(
                                   obj_dir, file
                                 ))
              object[:name] =
                File.basename(file, '.toml')
              objects[name].push object
            end
          end
        end
        objects[name] = objects[name].sort do |a, b|
          (a['order'] || a[:name]).to_s <=> (b['order'] || b[:name]).to_s
        end
      end
      @variables['objects'] = objects
    end

    def read_toml(file_path)
      Tomlrb.load_file(file_path)
    end

    def set_var(name, value)
      @variables[name] = value
    end

    def render(page)
      template = @page_templates[page]
      template.render(@variables)
    end

    def write_all
      Dir.mkdir(@config.build_dir) unless File.exist? @config.build_dir
      @page_templates.each_key do |template|
        out_dir = File.join(@config.build_dir,
                            File.dirname(template))
        Dir.mkdir(out_dir) unless File.exist? out_dir
        out_path = File.join(@config.build_dir,
                             "#{template}.html")
        File.open(out_path, 'w+') do |file|
          file.write(render(template))
        end
      end
      return if @config.dev_mode

      # in production, also copy all assets to the dist folder.
      @config.assets_dirs.each do |asset_dir|
        FileUtils.copy_entry File.join(@config.root, asset_dir),
                             File.join(@config.build_dir, asset_dir)
      end
    end

    private

    def dev_mode_content
      "<script src=\"http://localhost:#{@config.helper_port}/js/archival-helper.js\" type=\"application/javascript\"></script>" # rubocop:disable Layout/LineLength
    end
  end
end