module Harbr
  class Job
    include SuckerPunch::Job

    def load_manifest(container,version)
      manifest_path = "/var/harbr/#{container}/versions/#{version}/config/manifest.yml"
      raise "Manifest not found at #{manifest_path}" unless File.exist?(manifest_path)
      manifest_data = YAML.load_file(manifest_path)
      OpenStruct.new(manifest_data)
    end

    def create_traefik_config(containers)
      config = {
        "http" => {
          "routers" => {
            "traefik-dashboard" => {
              "rule" => "Host(`traefik.harbr.zero2one.ee`)",
              "service" => "api@internal"
            }
          },
          "services" => {}
        }
      }

      containers.each do |container|
        container.ip = "127.0.0.1"

        config["http"]["routers"]["#{container.name}-router"] = {
          "rule" => "Host(`#{container.host_header}`)",
          "service" => "#{container.name}-service"
        }
        config["http"]["services"]["#{container.name}-service"] = {
          "loadBalancer" => {
            "servers" => [{"url" => "http://#{container.ip}:#{container.port}"}]
          }
        }
      end

      File.write("/etc/traefik/harbr.toml", TomlRB.dump(config))
      puts "Traefik configuration written to /etc/traefik/harbr.toml"
    end

    def create_run_script(container_name, port)
      service_dir = "/etc/sv/harbr/#{container_name}"
      if File.directory?(service_dir)
        puts "Directory already exists: #{service_dir}"
        return
      end

      script_template = <<~SCRIPT
        #!/bin/sh
        exec 2>&1
        cd /var/harbr/#{container_name}/current
        exec bundle exec puma -p #{port}
      SCRIPT

      service_dir = "/etc/sv/harbr/#{container_name}"
      FileUtils.mkdir_p(service_dir)

      File.write("#{service_dir}/run", script_template)
      FileUtils.chmod("+x", "#{service_dir}/run")
      puts "Run script created and made executable for container: #{container_name}"
    end

    def create_log_script(container_name)
      log_dir = "/var/log/harbr/#{container_name}"

      if File.directory?(log_dir)
        puts "Directory already exists: #{log_dir}"
        return
      end

      FileUtils.mkdir_p(log_dir)

      script_template = <<~SCRIPT
        #!/bin/sh
        exec svlogd -tt #{log_dir}/
      SCRIPT

      dir_path = "/etc/sv/harbr/#{container_name}/log"
      FileUtils.mkdir_p(dir_path)

      File.write("#{dir_path}/run", script_template)
      FileUtils.chmod("+x", "#{dir_path}/run")
      puts "Log script created and made executable for container: #{container_name}"
    end

    def create_a_service(container_name, port)
      create_run_script(container_name, port)
      create_log_script(container_name)
      system("ln -s /etc/sv/harbr/#{container_name} /etc/service/#{container_name}") unless File.exist?("/etc/service/#{container_name}")
    end

    def run_container(manifest)
      puts "Starting container: #{manifest.name}"
      create_a_service(manifest.name, manifest.port)

      container = Container.new
      containers = Container::Repository.new

      container.name = manifest.name
      container.host_header = manifest.host
      container.ip = "127.0.0.1"
      container.port = manifest.port
      
      if containers.find_by_header(manifest.host)
        containers.update(container)  
      else
        containers.create(container)
      end

      system("cd /var/harbr/#{manifest.name}/current && bundle install")
      system("sv restart #{manifest.name}")
      puts "Started container: #{manifest.name}"
      create_traefik_config(containers.all)
    end

    def perform(container, version)
      puts "Running tasks for container: '#{container}', Version: '#{version}'"
      manifest = load_manifest(container, version)
      puts "Manifest: #{manifest}"
      run_container(manifest)
    end
  end
end