class Fradium

# File lib/fradium.rb, line 119
def self.generate_random_password(length=10)
  r = SecureRandom.urlsafe_base64.delete('-_')
  while r.length < length
    r << SecureRandom.urlsafe_base64.delete('-_')
  end
  r[0..length-1]
end

# File lib/fradium.rb, line 12
def initialize(params)
  @params = params
  @sequel = Sequel.connect(@params)
end

# File lib/fradium.rb, line 21
def all_users
  @sequel[:radcheck].where{attribute.like '%-Password'}
end

# File lib/fradium.rb, line 25
def create_user(username)
  raise UsernameEmptyError if username&.empty?
  raise UserAlreadyExistsError if user_exists?(username)
  password = Fradium.generate_random_password

  @sequel[:radcheck].insert(username: username,
                            attribute: 'Cleartext-Password',
                            op: ':=',
                            value: password)
end

# File lib/fradium.rb, line 98
def dbconsole
  case @sequel.adapter_scheme
  when :mysql2
    # I know this is not safe.
    Kernel.exec({'MYSQL_PWD' => @params['password']},
                'mysql',
                "--pager=less -SF",
                "--user=#{@params['username']}",
                "--host=#{@params['host']}" ,
                "#{@params['database']}")
  when :postgres
    Kernel.exec({'PGPASSWORD' => @params['password']},
                'psql',
                "--username=#{@params['username']}",
                "--dbname=#{@params['database']}",
                "--host=#{@params['host']}")
  when :sqlite
    Kernel.exec('sqlite3', @params)
  end
end

# File lib/fradium.rb, line 62
def expire_user(username)
  set_expiration(username, Time.now)
end

# File lib/fradium.rb, line 55
def expired_users
  now = Time.now
  # a little bit redundant but for consistency
  id = @sequel[:radcheck].where(attribute: 'Expiration').collect{|e| e[:id] if Time.now > Time.parse(e[:value])}
  @sequel[:radcheck].where(id: id)
end

# File lib/fradium.rb, line 51
def expiry
  @sequel[:radcheck].where(attribute: 'Expiration')
end

# File lib/fradium.rb, line 36
def find_user(username)
  raise UsernameEmptyError if username&.empty?
  reult = @sequel[:radcheck].where(username: username).where{attribute.like '%-Password'}
end

# File lib/fradium.rb, line 72
def is_expired?(username)
  expiration_date = query_expiration(username)&.fetch('value')
  return false if expiration_date.nil? || expiration_date.empty? # if expiration info not found, not expired yet
  Time.now > Time.parse(expiration_date)
end

# File lib/fradium.rb, line 41
def modify_user(username)
  raise UsernameEmptyError if username&.empty?
  raise UserNotFoundError unless user_exists?(username)
  password = Fradium.generate_random_password

  target = find_user(username)
  raise CorruptedUserDatabaseError if target.count > 1
  target.update(value: password, attribute: 'Cleartext-Password')
end

# File lib/fradium.rb, line 129
def query_expiration(username)
  raise UsernameEmptyError if username&.empty?
  raise UserNotFoundError unless user_exists?(username)

  r = @sequel[:radcheck].where(username: username, attribute: 'Expiration')

  raise CorruptedUserDatabaseError if r.count > 1
  return nil if r.count == 0 # if no expiration info found
  return nil if r&.first[:value]&.empty?

  r
end

# File lib/fradium.rb, line 78
def set_expiration(username, expiration_date)
  expiration_info = query_expiration(username)

  value = ''
  if expiration_date.instance_of?(Time)
    value = expiration_date.strftime("%d %b %Y %H:%M:%S")
  else
    value = Time.parse(expiration_date).strftime("%d %b %Y %H:%M:%S")
  end

  if expiration_info.nil? # add new entry
    @sequel[:radcheck].insert(username: username,
                              attribute: 'Expiration',
                              op: ':=',
                              value: value)
  else # update existing entry
    expiration_info.update(value: value)
  end
end

# File lib/fradium.rb, line 66
def unexpire_user(username)
  raise UsernameEmptyError if username&.empty?
  raise UserNotFoundError unless user_exists?(username)
  @sequel[:radcheck].where(username: username, attribute: 'Expiration').delete
end

# File lib/fradium.rb, line 17
def user_exists?(username)
  find_user(username).count > 0
end