require "pathname"
require "digest/md5"

require "spring_standalone/version"
require "spring_standalone/sid"
require "spring_standalone/configuration"

module SpringStandalone
  IGNORE_SIGNALS = %w(INT QUIT)
  STOP_TIMEOUT = 2 # seconds

  class Env
    attr_reader :log_file

    def initialize(options = {})
      @root         = options[:root]
      @project_root = options[:root]
      @log_file     = options[:log_file] || File.open(ENV["SPRING_LOG"] || File::NULL, "a")
    end

    def root
      @root ||= SpringStandalone.application_root_path
    end

    def project_root
      @project_root ||= SpringStandalone.project_root_path
    end

    def version
      SpringStandalone::VERSION
    end

    def tmp_path
      require "tmpdir"
      path = Pathname.new(
        ENV["SPRING_TMP_PATH"] ||
          File.join(ENV['XDG_RUNTIME_DIR'] || Dir.tmpdir, "spring-#{Process.uid}")
      )
      require "fileutils"
      FileUtils.mkdir_p(path) unless path.exist?
      path
    end

    def application_id
      ENV["SPRING_APPLICATION_ID"] || Digest::MD5.hexdigest(RUBY_VERSION + project_root.to_s)
    end

    def socket_path
      Pathname.new(ENV["SPRING_SOCKET"] || tmp_path.join(application_id))
    end

    def socket_name
      socket_path.to_s
    end

    def pidfile_path
      Pathname.new(ENV["SPRING_PIDFILE"] || socket_path.dirname.join("#{socket_path.basename(".*")}.pid"))
    end

    def pid
      pidfile_path.exist? ? pidfile_path.read.to_i : nil
    rescue Errno::ENOENT
      # This can happen if the pidfile is removed after we check it
      # exists
    end

    def app_name
      root.basename
    end

    def server_running?
      pidfile = pidfile_path.open('r+')
      !pidfile.flock(File::LOCK_EX | File::LOCK_NB)
    rescue Errno::ENOENT
      false
    ensure
      if pidfile
        pidfile.flock(File::LOCK_UN)
        pidfile.close
      end
    end

    def log(message)
      log_file.puts "[#{Time.now}] [#{Process.pid}] #{message}"
      log_file.flush
    end

    def stop
      if server_running?
        timeout = Time.now + STOP_TIMEOUT
        kill 'TERM'
        sleep 0.1 until !server_running? || Time.now >= timeout

        if server_running?
          kill 'KILL'
          :killed
        else
          :stopped
        end
      else
        :not_running
      end
    end

    def kill(sig)
      pid = self.pid
      Process.kill(sig, pid) if pid
    rescue Errno::ESRCH
      # already dead
    end

    def server_command
      ENV["SPRING_SERVER_COMMAND"] || "#{File.expand_path("../../../bin/spring_sa", __FILE__)} server --background"
    end
  end
end