A fake auction house I made for a school project.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
the_auctionhouse/src/db_models.rb

100 lines
2.7 KiB

# User table model
3 years ago
class User < EntityModel
attr_reader :email, :name, :bio_text, :balance, :avatar_url, :reputation, :pw_hash
3 years ago
def initialize(user_info)
super user_info
@email = user_info["email"]
@name = user_info["name"]
@bio_text = user_info["bio_text"]
@balance = user_info["balance"]
@avatar_url = user_info["avatar_url"]
@reputation = user_info["reputation"]
@pw_hash = user_info["pw_hash"]
3 years ago
end
def avatar
gravatar = nil # Gravatar.src @email
if gravatar then
return gravatar
else
return @avatar_url
end
end
# Find user by ID, returns a user object
def self.find_by_id(id)
data = self.get("*", "id = ?", id).first
data && User.new(data)
end
# Find user by email, same as above but for emails.
def self.find_by_email(email)
data = self.get("*", "email = ?", email).first
data && User.new(data)
end
def self.validate_register_creds(email, name, password, password_confirm)
# Field check
check_all_fields = email != "" && name != "" && password != "" && password_confirm != ""
# Check email
check_email_dupe = self.find_by_email(email)
check_email_valid = email.match(EMAIL_REGEX) != nil
# Name
check_name_len = name.length >= MIN_NAME_LEN
# Password
check_pass_equals = password == password_confirm
check_pass_len = password.length >= MIN_PASSWORD_LEN
# This code is really ugly
3 years ago
return false, REGISTER_ERRORS[:fields] unless check_all_fields
return false, REGISTER_ERRORS[:email_dupe] unless check_email_dupe
return false, REGISTER_ERRORS[:email_fake] unless check_email_valid
return false, REGISTER_ERRORS[:name_len] unless check_name_len
return false, REGISTER_ERRORS[:pass_notequals] unless check_pass_equals
return false, REGISTER_ERRORS[:pass_len] unless check_pass_len
return true, ""
end
3 years ago
def self.validate_password(pw_hash, password)
BCrypt::Password.new(pw_hash) == password
end
# Register a new user
# Returns: success?, data
def self.register(email, name, password, password_confirm)
check, errorstr = self.validate_register_creds(email, name, password, password_confirm)
3 years ago
if check then
pw_hash = BCrypt::Password.create password
data = { # payload
name: name,
email: email,
pw_hash: pw_hash
}
resp = self.insert data # insert into the db
return check, resp
else
return check, errorstr
end
end
3 years ago
# Log in user
# Returns: success?, user id
def self.login(email, password)
user = self.find_by_email email # get the user info
return false, LOGIN_ERRORS[:fail] unless user # Verify that the user exists
3 years ago
pw_check = self.validate_password(user.pw_hash, password)
3 years ago
return false, LOGIN_ERRORS[:fail] unless pw_check # Verify password
return true, user.id
end
end