160 lines
3.3 KiB
Ruby
Executable file
160 lines
3.3 KiB
Ruby
Executable file
require 'rubygems'
|
|
require 'sinatra'
|
|
require 'sequel'
|
|
require 'uri'
|
|
|
|
configure do
|
|
Sequel::Model.plugin(:schema)
|
|
|
|
Sequel.connect(ENV['DATABASE_URL'] || 'sqlite://shorten.db')
|
|
|
|
require 'ostruct'
|
|
Shorten = OpenStruct.new(
|
|
:base_url => ENV['url'],
|
|
:service_name => "amd.im",
|
|
:button_text => "shorten",
|
|
:path_size => 4
|
|
)
|
|
|
|
$LOAD_PATH.unshift(File.dirname(__FILE__) + '/lib')
|
|
|
|
end
|
|
|
|
$LOAD_PATH.unshift(File.dirname(__FILE__) + '/lib')
|
|
|
|
require 'shortenurl'
|
|
|
|
helpers do
|
|
def show_information
|
|
erb :information, :layout => false
|
|
end
|
|
|
|
def validate_link(link)
|
|
halt 401, 'We do not accept local URLs to be shortened' unless valid_url? link
|
|
end
|
|
|
|
# Determine if a URL is valid. We run it through
|
|
def valid_url?(url)
|
|
if url.include? "%3A"
|
|
url = URI.unescape(url)
|
|
end
|
|
|
|
retval = true
|
|
|
|
begin
|
|
uri = URI.parse(URI.escape(url))
|
|
if uri.class != URI::HTTP
|
|
retval = false
|
|
end
|
|
|
|
host = (URI.split(url))[2]
|
|
if host =~ /^(localhost|192\.168\.\d{1,3}\.\d{1,3}|127\.0\.0\.1|172\.((1[6-9])|(2[0-9])|(3[0-1])).\d{1,3}\.\d{1,3}|10.\d{1,3}\.\d{1,3}.\d{1,3})/
|
|
retval = false
|
|
end
|
|
rescue URI::InvalidURIError
|
|
retval = false
|
|
end
|
|
|
|
retval
|
|
end
|
|
end
|
|
|
|
get '/' do
|
|
@information = show_information
|
|
erb :index
|
|
end
|
|
|
|
get '/new' do
|
|
@information = show_information
|
|
erb :new, :locals => { :type => "main" }
|
|
end
|
|
|
|
get %r(/(api-){0,1}create/(.*)) do |api, link|
|
|
validate_link link
|
|
|
|
url = ShortenUrl.create_url(link)
|
|
|
|
if api == "api-"
|
|
"#{url.short_url}"
|
|
else
|
|
erb :finished, :locals => { :url => url, :type => "finished" }
|
|
end
|
|
end
|
|
|
|
get %r(/(api-){0,1}create) do |api|
|
|
if request['url']
|
|
validate_link request['url']
|
|
url = ShortenUrl.create_url(request['url'])
|
|
if api == "api-"
|
|
"#{url.short_url}"
|
|
else
|
|
erb :finished, :locals => { :url => url, :type => "finished" }
|
|
end
|
|
end
|
|
end
|
|
|
|
post '/' do
|
|
validate_link params[:url]
|
|
|
|
url = ShortenUrl.create_url(params[:url], params[:image])
|
|
|
|
erb :finished, :locals => { :url => url, :type => "finished", :image => params[:image] }
|
|
end
|
|
|
|
get '/upload' do
|
|
|
|
erb :upload
|
|
|
|
end
|
|
|
|
post '/upload' do
|
|
require 'aws/s3'
|
|
|
|
# establish connection
|
|
AWS::S3::Base.establish_connection!(
|
|
:access_key_id => ENV['s3_key'],
|
|
:secret_access_key => ENV['s3_secret']
|
|
)
|
|
|
|
# generate key and check uniqueness
|
|
key = Anybase::Base62.random(Shorten.path_size)
|
|
key_check = ShortenUrl.filter(:url => key).first
|
|
|
|
while key_check
|
|
key = Anybase::Base62.random(Shorten.path_size)
|
|
key_check = ShortenUrl.filter(:url => key).first
|
|
end
|
|
|
|
# merge key and extension
|
|
ext = File.extname(params[:file][:filename])
|
|
filename = key + ext
|
|
#filename = params[:file][:filename]
|
|
|
|
# upload to S3
|
|
AWS::S3::S3Object.store(filename, open(params[:file][:tempfile]), 'amdim', :access => :public_read)
|
|
object_url = AWS::S3::S3Object.url_for(filename, 'shorten', :authenticated => false)
|
|
|
|
# generate shorturl
|
|
url = ShortenUrl.new(:url => object_url, :key => key, :image => params[:image])
|
|
url.save
|
|
#url = ShortenUrl.create_url(object_url, params[:image])
|
|
|
|
erb :finished, :locals => { :url => url, :type => "finished", :image => params[:image] }
|
|
end
|
|
|
|
get '/:short' do
|
|
|
|
url = ShortenUrl.find(:key => params[:short])
|
|
|
|
halt 404, "Page not found" unless url
|
|
|
|
if url.image == true
|
|
erb :image, :locals => {:url => url.url}
|
|
else
|
|
redirect url.url
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|