Get back to use Sinatra settings
[coquelicot.git] / coquelicot_app.rb
1 $:.unshift File.join(File.dirname(__FILE__), 'lib')
2
3 require 'sinatra/base'
4 require 'haml'
5 require 'sass'
6 require 'digest/sha1'
7 require 'gettext'
8 require 'coquelicot'
9 require 'haml_gettext'
10
11 module Coquelicot
12   class StoredFile
13     def lockfile
14       @lockfile ||= Lockfile.new "#{File.expand_path(@path)}.lock", :timeout => 4
15     end
16
17     def each
18       # output content
19       yield @initial_content
20       @initial_content = nil
21       until (buf = @file.read(BUFFER_LEN)).nil?
22         yield @cipher.update(buf)
23       end
24       yield @cipher.final
25       @fully_sent = true
26     end
27
28     def close
29       if @cipher
30         @cipher.reset
31         @cipher = nil
32       end
33       @file.close
34       if one_time_only?
35         empty! if @fully_sent
36         lockfile.unlock
37       end
38     end
39   end
40
41   class << self
42     def settings
43       (class << self; Application; end)
44     end
45     def depot
46       @depot = Depot.new(settings.depot_path) if @depot.nil? || settings.depot_path != @depot.path
47       @depot
48     end
49   end
50
51   class Application < Sinatra::Base
52     set :app_file, __FILE__
53     set :upload_password, '0e5f7d398e6f9cd1f6bac5cc823e363aec636495'
54     set :default_expire, 60
55     set :maximum_expire, 60 * 24 * 30 # 1 month
56     set :gone_period, 10080
57     set :filename_length, 20
58     set :random_pass_length, 16
59     set :depot_path, Proc.new { File.join(root, 'files') }
60
61     def password_match?(password)
62       return TRUE if settings.upload_password.nil?
63       (not password.nil?) && Digest::SHA1.hexdigest(password) == settings.upload_password
64     end
65
66     GetText::bindtextdomain('coquelicot')
67     before do
68       GetText::set_current_locale(params[:lang] || request.env['HTTP_ACCEPT_LANGUAGE'] || 'en')
69     end
70
71     get '/style.css' do
72       content_type 'text/css', :charset => 'utf-8'
73       sass :style
74     end
75
76     get '/' do
77       haml :index
78     end
79
80     get '/random_pass' do
81       "#{Coquelicot.gen_random_pass}"
82     end
83
84     get '/ready/:link' do |link|
85       link, pass = link.split '-' if link.include? '-'
86       begin
87         file = Coquelicot.depot.get_file(link, nil)
88       rescue Errno::ENOENT => ex
89         not_found
90       end
91       @expire_at = file.expire_at
92       @base = request.url.gsub(/\/ready\/[^\/]*$/, '')
93       @name = "#{link}"
94       unless pass.nil?
95         @name << "-#{pass}"
96         @unprotected = true
97       end
98       @url = "#{@base}/#{@name}"
99       haml :ready
100     end
101
102     post '/authenticate' do
103       pass unless request.xhr?
104       unless password_match? params[:upload_password] then
105         error 403, "Forbidden"
106       end
107       'OK'
108     end
109
110     post '/upload' do
111       unless password_match? params[:upload_password] then
112         error 403
113       end
114       if params[:file] then
115         tmpfile = params[:file][:tempfile]
116         name = params[:file][:filename]
117       end
118       if tmpfile.nil? || name.nil? then
119         @error = "No file selected"
120         return haml(:index)
121       end
122       if params[:expire].nil? or params[:expire].to_i == 0 then
123         params[:expire] = settings.default_expire
124       elsif params[:expire].to_i > settings.maximum_expire then
125         error 403
126       end
127       expire_at = Time.now + 60 * params[:expire].to_i
128       one_time_only = params[:one_time] and params[:one_time] == 'true'
129       if params[:file_key].nil? or params[:file_key].empty?then
130         pass = Coquelicot.gen_random_pass
131       else
132         pass = params[:file_key]
133       end
134       src = params[:file][:tempfile]
135       link = Coquelicot.depot.add_file(
136          src, pass,
137          { "Expire-at" => expire_at.to_i,
138            "One-time-only" => one_time_only,
139            "Filename" => params[:file][:filename],
140            "Length" => src.stat.size,
141            "Content-Type" => params[:file][:type],
142          })
143       redirect "ready/#{link}-#{pass}" if params[:file_key].nil? or params[:file_key].empty?
144       redirect "ready/#{link}"
145     end
146
147     def expired
148       throw :halt, [410, haml(:expired)]
149     end
150
151     def send_stored_file(file)
152       last_modified file.created_at.httpdate
153       attachment file.meta['Filename']
154       response['Content-Length'] = "#{file.meta['Length']}"
155       response['Content-Type'] = file.meta['Content-Type'] || 'application/octet-stream'
156       throw :halt, [200, file]
157     end
158
159     def send_link(link, pass)
160       file = Coquelicot.depot.get_file(link, pass)
161       return false if file.nil?
162       return expired if file.expired?
163
164       if file.one_time_only?
165         begin
166           # unlocking done in file.close
167           file.lockfile.lock
168         rescue Lockfile::TimeoutLockError
169           error 409, "Download currently in progress"
170         end
171       end
172       send_stored_file(file)
173     end
174
175     get '/:link-:pass' do |link, pass|
176       link = Coquelicot.remap_base32_extra_characters(link)
177       pass = Coquelicot.remap_base32_extra_characters(pass)
178       not_found unless send_link(link, pass)
179     end
180
181     get '/:link' do |link|
182       link = Coquelicot.remap_base32_extra_characters(link)
183       not_found unless Coquelicot.depot.file_exists? link
184       @link = link
185       haml :enter_file_key
186     end
187
188     post '/:link' do |link|
189       pass = params[:file_key]
190       return 403 if pass.nil? or pass.empty?
191       begin
192         # send Forbidden even if file is not found
193         return 403 unless send_link(link, pass)
194       rescue Coquelicot::BadKey => ex
195         403
196       end
197     end
198
199     helpers do
200       def base_href
201         url = request.scheme + "://"
202         url << request.host
203         if request.scheme == "https" && request.port != 443 ||
204             request.scheme == "http" && request.port != 80
205           url << ":#{request.port}"
206         end
207         url << request.script_name
208         "#{url}/"
209       end
210     end
211   end
212 end
213
214 Coquelicot::Application.run! if __FILE__ == $0