|
| 1 | +require 'thor' |
| 2 | +require 'colorize' |
| 3 | +require 'sequel' |
| 4 | +require 'bcrypt' |
| 5 | +require 'readline' |
| 6 | +require 'yaml' |
| 7 | +require_relative '../ruby/db.rb' |
| 8 | + |
| 9 | +settings = YAML.load_file(File.join(File.dirname(__FILE__), '../config/settings.yml')) |
| 10 | +$host = ENV['DB_HOST'] || settings['database']['host'].to_s |
| 11 | +$user = ENV['DB_USERNAME'] || settings['database']['username'].to_s |
| 12 | +$password = ENV['DB_PASSWORD'] || settings['database']['password'].to_s |
| 13 | +$database = ENV['DB_DATABASE'] || settings['database']['dbname'].to_s |
| 14 | + |
| 15 | +class RubyCLI < Thor |
| 16 | + desc "create", "Create a new user" |
| 17 | + def create |
| 18 | + puts "Creating a new user...".red |
| 19 | + username = Readline.readline("Username: ", true) |
| 20 | + username = username.gsub(/[^0-9A-Za-z]/, '') |
| 21 | + username = username.downcase |
| 22 | + while true |
| 23 | + password = Readline.readline("Password: ", true) |
| 24 | + passwordConfirm = Readline.readline("Retype password: ", true) |
| 25 | + if password != passwordConfirm |
| 26 | + puts "\e[H\e[2J" |
| 27 | + puts "Passwords do not match! Try again.".red |
| 28 | + else |
| 29 | + break |
| 30 | + end |
| 31 | + end |
| 32 | + |
| 33 | + db = connectDB($host, $user, $password, $database) |
| 34 | + hashedPassword = BCrypt::Password.create(password) |
| 35 | + if db[:users].where(username: username).count > 0 |
| 36 | + puts "User already exists!".red |
| 37 | + db.disconnect |
| 38 | + exit |
| 39 | + else |
| 40 | + db[:users].insert(username: username, password: hashedPassword, admin: false) |
| 41 | + puts "User created!".blue |
| 42 | + db.disconnect |
| 43 | + end |
| 44 | + |
| 45 | + end |
| 46 | + desc "delete", "Delete a user" |
| 47 | + def delete |
| 48 | + puts "Deleting a user...".red |
| 49 | + username = Readline.readline("Username: ", true) |
| 50 | + username = username.gsub(/[^0-9A-Za-z]/, '') |
| 51 | + username = username.downcase |
| 52 | + db = connectDB($host, $user, $password, $database) |
| 53 | + while true |
| 54 | + usernameConfirm = Readline.readline("Are you sure you want to delete #{username}? (y/n): ", true).downcase |
| 55 | + if usernameConfirm == "y" || usernameConfirm == "yes" |
| 56 | + if db[:users].where(username: username).count > 0 |
| 57 | + db[:users].where(username: username).delete |
| 58 | + puts "User deleted!".blue |
| 59 | + db.disconnect |
| 60 | + else |
| 61 | + puts "User does not exist! (use the list command to see all users)".red |
| 62 | + db.disconnect |
| 63 | + end |
| 64 | + exit |
| 65 | + elsif usernameConfirm == "n" || usernameConfirm == "no" |
| 66 | + puts "Ok, exiting...".blue |
| 67 | + db.disconnect |
| 68 | + exit |
| 69 | + end |
| 70 | + end |
| 71 | + end |
| 72 | + desc "list", "List all users" |
| 73 | + def list |
| 74 | + db = connectDB($host, $user, $password, $database) |
| 75 | + puts "Listing all users...".red |
| 76 | + users = db[:users] |
| 77 | + users.each{|user| puts user[:username]} |
| 78 | + end |
| 79 | + desc "reset", "Reset a user's password" |
| 80 | + def reset |
| 81 | + puts "Resetting a user's password...".red |
| 82 | + username = Readline.readline("Username: ", true) |
| 83 | + username = username.gsub(/[^0-9A-Za-z]/, '') |
| 84 | + username = username.downcase |
| 85 | + db = connectDB($host, $user, $password, $database) |
| 86 | + if db[:users].where(username: username).count > 0 |
| 87 | + while true |
| 88 | + password = Readline.readline("New Password: ", true) |
| 89 | + passwordConfirm = Readline.readline("Retype new password: ", true) |
| 90 | + if password != passwordConfirm |
| 91 | + puts "\e[H\e[2J" |
| 92 | + puts "Passwords do not match! Try again.".red |
| 93 | + else |
| 94 | + break |
| 95 | + end |
| 96 | + end |
| 97 | + hashedPassword = BCrypt::Password.create(password) |
| 98 | + db[:users].where(username: username).update(password: hashedPassword) |
| 99 | + puts "Password reset!".blue |
| 100 | + db.disconnect |
| 101 | + else |
| 102 | + puts "User does not exist! (use the list command to see all users)".red |
| 103 | + db.disconnect |
| 104 | + end |
| 105 | + end |
| 106 | +end |
| 107 | + |
| 108 | +RubyCLI.start(ARGV) |
0 commit comments