cartodb/lib/tasks/setup.rake

namespace :cartodb do
  namespace :db do
    task :create, [:env] do |_t, args|
      # When the db:create task runs, there is no prior database. Since we use models in our initializers, we need to
      # skip the :environment dependency or Sequel will error out when trying to load the schema for the model table.
      # This task overrides the default Sequel task, loading the bare minimum (db config) instead of all initializers
      args.with_defaults(env: Rails.env)

      config = SequelRails::Configuration.for(Rails.root, Carto::Conf.new.db_config)

      unless SequelRails::Storage.create_environment(config.environment_for(args.env))
        abort "Could not create database for #{args.env}."
      end
    end
  end

  namespace :test do
    task :prepare do
      if (ENV['RAILS_ENV'] == "test")
        begin
          Rake::Task['db:drop'].invoke
        rescue Sequel::DatabaseConnectionError
          puts "Not dropping DB (did not exist)"
        end
        Rake::Task['db:create'].invoke &&
        Rake::Task['db:migrate'].invoke &&
        Rake::Task['cartodb:db:create_publicuser'].invoke
      else
        system("rake cartodb:test:prepare RAILS_ENV=test") || raise("Something went wrong")
      end
    end
  end
  namespace :db do
    desc <<-DESC
Setup cartodb database and creates a new user from environment variables:
  - ENV['EMAIL']: user e-mail
  - ENV['PASSWORD']: user password
  - ENV['SUBDOMAIN']: user subdomain
DESC
    task :setup => ["rake:db:create", "rake:db:migrate", "cartodb:db:create_publicuser"] do
    end

    task :setup_user => :environment do
      raise "You should provide a valid e-mail"    if ENV['EMAIL'].blank?
      raise "You should provide a valid password"  if ENV['PASSWORD'].blank?
      raise "You should provide a valid subdomain" if ENV['SUBDOMAIN'].blank?

      # Reload User model schema to avoid errors
      # when running this task along with db:migrate
      ::User.set_dataset :users

      u = ::User.new
      u.email = ENV['EMAIL']
      u.password = ENV['PASSWORD']
      u.password_confirmation = ENV['PASSWORD']
      u.username = ENV['SUBDOMAIN']
      u.database_host = ENV['DATABASE_HOST'] || ::SequelRails.configuration.environment_for(Rails.env)['host']

      if ENV['BUILDER_ENABLED'] == "true"
        u.builder_enabled = true
      end

      u.save

      raise u.errors.inspect if u.new?
      puts "User #{u.username} created successfully"
    end

    task :create_dev_user =>
      ["cartodb:db:create_publicuser"] do
      raise "You should provide a valid e-mail"    if ENV['EMAIL'].blank?
      raise "You should provide a valid password"  if ENV['PASSWORD'].blank?
      raise "You should provide a valid subdomain" if ENV['SUBDOMAIN'].blank?

      # Reload User model schema to avoid errors
      # when running this task along with db:migrate
      ::User.set_dataset :users

      user = ::User.new
      user.email = ENV['EMAIL']
      user.password = ENV['PASSWORD']
      user.password_confirmation = ENV['PASSWORD']
      user.username = ENV['SUBDOMAIN']
      user.database_host = ENV['DATABASE_HOST'] || ::SequelRails.configuration.environment_for(Rails.env)['host']

      if ENV['BUILDER_ENABLED'] == "true"
        user.builder_enabled = true
      end

      unless user.valid?
        puts
        puts 'There are some problems with the info that you provided to create the new user:'
        user.errors.full_messages.each do |error|
          puts "\t> #{error}"
        end
        puts

        exit
      end

      user.save
      puts "User #{user.username} created successfully"

      # 10 Gb of quota
      quota = 1073741824
      user.update(:quota_in_bytes => quota)

      user.db_service.rebuild_quota_trigger
      puts "User: #{user.username} quota updated to: 10 GB. #{user.tables.count} tables updated."

      user.update(:table_quota => nil)
      puts "User: #{user.username} table quota updated to: unlimited"

      user.update(:private_tables_enabled => true)
      puts "User: #{user.username} private tables enabled: true"

      user.update(:account_type => '[DEDICATED]')
      puts "User: #{user.username} table account type updated to: [DEDICATED]"

      user.update(:sync_tables_enabled => true)
      puts "User: #{user.username} sync tables enabled"

      file_size_quota = 1500*1024*1024
      row_count_quota = 5000000

      user.update(:max_import_file_size => file_size_quota) if file_size_quota > user.max_import_file_size
      puts "Twitter import user: #{user.username} max_import_file_size is now #{user.max_import_file_size}"

      user.update(:max_import_table_row_count => row_count_quota) if row_count_quota > user.max_import_table_row_count
      puts "Twitter import user: #{user.username} max_import_table_row_count is now #{user.max_import_table_row_count}"
    end

    desc "make public and tile users"
    task :create_publicuser => :environment do
      [CartoDB::PUBLIC_DB_USER, CartoDB::TILE_DB_USER].each do |u|
        puts "Creating user #{u}"
        ::SequelRails.connection.run("DO $$
        BEGIN
          IF NOT EXISTS ( SELECT * FROM pg_user WHERE usename = '#{u}') THEN
            CREATE USER #{u}; -- TODO: with password '...'
          ELSE
            RAISE NOTICE 'User #{u} already exists';
          END IF;
          RETURN;
        END;
        $$ LANGUAGE 'plpgsql';")
      end
    end

    # TODO: remove text bit and just use env
    desc "Create a plain user account"
    task :create_user => :environment do
      begin
        raise "You should provide a valid e-mail"    if ENV['EMAIL'].blank?
        raise "You should provide a valid password"  if ENV['PASSWORD'].blank?
        raise "You should provide a valid subdomain" if ENV['SUBDOMAIN'].blank?

        u = ::User.new
        u.email = ENV['EMAIL']
        u.password = ENV['PASSWORD']
        u.password_confirmation = ENV['PASSWORD']
        u.username = ENV['SUBDOMAIN']
        if ENV['DATABASE_HOST'].blank?
          u.database_host = ::SequelRails.configuration.environment_for(Rails.env)['host']
        else
          u.database_host = ENV['DATABASE_HOST']
        end

        if ENV['BUILDER_ENABLED'] == "true"
          u.builder_enabled = true
        end

        u.save
        if u.new?
          raise u.errors.inspect
        end
        puts "USER_ID #{u.id}"
      rescue => e
        puts e.inspect
      end
    end

    desc "Set the password of the user in the USER_NAME environment variable to the value of the USER_PASSWORD environment variable"
    task :change_user_password => :environment do
      raise "Set USER_NAME environment variable" if ENV['USER_NAME'].blank?
      raise "Set USER_PASSWORD environment variable" if ENV['USER_PASSWORD'].blank?
      password = ENV['USER_PASSWORD']

      users = ::User.filter(:username => ENV['USER_NAME']).all
      if users.empty?
        raise "User doesn't exist"
      else
        u = users.first
        u.password = password
        u.password_confirmation = password
        if !u.save
          rais u.errors.inspect
        else
          puts "Password changed"
        end
      end
    end
  end
end