#!/usr/bin/env ruby

require 'siteleaf'
require 'fileutils'
require 'open-uri'
require 'digest/md5'
require 'securerandom'
require 'zip'

def help
  %Q(
Usage: siteleaf [COMMAND] [OPTIONS]

Commands:
  auth                Login in with your credentials
  s, server           Starts a local server.
  c, config DOMAIN    Configure an existing directory
  n, new DOMAIN       Creates new site on siteleaf.net
  pull theme          Pulls theme files for configured site from Siteleaf
  push theme          Pushes all files in dir as theme to configured site.
  help                Prints this help document
  version             Prints the siteleaf gem version

Options:
  -h, --help          Prints this help document
  -v, --version       Prints the siteleaf gem version
  -p, --port PORT     Binds local server to PORT (default: 9292)

See https://github.com/siteleaf/siteleaf-gem for additional documentation.
)
end

def auth(re_auth = false)
  Siteleaf.load_settings if !re_auth

  if re_auth or !Siteleaf.api_key
    print 'Enter your Siteleaf e-mail: '
    email = $stdin.gets.chomp

    print 'Enter your Siteleaf password: '
    system 'stty -echo'
    password = $stdin.gets.chomp
    system 'stty echo'

    puts "\nAuthorizing..."

    if auth = Siteleaf::Client.auth(email, password) and auth.is_a?(Hash) and auth.has_key?('api_key')
      File.open(Siteleaf.settings_file,'w') do|file|
        Marshal.dump({:api_key => auth['api_key'], :api_secret => auth['api_secret']}, file)
      end
      puts "=> Gem authorized." if re_auth
    else
      puts "Could not authorize, check your e-mail or password."
    end
  end
end

def config(site)
  File.open('config.ru', 'w') { |file| file.write "# Intended for development purposes only, do not upload or use in production.
# See https://github.com/siteleaf/siteleaf-gem for documentation.
require 'rubygems'
require 'siteleaf'
run Siteleaf::Server.new(:site_id => '#{site.id}')" }

  pow_path = File.expand_path('~/.pow')
  if File.directory?(pow_path)
    site_no_tld = site.domain.gsub(/\.[a-z]{0,4}$/i,'')
    site_symlink = "#{pow_path}/#{site_no_tld}"
    FileUtils.rm(site_symlink) if File.symlink?(site_symlink)
    FileUtils.symlink(File.expand_path('.'), site_symlink)
    puts "=> Site configured with Pow, open `http://#{site_no_tld}.dev` to test site locally.\n"
  else
    puts "=> Site configured, run `siteleaf server` to test site locally.\n"
  end
end

def get_site_id
  if File.exist?('config.ru')
    if match = /:site_id => '([a-z0-9]{24})'/i.match(File.read('config.ru'))
      return match[1]
    end
  end
end

def get_theme_assets(site_id)
  if assets = Siteleaf::Client.get("sites/#{site_id}/theme/assets")
    updated_count = 0
    assets.each do |asset|
      if File.exist?(asset['filename']) && (file = File.read(asset['filename'])) && (Digest::MD5.hexdigest(file) == asset['checksum'])
        # file is up to date
      else
        file = open("#{asset['file']['url']}?#{SecureRandom.hex(16)}", /^1\.8/.match(RUBY_VERSION) ? 'r' : 'r:UTF-8') { |f| f.read }
        FileUtils.mkdir_p(File.dirname(asset['filename']))
        File.open(asset['filename'], /^1\.8/.match(RUBY_VERSION) ? 'w' : 'w:UTF-8') { |f| f.write(file) }
        updated_count += 1
      end
    end
    puts "=> #{updated_count} asset(s) downloaded.\n"
  end
end

def put_theme_assets(site_id)
  theme = 'theme.zip'
  FileUtils.rm(theme) if File.exist?(theme)
  Zip::ZipFile.open(theme, Zip::ZipFile::CREATE) do |zip|
    Dir.glob("**/*").each do |path|
      zip.add(path, path) if !['config.ru'].include?(path) && !path.start_with?('.')
    end
  end
  if response = Siteleaf::Client.post("sites/#{site_id}/theme/assets", {:file => File.new(theme), :replace => true})
    updated_count = response.is_a?(Array) ? response.size : 1
    puts "=> #{updated_count} asset(s) uploaded.\n"
  else
    puts "=> Could not upload assets.\n"
  end
  FileUtils.rm(theme) if File.exist?(theme)
end

case ARGV[0]
when '-v', '--version', 'version'
  puts Siteleaf::VERSION
when '-h', '--help', 'help'
  puts help
when 's', 'server'
  if File.exist?('config.ru')
    port = ARGV[2] if %w[-p --port].include?(ARGV[1]) && ARGV[1]
    `rackup config.ru -p #{port || '9292'} >&2`
  else
    puts "No config found, run `siteleaf config yoursite.com`.\n"
  end
when 'auth'
  auth true
when 'c', 'config', 'setup'
  auth
  if site = Siteleaf::Site.find_by_domain(ARGV[1])
    config site
  else
    puts "No site found for `#{ARGV[1]}`, run `siteleaf new #{ARGV[1]}` to create it.\n"
  end
when 'n', 'new'
  auth
  if site = Siteleaf::Site.create(:title => ARGV[1], :domain => ARGV[1])
    dir = ARGV.size >= 3 ? ARGV[2] : ARGV[1]
    Dir.mkdir(dir) unless File.directory?(dir)
    Dir.chdir(dir)
    config site
  else
    puts "Could not create site `#{ARGV[1]}`.\n"
  end
when 'pull'
  case ARGV[1]
  when 'theme'
    auth
    if site_id = get_site_id
      get_theme_assets(site_id)
    else
      puts "Site not configured, run `siteleaf config yoursite.com`.\n"
    end
  else
    puts "`#{ARGV[0]}` command not found.\n"
  end
when 'push'
  case ARGV[1]
  when 'theme'
    auth
    if site_id = get_site_id
      put_theme_assets(site_id)
    else
      puts "Site not configured, run `siteleaf config yoursite.com`.\n"
    end
  else
    puts "`#{ARGV[0]}` command not found.\n"
  end
else
  puts "`#{ARGV[0]}` command not found.\n"
  puts help
end