klepto 0.7.2 → 0.7.3

data/lib/klepto/bot.rb

@@ -1,9 +1,24 @@
 module Klepto
   class Bot
-    attr_reader :config
+    attr_reader :config, :vars
+
+    def set(hash)
+      @vars = vars.merge(hash)
+    end 
+
+    def get(key)
+      val = vars[key]
+      
+      if val.is_a? Proc
+        val = val.call(@browser.page)
+        set(key => val)
+      end
+      val
+    end
 
     def initialize(url=nil, &block)
       @config = Klepto::Config.new
+      @vars   = {}
       @config.url url
       @queue  = []
       @browser = Klepto::Browser.new
@@ -92,6 +107,7 @@ EOS
 
     def __structure(context)
       structure = Structure.new(context)
+      structure._bot = self
 
       # A queue of DSL instructions
       queue.each do |instruction|

data/lib/klepto/structure.rb

@@ -9,6 +9,7 @@ module Klepto
     attr_reader :_parent
     attr_reader :_hash
     attr_reader :_context
+    attr_accessor :_bot
 
     def initialize(_context=nil, _parent=nil)
       Klepto.logger.debug("\tnew Structure (#{_parent}) -> (#{_context})")
@@ -32,6 +33,10 @@ module Klepto
       options[:parser]  ||= nil
       selector          = args.shift
 
+      if selector.is_a?(Proc)
+        selector = selector.call(_bot)
+      end
+
       if !block_given? && options[:parser]
         block = options[:parser].new
       end

data/lib/klepto/version.rb

@@ -1,3 +1,3 @@
 module Klepto
-  VERSION = "0.7.2"
+  VERSION = "0.7.3"
 end

data/spec/lib/klepto/bot_spec.rb

@@ -2,6 +2,48 @@ require 'spec_helper'
 
 describe Klepto::Bot do  
   describe 'Klepto::Bot.new' do
+    describe 'vars' do
+      describe 'structuring with a variable' do
+        before(:each) do
+          @bot = Klepto::Bot.new("https://www.twitter.com/justinbieber"){
+            name      'h1.fullname'
+            set :status_id => lambda{|doc|
+              doc.first('.stream-items li.stream-item')['data-item-id'].to_i
+            }
+
+            got_the_node(lambda{|bot| "[data-item-id='#{bot.get(:status_id)}']"}) do |node|
+              true
+            end
+            get_the_id 'a' do |node|
+              get(:status_id)
+            end
+          }      
+        end
+
+        it 'should have the variables' do
+          @bot.get(:status_id).should be_kind_of(Integer)
+          @bot.get(:status_id).should_not be(0)
+        end   
+
+        it 'should be able to structure with the variable' do
+          @bot.structure[:get_the_id].should be @bot.get(:status_id)
+          @bot.structure[:got_the_node].should be true
+        end   
+      end      
+      describe 'setting a variable' do
+        before(:each) do
+          @bot = Klepto::Bot.new("https://www.twitter.com/justinbieber"){
+            name      'h1.fullname'
+            set :cool => 'chickens'
+          }      
+        end
+
+        it 'should have the variables' do
+          @bot.get(:cool).should eq 'chickens'
+        end      
+      end
+    end
+
     describe 'create a bot with a redirect' do
       describe 'that aborts on redirect' do
         before(:each) do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: klepto
 version: !ruby/object:Gem::Version
-  version: 0.7.2
+  version: 0.7.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-07-15 00:00:00.000000000 Z
+date: 2013-07-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: poltergeist