require 'yaml'
require 'ostruct'
require 'sucker_punch'

require 'harbr'
module Harbr
  module Next
    class Job
      include SuckerPunch::Job

      def highest_numbered_directory(path)
        directories = Dir.glob("#{path}/*").select { |entry| File.directory?(entry) }
        directories.max_by { |entry| entry[/\d+/].to_i }
      end

      def get_container_name(path)
        File.basename(path)
      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"
          name = container.name.gsub(".","-")
      
          config["http"]["routers"]["#{name}-router"] = {
            "rule" => "Host(`#{container.host_header}`)",
            "service" => "#{name}-service"
          }
          config["http"]["services"]["#{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 collate_containers(name,port, version)
      
        containers = Harbr::Container::Repository.new
        container = containers.find_by_name(name)
      
        container.port = port
        container.next_version = version
        container.previous_version = container.current_version     
        containers.update(container)

        containers.all
      end
      
      
      module Runit
      
        class Run
          def initialize(container, port)
            @container_name = container
            @port = port
          end
      
          def to_s
            script_template = <<~SCRIPT
                #!/bin/sh
                exec 2>&1
                cd /var/harbr/#{@container_name}/current
                exec ./exe/run #{@port}
            SCRIPT
          end
      
          def link
            "ln -s /etc/sv/harbr/#{@container_name} /etc/service/#{@container_name}"
          end
        end
      
      
        class Finish
          def initialize(port)
            @port = port
          end
      
          def to_s
            script_template = <<~SCRIPT
                #!/bin/sh
                sleep 3
                `lsof -i :#{@port} | awk 'NR!=1 {print $2}' | xargs kill`
            SCRIPT
          end
        end
      
        class Log
          def initialize(container, port)
            @container_name = container
          end
      
          def to_s
            script_template = <<~SCRIPT
                #!/bin/sh
                exec svlogd -tt /var/log/harbr/#{@container_name}/next/
            SCRIPT
          end
      
        end
      
        module Next
      
          class Run
            def initialize(container, port)
              @container_name = container
              @port = port
            end
      
            def to_s
              script_template = <<~SCRIPT
                  #!/bin/sh
                  exec 2>&1
                  cd /var/harbr/containers/#{@container_name}/next
                  exec ./exe/run #{@port}
              SCRIPT
            end
          end
      
          class Log
            def initialize(container)
              @container_name = container
            end
      
            def to_s
              script_template = <<~SCRIPT
                  #!/bin/sh
                  exec svlogd -tt /var/log/harbr/#{@container_name}/next/
              SCRIPT
            end
          end
      
        end
      end
      
      def write_to_file(path, contents)
        File.open(path, 'w') do |file|
          file.write(contents)
        end
      end
      
      def load_manifest(container, version)
        manifest_path = "/var/harbr/containers/#{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 perform(name, version)
        `bundle config set --local path 'vendor/bundle'`
        
        manifest = load_manifest(name,version)
        
        current_path = "/var/harbr/containers/#{name}/current"
      
        port = `port assign next.#{manifest.port}`.strip
    
        Dir.chdir current_path do
    
          system "sv stop next.#{name}"
          system "bundle install" 
    
          `mkdir -p /etc/sv/harbr/#{name}/next`
          `mkdir -p /etc/sv/harbr/#{name}/next/log`
          `mkdir -p /var/log/harbr/#{name}/next/log`
    
          write_to_file "/etc/sv/harbr/#{name}/next/run", Runit::Next::Run.new(name,port).to_s
          write_to_file "/etc/sv/harbr/#{name}/next/finish", Runit::Finish.new(port).to_s
          write_to_file "/etc/sv/harbr/#{name}/next/log/run", Runit::Next::Log.new(name).to_s
          
          `chmod +x /etc/sv/harbr/#{name}/next/run`
          `chmod +x /etc/sv/harbr/#{name}/next/log/run`
          `chmod +x /etc/sv/harbr/#{name}/next/finish`
          
          
          system "rm /etc/service/next.#{name}"
          system "rm /var/harbr/containers/#{name}/next"
          
          system "ln -sf /var/harbr/containers/#{name}/versions/#{version} /var/harbr/containers/#{name}/next"
          system "ln -sf /etc/sv/harbr/#{name}/next /etc/service/next.#{name}"
          system "sv restart next.#{name}"
          
          containers = collate_containers("next.#{name}",port,version)
          create_traefik_config(containers)
        end
      
        puts "process #{version} of #{name}"         
        
      end

    end
    
  end
end