From d72d635c6804702681e6f6905a79a52b9ca80b9a Mon Sep 17 00:00:00 2001
From: Alex Ling <hkalexling@gmail.com>
Date: Sun, 31 May 2020 14:30:45 +0000
Subject: [PATCH] Add `admin/user` sub-command

---
 src/mango.cr | 28 ++++++++++++++++++++++++++--
 1 file changed, 26 insertions(+), 2 deletions(-)

diff --git a/src/mango.cr b/src/mango.cr
index 53a281c..7bd0829 100644
--- a/src/mango.cr
+++ b/src/mango.cr
@@ -45,26 +45,50 @@ class CLI < Clim
         usage "mango admin user [arguments] [options]"
         help short: "-h"
         argument "action", type: String,
-          desc: "Action to make. Can be add/delete/update", required: true
+          desc: "Action to perform. Can be add/delete/update/list"
         argument "username", type: String,
           desc: "Username to update or delete"
         option "-u USERNAME", "--username=USERNAME", type: String,
           desc: "Username"
         option "-p PASSWORD", "--password=PASSWORD", type: String,
           desc: "Password"
-        option "--admin", desc: "Admin flag", type: Bool, default: false
+        option "-a", "--admin", desc: "Admin flag", type: Bool, default: false
         common_option
         run do |opts, args|
           Config.load(opts.config).set_current
+          storage = Storage.new nil, false
 
           case args.action
           when "add"
             throw "Options `-u` and `-p` required." if opts.username.nil? ||
                                                        opts.password.nil?
+            storage.new_user opts.username.not_nil!,
+              opts.password.not_nil!, opts.admin
           when "delete"
             throw "Argument `username` required." if args.username.nil?
+            storage.delete_user args.username
           when "update"
             throw "Argument `username` required." if args.username.nil?
+            username = opts.username || args.username
+            password = opts.password || ""
+            storage.update_user args.username, username.not_nil!,
+              password.not_nil!, opts.admin
+          when "list"
+            users = storage.list_users
+            name_length = users.map(&.[0].size).max
+            l_cell_width = ["username".size, name_length].max
+            r_cell_width = "admin access".size
+            header = " #{"username".ljust l_cell_width} | admin access "
+            puts "-" * header.size
+            puts header
+            puts "-" * header.size
+            users.each do |name, admin|
+              puts " #{name.ljust l_cell_width} | " \
+                   "#{admin.to_s.ljust r_cell_width} "
+            end
+            puts "-" * header.size
+          when nil
+            puts opts.help_string
           else
             throw "Unknown action \"#{args.action}\"."
           end