guillotine 0.0.1
Sign up to get free protection for your applications and to get access to all the features.
- data/Gemfile +27 -0
- data/LICENSE +22 -0
- data/Rakefile +135 -0
- data/config/haproxy.riak.cfg +26 -0
- data/guillotine.gemspec +73 -0
- data/lib/guillotine/adapters/active_record_adapter.rb +58 -0
- data/lib/guillotine/adapters/memory_adapter.rb +41 -0
- data/lib/guillotine/adapters/riak_adapter.rb +50 -0
- data/lib/guillotine/adapters/sequel_adapter.rb +56 -0
- data/lib/guillotine/app.rb +28 -0
- data/lib/guillotine.rb +47 -0
- data/test/active_record_adapter_test.rb +38 -0
- data/test/app_test.rb +49 -0
- data/test/helper.rb +6 -0
- data/test/memory_adapter_test.rb +29 -0
- data/test/riak_adapter_test.rb +59 -0
- data/test/sequel_adapter_test.rb +38 -0
- metadata +88 -0
data/Gemfile
ADDED
@@ -0,0 +1,27 @@
|
|
1
|
+
source 'http://rubygems.org'
|
2
|
+
|
3
|
+
gem 'sinatra'
|
4
|
+
|
5
|
+
group :test do
|
6
|
+
gem 'rack-test'
|
7
|
+
end
|
8
|
+
|
9
|
+
# Bundler isn't designed to provide optional functionality like this. You're
|
10
|
+
# on your own
|
11
|
+
#
|
12
|
+
#group :riak do
|
13
|
+
# gem 'riak-client'
|
14
|
+
#end
|
15
|
+
#
|
16
|
+
#group :sequel do
|
17
|
+
# gem 'sequel'
|
18
|
+
#end
|
19
|
+
#
|
20
|
+
#group :active_record do
|
21
|
+
# gem 'activerecord'
|
22
|
+
#end
|
23
|
+
#
|
24
|
+
#group :sequel, :active_record do
|
25
|
+
# gem 'sqlite3'
|
26
|
+
#end
|
27
|
+
|
data/LICENSE
ADDED
@@ -0,0 +1,22 @@
|
|
1
|
+
The MIT License
|
2
|
+
|
3
|
+
Copyright (c) Rick Olson
|
4
|
+
|
5
|
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
6
|
+
of this software and associated documentation files (the "Software"), to deal
|
7
|
+
in the Software without restriction, including without limitation the rights
|
8
|
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
9
|
+
copies of the Software, and to permit persons to whom the Software is
|
10
|
+
furnished to do so, subject to the following conditions:
|
11
|
+
|
12
|
+
The above copyright notice and this permission notice shall be included in
|
13
|
+
all copies or substantial portions of the Software.
|
14
|
+
|
15
|
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
16
|
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
17
|
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
18
|
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
19
|
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
20
|
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
21
|
+
THE SOFTWARE.
|
22
|
+
|
data/Rakefile
ADDED
@@ -0,0 +1,135 @@
|
|
1
|
+
require 'rubygems'
|
2
|
+
require 'rake'
|
3
|
+
require 'date'
|
4
|
+
|
5
|
+
#############################################################################
|
6
|
+
#
|
7
|
+
# Helper functions
|
8
|
+
#
|
9
|
+
#############################################################################
|
10
|
+
|
11
|
+
def name
|
12
|
+
@name ||= Dir['*.gemspec'].first.split('.').first
|
13
|
+
end
|
14
|
+
|
15
|
+
def version
|
16
|
+
line = File.read("lib/#{name}.rb")[/^\s*VERSION\s*=\s*.*/]
|
17
|
+
line.match(/.*VERSION\s*=\s*['"](.*)['"]/)[1]
|
18
|
+
end
|
19
|
+
|
20
|
+
def date
|
21
|
+
Date.today.to_s
|
22
|
+
end
|
23
|
+
|
24
|
+
def rubyforge_project
|
25
|
+
name
|
26
|
+
end
|
27
|
+
|
28
|
+
def gemspec_file
|
29
|
+
"#{name}.gemspec"
|
30
|
+
end
|
31
|
+
|
32
|
+
def gem_file
|
33
|
+
"#{name}-#{version}.gem"
|
34
|
+
end
|
35
|
+
|
36
|
+
def replace_header(head, header_name)
|
37
|
+
head.sub!(/(\.#{header_name}\s*= ').*'/) { "#{$1}#{send(header_name)}'"}
|
38
|
+
end
|
39
|
+
|
40
|
+
#############################################################################
|
41
|
+
#
|
42
|
+
# Standard tasks
|
43
|
+
#
|
44
|
+
#############################################################################
|
45
|
+
|
46
|
+
task :default => :test
|
47
|
+
|
48
|
+
require 'rake/testtask'
|
49
|
+
Rake::TestTask.new(:test) do |test|
|
50
|
+
test.libs << 'lib' << 'test'
|
51
|
+
test.pattern = 'test/**/*_test.rb'
|
52
|
+
test.verbose = true
|
53
|
+
end
|
54
|
+
|
55
|
+
desc "Open an irb session preloaded with this library"
|
56
|
+
task :console do
|
57
|
+
sh "irb -rubygems -r ./lib/#{name}.rb"
|
58
|
+
end
|
59
|
+
|
60
|
+
#############################################################################
|
61
|
+
#
|
62
|
+
# Custom tasks (add your own tasks here)
|
63
|
+
#
|
64
|
+
#############################################################################
|
65
|
+
|
66
|
+
|
67
|
+
|
68
|
+
#############################################################################
|
69
|
+
#
|
70
|
+
# Packaging tasks
|
71
|
+
#
|
72
|
+
#############################################################################
|
73
|
+
|
74
|
+
desc "Create tag v#{version} and build and push #{gem_file} to Rubygems"
|
75
|
+
task :release => :build do
|
76
|
+
unless `git branch` =~ /^\* master$/
|
77
|
+
puts "You must be on the master branch to release!"
|
78
|
+
exit!
|
79
|
+
end
|
80
|
+
sh "git commit --allow-empty -a -m 'Release #{version}'"
|
81
|
+
sh "git tag v#{version}"
|
82
|
+
sh "git push origin master"
|
83
|
+
sh "git push origin v#{version}"
|
84
|
+
sh "gem push pkg/#{name}-#{version}.gem"
|
85
|
+
end
|
86
|
+
|
87
|
+
desc "Build #{gem_file} into the pkg directory"
|
88
|
+
task :build => :gemspec do
|
89
|
+
sh "mkdir -p pkg"
|
90
|
+
sh "gem build #{gemspec_file}"
|
91
|
+
sh "mv #{gem_file} pkg"
|
92
|
+
end
|
93
|
+
|
94
|
+
desc "Generate #{gemspec_file}"
|
95
|
+
task :gemspec => :validate do
|
96
|
+
# read spec file and split out manifest section
|
97
|
+
spec = File.read(gemspec_file)
|
98
|
+
head, manifest, tail = spec.split(" # = MANIFEST =\n")
|
99
|
+
|
100
|
+
# replace name version and date
|
101
|
+
replace_header(head, :name)
|
102
|
+
replace_header(head, :version)
|
103
|
+
replace_header(head, :date)
|
104
|
+
#comment this out if your rubyforge_project has a different name
|
105
|
+
replace_header(head, :rubyforge_project)
|
106
|
+
|
107
|
+
# determine file list from git ls-files
|
108
|
+
files = `git ls-files`.
|
109
|
+
split("\n").
|
110
|
+
sort.
|
111
|
+
reject { |file| file =~ /^\./ }.
|
112
|
+
reject { |file| file =~ /^(rdoc|pkg)/ }.
|
113
|
+
map { |file| " #{file}" }.
|
114
|
+
join("\n")
|
115
|
+
|
116
|
+
# piece file back together and write
|
117
|
+
manifest = " s.files = %w[\n#{files}\n ]\n"
|
118
|
+
spec = [head, manifest, tail].join(" # = MANIFEST =\n")
|
119
|
+
File.open(gemspec_file, 'w') { |io| io.write(spec) }
|
120
|
+
puts "Updated #{gemspec_file}"
|
121
|
+
end
|
122
|
+
|
123
|
+
desc "Validate #{gemspec_file}"
|
124
|
+
task :validate do
|
125
|
+
libfiles = Dir['lib/*'] - ["lib/#{name}.rb", "lib/#{name}"]
|
126
|
+
unless libfiles.empty?
|
127
|
+
puts "Directory `lib` should only contain a `#{name}.rb` file and `#{name}` dir."
|
128
|
+
exit!
|
129
|
+
end
|
130
|
+
unless Dir['VERSION*'].empty?
|
131
|
+
puts "A `VERSION` file at root level violates Gem best practices."
|
132
|
+
exit!
|
133
|
+
end
|
134
|
+
end
|
135
|
+
|
@@ -0,0 +1,26 @@
|
|
1
|
+
# Sample haproxy config for riak protobuf api
|
2
|
+
# Assumes you have 3 riak instances setup like the fast track
|
3
|
+
# http://wiki.basho.com/Building-a-Development-Environment.html
|
4
|
+
#
|
5
|
+
# Adapted from:
|
6
|
+
# http://www.mail-archive.com/riak-users@lists.basho.com/msg03502.html
|
7
|
+
global
|
8
|
+
user technoweenie # set this to your user :)
|
9
|
+
group staff
|
10
|
+
#daemon
|
11
|
+
maxconn 100000
|
12
|
+
|
13
|
+
defaults
|
14
|
+
mode tcp
|
15
|
+
balance leastconn
|
16
|
+
clitimeout 60000
|
17
|
+
srvtimeout 60000
|
18
|
+
contimeout 5000
|
19
|
+
retries 3
|
20
|
+
option redispatch
|
21
|
+
|
22
|
+
listen riak localhost:8080
|
23
|
+
server riak1 localhost:8081 weight 1 maxconn 1000
|
24
|
+
server riak2 localhost:8082 weight 1 maxconn 1000
|
25
|
+
server riak3 localhost:8083 weight 1 maxconn 1000
|
26
|
+
|
data/guillotine.gemspec
ADDED
@@ -0,0 +1,73 @@
|
|
1
|
+
## This is the rakegem gemspec template. Make sure you read and understand
|
2
|
+
## all of the comments. Some sections require modification, and others can
|
3
|
+
## be deleted if you don't need them. Once you understand the contents of
|
4
|
+
## this file, feel free to delete any comments that begin with two hash marks.
|
5
|
+
## You can find comprehensive Gem::Specification documentation, at
|
6
|
+
## http://docs.rubygems.org/read/chapter/20
|
7
|
+
Gem::Specification.new do |s|
|
8
|
+
s.specification_version = 2 if s.respond_to? :specification_version=
|
9
|
+
s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
|
10
|
+
s.rubygems_version = '1.3.5'
|
11
|
+
|
12
|
+
## Leave these as is they will be modified for you by the rake gemspec task.
|
13
|
+
## If your rubyforge_project name is different, then edit it and comment out
|
14
|
+
## the sub! line in the Rakefile
|
15
|
+
s.name = 'guillotine'
|
16
|
+
s.version = '0.0.1'
|
17
|
+
s.date = '2011-08-08'
|
18
|
+
s.rubyforge_project = 'guillotine'
|
19
|
+
|
20
|
+
## Make sure your summary is short. The description may be as long
|
21
|
+
## as you like.
|
22
|
+
s.summary = "Adaptable private URL shortener"
|
23
|
+
s.description = "Adaptable private URL shortener"
|
24
|
+
|
25
|
+
## List the primary authors. If there are a bunch of authors, it's probably
|
26
|
+
## better to set the email to an email list or something. If you don't have
|
27
|
+
## a custom homepage, consider using your GitHub URL or the like.
|
28
|
+
s.authors = ["Rick Olson"]
|
29
|
+
s.email = 'technoweenie@gmail.com'
|
30
|
+
s.homepage = 'https://github.com/technoweenie/gitio'
|
31
|
+
|
32
|
+
## This gets added to the $LOAD_PATH so that 'lib/NAME.rb' can be required as
|
33
|
+
## require 'NAME.rb' or'/lib/NAME/file.rb' can be as require 'NAME/file.rb'
|
34
|
+
s.require_paths = %w[lib]
|
35
|
+
|
36
|
+
## List your runtime dependencies here. Runtime dependencies are those
|
37
|
+
## that are needed for an end user to actually USE your code.
|
38
|
+
s.add_dependency('sinatra', "~> 1.2.6")
|
39
|
+
|
40
|
+
## List your development dependencies here. Development dependencies are
|
41
|
+
## those that are only needed during development
|
42
|
+
s.add_development_dependency('rack-test')
|
43
|
+
|
44
|
+
## Leave this section as-is. It will be automatically generated from the
|
45
|
+
## contents of your Git repository via the gemspec task. DO NOT REMOVE
|
46
|
+
## THE MANIFEST COMMENTS, they are used as delimiters by the task.
|
47
|
+
# = MANIFEST =
|
48
|
+
s.files = %w[
|
49
|
+
Gemfile
|
50
|
+
LICENSE
|
51
|
+
Rakefile
|
52
|
+
config/haproxy.riak.cfg
|
53
|
+
guillotine.gemspec
|
54
|
+
lib/guillotine.rb
|
55
|
+
lib/guillotine/adapters/active_record_adapter.rb
|
56
|
+
lib/guillotine/adapters/memory_adapter.rb
|
57
|
+
lib/guillotine/adapters/riak_adapter.rb
|
58
|
+
lib/guillotine/adapters/sequel_adapter.rb
|
59
|
+
lib/guillotine/app.rb
|
60
|
+
test/active_record_adapter_test.rb
|
61
|
+
test/app_test.rb
|
62
|
+
test/helper.rb
|
63
|
+
test/memory_adapter_test.rb
|
64
|
+
test/riak_adapter_test.rb
|
65
|
+
test/sequel_adapter_test.rb
|
66
|
+
]
|
67
|
+
# = MANIFEST =
|
68
|
+
|
69
|
+
## Test files will be grabbed from the file list. Make sure the path glob
|
70
|
+
## matches what you actually use.
|
71
|
+
s.test_files = s.files.select { |path| path =~ /^test\/.*_test\.rb/ }
|
72
|
+
end
|
73
|
+
|
@@ -0,0 +1,58 @@
|
|
1
|
+
require 'active_record'
|
2
|
+
|
3
|
+
module Guillotine
|
4
|
+
module Adapters
|
5
|
+
class ActiveRecordAdapter < Adapter
|
6
|
+
class Url < ActiveRecord::Base; end
|
7
|
+
|
8
|
+
def initialize(config)
|
9
|
+
Url.establish_connection config
|
10
|
+
end
|
11
|
+
|
12
|
+
# Public: Stores the shortened version of a URL.
|
13
|
+
#
|
14
|
+
# url - The String URL to shorten and store.
|
15
|
+
# code - Optional String code for the URL.
|
16
|
+
#
|
17
|
+
# Returns the unique String code for the URL. If the URL is added
|
18
|
+
# multiple times, this should return the same code.
|
19
|
+
def add(url, code = nil)
|
20
|
+
if row = Url.select(:code).where(:url => url).first
|
21
|
+
row[:code]
|
22
|
+
else
|
23
|
+
code ||= shorten url
|
24
|
+
begin
|
25
|
+
Url.create :url => url, :code => code
|
26
|
+
rescue ActiveRecord::RecordNotUnique
|
27
|
+
row = Url.select(:url).where(:code => code).first
|
28
|
+
existing_url = row && row[:url]
|
29
|
+
raise DuplicateCodeError.new(existing_url, url, code)
|
30
|
+
end
|
31
|
+
code
|
32
|
+
end
|
33
|
+
end
|
34
|
+
|
35
|
+
# Public: Retrieves a URL from the code.
|
36
|
+
#
|
37
|
+
# code - The String code to lookup the URL.
|
38
|
+
#
|
39
|
+
# Returns the String URL.
|
40
|
+
def find(code)
|
41
|
+
if row = Url.select(:url).where(:code => code).first
|
42
|
+
row[:url]
|
43
|
+
end
|
44
|
+
end
|
45
|
+
|
46
|
+
def setup
|
47
|
+
conn = Url.connection
|
48
|
+
conn.create_table :urls do |t|
|
49
|
+
t.string :url
|
50
|
+
t.string :code
|
51
|
+
end
|
52
|
+
|
53
|
+
conn.add_index :urls, :url, :unique => true
|
54
|
+
conn.add_index :urls, :code, :unique => true
|
55
|
+
end
|
56
|
+
end
|
57
|
+
end
|
58
|
+
end
|
@@ -0,0 +1,41 @@
|
|
1
|
+
module Guillotine
|
2
|
+
module Adapters
|
3
|
+
# Stores shortened URLs in memory. Totally scales.
|
4
|
+
class MemoryAdapter < Adapter
|
5
|
+
def initialize
|
6
|
+
@hash = {}
|
7
|
+
@urls = {}
|
8
|
+
end
|
9
|
+
|
10
|
+
# Public: Stores the shortened version of a URL.
|
11
|
+
#
|
12
|
+
# url - The String URL to shorten and store.
|
13
|
+
# code - Optional String code for the URL.
|
14
|
+
#
|
15
|
+
# Returns the unique String code for the URL. If the URL is added
|
16
|
+
# multiple times, this should return the same code.
|
17
|
+
def add(url, code = nil)
|
18
|
+
if existing_code = @urls[url]
|
19
|
+
existing_code
|
20
|
+
else
|
21
|
+
code ||= shorten(url)
|
22
|
+
if existing_url = @hash[code]
|
23
|
+
raise DuplicateCodeError.new(existing_url, url, code) if url != existing_url
|
24
|
+
end
|
25
|
+
@hash[code] = url
|
26
|
+
@urls[url] = code
|
27
|
+
code
|
28
|
+
end
|
29
|
+
end
|
30
|
+
|
31
|
+
# Public: Retrieves a URL from the code.
|
32
|
+
#
|
33
|
+
# code - The String code to lookup the URL.
|
34
|
+
#
|
35
|
+
# Returns the String URL.
|
36
|
+
def find(code)
|
37
|
+
@hash[code]
|
38
|
+
end
|
39
|
+
end
|
40
|
+
end
|
41
|
+
end
|
@@ -0,0 +1,50 @@
|
|
1
|
+
require 'digest/sha1'
|
2
|
+
|
3
|
+
module Guillotine
|
4
|
+
module Adapters
|
5
|
+
# Stores shortened URLs in Riak. Totally scales.
|
6
|
+
class RiakAdapter < Adapter
|
7
|
+
def initialize(bucket)
|
8
|
+
@bucket = bucket
|
9
|
+
@client = bucket.client
|
10
|
+
end
|
11
|
+
|
12
|
+
# Public: Stores the shortened version of a URL.
|
13
|
+
#
|
14
|
+
# url - The String URL to shorten and store.
|
15
|
+
# code - Optional String code for the URL.
|
16
|
+
#
|
17
|
+
# Returns the unique String code for the URL. If the URL is added
|
18
|
+
# multiple times, this should return the same code.
|
19
|
+
def add(url, code = nil)
|
20
|
+
sha = Digest::SHA1.hexdigest url
|
21
|
+
url_obj = @bucket.get_or_new sha, :r => 1
|
22
|
+
url_obj.data || begin
|
23
|
+
code ||= shorten url
|
24
|
+
code_obj = @bucket.get_or_new code
|
25
|
+
if existing_url = code_obj.data # key exists
|
26
|
+
raise DuplicateCodeError.new(existing_url, url, code) if existing_url != url
|
27
|
+
end
|
28
|
+
code_obj.content_type = url_obj.content_type = 'text/plain'
|
29
|
+
code_obj.data = url
|
30
|
+
url_obj.data = code
|
31
|
+
code_obj.store
|
32
|
+
url_obj.store
|
33
|
+
code
|
34
|
+
end
|
35
|
+
end
|
36
|
+
|
37
|
+
# Public: Retrieves a URL from the code.
|
38
|
+
#
|
39
|
+
# code - The String code to lookup the URL.
|
40
|
+
#
|
41
|
+
# Returns the String URL.
|
42
|
+
def find(code)
|
43
|
+
if obj = @bucket.get(code, :r => 1)
|
44
|
+
obj.data
|
45
|
+
end
|
46
|
+
end
|
47
|
+
end
|
48
|
+
end
|
49
|
+
end
|
50
|
+
|
@@ -0,0 +1,56 @@
|
|
1
|
+
module Guillotine
|
2
|
+
module Adapters
|
3
|
+
class SequelAdapter < Adapter
|
4
|
+
def initialize(db)
|
5
|
+
@db = db
|
6
|
+
@table = @db[:urls]
|
7
|
+
end
|
8
|
+
|
9
|
+
# Public: Stores the shortened version of a URL.
|
10
|
+
#
|
11
|
+
# url - The String URL to shorten and store.
|
12
|
+
# code - Optional String code for the URL.
|
13
|
+
#
|
14
|
+
# Returns the unique String code for the URL. If the URL is added
|
15
|
+
# multiple times, this should return the same code.
|
16
|
+
def add(url, code = nil)
|
17
|
+
if row = @table.select(:code).where(:url => url).first
|
18
|
+
row[:code]
|
19
|
+
else
|
20
|
+
code ||= shorten url
|
21
|
+
begin
|
22
|
+
@table << {:url => url, :code => code}
|
23
|
+
rescue Sequel::DatabaseError
|
24
|
+
if existing_url = @table.select(:url).where(:code => code).first
|
25
|
+
raise DuplicateCodeError.new(existing_url, url, code)
|
26
|
+
else
|
27
|
+
raise
|
28
|
+
end
|
29
|
+
end
|
30
|
+
code
|
31
|
+
end
|
32
|
+
end
|
33
|
+
|
34
|
+
# Public: Retrieves a URL from the code.
|
35
|
+
#
|
36
|
+
# code - The String code to lookup the URL.
|
37
|
+
#
|
38
|
+
# Returns the String URL.
|
39
|
+
def find(code)
|
40
|
+
if row = @table.select(:url).where(:code => code).first
|
41
|
+
row[:url]
|
42
|
+
end
|
43
|
+
end
|
44
|
+
|
45
|
+
def setup
|
46
|
+
@db.create_table :urls do
|
47
|
+
string :url
|
48
|
+
string :code
|
49
|
+
|
50
|
+
unique :url
|
51
|
+
unique :code
|
52
|
+
end
|
53
|
+
end
|
54
|
+
end
|
55
|
+
end
|
56
|
+
end
|
@@ -0,0 +1,28 @@
|
|
1
|
+
require 'sinatra/base'
|
2
|
+
|
3
|
+
module Guillotine
|
4
|
+
class App < Sinatra::Base
|
5
|
+
get "/:code" do
|
6
|
+
code = params[:code]
|
7
|
+
if url = settings.db.find(code)
|
8
|
+
redirect url
|
9
|
+
else
|
10
|
+
halt 404, "No url found for #{code}"
|
11
|
+
end
|
12
|
+
end
|
13
|
+
|
14
|
+
post "/" do
|
15
|
+
url = params[:url]
|
16
|
+
code = params[:code]
|
17
|
+
begin
|
18
|
+
if code = settings.db.add(url, code)
|
19
|
+
redirect code
|
20
|
+
else
|
21
|
+
halt 422, "Unable to shorten #{url}"
|
22
|
+
end
|
23
|
+
rescue Guillotine::DuplicateCodeError => err
|
24
|
+
halt 422, err.to_s
|
25
|
+
end
|
26
|
+
end
|
27
|
+
end
|
28
|
+
end
|
data/lib/guillotine.rb
ADDED
@@ -0,0 +1,47 @@
|
|
1
|
+
require 'base64'
|
2
|
+
require 'digest/md5'
|
3
|
+
|
4
|
+
module Guillotine
|
5
|
+
VERSION = "0.0.1"
|
6
|
+
|
7
|
+
dir = File.expand_path '../guillotine', __FILE__
|
8
|
+
autoload :App, "#{dir}/app"
|
9
|
+
|
10
|
+
class DuplicateCodeError < StandardError
|
11
|
+
attr_reader :existing_url, :new_url, :code
|
12
|
+
|
13
|
+
def initialize(existing_url, new_url, code)
|
14
|
+
@existing_url = existing_url
|
15
|
+
@new_url = new_url
|
16
|
+
@code = code
|
17
|
+
super "#{@new_url.inspect} was supposed to be shortened to #{@code.inspect}, but #{@existing_url.inspect} already is!"
|
18
|
+
end
|
19
|
+
end
|
20
|
+
|
21
|
+
module Adapters
|
22
|
+
dir = File.expand_path '../guillotine/adapters', __FILE__
|
23
|
+
autoload :MemoryAdapter, "#{dir}/memory_adapter"
|
24
|
+
autoload :SequelAdapter, "#{dir}/sequel_adapter"
|
25
|
+
autoload :RiakAdapter, "#{dir}/riak_adapter"
|
26
|
+
autoload :ActiveRecordAdapter, "#{dir}/active_record_adapter"
|
27
|
+
|
28
|
+
# Adapters handle the storage and retrieval of URLs in the system. You can
|
29
|
+
# use whatever you want, as long as it implements the #add and #find
|
30
|
+
# methods. See MemoryAdapter for a simple solution.
|
31
|
+
class Adapter
|
32
|
+
# Public: Shortens a given URL to a short code.
|
33
|
+
#
|
34
|
+
# 1) MD5 hash the URL to the hexdigest
|
35
|
+
# 2) Convert it to a Bignum
|
36
|
+
# 3) Pack it into a bitstring as a big-endian int
|
37
|
+
# 4) base64-encode the bitstring, remove the trailing junk
|
38
|
+
#
|
39
|
+
# url - String URL to shorten.
|
40
|
+
#
|
41
|
+
# Returns a unique String code for the URL.
|
42
|
+
def shorten(url)
|
43
|
+
Base64.urlsafe_encode64([Digest::MD5.hexdigest(url).to_i(16)].pack("N")).sub(/==\n?$/, '')
|
44
|
+
end
|
45
|
+
end
|
46
|
+
end
|
47
|
+
end
|
@@ -0,0 +1,38 @@
|
|
1
|
+
require File.expand_path('../helper', __FILE__)
|
2
|
+
|
3
|
+
begin
|
4
|
+
require 'active_record'
|
5
|
+
class ActiveRecordAdapterTest < Guillotine::TestCase
|
6
|
+
ADAPTER = Guillotine::Adapters::ActiveRecordAdapter.new :adapter => 'sqlite3', :database => ':memory:'
|
7
|
+
ADAPTER.setup
|
8
|
+
|
9
|
+
def setup
|
10
|
+
@db = ADAPTER
|
11
|
+
end
|
12
|
+
|
13
|
+
def test_adding_a_link_returns_code
|
14
|
+
code = @db.add 'abc'
|
15
|
+
assert_equal 'abc', @db.find(code)
|
16
|
+
end
|
17
|
+
|
18
|
+
def test_adding_duplicate_link_returns_same_code
|
19
|
+
code = @db.add 'abc'
|
20
|
+
assert_equal code, @db.add('abc')
|
21
|
+
end
|
22
|
+
|
23
|
+
def test_adds_url_with_custom_code
|
24
|
+
assert_equal 'code', @db.add('def', 'code')
|
25
|
+
assert_equal 'def', @db.find('code')
|
26
|
+
end
|
27
|
+
|
28
|
+
def test_clashing_urls_raises_error
|
29
|
+
code = @db.add '123'
|
30
|
+
assert_raises Guillotine::DuplicateCodeError do
|
31
|
+
code = @db.add '456', code
|
32
|
+
end
|
33
|
+
end
|
34
|
+
end
|
35
|
+
rescue LoadError
|
36
|
+
puts "skipping ActiveRecord tests: #{$!}"
|
37
|
+
end
|
38
|
+
|
data/test/app_test.rb
ADDED
@@ -0,0 +1,49 @@
|
|
1
|
+
require File.expand_path('../helper', __FILE__)
|
2
|
+
|
3
|
+
class AppTest < Guillotine::TestCase
|
4
|
+
ADAPTER = Guillotine::Adapters::MemoryAdapter.new
|
5
|
+
Guillotine::App.set :db, ADAPTER
|
6
|
+
|
7
|
+
include Rack::Test::Methods
|
8
|
+
|
9
|
+
def test_adding_a_link_returns_code
|
10
|
+
url = 'http://github.com'
|
11
|
+
post '/', :url => url
|
12
|
+
assert code_url = last_response.headers['Location']
|
13
|
+
code = code_url.gsub(/.*\//, '')
|
14
|
+
|
15
|
+
get "/#{code}"
|
16
|
+
assert_equal 302, last_response.status
|
17
|
+
assert_equal url, last_response.headers['Location']
|
18
|
+
end
|
19
|
+
|
20
|
+
def test_adding_duplicate_link_returns_same_code
|
21
|
+
url = 'http://github.com'
|
22
|
+
code = ADAPTER.add url
|
23
|
+
|
24
|
+
post '/', :url => url
|
25
|
+
assert code_url = last_response.headers['Location']
|
26
|
+
assert_equal code, code_url.gsub(/.*\//, '')
|
27
|
+
end
|
28
|
+
|
29
|
+
def test_adds_url_with_custom_code
|
30
|
+
url = 'http://github.com/abc'
|
31
|
+
post '/', :url => url, :code => 'code'
|
32
|
+
assert code_url = last_response.headers['Location']
|
33
|
+
assert_match /\/code$/, code_url
|
34
|
+
|
35
|
+
get "/code"
|
36
|
+
assert_equal 302, last_response.status
|
37
|
+
assert_equal url, last_response.headers['Location']
|
38
|
+
end
|
39
|
+
|
40
|
+
def test_clashing_urls_raises_error
|
41
|
+
code = ADAPTER.add '123'
|
42
|
+
post '/', :url => '456', :code => code
|
43
|
+
assert_equal 422, last_response.status
|
44
|
+
end
|
45
|
+
|
46
|
+
def app
|
47
|
+
Guillotine::App
|
48
|
+
end
|
49
|
+
end
|
data/test/helper.rb
ADDED
@@ -0,0 +1,29 @@
|
|
1
|
+
require File.expand_path('../helper', __FILE__)
|
2
|
+
|
3
|
+
class MemoryAdapterTest < Guillotine::TestCase
|
4
|
+
def setup
|
5
|
+
@db = Guillotine::Adapters::MemoryAdapter.new
|
6
|
+
end
|
7
|
+
|
8
|
+
def test_adding_a_link_returns_code
|
9
|
+
code = @db.add 'abc'
|
10
|
+
assert_equal 'abc', @db.find(code)
|
11
|
+
end
|
12
|
+
|
13
|
+
def test_adding_duplicate_link_returns_same_code
|
14
|
+
code = @db.add 'abc'
|
15
|
+
assert_equal code, @db.add('abc')
|
16
|
+
end
|
17
|
+
|
18
|
+
def test_adds_url_with_custom_code
|
19
|
+
assert_equal 'code', @db.add('def', 'code')
|
20
|
+
assert_equal 'def', @db.find('code')
|
21
|
+
end
|
22
|
+
|
23
|
+
def test_clashing_urls_raises_error
|
24
|
+
code = @db.add 'abc'
|
25
|
+
assert_raises Guillotine::DuplicateCodeError do
|
26
|
+
@db.add 'def', code
|
27
|
+
end
|
28
|
+
end
|
29
|
+
end
|
@@ -0,0 +1,59 @@
|
|
1
|
+
require File.expand_path('../helper', __FILE__)
|
2
|
+
|
3
|
+
begin
|
4
|
+
require 'riak/client'
|
5
|
+
|
6
|
+
# Assumes a local dev install of riak is setup
|
7
|
+
#
|
8
|
+
# http://wiki.basho.com/Building-a-Development-Environment.html
|
9
|
+
#
|
10
|
+
# Riak should be accessible from:
|
11
|
+
#
|
12
|
+
# http://localhost:8091/riak/guillotine-test
|
13
|
+
class RiakAdapterTest < Guillotine::TestCase
|
14
|
+
BUCKET = Riak::Client.new(:http_port => 8091)['guillotine-test']
|
15
|
+
ADAPTER = Guillotine::Adapters::RiakAdapter.new BUCKET
|
16
|
+
|
17
|
+
def setup
|
18
|
+
@db = ADAPTER
|
19
|
+
end
|
20
|
+
|
21
|
+
def test_adding_a_link_returns_code
|
22
|
+
code = @db.add 'abc'
|
23
|
+
assert_equal 'abc', @db.find(code)
|
24
|
+
|
25
|
+
BUCKET.delete Digest::SHA1.hexdigest('abc')
|
26
|
+
BUCKET.delete code
|
27
|
+
end
|
28
|
+
|
29
|
+
def test_adding_duplicate_link_returns_same_code
|
30
|
+
code = @db.add 'abc'
|
31
|
+
assert_equal code, @db.add('abc')
|
32
|
+
|
33
|
+
BUCKET.delete Digest::SHA1.hexdigest('abc')
|
34
|
+
BUCKET.delete code
|
35
|
+
end
|
36
|
+
|
37
|
+
def test_adds_url_with_custom_code
|
38
|
+
assert_equal 'code', @db.add('def', 'code')
|
39
|
+
assert_equal 'def', @db.find('code')
|
40
|
+
|
41
|
+
BUCKET.delete Digest::SHA1.hexdigest('def')
|
42
|
+
BUCKET.delete 'code'
|
43
|
+
end
|
44
|
+
|
45
|
+
def test_clashing_urls_raises_error
|
46
|
+
code = @db.add 'abc'
|
47
|
+
assert_raises Guillotine::DuplicateCodeError do
|
48
|
+
@db.add 'def', code
|
49
|
+
end
|
50
|
+
|
51
|
+
BUCKET.delete Digest::SHA1.hexdigest('abc')
|
52
|
+
BUCKET.delete Digest::SHA1.hexdigest('def')
|
53
|
+
BUCKET.delete code
|
54
|
+
end
|
55
|
+
end
|
56
|
+
rescue LoadError
|
57
|
+
puts "Skipping Riak tests: #{$!}"
|
58
|
+
end
|
59
|
+
|
@@ -0,0 +1,38 @@
|
|
1
|
+
require File.expand_path('../helper', __FILE__)
|
2
|
+
|
3
|
+
begin
|
4
|
+
require 'sequel'
|
5
|
+
class SequelAdapterTest < Guillotine::TestCase
|
6
|
+
ADAPTER = Guillotine::Adapters::SequelAdapter.new Sequel.sqlite
|
7
|
+
ADAPTER.setup
|
8
|
+
|
9
|
+
def setup
|
10
|
+
@db = ADAPTER
|
11
|
+
end
|
12
|
+
|
13
|
+
def test_adding_a_link_returns_code
|
14
|
+
code = @db.add 'abc'
|
15
|
+
assert_equal 'abc', @db.find(code)
|
16
|
+
end
|
17
|
+
|
18
|
+
def test_adding_duplicate_link_returns_same_code
|
19
|
+
code = @db.add 'abc'
|
20
|
+
assert_equal code, @db.add('abc')
|
21
|
+
end
|
22
|
+
|
23
|
+
def test_adds_url_with_custom_code
|
24
|
+
assert_equal 'code', @db.add('def', 'code')
|
25
|
+
assert_equal 'def', @db.find('code')
|
26
|
+
end
|
27
|
+
|
28
|
+
def test_clashing_urls_raises_error
|
29
|
+
code = @db.add '123'
|
30
|
+
assert_raises Guillotine::DuplicateCodeError do
|
31
|
+
code = @db.add '456', code
|
32
|
+
end
|
33
|
+
end
|
34
|
+
end
|
35
|
+
rescue LoadError
|
36
|
+
puts "Skipping sequel tests"
|
37
|
+
end
|
38
|
+
|
metadata
ADDED
@@ -0,0 +1,88 @@
|
|
1
|
+
--- !ruby/object:Gem::Specification
|
2
|
+
name: guillotine
|
3
|
+
version: !ruby/object:Gem::Version
|
4
|
+
version: 0.0.1
|
5
|
+
prerelease:
|
6
|
+
platform: ruby
|
7
|
+
authors:
|
8
|
+
- Rick Olson
|
9
|
+
autorequire:
|
10
|
+
bindir: bin
|
11
|
+
cert_chain: []
|
12
|
+
date: 2011-08-08 00:00:00.000000000Z
|
13
|
+
dependencies:
|
14
|
+
- !ruby/object:Gem::Dependency
|
15
|
+
name: sinatra
|
16
|
+
requirement: &70305960198340 !ruby/object:Gem::Requirement
|
17
|
+
none: false
|
18
|
+
requirements:
|
19
|
+
- - ~>
|
20
|
+
- !ruby/object:Gem::Version
|
21
|
+
version: 1.2.6
|
22
|
+
type: :runtime
|
23
|
+
prerelease: false
|
24
|
+
version_requirements: *70305960198340
|
25
|
+
- !ruby/object:Gem::Dependency
|
26
|
+
name: rack-test
|
27
|
+
requirement: &70305960197960 !ruby/object:Gem::Requirement
|
28
|
+
none: false
|
29
|
+
requirements:
|
30
|
+
- - ! '>='
|
31
|
+
- !ruby/object:Gem::Version
|
32
|
+
version: '0'
|
33
|
+
type: :development
|
34
|
+
prerelease: false
|
35
|
+
version_requirements: *70305960197960
|
36
|
+
description: Adaptable private URL shortener
|
37
|
+
email: technoweenie@gmail.com
|
38
|
+
executables: []
|
39
|
+
extensions: []
|
40
|
+
extra_rdoc_files: []
|
41
|
+
files:
|
42
|
+
- Gemfile
|
43
|
+
- LICENSE
|
44
|
+
- Rakefile
|
45
|
+
- config/haproxy.riak.cfg
|
46
|
+
- guillotine.gemspec
|
47
|
+
- lib/guillotine.rb
|
48
|
+
- lib/guillotine/adapters/active_record_adapter.rb
|
49
|
+
- lib/guillotine/adapters/memory_adapter.rb
|
50
|
+
- lib/guillotine/adapters/riak_adapter.rb
|
51
|
+
- lib/guillotine/adapters/sequel_adapter.rb
|
52
|
+
- lib/guillotine/app.rb
|
53
|
+
- test/active_record_adapter_test.rb
|
54
|
+
- test/app_test.rb
|
55
|
+
- test/helper.rb
|
56
|
+
- test/memory_adapter_test.rb
|
57
|
+
- test/riak_adapter_test.rb
|
58
|
+
- test/sequel_adapter_test.rb
|
59
|
+
homepage: https://github.com/technoweenie/gitio
|
60
|
+
licenses: []
|
61
|
+
post_install_message:
|
62
|
+
rdoc_options: []
|
63
|
+
require_paths:
|
64
|
+
- lib
|
65
|
+
required_ruby_version: !ruby/object:Gem::Requirement
|
66
|
+
none: false
|
67
|
+
requirements:
|
68
|
+
- - ! '>='
|
69
|
+
- !ruby/object:Gem::Version
|
70
|
+
version: '0'
|
71
|
+
required_rubygems_version: !ruby/object:Gem::Requirement
|
72
|
+
none: false
|
73
|
+
requirements:
|
74
|
+
- - ! '>='
|
75
|
+
- !ruby/object:Gem::Version
|
76
|
+
version: '0'
|
77
|
+
requirements: []
|
78
|
+
rubyforge_project: guillotine
|
79
|
+
rubygems_version: 1.8.6
|
80
|
+
signing_key:
|
81
|
+
specification_version: 2
|
82
|
+
summary: Adaptable private URL shortener
|
83
|
+
test_files:
|
84
|
+
- test/active_record_adapter_test.rb
|
85
|
+
- test/app_test.rb
|
86
|
+
- test/memory_adapter_test.rb
|
87
|
+
- test/riak_adapter_test.rb
|
88
|
+
- test/sequel_adapter_test.rb
|