moviepilot/old-maid

require 'colorize'
require 'httparty'
require 'open-uri'

class Zeta::LocalOrRemoteFile
  def initialize(options)
    @options = options
  end

  def read
    if @options[:path]
      read_local
    elsif @options[:github]
      read_from_github
    else
      raise "Unknown file location #{@options}"
    end
  end

  private

  def read_local
    open(File.join(@options[:path], @options[:file])).read
  end

  def read_from_github
    self.class.http_get(github_url, verbose?)
  end

  def self.http_get(url, verbose)
    retries ||= 3
    masked_url = ENV['ZETA_HTTP_PASSWORD'].blank? ? url : url.sub(ENV['ZETA_HTTP_PASSWORD'], '***')
    print "GET #{masked_url}... " if verbose
    result = HTTParty.get url
    raise "Error #{result.code}: #{result}" unless result.code == 200
    print "OK\n".green if verbose
    result.to_s
  rescue
    print "ERROR\n".blue if verbose
    raise if (retries -= 1).zero?
    sleep 1
    retry
  end

  def verbose?
    !!@options[:verbose]
  end

  # In order not to have git as a dependency, we'll fetch from
  # raw.githubusercontent.com as long as we get away with it.
  def github_url
    repo   = @options[:github][:repo]
    branch = @options[:github][:branch]
    path   = @options[:github][:path]
    file   = @options[:file]

    uri = [branch, path, file].compact.join('/')
    u = ENV['ZETA_HTTP_USER']
    p = ENV['ZETA_HTTP_PASSWORD']
    if p
      "https://#{u}:#{p}@raw.githubusercontent.com/#{repo}/#{uri}"
    else
      "https://raw.githubusercontent.com/#{repo}/#{uri}"
    end
  end
end