petit 0.0.1 → 0.0.2

data/lib/petit/core_ext.rb

@@ -0,0 +1,5 @@
+class Hash
+  def rename_key(old,new)
+    self[new] = self.delete(old)
+  end
+end

data/lib/petit/router.rb

@@ -8,9 +8,9 @@ module Petit
       #   +links+ a hash of links to redirect
       def builder(links = Petit.links)
         @builder ||= Rack::Builder.new do
-          links.each do |key,url|
-            map "/#{key}" do
-              run Proc.new {|env| RackHelper.redirect_response(url)}
+          links.each do |source,dest|
+            map source do
+              run Proc.new {|env| RackHelper.redirect_response(dest)}
            end
           end
         end

data/lib/petit.rb

@@ -1,4 +1,5 @@
 require 'yaml'
+require 'petit/core_ext'
 
 module Petit
 
@@ -16,7 +17,15 @@ module Petit
     end
 
     def links(source = link_source)
-      @links ||= YAML::load(File.open(source))
+      unless @links
+        @links = {}
+        links = YAML::load(File.open(source))
+        links.each do |key,url|
+          @links["/#{key}"] = url
+        end
+        @links.rename_key("/root","/")
+      end
+      @links
     end
 
     # Default way to setup petit

data/petit.gemspec

@@ -3,7 +3,7 @@ $:.push File.expand_path("../lib", __FILE__)
 
 Gem::Specification.new do |s|
   s.name        = "petit"
-  s.version     = "0.0.1"
+  s.version     = "0.0.2"
   s.authors     = ["Greg Osuri"]
   s.email       = ["gosuri@gmail.com"]
   s.homepage    = "http://gosuri.github.com/petit"

data/test/links.yml

@@ -1,4 +1,4 @@
+root: http://root.com
 foo:  http://bar.com
-beth: http://beatty.net
-m:    http://morissettedibbert.name
-h:    http://hamill.name
+a:    http://apple.com
+g:    http://google.com

data/test/sample_app/links.yml

@@ -1,4 +1,4 @@
-g:   http://gregosuri.com
-gb:  http://gridbag.com
+root:   http://gregosuri.com
+gb:     http://gridbag.com
 # Links go here
 # foo: http://bar.com   # => will redirect yourdomain/foo to http://bar.com

data/test/test_petit.rb

@@ -7,7 +7,12 @@ class TestPetit < Test::Unit::TestCase
   end
 
   def test_load_link_files
-    assert       @links.has_key? 'foo'
-    assert_equal @links['foo'], 'http://bar.com'
+    assert       @links.has_key? '/foo'
+    assert_equal @links['/foo'], 'http://bar.com'
+  end
+
+  def test_root_link
+    assert_equal @links['/'], 'http://root.com'
   end
 end
+

data/test/test_router.rb

@@ -4,7 +4,7 @@ require 'rack/mock'
 class TestRouter < Test::Unit::TestCase
 
   def setup
-    @links = {"foo" => "http://bar.com"}
+    @links = {"/foo" => "http://bar.com"}
   end
 
   def test_builder

metadata

@@ -2,7 +2,7 @@
 name: petit
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors: 
 - Greg Osuri
@@ -51,6 +51,7 @@ files:
 - Rakefile
 - bin/petit
 - lib/petit.rb
+- lib/petit/core_ext.rb
 - lib/petit/rack_helper.rb
 - lib/petit/router.rb
 - petit.gemspec