atmos-singem 0.0.5

data/lib/generators/twitter/templates/features/basics.feature.erb

@@ -0,0 +1,11 @@
+Feature: Visiting the Home Page
+  In order to display a splash page
+
+  Scenario: Seeing the splash page unauthenticated
+    When I visit the home page
+    Then I should be prompted to try the app
+
+  Scenario: Seeing the splash page authenticated
+    Given I am logged in
+    When I visit the home page
+    Then I should be greeted

data/lib/generators/twitter/templates/features/step_definitions/basics.rb.erb

@@ -0,0 +1,15 @@
+When /^I visit the home page$/ do
+  get '/'
+end
+
+Then /^I should be greeted$/ do
+  last_response.should have_selector("h2:contains('Hello There, you little twitterer!')")
+end
+
+Then /^I should be prompted to try the app$/ do
+  last_response.should have_selector("p:contains('Would you mind trying out my new app')")
+end
+
+Given /^I am logged in$/ do
+  login_quentin
+end

data/lib/generators/twitter/templates/features/support/env.rb.erb

@@ -0,0 +1,20 @@
+require 'spec'
+require 'spec/mocks'
+require 'webrat'
+require 'rack/test'
+require 'dm-sweatshop'
+require 'pp'
+
+require File.dirname(__FILE__)+'/../../lib/<%= name %>'
+require File.dirname(__FILE__)+'/../../spec/helpers'
+
+World(Rack::Test::Methods)
+World(Spec::Mocks::ExampleMethods)
+World(Webrat::Methods)
+World(Webrat::Matchers)
+World(<%= name.camelize %>::AppHelpers)
+
+Before do
+  DataMapper.setup(:default, "sqlite3://:memory:")
+  DataMapper.auto_migrate!
+end

data/lib/generators/twitter/templates/features/viewing_my_wish_list.feature

@@ -0,0 +1,14 @@
+Feature: Viewing my wish list
+  In order to display a users' preferences
+  As a twitter user
+
+  Background:
+    Given I am logged in
+    And I have created a "53cm Pistas" category
+
+  Scenario: After Authenticating
+    When I visit the home page
+    Then I should see my list of categories
+    When I enter "Prostitutes" for a new category
+    And I click submit
+    Then I should see the "Prostitutes" list

data/lib/generators/twitter/templates/features/viewing_the_site_for_the_first_time.feature

@@ -0,0 +1,6 @@
+Feature: Visiting the Home Page
+  In order to display a splash page
+  As an anonymous user
+    Scenario: Seeing the splash page
+    When I visit the home page
+    Then I should be greeted

data/lib/generators/twitter/templates/lib/templates.rb.erb

@@ -0,0 +1,56 @@
+gem 'oauth'
+require 'oauth'
+gem 'json'
+require 'json'
+gem 'haml', '~>2.0.9'
+require 'haml/util'
+require 'haml/engine'
+gem 'curb'
+require 'curb'
+require 'logger'
+
+gem 'data_objects', '~>0.9.11'
+gem 'dm-core', '~>0.9.10'
+gem 'dm-types', '~>0.9.10'
+gem 'dm-validations', '~>0.9.10'
+gem 'dm-timestamps', '~>0.9.10'
+require 'dm-core'
+require 'dm-types'
+require 'dm-validations'
+require 'dm-timestamps'
+require 'sinatra/base'
+
+module <%= name.camelize %>
+  module Log
+    def self.logger
+      if @logger.nil?
+        @logger        = Logger.new("<%= name %>.log")
+        @logger.level  = Logger::INFO 
+      end
+      @logger
+    end
+  end
+
+  module OAuth
+    def self.consumer
+      ::OAuth::Consumer.new(ENV['<%= name.upcase %>_READKEY'],
+                            ENV['<%= name.upcase %>_READSECRET'],
+                            {:site => 'http://twitter.com'})
+    end
+  end
+
+  def self.retryable(options = {}, &block)
+    opts = { :tries => 1, :on => StandardError }.merge(options)
+    retry_exception, retries = opts[:on], opts[:tries]
+
+    begin
+      return yield
+    rescue retry_exception
+      retry if (retries -= 1) > 0
+    end
+    yield
+  end
+end
+
+require File.dirname(__FILE__)+'/<%= name %>/models/user'
+require File.dirname(__FILE__)+'/<%= name %>/sinatra/app'

data/lib/generators/twitter/templates/lib/templates/models/user.rb.erb

@@ -0,0 +1,38 @@
+module <%= name.camelize %>
+  class User
+    include DataMapper::Resource
+    class  UserCreationError < StandardError; end
+    storage_names[:default] = '<%= name.camelize %>_users'
+
+    property :id, Serial
+    property :twitter_id, Integer, :nullable => false, :unique => true
+    property :name, String
+    property :token, String
+    property :secret, String
+
+    property :url, String, :length => 512
+    property :avatar, String, :length => 512, :default => 'http://static.twitter.com/images/default_profile_normal.png'
+
+    timestamps :at
+
+    def access_token
+      ::OAuth::AccessToken.new(::<%= name.camelize %>::OAuth.consumer, token, secret)
+    end
+
+    def self.create_twitter_user(twitter_id)
+      content = Curl::Easy.perform("http://twitter.com/users/show/#{twitter_id}.json") do |curl|
+        curl.timeout = 30
+      end
+      user_info = JSON.parse(content.body_str)
+      raise UserCreationError.new("Unable to find '#{twitter_id}'") if(user_info['error'] == 'Not found')
+      result = unless user_info['error']
+        self.first_or_create({:twitter_id => user_info['id']},{
+                              :name       => user_info['name'],
+                              :avatar     => user_info['profile_image_url'],
+                              :url        => 'http://twitter.com/'+user_info['screen_name']})
+      end
+    rescue JSON::ParserError
+      raise UserCreationError.new("Unable to find '#{twitter_id}'")
+    end
+  end
+end

data/lib/generators/twitter/templates/lib/templates/sinatra/app.rb.erb

@@ -0,0 +1,66 @@
+module <%= name.camelize %>
+  class App < Sinatra::Default
+    set    :views, File.expand_path(File.dirname(__FILE__)+'/../views')
+    enable :sessions
+    enable :methodoverride
+
+    helpers do
+      def oauth_consumer
+        ::<%= name.camelize %>::OAuth.consumer
+      end
+
+      def current_user
+        session[:user_id].nil? ? nil : ::<%= name.camelize %>::User.get(session[:user_id])
+      end
+    end
+
+    error do
+      <%= name.camelize %>::Log.logger.info env['sinatra.error'].message
+      <%= name.camelize %>::Log.logger.info env['sinatra.error'].backtrace.join("\n")
+      haml :failed
+    end
+
+    get '/' do
+      haml current_user.nil? ? :about : :home
+    end
+
+    get '/authenticate' do
+      ::<%= name.camelize %>.retryable(:times => 2) do
+        request_token = oauth_consumer.get_request_token
+        session[:request_token] = request_token.token
+        session[:request_token_secret] = request_token.secret
+        redirect request_token.authorize_url
+      end
+    end
+
+    get '/peace' do
+      session.clear
+      redirect '/'
+    end
+
+    get '/callback' do
+      @request_token = ::OAuth::RequestToken.new(oauth_consumer,
+                                                 session[:request_token],
+                                                 session[:request_token_secret])
+
+      access_token = @request_token.get_access_token
+
+      oauth_response = oauth_consumer.request(:get, '/account/verify_credentials.json',
+                                              access_token, { :scheme => :query_string })
+      case oauth_response
+      when Net::HTTPSuccess
+        user_info = JSON.parse(oauth_response.body)
+        user = ::<%= name.camelize %>::User.first_or_create(:twitter_id  => user_info['id'])  # really wish first_or_create behaved sanely
+        user.name,  user.avatar = user_info['name'], user_info['profile_image_url']
+        user.token, user.secret = access_token.token, access_token.secret
+        user.url                = 'http://twitter.com/'+user_info['screen_name']
+        user.save
+
+        session[:user_id] = user.id
+        redirect '/'
+      else
+        raise ArgumentError.new('Unhandled HTTP Response')
+      end
+    end
+  end
+end

data/lib/generators/twitter/templates/lib/templates/views/about.haml

@@ -0,0 +1,5 @@
+%h2 Do you use twitter?
+%p
+  Would you mind trying out my new app?
+%p
+  Enter your custom stuff here.

data/lib/generators/twitter/templates/lib/templates/views/failed.haml

@@ -0,0 +1,4 @@
+%p
+  Something went wonky.
+  a{:href => '/'} Try again
+  ?

data/lib/generators/twitter/templates/lib/templates/views/home.haml

@@ -0,0 +1 @@
+%h2 Hello There, you little twitterer!

data/lib/generators/twitter/templates/spec/helpers.rb.erb

@@ -0,0 +1,43 @@
+class Net::HTTPResponse 
+  def body=(content) 
+    @body = content 
+    @read = true 
+  end
+end
+
+Webrat.configure do |config|
+  config.mode = :rack_test
+  config.application_port = 4567
+end
+
+module <%= name.camelize %>::AppHelpers
+  def app
+    @app = Rack::Builder.new do
+      run <%= name.camelize %>::App
+    end
+  end
+
+  def login_quentin
+    response = Net::HTTPSuccess.new('1.0', 200, nil)
+    response.body = "{\"description\":\"lulz\",\"profile_background_image_url\":\"http:\\/\\/static.twitter.com\\/images\\/themes\\/theme3\\/bg.gif\",\"utc_offset\":-25200,\"friends_count\":157,\"profile_background_color\":\"EDECE9\",\"profile_text_color\":\"634047\",\"url\":\"http:\\/\\/example.org\",\"name\":\"Quentin Blake\",\"favourites_count\":6,\"profile_link_color\":\"088253\",\"protected\":false,\"status\":{\"truncated\":false,\"in_reply_to_status_id\":null,\"text\":\"stu stu studio\",\"in_reply_to_user_id\":null,\"favorited\":false,\"created_at\":\"Tue Mar 31 19:02:12 +0000 2009\",\"id\":1426242614,\"source\":\"<a href=\\\"http:\\/\\/iconfactory.com\\/software\\/twitterrific\\\">twitterrific<\\/a>\"},\"created_at\":\"Sun Mar 18 20:07:13 +0000 2007\",\"statuses_count\":2560,\"profile_background_tile\":false,\"time_zone\":\"Mountain Time (US & Canada)\",\"profile_sidebar_fill_color\":\"E3E2DE\",\"profile_image_url\":\"http:\\/\\/static.twitter.com\\/images\\/default_profile_normal.png\",\"notifications\":false,\"profile_sidebar_border_color\":\"D3D2CF\",\"location\":\"Boulder, Colorado\",\"id\":1484261,\"following\":false,\"followers_count\":368,\"screen_name\":\"caboose\"}"
+    login(response)
+  end
+
+  def unauthorized_quentin
+    response = Net::HTTPUnauthorized.new('1.0', 401, nil)
+    response.body = "Unauthorized"
+    lambda { login(response) }.should raise_error(ArgumentError)
+  end
+
+  def login(response)
+    token = 'oU5W1XD2TTZhWT6Snfii9JbVBUkJOurCKhWQHz98765' 
+
+    consumer = mock('Consumer', {:request => response})
+    request_token = mock('RequestToken', {:get_access_token => mock('AccessToken', {:token => 'foo', :secret => 'bar'})})
+
+    OAuth::Consumer.stub!(:new).and_return(consumer)
+    OAuth::RequestToken.stub!(:new).and_return(request_token)
+
+    visit "/callback?oauth_token=#{token}"
+  end
+end

data/lib/generators/twitter/templates/spec/spec_helper.rb.erb

@@ -0,0 +1,38 @@
+$TESTING=true
+$:.push File.join(File.dirname(__FILE__), '..', 'lib')
+require 'rubygems'
+require 'randexp'
+require '<%= name %>'
+require 'rack/test'
+require 'webrat'
+require 'dm-sweatshop'
+gem 'fakeweb', '~>1.2.4'
+require 'fakeweb'
+require 'pp'
+
+FakeWeb.allow_net_connect = false
+ENV['<%= name.upcase %>_READKEY'] = /\w{18}/.gen
+ENV['<%= name.upcase %>_READSECRET'] = /\w{18}/.gen
+
+require File.dirname(__FILE__)+'/fixtures'
+require File.dirname(__FILE__)+'/helpers'
+
+DataMapper.setup(:default, 'sqlite3::memory:')
+
+Spec::Runner.configure do |config|
+  config.include(Rack::Test::Methods)
+  config.include(Webrat::Methods)
+  config.include(Webrat::Matchers)
+  config.include(<%= name.camelize %>::AppHelpers)
+
+  config.before(:each) do
+    ::<%= name.camelize %>::App.environment = :development
+    DataMapper.auto_migrate!
+    FakeWeb.clean_registry
+    FakeWeb.register_uri(:post, "http://twitter.com:80/oauth/request_token",
+      [{:body => "oauth_token=requestkey&oauth_token_secret=requestsecret", :status => ["200", "OK"]},
+       {:body => "",            :status => ["401", "Unauthorized"]},
+       {:body => "",            :status => ["403", "Forbidden"]},
+       {:body => "Bad Gateway", :status => ["502", "Bad Gateway"]} ])
+  end
+end

data/lib/generators/twitter/templates/spec/templates_spec.rb.erb

@@ -0,0 +1,59 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+describe "visiting /" do
+  describe "when unauthenticated" do
+    it "should display the about page" do
+      get '/'
+      last_response.should have_selector("h2:contains('Do you use twitter?')")
+    end
+  end
+  describe "when unauthorized" do
+    it "should raise an error" do
+      lambda { unauthorized_quentin }.should raise_error
+    end
+  end
+  describe "when authenticated" do
+    it "should display a greeting" do
+      login_quentin
+      get '/'
+      last_response.should have_selector("h2:contains('Hello There, you little twitterer!')")
+    end
+  end
+end
+
+describe "visiting /peace" do
+  describe "when authenticated" do
+    it "should clear the session and redirect to the home page" do
+      login_quentin
+      get '/peace'
+      last_response.headers['Location'].should eql('/')
+
+      get last_response.headers['Location']
+      last_response.should have_selector("h2:contains('Do you use twitter?')")
+    end
+  end
+end
+
+describe "triggering a 404" do
+  it "should display the failed page" do
+    pending
+    get '/lost'
+    last_response.should have_selector("p:contains('Something went wonky.')")
+  end
+end
+
+describe "visiting /authenticate" do
+  it "should redirect to twitter for authentication" do
+    get '/authenticate'
+    last_response.headers['Location'].should eql('http://twitter.com/oauth/authorize?oauth_token=requestkey')
+  end
+end
+
+describe "<%= name.camelize %>::User" do
+  it "can create users from twitter ids that exist" do
+    lambda { <%= name.camelize %>::User.create_twitter_user('atmos') }.should_not raise_error
+  end
+  it "can't create users from twitter ids that do not exist" do
+    lambda { <%= name.camelize %>::User.create_twitter_user(/\w{18}/.gen) }.should raise_error(<%= name.camelize %>::User::UserCreationError)
+  end
+end

data/lib/generators/twitter/twitter_generator.rb

@@ -0,0 +1,90 @@
+class TwitterGenerator < RubiGen::Base
+  DEFAULT_SHEBANG = File.join(Config::CONFIG['bindir'],
+                              Config::CONFIG['ruby_install_name'])
+
+  default_options :author => nil
+  attr_reader :name
+
+  def initialize(runtime_args, runtime_options = {})
+    super
+    usage if args.empty?
+    @destination_root = File.expand_path(args.shift)
+    @name = base_name
+    extract_options
+  end
+
+  def manifest
+    record do |m|
+      # Ensure appropriate folder(s) exists
+      m.directory ''
+      BASEDIRS.each { |path| m.directory path }
+      m.directory "lib/#{name}"
+      m.directory "features/support"
+      m.directory "features/step_definitions"
+      m.directory "lib/#{name}/sinatra"
+      m.directory "lib/#{name}/models"
+      m.directory "lib/#{name}/views"
+
+      # Create stubs
+      m.template "config.ru.erb", "config.ru.example"
+      m.template "lib/templates.rb.erb", "lib/#{name}.rb"
+      m.template "lib/templates/sinatra/app.rb.erb", "lib/#{name}/sinatra/app.rb"
+      m.template "lib/templates/models/user.rb.erb", "lib/#{name}/models/user.rb"
+
+      # cucumber stubs
+      m.template "features/support/env.rb.erb", "features/support/env.rb"
+      m.template "features/basics.feature.erb", "features/#{name}.feature"
+      m.template "features/step_definitions/basics.rb.erb", "features/step_definitions/#{name}.rb"
+
+      # rspec stubs
+      m.template "spec/spec_helper.rb.erb", "spec/spec_helper.rb"
+      m.template "spec/helpers.rb.erb", "spec/helpers.rb"
+      m.template "spec/templates_spec.rb.erb", "spec/#{name}_spec.rb"
+      m.template "spec/fixtures.rb.erb", "spec/fixtures.rb"
+
+      %w(LICENSE Rakefile README.md).each do |file| 
+        m.template file, file
+      end
+      %w(TODO).each do |file| 
+        m.file file, file
+      end
+      %w(about failed home).each do |file|
+        m.file "lib/templates/views/#{file}.haml", "lib/#{name}/views/#{file}.haml"
+      end
+    end
+  end
+
+  protected
+    def banner
+      <<-EOS
+Creates a simple RubyGems scaffold.
+
+USAGE: #{spec.name} name --simple"
+EOS
+    end
+
+    def add_options!(opts)
+      opts.separator ''
+      opts.separator 'Options:'
+      # For each option below, place the default
+      # at the top of the file next to "default_options"
+      # opts.on("-a", "--author=\"Your Name\"", String,
+      #         "Some comment about this option",
+      #         "Default: none") { |x| options[:author] = x }
+      opts.on("-v", "--version", "Show the #{File.basename($0)} version number and quit.")
+    end
+
+    def extract_options
+      # for each option, extract it into a local variable (and create an "attr_reader :author" at the top)
+      # Templates can access these value via the attr_reader-generated methods, but not the
+      # raw instance variable value.
+      # @author = options[:author]
+    end
+
+    # Installation skeleton.  Intermediate directories are automatically
+    # created so don't sweat their absence here.
+    BASEDIRS = %w(
+      lib
+      spec
+    )
+end