cli.rb

require 'annoy'
require 'drydock'
require 'onetime'
require 'familia/tools'

#::SCRIPT_LINES__ = {} unless defined? ::SCRIPT_LINES__

class OT::CLI < Drydock::Command

  def init
    OT.load! :cli
  end

  def register_build
    begin
      Onetime::VERSION.increment! argv.first
      puts Onetime::VERSION.inspect
    rescue => ex
      puts ex.message
      exit 1
    end
  end

  def entropy
    puts OT::Entropy.count
  end

  def clear_entropy
    require_sudo
    OT::Entropy.values.clear
    entropy
  end

  def generate_entropy
    #require_sudo
    option.count = 100_000 if option.count.to_i > 100_000
    OT::Entropy.generate option.count
    entropy
  end

  def move_keys
    sourcedb, targetdb, filter = *argv
    raise "No target database supplied" unless sourcedb && targetdb
    raise "No filter supplied" unless filter
    source_uri = URI.parse Familia.uri.to_s
    target_uri = URI.parse Familia.uri.to_s
    source_uri.db, target_uri.db = sourcedb, targetdb
    Familia::Tools.move_keys filter, source_uri, target_uri do |idx, type, key, ttl|
      if global.verbose > 0
        puts '%4d (%6s, %4d): %s' % [idx+1, type, ttl, key]
      else
        print "\rMoved #{idx+1} keys"
      end
    end
    puts
  end

  def customers
    puts '%d customers' % OT::Customer.values.size
  end

  def require_sudo
    unless Process.uid.zero?
      raise RuntimeError, "Must run as root or with sudo"
    end
  end
  def redis
    require_sudo
    y Familia.redis.info
  end

  def redis_start
    require_sudo
    STDERR.puts 'RUN THIS:'
    puts 'redis-server %s' % [OT.conf[:redis][:config] || '[no config set]']
  end

  def redis_stop
    require_sudo
    uptime = Familia.redis.info['uptime_in_seconds']
    # In some cases SHUTDOWN does not call SAVE so we call it to be sure.
    # (If there are no SAVE lines in redis.conf for example.)
    puts "Saving..."
    redis_save
    # SHUTDOWN does the following:
    #   Stop all the clients.
    #   Perform a blocking SAVE if at least one save point is configured.
    #   Flush the Append Only File if AOF is enabled.
    #   Quit the server.
    puts( "Shutting down... (up for %d hours)" % [uptime.to_i/3600])
    Familia.redis.shutdown
  end

  def redis_save
    require_sudo
    Familia.redis.save
  end

  def redis_bgsave
    require_sudo
    Familia.redis.bgsave
  end

  def redis_bgrewriteaof
    require_sudo
    Familia.redis.bgrewriteaof
  end

end