pat-maddox-twinkies 0.1.0

data/VERSION.yml

@@ -0,0 +1,4 @@
+--- 
+:patch: 0
+:major: 0
+:minor: 1

data/bin/twinkies

@@ -0,0 +1,41 @@
+#!/usr/bin/env ruby
+require 'rubygems'
+require 'ostruct'
+$:.unshift File.dirname(__FILE__) + '/../lib'
+require 'twinkies'
+
+def run_app(base_dir)
+  config = OpenStruct.new
+  Dir.chdir(base_dir) do
+    eval File.read('twinkies_config.rb')
+    config.tweet_db = File.expand_path(config.tweet_db)
+  end
+  require 'twinkies/server'
+  Sinatra::Application.default_options[:env] = :production
+  Twinkies::Server.new(config).run
+end
+
+if ARGV.size == 1
+  username = ARGV.first
+
+  if File.directory?(username)
+    run_app username
+  else
+    FileUtils.mkdir(username)
+    Dir.chdir(username) do
+      File.open('twinkies_config.rb', 'w+') do |file|
+        file << "config.username='#{username}'\n"
+        file << "config.password='CHANGE_ME'\n"
+        file << "config.tweet_db='#{username}_tweets.db'\n"
+      end
+    end
+  end
+else
+  if File.exists?('twinkies_config.rb')
+    run_app '.'
+  else
+    puts "Usage: twinkies [username]"
+    puts "\ttwinkies padillac - creates a new dir named padillac with a config file if none exists, or runs a twinkie instance with the config in that dir"
+    puts "\ttwinkies - will run a twinkies instance with the config in the current dir"
+  end
+end

data/lib/twinkies/friend_searcher.rb

@@ -0,0 +1,14 @@
+module Twinkies
+  class FriendSearcher
+    def initialize(username, password)
+      @username = username
+      @password = password
+    end
+
+    def search(term)
+      Twitter::Base.new(@username, @password).timeline.select do |tweet|
+        tweet.text.include?('http')
+      end
+    end
+  end
+end

data/lib/twinkies/item.rb

@@ -0,0 +1,41 @@
+module Twinkies
+  class Item
+    include DataMapper::Resource
+    property :id, Integer, :serial => true
+    property :twitter_id, Integer
+    property :link, String
+    property :user, String
+    property :created_at, DateTime
+    property :text, Text, :lazy => false
+
+    def tweet=(tweet)
+      self.twitter_id = tweet.id
+      self.text = tweet.text
+      self.user = tweet.user.screen_name
+      self.created_at = tweet.created_at
+    end
+
+    def save
+      @existing = Item.first(:twitter_id => twitter_id)
+      if @existing && @existing.link == link
+        self.id = @existing.id
+        true
+      else
+        super
+      end
+    end
+
+    def self.latest
+      all :order => [:twitter_id.desc], :limit => 100
+    end
+
+    def self.refresh(username, password)
+      raise "Must provide credentials" if username.nil? || password.nil?
+
+      tweets = FriendSearcher.new(username, password).search('http')
+      UrlList.new(tweets) {|t| t.text}.each do |item|
+        create :link => item[:url], :tweet => item[:item]
+      end
+    end
+  end
+end

data/lib/twinkies/server.rb

@@ -0,0 +1,58 @@
+require 'sinatra'
+
+module Twinkies
+  class Server
+    def initialize(config)
+      @tweet_db = config.tweet_db
+      @username = config.username
+      @password = config.password
+    end
+
+    def run
+      setup_db
+      start_tweet_refresher
+      handle_request
+    end
+
+    def setup_db
+      DataMapper.setup(:default, "sqlite3://#{@tweet_db}")
+      DataMapper.auto_upgrade!
+    end
+
+    def start_tweet_refresher
+      Thread.abort_on_exception = true
+      Thread.new do
+        loop do
+          puts "refreshing tweets..."
+          Item.refresh @username, @password
+          sleep 60
+        end
+      end
+    end
+
+    def handle_request
+      username = @username # builder must instance_eval?
+      get '/feed.xml' do
+        builder do |xml|
+          xml.instruct!
+          xml.rss :version => '2.0' do
+            xml.channel do
+              xml.title "#{username}'s twitter URL feed"
+              xml.description "#{username}'s twitter URL feed"
+              xml.link "http://twitter.com/#{username}"
+
+              Item.latest.each do |tweet|
+                xml.item do
+                  xml.title "#{tweet.user} - #{tweet.text}"
+                  xml.link tweet.link
+                  xml.pubDate tweet.created_at
+                  xml.guid tweet.id, :isPermaLink => false
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/twinkies/url_list.rb

@@ -0,0 +1,33 @@
+module Twinkies
+  class UrlList
+    include Enumerable
+
+    def initialize(*strings)
+      strings = strings.first if strings.size == 1 && strings.first.is_a?(Array)
+      @urls = strings.inject([]) do |urls, string|
+        if block_given?
+          yield(string).scan(/http:\/\/\S+/) {|s| urls << {:item => string, :url => s}}
+          urls
+        else
+          urls += string.scan(/http:\/\/\S+/)
+        end
+      end
+    end
+
+    def each(&block)
+      @urls.each &block
+    end
+
+    def inspect
+      @urls.inspect
+    end
+
+    def to_s
+      @urls.to_s
+    end
+
+    def ==(other)
+      @urls == other
+    end
+  end
+end

data/lib/twinkies.rb

@@ -0,0 +1,6 @@
+$:.unshift File.dirname(__FILE__)
+require 'dm-core'
+require 'twitter'
+require 'twinkies/friend_searcher'
+require 'twinkies/url_list'
+require 'twinkies/item'

data/spec/item_spec.rb

@@ -0,0 +1,46 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+module Twinkies
+  describe Item do
+    describe "created" do
+      before(:each) do
+        @tweet = Twitter::Status.new
+        @tweet.id = 123
+        @tweet.text = "hello mang!"
+        @tweet.created_at = DateTime.parse('7/24/1985')
+        @user = Twitter::User.new
+        @user.screen_name = 'padillac'
+        @tweet.user = @user
+        @item = Item.create :tweet => @tweet, :link => "http://foo"
+      end
+
+      it "should get the twitter ID from the tweet id" do
+        @item.twitter_id.should == 123
+      end
+
+      it "should get the text from the tweet" do
+        @item.text.should == "hello mang!"
+      end
+
+      it "should get the screen name from the tweet" do
+        @item.user.should == 'padillac'
+      end
+
+      it "should get the date from the tweet" do
+        @item.created_at.should == DateTime.parse('7/24/1985')
+      end
+
+      describe "created with the same twitter ID" do
+        it "should return existing record if URLs are the same" do
+          i = Item.create :tweet => @tweet, :link => "http://foo"
+          i.should == @item
+        end
+
+        it "should create a new item if URLs are different" do
+          i = Item.create :tweet => @tweet, :link => "http://bar"
+          i.should_not == @item
+        end
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,10 @@
+require 'rubygems'
+require 'spec'
+$:.unshift File.expand_path(File.dirname(__FILE__) + '/../lib')
+require 'twinkies'
+
+DataMapper.setup(:default, 'sqlite3::memory:')
+
+Spec::Runner.configure do |config|
+  config.before(:each) { DataMapper.auto_migrate! }
+end

data/spec/url_list_spec.rb

@@ -0,0 +1,48 @@
+require File.dirname(__FILE__) + '/spec_helper'
+require 'ostruct'
+
+module Twinkies
+  describe UrlList do
+    it "should find a url from a string" do
+      UrlList.new('http://foo').should include('http://foo')
+    end
+
+    it "should find a url from a list" do
+      UrlList.new('http://foo', 'http://bar').
+        should include('http://foo', 'http://bar')
+    end
+
+    it "should find a url from an array" do
+      UrlList.new(['http://foo', 'http://bar']).
+        should include('http://foo', 'http://bar')
+    end
+
+    it "should find a url from a string with extra chars" do
+      UrlList.new('I like http://foo !!').should include('http://foo')
+    end
+
+    it "should not find extra characters" do
+      UrlList.new('I like http://foo !!').
+        should_not include('I', 'like')
+    end
+
+    it "should find multiple urls from a single string" do
+      UrlList.new('here: http://foo & http://bar').
+        should include('http://foo', 'http://bar')
+    end
+
+    it "should have the enumerable methods" do
+      UrlList.new('http://foo http://bar', 'http://baz').
+        map.should == ['http://foo', 'http://bar', 'http://baz']
+    end
+
+    describe "working with objects instead of strings" do
+      it "should get the strings based on a block" do
+        tweet = OpenStruct.new(:text => 'http://foo http://bar')
+        UrlList.new(tweet) {|t| t.text }.
+          should == [{:item => tweet, :url => 'http://foo'},
+                     {:item => tweet, :url => 'http://bar'}]
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,89 @@
+--- !ruby/object:Gem::Specification 
+name: pat-maddox-twinkies
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Pat Maddox
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-01-09 00:00:00 -08:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: sinatra
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: dm-core
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: do_sqlite3
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+description: TODO
+email: pat.maddox@gmail.com
+executables: 
+- twinkies
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- VERSION.yml
+- bin/twinkies
+- lib/twinkies.rb
+- lib/twinkies
+- lib/twinkies/friend_searcher.rb
+- lib/twinkies/server.rb
+- lib/twinkies/url_list.rb
+- lib/twinkies/item.rb
+- spec/spec_helper.rb
+- spec/url_list_spec.rb
+- spec/item_spec.rb
+has_rdoc: false
+homepage: http://github.com/pat-maddox/twinkies
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: TODO
+test_files: []
+