phoseum-cli 0.0.9 → 0.0.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7282f4986b7a9f813c002c354d5fae25c6b0e29fd6274c4cd8ee5fa2a75cc89b
-  data.tar.gz: 3e22b900db8134c6a90b9cd54c55d672ddef8af86195c0148f7458fe924d47c0
+  metadata.gz: a9472d2d52deb75c37183130fce537c43f87e25d2c5366f73d34e2e39f4b2f67
+  data.tar.gz: 6e37a6d2768e87133c97e5a7d5257595e39b9a97483be4eddc109f38f15e81e6
 SHA512:
-  metadata.gz: 6f23596ae34b88ba0bbb537672304490c2e065e448dfbb9b434052f73dc73080c0dbe89cbbf9b2b149c844b8a5695be19472f669e91fbeaf90e6a89cbe5d773c
-  data.tar.gz: c044c7b0922aacc1f0830e0592f6b7357266664f9c4ecbc5418a69ad2e86cb0f20cec3067708aa0e3321e76c2b8f5d0c096bdf494717813866e55007eda83ce0
+  metadata.gz: a249f7bebf5c8621067a8f52a560ad6c24867d09956acd7e1323c79aa60628d8261ae06f73e45e54e1355ba07b673c883176c13273616c7c66683f90536f2f74
+  data.tar.gz: ea4973b4317ee70af953b628fefda619bbc35bc355434bf9c61f2d037ba058194a57959ffa6ef5c976728ae6298bc966af114e269407bf34dd82eca680de569c

data/bin/phoseum-cli

@@ -24,16 +24,16 @@ $MULTI = options[:multiple] ? true : false
 client_checks
 
 def upload(album)
-
-  if $config['TOKEN']
-    headers={ "bearer" => "#{$config['TOKEN']}" }
-  else
+  headers = {}
+  if !validate_token($config['TOKEN'])
     if token = user_login()
       headers = { "bearer" => token }
     else
       puts "Login Failed".red
       exit 1
     end
+  else
+    headers={ "bearer" => "#{$config['TOKEN']}" }
   end
 
   list=Dir.entries("./")
@@ -154,15 +154,15 @@ end
 
 def health(what='',name='')
   headers = {}
-  if $config['TOKEN']
-    headers={ "bearer" => "#{$config['TOKEN']}" }
-  else
+  if !validate_token($config['TOKEN'])
     if token = user_login()
       headers = { "bearer" => token }
     else
       puts "Login Failed".red
       exit 1
     end
+  else
+    headers={ "bearer" => "#{$config['TOKEN']}" }
   end
   some_uri=''
   if what == "album"
@@ -264,29 +264,28 @@ def user_mgmt(action,user,pass='',role='')
   exit 0
 end
 
-def user_login()
+def user_login(puser='',ppass='')
   post_body={}
-  user=''
-  pass=''
+  user= !puser ? '' : puser
+  pass= !ppass ? '' : ppass
   if $config['USER']
     user=$config['USER']
   else
     print "Username: "
-    user=STDIN.noecho(&:gets).chomp
+    user=STDIN.gets.chomp
   end
   if $config['PASS']
     pass=$config['PASS']
-    puts "Keeping your password on the configuration file is UNSAFE".red
+    puts "\nKeeping your password on the configuration file is UNSAFE".red
   else
-    print "\nPassword: "
-    pass=STDIN.noecho(&:gets).chomp
+    pass=STDIN.getpass('Password: ')
   end
   post_body=JSON.generate({"action" => "user-login", "user" => "#{user}", "password" => "#{pass}"})
   base = URI.parse("#{$config['SERVERURL']}")
   puts "\nLogging to: #{$config['SERVERURL']}".yellow if !$QUIET
   request = Net::HTTP::Post.new(base)
   request.body = post_body
-  request.basic_auth("auth", $config['DEFAULT_SECRET'])
+  request.basic_auth("cli", "loginNOauth")
   response = Net::HTTP.start(base.hostname, $config['PORT'],
                              :timeout => $config['CALL_TIMEOUT'],
                              :use_ssl => base.scheme == "https",
@@ -323,50 +322,40 @@ end
 
 
 def delete(what='',path='')
-  puts "You must write 'YES' to confirm, otherwise NO is assumed".yellow
-  print "Are you sure you want to delete the #{what} #{path} :[YES/NO]: "
-  confirm=STDIN.gets.chomp
-
-  if confirm == "YES"
-    headers={}
-    if $config['TOKEN']
-      headers={ "bearer" => "#{$config['TOKEN']}" }
+  if !validate_token($config['TOKEN'])
+    if token = user_login()
+      headers = { "bearer" => token }
     else
-      if token = user_login()
-        headers = { "bearer" => token }
-      else
-        puts "Login Failed".red
-        exit 1
-      end
-    end
-
-    base = URI.parse("#{$config['SERVERURL']}/#{path}")
-    puts "Connecting to: #{$config['SERVERURL']}".yellow if !$QUIET
-    request = Net::HTTP::Delete.new(base,headers)
-    response = Net::HTTP.start(base.hostname, $config['PORT'],
-                                :timeout => $config['CALL_TIMEOUT'],
-                                :use_ssl => base.scheme == "https",
-                                :verify_mode => OpenSSL::SSL::VERIFY_PEER,
-                                :ca_file => $config['CA_TRUST']
-                                ) do |http|
-      http.request(request)
-    end
-    begin
-      list = JSON.parse(response.body)
-      if list['error']
-        puts list['error'].red
-        exit 1
-      else
-        puts list['success'].green
-        exit 0
-      end
-    rescue
-      puts "\nThe server sent out an Error:".red
-      puts clean_html(response.body)
+      puts "Login Failed".red
       exit 1
     end
   else
-    puts "Deleting cancelled.".green
+    headers={ "bearer" => "#{$config['TOKEN']}" }
+  end
+  base = URI.parse("#{$config['SERVERURL']}/#{path}")
+  puts "Connecting to: #{$config['SERVERURL']}".yellow if !$QUIET
+  request = Net::HTTP::Delete.new(base,headers)
+  response = Net::HTTP.start(base.hostname, $config['PORT'],
+                              :timeout => $config['CALL_TIMEOUT'],
+                              :use_ssl => base.scheme == "https",
+                              :verify_mode => OpenSSL::SSL::VERIFY_PEER,
+                              :ca_file => $config['CA_TRUST']
+                              ) do |http|
+    http.request(request)
+  end
+  begin
+    list = JSON.parse(response.body)
+    if list['error']
+      puts list['error'].red
+      exit 1
+    else
+      puts list['success'].green
+      exit 0
+    end
+  rescue
+    puts "\nThe server sent out an Error:".red
+    puts clean_html(response.body)
+    exit 1
   end
 end
 
@@ -387,10 +376,10 @@ case options
         puts "You need to select one of the valid Roles".red
         exit 1
       else
-        print "Please type the password twice for user #{value}.\n\tPassword: "
-        new_password=STDIN.noecho(&:gets).chomp
-        print "\n\tConfirm: "
-        confirm=STDIN.noecho(&:gets).chomp
+        print "Please type the password twice for user #{value}.\n\t"
+        new_password=STDIN.getpass('Password: ')
+        print "\t"
+        confirm=STDIN.getpass('Confirm: ')
         if new_password == confirm
           user_mgmt(what,value,new_password,options[:role])
         end
@@ -405,19 +394,66 @@ case options
         puts "It is Not possible to delete base album".red
         exit 1
       end
+      if !confirm_action("Are you sure you want to delete the #{what} #{value}")
+        puts "Deleting #{what} cancelled.".green
+      end
     end
     if options[:image]
       if _found = check_image_exists(options[:image])
-        img_data = JSON.parse(_found)
-        value="#{img_data['web_path']}/#{img_data['image']}"
+        if _found =~ /^Signature Found at \[\"(.*)\"\]/
+          capt = _found.match(/^Signature Found at \[\"(.*)\"\]/).captures
+          capt_part = capt[0].match(/.*(Albums.*)$/).captures
+          value = capt_part[0]
+        else
+          img_data = JSON.parse(_found)
+          value="#{img_data['web_path']}/#{img_data['image']}"
+        end
         what='image'
       else
         puts "Image not found on server".red
         exit 1
       end
+      if confirm_action("Are you sure you want to delete the #{what} #{value}")
+        puts "Deleting #{what} #{value}".green if !$QUIET
+        delete(what,value)
+      else
+        puts "Deleting #{what} cancelled.".green
+      end
+    end
+    if options[:user]
+      what='user'
+      value=options[:user]
+      if confirm_action("Are you sure you want to delete the #{what} #{value}")
+        puts "Deleting #{what} #{value}".green if !$QUIET
+        if delete_user(value)
+          puts "#{value} successfully deleted.".green
+        else
+          puts "#{value} failed to be deleted.".red
+        end
+      else
+        puts "Deleting #{what} cancelled.".green
+      end
+    end
+    if !what
+      puts "Got no object to delete. Exiting.".red
+      exit 1
+    end
+  when -> (o) { o[:options] }
+    if options[:user]
+      print "Please confirm #{options[:user]} password.\n\t"
+      current_password=STDIN.getpass('Password: ')
+      if token = user_login(options[:user],current_password)
+        print "\n Change [P]assword, [U]sername, [R]ole :[P/U/R]: "
+        confirm=STDIN.gets.chomp
+      else
+        puts "User Check Failed".red
+        exit 1
+      end
+    else
+    # if !what
+      puts "Got no user to work with. Exiting.".red
+      exit 1
     end
-    puts "Deleting #{what} #{value}".green if !$QUIET
-    delete(what,value)
   when -> (u) { u[:upload] }
     puts "Preparing to upload images to API server".green if !$QUIET
     if !options.key?(:album)

data/lib/phoseum/phoseum-cli-lib.rb

@@ -44,6 +44,10 @@ def option_parser(opts)
       options[:role] = r
     end
 
+    opts.on("-o", "--options", "Update options from object. Use with: [user]") do |o|
+      options[:options] = o
+    end
+
     opts.on("-q", "--quiet", "Avoid regular info, useful for JSON output") do |q|
       options[:quiet] = q
     end
@@ -56,6 +60,10 @@ def option_parser(opts)
       options[:upload] = u
     end
 
+    opts.on("-U", "--User USERNAME", "Username for delete or change password / role. Use with: [delete, options]") do |usr|
+      options[:user] = usr
+    end
+
     opts.on("-h", "-h", "That is this...") do |h|
       options[:help] = h
     end
@@ -64,6 +72,17 @@ def option_parser(opts)
   return options
 end
 
+def confirm_action(msg)
+  puts "You must write 'YES' to confirm, otherwise NO is assumed".yellow
+  print "#{msg} :[YES/NO]: "
+  confirm=STDIN.gets.chomp
+  if confirm == "YES"
+    return true
+  else
+    return false
+  end
+end
+
 class String
   # colorization
   def colorize(color_code)
@@ -145,16 +164,15 @@ def check_image_exists(fpath)
 end
 
 def search_image(sign,album='')
-  headers = {}
-  if $config['TOKEN']
-    headers={ "bearer" => "#{$config['TOKEN']}" }
-  else
+  if !validate_token($config['TOKEN'])
     if token = user_login()
       headers = { "bearer" => token }
     else
       puts "Login Failed".red
       exit 1
     end
+  else
+    headers={ "bearer" => "#{$config['TOKEN']}" }
   end
   some_uri= album == '' ? "?signature=#{sign}" : "?signature=#{sign}&inside_album=#{album}"
   base = URI.parse("#{$config['SERVERURL']}#{some_uri}")
@@ -182,3 +200,69 @@ def search_image(sign,album='')
     exit 1
   end
 end
+
+def validate_token(token)
+  headers = {}
+  if $config['TOKEN']
+    headers={ "bearer" => "#{$config['TOKEN']}" }
+  else
+    return false
+  end
+  base = URI.parse("#{$config['SERVERURL']}")
+  request = Net::HTTP::Post.new(base,headers)
+  request.body = JSON.generate({"action" => "check-token" })
+  response = Net::HTTP.start(base.hostname, $config['PORT'],
+                             :timeout => $config['CALL_TIMEOUT'],
+                             :use_ssl => base.scheme == "https",
+                             :verify_mode => OpenSSL::SSL::VERIFY_PEER,
+                             :ca_file => $config['CA_TRUST']
+                             ) do |http|
+    http.request(request)
+  end
+  begin
+    list = JSON.parse(response.body)
+    if list['error']
+      return false
+    end
+    if list['success']
+      return true
+    end
+  rescue
+    puts "\nThe server sent out an Error:".red
+    puts clean_html(response.body)
+    exit 1
+  end
+end
+
+def delete_user(username)
+  headers = {}
+  if $config['TOKEN']
+    headers={ "bearer" => "#{$config['TOKEN']}" }
+  else
+    return false
+  end
+  base = URI.parse("#{$config['SERVERURL']}")
+  request = Net::HTTP::Post.new(base,headers)
+  request.body = JSON.generate({"action" => "delete-user", "username" => username })
+  response = Net::HTTP.start(base.hostname, $config['PORT'],
+                             :timeout => $config['CALL_TIMEOUT'],
+                             :use_ssl => base.scheme == "https",
+                             :verify_mode => OpenSSL::SSL::VERIFY_PEER,
+                             :ca_file => $config['CA_TRUST']
+                             ) do |http|
+    http.request(request)
+  end
+  begin
+    list = JSON.parse(response.body)
+    if list['error']
+      return false
+    end
+    if list['success']
+      return true
+    end
+  rescue
+    puts "\nThe server sent out an Error:".red
+    puts clean_html(response.body)
+    exit 1
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: phoseum-cli
 version: !ruby/object:Gem::Version
-  version: 0.0.9
+  version: 0.0.10
 platform: ruby
 authors:
 - Julio C Hegedus
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2020-09-07 00:00:00.000000000 Z
+date: 2020-09-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: yaml