require "httparty"
require "awesome_print"
require "influxdb"
require "yaml"

def processs_org(org)
  arch = 'x86'
  case(org)
  when 'linuxserver' then
    arch = 'x86'
  when 'lsioarmhf' then
    arch = 'armhf'
  end

  res = HTTParty.get(
    "https://hub.docker.com/v2/repositories/#{org}/",
    query: { page_size: 100   })

  pull_counts = {}
  continue = true
  while continue
    data = JSON.parse(res.body)
    pull_counts.merge!(process_counts(data, arch))

    if data.include?('next') && data['next'].nil? == false
      url = data['next']
      res = HTTParty.get(url)
    else
      continue = false
    end

  end

  pull_counts
end

def process_counts(data, arch)
  pull_counts = {}
  data['results'].each do |repo|
    name = repo['name']
    count = repo['pull_count']
    pull_counts["#{name}-#{arch}"] = { name: name, arch: arch, pull_count: count}
  end

  pull_counts
end

def insert_counts(data)
  port = 8086
  port = ENV['INFLUXDB_PORT'].to_i unless ENV['INFLUXDB_PORT'].nil?

  influxdb = InfluxDB::Client.new database: "dockerhub_stats", port: port
  data.each do |key, repo_data|
    influx_data = {
      values: { pull_count: repo_data[:pull_count] },
      tags:   { repo: repo_data[:name], arch: repo_data[:arch], org: repo_data[:org] }
    }

    influxdb.write_point("dockerhub_stats", influx_data)
  end

end

def run
  orgs = nil

  # pull orgs from env var
  unless ENV['DOCKERHUB_ORGS'].nil?
    orgs = ENV['DOCKERHUB_ORGS'].split(',')
  end

  # pull orgs from settings file if not specified in env var
  if orgs.nil?
    config_file = File.dirname(__FILE__) + "/config/settings.yml"
    config = YAML.load_file(config_file)

    orgs = config["orgs"]
  end

  counts = {}
  orgs.each do |org|
    org_counts = processs_org(org)
    org_counts.each do |org_count|
      org_count[1][:org] = org
    end

    counts.merge!(org_counts)
  end

  insert_counts(counts)
end

run