initial plan structure, and some misc fixes

Gemfile

@@ -22,6 +22,7 @@ gem 'cocaine'
 gem 'zipruby'
 gem 'sass', require: nil
 gem 'dav4rack'
+gem 'filesize'
 
 platform :mri do
   gem 'magic' # sudo apt-get install file, For OSX: brew install libmagic

Gemfile.lock

@@ -57,6 +57,7 @@ GEM
     faraday (0.9.0)
       multipart-post (>= 1.2, < 3)
     ffi (1.9.3)
+    filesize (0.0.3)
     google-api-client (0.7.1)
       addressable (>= 2.3.2)
       autoparse (>= 0.3.3)
@@ -215,6 +216,7 @@ DEPENDENCIES
   erubis
   fabrication
   faker
+  filesize
   google-api-client
   hiredis
   jdbc-postgres

Rakefile

@@ -116,3 +116,11 @@ task :buildssl => [:environment] do
     ar.add_buffer 'sslsites.json', payload.to_json
   end
 end
+
+desc 'Set existing stripe customers to internal supporter plan'
+task :primenewstriperunonlyonce => [:environment] do
+  Site.exclude(stripe_customer_id: nil).all.each do |site|
+    site.plan_type = 'supporter'
+    site.save_changes validate: false
+  end
+end

app.rb

@@ -348,7 +348,7 @@ post '/tags/add' do
   if current_site.valid?
     current_site.save
   else
-    flash[:errors] = current_site.errors[:tags].first
+    flash[:errors] = current_site.errors.first
   end
 
   redirect request.referer
@@ -713,15 +713,16 @@ end
 
 post '/change_email' do
   require_login
-  current_site.email = params[:email]
-  current_site.email_confirmation_token = SecureRandom.hex 3
-  current_site.email_confirmed = false
   
   if params[:email] == current_site.email
     current_site.errors.add :email, 'You are already using this email address for this account.'
     halt erb(:settings)
   end
 
+  current_site.email = params[:email]
+  current_site.email_confirmation_token = SecureRandom.hex 3
+  current_site.email_confirmed = false
+
   if current_site.valid?
     current_site.save_changes
     send_confirmation_email
@@ -919,7 +920,7 @@ post %r{\/site_files\/save\/(.+)} do
   tempfile.close
 
   if current_site.file_size_too_large? tempfile.size
-    halt 'File is too large to fit in your space, it has NOT been saved. Please make a local copy and then try to reduce the size.'
+    halt 'File is too large to fit in your space, it has NOT been saved. You will need to reduce the size or upgrade to a new plan.'
   end
 
   current_site.store_file filename, tempfile

migrations/044_add_plan_type.rb

@@ -0,0 +1,9 @@
+Sequel.migration do
+  up {
+    DB.add_column :sites, :plan_type, :text
+  }
+
+  down {
+    DB.drop_column :sites, :plan_type
+  }
+end

models/site.rb

@@ -78,6 +78,46 @@ class Site < Sequel::Model
 
   COMMENTING_ALLOWED_UPDATED_COUNT = 2
 
+  PLAN_FEATURES = {
+    fatcat: {
+      name: 'Fat Cat',
+      space: Filesize.from('100GB').to_i,
+      bandwidth: Filesize.from('2TB').to_i,
+      custom_domains: true,
+      custom_ssl_certificates: true,
+      global_cdn: true,
+      ddos_mitigation: true,
+      unlimited_site_creation: true,
+      site_mounting: true,
+      no_file_restrictions: true
+    }
+  }
+
+  PLAN_FEATURES[:catbus] = PLAN_FEATURES[:fatcat].merge(
+    name: 'Cat Bus',
+    space: Filesize.from('10GB').to_i,
+    bandwidth: Filesize.from('500GB').to_i
+  )
+
+  PLAN_FEATURES[:supporter] = PLAN_FEATURES[:catbus].merge(
+    name: 'Supporter',
+    space: Filesize.from('1GB').to_i,
+    bandwidth: Filesize.from('100GB').to_i,
+    unlimited_site_creation: false,
+    custom_ssl_certificates: false,
+    no_file_restrictions: false
+  )
+
+  PLAN_FEATURES[:free] = PLAN_FEATURES[:supporter].merge(
+    name: 'Free',
+    space: Filesize.from('30MB').to_i,
+    bandwidth: Filesize.from('10GB').to_i,
+    custom_domains: false,
+    global_cdn: false,
+    ddos_mitigation: false,
+    site_mounting: false
+  )
+
   many_to_one :server
 
   many_to_many :tags
@@ -383,8 +423,11 @@ class Site < Sequel::Model
 
       if new_title.length < TITLE_MAX
         self.title = new_title
-        save_changes(validate: false)
       end
+
+      self.site_changed = true
+
+      save_changes(validate: false)
     end
 
     dirname = pathname.dirname.to_s
@@ -408,11 +451,6 @@ class Site < Sequel::Model
 
     SiteChange.record self, relative_path
 
-    if self.site_changed != true
-      self.site_changed = true
-      save_changes(validate: false)
-    end
-
     true
   end
 
@@ -519,6 +557,11 @@ class Site < Sequel::Model
     super
   end
 
+  def email=(email)
+    @original_email = values[:email] unless new?
+    super
+  end
+
 #  def after_destroy
 #    FileUtils.rm_rf files_path
 #    super
@@ -549,11 +592,16 @@ class Site < Sequel::Model
       errors.add :email, 'An email address is required.'
     end
 
-    # Check for existing email
-    email_check = self.class.select(:id).filter(email: values[:email]).first
+    # Check for existing email if new or changing email.
+    if new? || @original_email
+      email_check = self.class.select(:id).filter(email: values[:email])
+      email_check.exclude!(id: self.id) unless new?
+      email_check = email_check.first
+
       if email_check && email_check.id != self.id
         errors.add :email, 'This email address already exists on Neocities, please use your existing account instead of creating a new one.'
       end
+    end
 
     unless values[:email] =~ EMAIL_SANITY_REGEX
       errors.add :email, 'A valid email address is required.'
@@ -676,7 +724,7 @@ class Site < Sequel::Model
     end
 
     list.select {|f| f[:is_directory]}.sort_by {|f| f[:name]} +
-    list.select {|f| f[:is_directory] == false}.sort_by{|f| f[:name]}
+    list.select {|f| f[:is_directory] == false}.sort_by{|f| f[:name].downcase}
   end
 
   def file_size_too_large?(size_in_bytes)
@@ -719,14 +767,18 @@ class Site < Sequel::Model
     !values[:stripe_customer_id].nil?
   end
 
-  # This will return false if they have ended their support plan.
+  # This will return false if they have ended their plan.
   def ended_supporter?
-    values[:ended_plan]
+    values[:plan_ended]
   end
 
   def plan_name
-    return 'Free Plan' if !supporter? || (supporter? && ended_supporter?)
-    'Supporter Plan'
+    PLAN_FEATURES[plan_type.to_sym][:name]
+  end
+
+  def plan_type
+    return 'free' if values[:plan_type].nil?
+    values[:plan_type]
   end
 
   def latest_events(current_page=1, limit=10)

models/tag.rb

@@ -13,7 +13,7 @@ class Tag < Sequel::Model
   end
 
   def self.suggestions(name, limit=3)
-    Tag.filter(name: /^#{name}/).
+    Tag.filter(name: /^#{name}/i).
     order(:name).
     limit(limit).
     all

tests/site_file_tests.rb

@@ -9,9 +9,9 @@ end
 
 describe 'site_files' do
   describe 'upload' do
-
     it 'succeeds with index.html file' do
       site = Fabricate :site
+      site.site_changed.must_equal false
       PurgeCacheWorker.jobs.clear
       ScreenshotWorker.jobs.clear
 
@@ -25,6 +25,7 @@ describe 'site_files' do
       args = ScreenshotWorker.jobs.first['args']
       args.first.must_equal site.username
       args.last.must_equal 'index.html'
+      site.reload.site_changed.must_equal true
     end
 
     it 'succeeds with valid file' do
@@ -48,6 +49,8 @@ describe 'site_files' do
       Site::THUMBNAIL_RESOLUTIONS.each do |resolution|
         File.exists?(site.thumbnail_path('test.jpg', resolution)).must_equal true
       end
+
+      site.site_changed.must_equal false
     end
 
     it 'works with directory path' do

tests/site_tests.rb

@@ -0,0 +1,29 @@
+require_relative './environment.rb'
+require 'rack/test'
+
+include Rack::Test::Methods
+
+def app
+  Sinatra::Application
+end
+
+describe 'site' do
+  describe 'plan_name' do
+    it 'should set to free for missing stripe_customer_id' do
+      site = Fabricate :site
+      site.reload.plan_type.must_equal 'free'
+    end
+
+    it 'should be free for no plan_type entry' do
+      site = Fabricate :site, stripe_customer_id: 'cust_derp'
+      site.plan_type.must_equal 'free'
+    end
+
+    it 'should match plan_type' do
+      %w{supporter neko catbus fatcat}.each do |plan_type|
+        site = Fabricate :site, plan_type: plan_type
+        site.plan_type.must_equal plan_type
+      end
+    end
+  end
+end

views/layout.erb

@@ -58,7 +58,7 @@
       $('#shareButton').popover({html: true})
 
       $('.typeahead').typeahead({
-        minLength: 3,
+        minLength: 2,
         highlight: true
       }, {
         name: 'tags',