require 'net/http'
require 'zip/zipfilesystem'

module AsProject

  class PlayerFetcher
  
    def initialize(url, target, zipped_file, debug=false)
      @debug = debug
      if(!debug)
        verify_download(target)
        fetch(url, target, zipped_file)
      end
    end
    
    def verify_download(target)
      puts <<EOF

Unable to find Flash Player at:
#{target}

Download now? [Yn]

EOF
      if(!$stdin.gets.chomp.downcase.index('y'))
        raise UsageError.new("Cannot find the Flash Player at #{target}, and automatic download was refused...")
      end
    end

    def fetch(url, target, zipped_file)
      puts "Fetching: #{url} to: #{target} now!"
      zip_name = url.split(File::SEPARATOR).pop
      parts = target.split(File::SEPARATOR)
      parts.pop
      zip_dir = parts.join(File::SEPARATOR)
      zip_file = File.join(zip_dir, zip_name)
      
      puts 'zip_file: ' + zip_file

      if(File.exists?(zip_file))
        result = extract(zip_file, zipped_file, target)
        write_file(target, result)
        return
      end

      response = nil
      host = 'download.macromedia.com'
        
      t = Thread.new {
        response = Net::HTTP.get_response(host, url)
      }

      while t.alive?
        print(".")
        $stdout.flush
        sleep(0.3)
      end
      
      if(response.nil?)
        msg1 = "Download failed for\n"
        msg1 << "http://#{host}#{url}\n"
        raise UsageError.new(msg1)
      end
      
      if(response.code != '200')
        msg = "Download failed for\n"
        msg << "http://#{host}#{url}\n"
        msg << "with HTTP Status code: #{response.code.to_s}\n"
        msg << response.message
        raise UsageError.new(msg)
      end
      
      puts ""
      puts "Downloaded: #{(response.body.size / 1024).to_s} KB"
      puts ""
      

      result = ''

      if(is_zip?(url))
        File.open(zip_file, 'w') do |f|
          f.write(response.body)
        end
        result = extract(zip_file, zipped_file, target)
      else
        result = response.body
      end

      write_file(target, result)
    end
    
    def write_file(file, content)
      File.open(file, 'w') do |f|
        f.write(content)
      end
      
      File.chmod(0755, file)
    end
    
    def extract(zip_file, zipped_file, unzipped_file)
      result = nil
      Zip::ZipFile.open(zip_file) do |file|
        entry = file.get_entry(zipped_file)
        result = file.get_input_stream(entry).read
      end
      return result
    end
    
    def is_zip?(file)
      return file.match(/\.zip$/)
    end
    
    def is_dmg?(file)
      return file.match(/\.dmg$/)
    end
  end
end