module Datadog
  module DemoEnv
    module_function

    def gem_spec(gem_name, defaults = {})
      args =  if local_gem(gem_name)
                [local_gem(gem_name)]
              elsif git_gem(gem_name)
                [git_gem(gem_name)]
              else
                []
              end

      yield(args) if block_given?

      args
    end

    def gem_env_name(gem_name)
      gem_name.upcase.gsub('-', '_')
    end

    def local_gem(gem_name)
      local_env_name = "DD_DEMO_ENV_GEM_LOCAL_#{gem_env_name(gem_name)}"

      return unless ENV.key?(local_env_name)

      { path: ENV[local_env_name] }
    end

    def git_gem(gem_name)
      git_env_name = "DD_DEMO_ENV_GEM_GIT_#{gem_env_name(gem_name)}"
      ref_env_name = "DD_DEMO_ENV_GEM_REF_#{gem_env_name(gem_name)}"

      return unless [git_env_name, ref_env_name].all? { |var| ENV.key?(var) && !ENV[var].empty? }

      { git: ENV[git_env_name], ref: ENV[ref_env_name] }
    end

    def process
      ENV['DD_DEMO_ENV_PROCESS']
    end

    def features
      (ENV['DD_DEMO_ENV_FEATURES'] || '').split(',')
    end

    def feature?(feature)
      features.include?(feature)
    end

    def print_env(header = 'Datadog test environment')
      puts "\n==== #{header} ===="
      puts "Process: #{process}"
      puts "Features: #{features}"
      puts "Rails env: #{ENV['RAILS_ENV']}" if ENV['RAILS_ENV']
      puts "PID: #{Process.pid}"
      if ddtrace = Gem.loaded_specs['ddtrace']
        puts "Runtime ID: #{Datadog::Runtime::Identity.id}" if defined?(Datadog::Runtime::Identity)
        puts "ddtrace version: #{ddtrace.version}"
        puts "ddtrace path: #{ddtrace.full_gem_path}"
        if git_spec = git_gem('ddtrace')
          puts "ddtrace git: #{git_spec[:git]}"
          puts "ddtrace ref: #{git_spec[:ref]}"
        end
      end
      puts "\n"
    end

    def profiler_file_transport(dump_path = "/data/profile-pid-#{Process.pid}.pprof")
      Datadog::Profiling::Transport::IO.default(
        write: ->(out, data) do
          result = nil
          puts "Writing pprof #{dump_path}..."
          File.open(dump_path, "w") { |f| result = f.write(data) }
          puts "Pprof #{dump_path} written!"
          result
        end
      )
    end

    def start_mem_dump!
      require 'objspace'
      ObjectSpace.trace_object_allocations_start
    end

    def finish_mem_dump!(dump_path = "/data/mem-pid-#{Process.pid}.dump")
      File.delete(dump_path) if File.exist?(dump_path)
      File.open(dump_path, "w") do |io|
        ObjectSpace.dump_all(output: io)
      end
    end

    def mem_dump!(dump_path = "/data/mem-pid-#{Process.pid}.dump")
      start_mem_dump!
      result = yield
      finish_mem_dump!
      result
    end
  end
end