nub 0.0.70 → 0.0.72

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1425045f9af6f1cf2a0d871c7199015ce855ec5a658717ee023f88e4b738fed9
-  data.tar.gz: 3f7241bfe87d221fd23e359bc4b7eddb114504d8a6204b71c3a2368a7c6dc3c6
+  metadata.gz: c381c27337a97c6edadf9e4a928a4dd7adee88e6af49d0437089362fb89862da
+  data.tar.gz: fcb68bf919c81141910718a1af78b84cff7b7ff936697d0610e69b8ca54da2a1
 SHA512:
-  metadata.gz: b86b8c2fced9f9dba5f0520c1d72f76b349e594ba7d74d50a4ac8b17fa6964356e577b2402231bb38df553defd284eb0242459ee6b2799092cc4f24f33414d78
-  data.tar.gz: 54f5ff52b47fb8827c98071ca0990a04750c77e175a6f68fa7a48399a46841491df83354930433cf7f4973dd3ff4e375766b4ca5cfe59d18ee09f3e99f6cbb3a
+  metadata.gz: 9750357d487da6f02a19a0b1664ae4c774cfea1207a7fad6218f60f69908c4aa7161a690f8d1a10d2fc835df9066171829842873d0a4341d02c8c491db0ddd02
+  data.tar.gz: 793329dd53c30bc8c90050f5f63bf1a18d254caeb058b8a447501572e14a2c32ae31bfbc4bceacb2cd0088234fcb7b205e426b11b601831203c0f57d5a3b615e

data/lib/nub/sys.rb

@@ -27,15 +27,6 @@ require_relative 'log'
 module Sys
   extend self
 
-  # Get the given environment variable by nam
-  # @param var [String] name of the environment var
-  # @param required [Bool] require that the variable exists by default
-  def env(var, required:true)
-    value = ENV[var]
-    Log.die("#{var} env variable is required!") if required && !value
-    return value
-  end
-
   # Wait for any key to be pressed  
   def any_key?
     begin
@@ -70,6 +61,60 @@ module Sys
     return OpenStruct.new(result: result, stdout: stdout.string, stderr: stderr.string)
   end
 
+  # Drop root privileges to original user
+  # Only affects ruby commands not system commands
+  # Params:
+  # +returns+:: uid, gid of prior user
+  def drop_privileges()
+    uid = gid = nil
+
+    if Process.uid.zero?
+      uid, gid = Process.uid, Process.gid
+      sudo_uid, sudo_gid = ENV['SUDO_UID'].to_i, ENV['SUDO_GID'].to_i
+      Process::Sys.setegid(sudo_uid)
+      Process::Sys.seteuid(sudo_gid)
+    end
+
+    return uid, gid
+  end
+
+  # Get the given environment variable by nam
+  # @param var [String] name of the environment var
+  # @param required [Bool] require that the variable exists by default
+  def env(var, required:true)
+    value = ENV[var]
+    Log.die("#{var} env variable is required!") if required && !value
+    return value
+  end
+
+  # Run the system command in an exception wrapper
+  # @param cmd [String/Array] cmd to execute
+  # @param env [Hash] optional environment variables to set
+  # @param die [Bool] fail on errors if true
+  # @returns boolean status, string message
+  def exec(cmd, env:{}, die:true)
+    result = false
+    puts("exec: #{cmd.is_a?(String) ? cmd : cmd * ' '}")
+
+    begin
+      if cmd.is_a?(String)
+        result = system(env, cmd, out: $stdout, err: :out)
+      else
+        result = system(env, *cmd, out: $stdout, err: :out)
+      end
+    rescue Exception => e
+      result = false
+      puts(e.message.colorize(:red))
+      puts(e.backtrace.inspect.colorize(:red))
+      exit if die
+    end
+
+    !puts("Error: failed to execute command properly".colorize(:red)) and
+      exit unless !die or result
+
+    return result
+  end
+
   # Read a password from stdin without echoing
   # @returns pass [String] the password read in
   def getpass
@@ -78,6 +123,74 @@ module Sys
     puts
     return pass
   end
+
+  # Update pacman database
+  def pacman_update
+    success = false
+    while not success
+      begin
+        Sys.exec("pacman -Sy")
+        success = true
+      rescue Exception => e
+        puts(e.message)
+      end
+    end
+  end
+
+  # Raise privileges if dropped earlier
+  # Only affects ruby commands not system commands
+  # Params:
+  # +uid+:: uid of user to assume
+  # +gid+:: gid of user to assume
+  def raise_privileges(uid, gid)
+    if uid and gid
+      Process::Sys.seteuid(uid)
+      Process::Sys.setegid(gid)
+    end
+  end
+
+  # Remove given dir or file
+  # Params:
+  # +path+:: path to delete
+  # +returns+:: path that was deleted
+  def rm_rf(path)
+    Sys.exec("rm -rf #{path}")
+    return path
+  end
+
+  # Unmount the given mount point
+  # Params:
+  # +mount+:: mount point to unmount
+  # +retries+:: number of times to retry
+  def umount(mount, retries:1)
+    check = ->(mnt){
+      match = `mount`.split("\n").find{|x| x.include?(mnt)}
+      match = match.split('overlay').first if match
+      return (match and match.include?(mnt))
+    }
+
+    success = false
+    while not success and retries > 0
+      if check[mount]
+        success = system('umount', '-fv', mount)
+      else
+        success = true
+      end
+
+      # Sleep for a second if failed
+      sleep(1) and retries -= 1 unless success
+    end
+
+    # Die if still mounted
+    !puts("Error: Failed to umount #{mount}".colorize(:red)) and
+      exit if check[mount]
+  end
+
+  # Get the current user taking into account sudo
+  # @returns the current user
+  def user
+    return Process.uid.zero? ? Etc.getpwuid(ENV['SUDO_UID'].to_i).name : ENV['USER']
+  end
 end
 
 # vim: ft=ruby:ts=2:sw=2:sts=2

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: nub
 version: !ruby/object:Gem::Version
-  version: 0.0.70
+  version: 0.0.72
 platform: ruby
 authors:
 - Patrick Crummett
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-05-24 00:00:00.000000000 Z
+date: 2018-06-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: colorize