wod 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in wod.gemspec
+gemspec

data/README.md

@@ -0,0 +1,47 @@
+The Wizard Of Dev center interacts with the Apple Dev Center web pages so you don't have to!
+
+Install
+---
+
+    $ gem install wod
+
+Usage
+---
+    $ wod help
+    
+    === General Commands
+
+    help                      # show this usage
+    version                   # show the gem version
+
+    login                     # log in with your apple credentials
+    logout                    # clear local authentication credentials
+
+    devices                   # list your registered devices
+    devices:add <name> <udid> # add a new device
+    devices:remove <name>     # remove device (Still counts against device limit)
+
+Examples
+---
+
+List Devices:
+
+    $ wod devices
+    Steve Jobs iPad 3  | 554f3fg54bc953547ry7a6bd62c678c11e912345
+    Jon Ive's iPhone 5 | 2d84d56ceg52c49379537413d3b9865ae2b12345
+
+Add Device:
+
+    $ wod devices:add "Dave's iPod Touch" 2d84d56ceg52c49379537413d3b9865ae2b12345
+    
+Remove Device:
+
+    $ wod devices:remove "Jon Ive's iPhone 5"
+    
+
+Roadmap
+---
+
+More features to come as I need them or as other people fork and add em ;)
+
+Proudly brought to you by the wizard of identity crises (whatupdave/snappycode/wizard of id)

data/Rakefile

@@ -0,0 +1,2 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks

data/bin/wod

@@ -0,0 +1,11 @@
+#!/usr/bin/env ruby
+
+$:.unshift File.join File.dirname(__FILE__), '../lib'
+
+require "wod"
+
+args = ARGV.dup
+ARGV.clear
+command = args.shift.strip rescue 'help'
+
+Wod::Command.run command, args

data/lib/wod.rb

@@ -0,0 +1,6 @@
+module Wod
+  class InvalidCredentials < RuntimeError; end
+end
+
+require 'wod/client'
+require 'wod/command'

data/lib/wod/client.rb

@@ -0,0 +1,81 @@
+require 'mechanize'
+require 'wod/helpers'
+require 'wod/version'
+
+module Wod
+  class DevCenterPage
+    attr_reader :page
+    
+    def initialize page
+      @page = page
+    end
+    
+    def logged_in?
+      (page.title =~ /sign in/i) == nil && (page.search("span").find {|s| s.text == "Log in"}) == nil
+    end
+    
+    def search arg
+      @page.search arg
+    end
+    
+    def form arg
+      @page.form arg
+    end
+  end
+
+  class Client
+    include Wod::Helpers
+  
+    attr_reader :name
+  
+    def initialize username, password
+      @username = username
+      @password = password
+    end
+  
+    def cookies_file
+      "#{home_directory}/.wod/cookie_jar"
+    end
+    
+    def create_agent
+      agent = Mechanize.new
+      agent.cookie_jar.load cookies_file if File.exists?(cookies_file)
+      agent
+    end
+  
+    def agent
+      @agent ||= create_agent
+    end
+    
+    def login_and_reopen url
+      page = DevCenterPage.new agent.get("https://developer.apple.com/devcenter/ios/index.action")
+
+      unless page.logged_in?
+        puts "Creating session"
+        login_page = page.page.links.find { |l| l.text == 'Log in'}.click
+
+        f = login_page.form("appleConnectForm")
+        f.theAccountName = @username
+        f.theAccountPW = @password
+        f.submit
+      end
+      
+      page = DevCenterPage.new agent.get url
+      raise InvalidCredentials unless page.logged_in?
+      agent.cookie_jar.save_as cookies_file
+      page
+    end
+
+    def get url
+      page = DevCenterPage.new agent.get(url)
+      page = login_and_reopen(url) unless page.logged_in?
+      page
+    end
+    
+    def logged_in?
+      page = get "https://developer.apple.com/devcenter/ios/index.action"
+      page.logged_in?
+    end
+  
+  end
+end

data/lib/wod/command.rb

@@ -0,0 +1,59 @@
+require 'wod/helpers'
+require 'wod/commands/base'
+
+Dir["#{File.dirname(__FILE__)}/commands/*.rb"].each { |c| require c }
+
+module Wod
+  module Command
+    class InvalidCommand < RuntimeError; end
+    class CommandFailed  < RuntimeError; end
+    
+    extend Wod::Helpers
+    
+    def self.run(command, args, retries=0)
+      begin
+        run_internal 'auth:reauthorize', args.dup if retries > 0
+        run_internal command, args.dup
+      rescue InvalidCommand
+        error "Unknown command. Run 'wod help' for usage information."
+      rescue Wod::InvalidCredentials
+        if retries < 3
+          STDERR.puts "Authentication failure"
+          run command, args, retries + 1
+        else
+          error "Authentication failure"
+        end
+      end
+      
+    end
+    
+    def self.run_internal(command, args, wod=nil)
+      klass, method = parse command
+      runner = klass.new args, wod
+      raise InvalidCommand unless runner.respond_to?(method)
+      runner.send method
+    end
+    
+    def self.parse(command)
+      parts = command.split(':')
+      case parts.size
+        when 1
+          begin
+            return eval("Wod::Command::#{command.capitalize}"), :index
+          rescue NameError, NoMethodError
+            return Wod::Command::App, command.to_sym
+          end
+        else
+          begin
+            const = Wod::Command
+            command = parts.pop
+            parts.each { |part| const = const.const_get(part.capitalize) }
+            return const, command.to_sym
+          rescue NameError
+            raise InvalidCommand
+          end
+      end
+    end
+    
+  end
+end

data/lib/wod/commands/app.rb

@@ -0,0 +1,15 @@
+require 'readline'
+
+module Wod::Command
+  class App < Base
+    def login
+      Wod::Command.run_internal "auth:reauthorize", args.dup
+    end
+    
+    def logout
+      Wod::Command.run_internal "auth:delete_credentials", args.dup
+      puts "Local credentials cleared."
+    end
+    
+  end
+end

data/lib/wod/commands/auth.rb

@@ -0,0 +1,113 @@
+require 'readline'
+
+module Wod::Command
+  class Auth < Base
+    attr_accessor :credentials
+    
+    def client
+      @client = Wod::Client.new(user, password)
+    end
+        
+    # just a stub; will raise if not authenticated
+    def check
+      client.logged_in?
+    end
+    
+    def reauthorize
+      @credentials = ask_for_and_save_credentials
+    end
+    
+    def user
+      get_credentials
+      @credentials[0]
+    end
+    
+    def password
+      get_credentials
+      @credentials[1]
+    end
+    
+    def credentials_file
+      "#{home_directory}/.wod/credentials"
+    end
+    
+    def get_credentials
+      return if @credentials
+      unless @credentials = read_credentials
+        ask_for_and_save_credentials
+      end
+      @credentials
+    end
+    
+    def read_credentials
+      File.exists?(credentials_file) and File.read(credentials_file).split("\n")
+    end
+    
+    def echo_off
+      system "stty -echo"
+    end
+    
+    def echo_on
+      system "stty echo"
+    end
+    
+    def ask_for_credentials
+      puts "Enter your apple credentials"
+      
+      print "Apple ID: "
+      user = ask
+      
+      print "Password: "
+      password = ask_for_password
+
+      [user, password]
+    end
+    
+    def ask_for_password
+      echo_off
+      password = ask
+      puts
+      echo_on
+      return password
+    end
+    
+    def ask_for_and_save_credentials
+      begin
+       @credentials = ask_for_credentials
+       write_credentials
+       check
+      rescue ::Wod::InvalidCredentials
+        delete_credentials
+        @client = nil
+        @credentials = nil
+        puts "Authentication failed."
+        return if retry_login?
+        exit 1
+      end
+    end
+    
+    def retry_login?
+      @login_attempts ||= 0
+      @login_attempts += 1
+      @login_attempts < 3
+    end
+    
+   def write_credentials
+      FileUtils.mkdir_p(File.dirname(credentials_file))
+      f = File.open(credentials_file, 'w')
+      f.chmod(0600)
+      f.puts self.credentials
+      f.close
+      set_credentials_permissions
+    end
+    
+    def set_credentials_permissions
+      FileUtils.chmod 0700, File.dirname(credentials_file)
+      FileUtils.chmod 0600, credentials_file
+    end
+
+    def delete_credentials
+      FileUtils.rm_rf "#{home_directory}/.wod/"
+    end
+  end
+end

data/lib/wod/commands/base.rb

@@ -0,0 +1,18 @@
+require 'fileutils'
+
+module Wod::Command
+  class Base
+    include Wod::Helpers
+    
+    attr_accessor :args
+    
+    def initialize(args, wod=nil)
+      @args = args
+      @wod = wod
+    end
+    
+    def wod
+      @wod ||= Wod::Command.run_internal('auth:client', args)
+    end
+  end
+end

data/lib/wod/commands/devices.rb

@@ -0,0 +1,50 @@
+
+module Wod::Command
+  class Devices < Base
+    def index
+      page = wod.get "https://developer.apple.com/ios/manage/devices/index.action"
+      
+      names = page.search("td.name span").map(&:text)
+      udids = page.search("td.id").map(&:text)
+    
+      devices_left = page.search(".devicesannounce strong").first.text
+      devices = names.map.with_index{|name, i| {:name => name, :udid => udids[i] } }
+      
+      display_formatted devices, [:name, :udid]
+      puts
+      puts "#{devices.size} devices registered. #{devices_left}"
+    end
+    
+    def add
+      name = args.shift
+      udid = args.shift
+      
+      page = wod.get "https://developer.apple.com/ios/manage/devices/add.action"
+      
+      form = page.form "add"
+      form["deviceNameList[0]"] = name
+      form["deviceNumberList[0]"] = udid
+      
+      form.submit
+    end
+    
+    def remove
+      name = args.shift
+      
+      page = wod.get "http://developer.apple.com/ios/manage/devices/index.action"
+      
+      device_span = page.search("span:contains('#{name}')")
+      if device_span.empty?
+        error "Device not found"
+      end
+      
+      tr = device_span.first.parent.parent
+      row_identifier = tr.search("input[name='__checkbox_selectedValues']").first[:value]
+      
+      form = page.form "removeDevice"
+      checkbox = form.checkboxes.find {|c| c[:value] == row_identifier}
+      checkbox.check
+      form.submit
+    end
+  end
+end

data/lib/wod/commands/help.rb

@@ -0,0 +1,76 @@
+module Wod::Command
+  class Help < Base
+    class HelpGroup < Array
+      attr_reader :title
+
+      def initialize(title)
+        @title = title
+      end
+
+      def command(name, description)
+        self << [name, description]
+      end
+
+      def space
+        self << ['', '']
+      end
+    end
+    
+    def self.groups
+      @groups ||= []
+    end
+
+    def self.group(title, &block)
+      groups << begin
+        group = HelpGroup.new(title)
+        yield group
+        group
+      end
+    end
+
+    def self.create_default_groups!
+      return if @defaults_created
+      @defaults_created = true
+      group 'General Commands' do |group|
+        group.command 'help',                         'show this usage'
+        group.command 'version',                      'show the gem version'
+        group.space
+        group.command 'login',                        'log in with your apple credentials'
+        group.command 'logout',                       'clear local authentication credentials'
+        group.space
+        group.command 'devices',                         'list your registered devices'
+        group.command 'devices:add <name> <udid>',      'add a new device'
+        group.command 'devices:remove <name>',      'remove device (Still counts against device limit)'
+      end
+    end    
+    
+    def index
+      puts usage
+    end
+    
+    def usage
+      longest_command_length = self.class.groups.map do |group|
+        group.map { |g| g.first.length }
+      end.flatten.max
+      
+      self.class.groups.inject(StringIO.new) do |output, group|
+        output.puts "=== %s" % group.title
+        output.puts
+      
+        group.each do |command, description|
+          if command.empty?
+            output.puts
+          else
+            output.puts "%-*s # %s" % [longest_command_length, command, description]
+          end
+        end
+      
+        output.puts
+        output
+      end.string
+    end
+
+  end
+end
+
+Wod::Command::Help.create_default_groups!

data/lib/wod/commands/version.rb

@@ -0,0 +1,7 @@
+module Wod::Command
+  class Version < Base
+    def index
+      puts Wod::Client.gem_version_string
+    end
+  end
+end

data/lib/wod/helpers.rb

@@ -0,0 +1,29 @@
+module Wod
+  module Helpers
+    def home_directory
+      ENV['HOME']
+    end
+        
+    def error(msg)
+      STDERR.puts(msg)
+      exit 1
+    end
+    
+    def ask
+      gets.strip
+    end
+    
+    def display_formatted hashes, columns
+      column_lengths = columns.map do |c|
+        hashes.map { |h| h[c].length }.max
+      end
+      sorted = hashes.sort_by {|h| h[columns.first] }
+      
+      sorted.each.with_index do |h, i|
+        puts " " + columns.map.with_index{ |c, i| h[c].ljust(column_lengths[i]) }.join(" | ")
+      end
+    end
+    
+    
+  end
+end

data/lib/wod/version.rb

@@ -0,0 +1,3 @@
+module Wod
+  VERSION = "0.0.1"
+end

data/wod.gemspec

@@ -0,0 +1,23 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "wod/version"
+
+Gem::Specification.new do |s|
+  s.name        = "wod"
+  s.version     = Wod::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Dave Newman"]
+  s.email       = ["dave@whatupdave.com"]
+  s.homepage    = "http://github.com/snappycode/wod"
+  s.summary     = %q{The Wizard Of Dev center}
+  s.description = %q{Command line tool for interacting with the Apple Dev Center}
+
+  s.rubyforge_project = "wod"
+  
+  s.add_dependency "mechanize"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+end

metadata

@@ -0,0 +1,83 @@
+--- !ruby/object:Gem::Specification 
+name: wod
+version: !ruby/object:Gem::Version 
+  prerelease: 
+  version: 0.0.1
+platform: ruby
+authors: 
+- Dave Newman
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-04-01 00:00:00 +11:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: mechanize
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+description: Command line tool for interacting with the Apple Dev Center
+email: 
+- dave@whatupdave.com
+executables: 
+- wod
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- README.md
+- Rakefile
+- bin/wod
+- lib/wod.rb
+- lib/wod/client.rb
+- lib/wod/command.rb
+- lib/wod/commands/app.rb
+- lib/wod/commands/auth.rb
+- lib/wod/commands/base.rb
+- lib/wod/commands/devices.rb
+- lib/wod/commands/help.rb
+- lib/wod/commands/version.rb
+- lib/wod/helpers.rb
+- lib/wod/version.rb
+- wod.gemspec
+has_rdoc: true
+homepage: http://github.com/snappycode/wod
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+requirements: []
+
+rubyforge_project: wod
+rubygems_version: 1.5.3
+signing_key: 
+specification_version: 3
+summary: The Wizard Of Dev center
+test_files: []
+