idnio 2.3.2b

data/lib/idnio/markdown.rb

@@ -0,0 +1,345 @@
+require 'fileutils'
+#
+# Markdown Writing Utility
+#
+module Markdown
+
+  @@buffer = ""
+  @@file = nil
+
+
+
+  # private_class_method :new
+
+  #def self.open( file )
+  #  open( file )
+  #end
+
+  def self.open( file )
+    @@file = file
+    if File.exist?( @@file )
+      FileUtils.rm_f( @@file )
+    end
+
+    FileUtils.mkdir_p File.dirname( file )
+  end
+
+  def self.toc
+    text = "\n[TOC]\n"
+    @@buffer << text
+    return text
+  end
+
+  def self.strong( text )
+    text = "**#{text}**"
+    @@buffer << text
+    return text
+  end
+
+  def self.h1( text )
+    text = "\n# #{text}\n"
+    @@buffer << text
+    return text
+  end
+
+  def self.h2( text )
+    text = "\n## #{text}\n"
+    @@buffer << text
+    return text
+  end
+
+  def self.h3( text )
+    text = "\n### #{text}\n"
+    @@buffer << text
+    return text
+  end
+
+  def self.h4( text )
+    text = "\n#### #{text}\n"
+    @@buffer << text
+    return text
+  end
+
+  def self.h5( text )
+    text = "\n##### #{text}\n"
+    @@buffer << text
+    return text
+  end
+
+  def self.link( link, label )
+    text = "[#{label}](#{link})"
+    @@buffer << text
+    return text
+  end
+
+  def self.image( file, alt_text )
+    text = "\n![#{alt_text}](#{file})\n"
+    @@buffer << text
+    return text
+  end
+
+  def self.code( text )
+    text = "\n~~~\n#{text}\n~~~\n"
+    @@buffer << text
+    return text
+  end
+
+  def self.json( text )
+    text = "\n~~~json\n#{text}\n~~~\n"
+    @@buffer << text
+    return text
+  end
+
+  def self.xml( text )
+    text = "\n~~~xml\n#{text}\n~~~\n"
+    @@buffer << text
+    return text
+  end
+
+  def self.text( text )
+    text = "#{text}"
+    @@buffer << text
+    return text
+  end
+
+  def self.line( text )
+    text = "\n***\n"
+    @@buffer << text
+    return text
+  end
+
+  def self.ul( list )
+    text = ""
+    list.each do |item|
+      text << "- #{item}\n"
+    end
+    @@buffer << text
+    return text
+  end
+
+  def self.ul2( list )
+    text = ""
+    list.each do |item|
+      text << "\t- #{item}\n"
+    end
+    @@buffer << text
+    return text
+  end
+
+  def self.ol( list )
+    text = ""
+    list.each_with_index do |item, index|
+      output << "#{index + 1}. #{item}\n"
+    end
+    @@buffer << text
+    return text
+  end
+
+  # Generate a Markdown table.
+  # labels and data are one and two-dimensional arrays, respectively.
+  # All input must have a to_s method.
+  # Pass align: 'l' for left alignment or 'r' for right  alignment. Anything
+  # else will result in cells being centered. Pass an array of align values
+  # to specify alignment per column.
+  # If is_rows is true, then each sub-array of data represents a row.
+  # Conversely, if is_rows is false, each sub-array of data represents a column.
+  # Empty cells can be given with nil or an empty string.
+  def self.make_table(labels, data, align: '', is_rows: false)
+    validate(labels, data, align, is_rows)
+
+    # Deep copy the arguments so we don't mutate the originals.
+    labels = Marshal.load(Marshal.dump(labels))
+    data = Marshal.load(Marshal.dump(data))
+
+    # Remove any breaking Markdown characters.
+    labels.map! {|label| sanitize(label)}
+    data.map! {|datum| datum.map {|cell| sanitize(cell)}}
+
+    # Convert align to something that other methods won't need to validate.
+    align.class == String && align = [align] * labels.length
+    align.map! {|a| a =~ /[lr]/i ? a.downcase : 'c'}
+
+    # Generate the column labels and alignment line.
+    header_line = labels.join('|')
+    alignment_line = parse_alignment(align, labels.length)
+
+    # Pad the data arrays so that it can be transposed if necessary.
+    max_len = data.map(&:length).max
+    data.map! {|datum| fill(datum, max_len)}
+
+    # Generate the table rows.
+    rows = (is_rows ? data : data.transpose).map {|row| row.join('|')}
+
+    text = [header_line, alignment_line, rows.join("\n")].join("\n")
+    @@buffer << text
+    return text
+  end
+
+  # Convert a Markdown table into human-readable form.
+  # def self.plain_text( md_table )
+  #   md_table !~ // && raise('Invalid input')
+  #
+  #   # Split the table into lines to get the labels, rows, and alignments.
+  #   lines = md_table.split("\n")
+  #   alignments = lines[1].split('|')
+  #   # labels or rows might have some empty values but alignments
+  #   # is guaranteed to be of the right width.
+  #   table_width = alignments.length
+  #   # '|||'.split('|') == [], so we need to manually add trailing empty cells.
+  #   # Leading empty cells are taken care of automatically.
+  #   labels = fill(lines[0].split('|'), table_width)
+  #   rows = lines[2..-1].map {|line| fill(line.split('|'), table_width)}
+  #
+  #   # Get the width for each column.
+  #   cols = rows.transpose
+  #   widths = cols.each_index.map {|i| column_width(cols[i].push(labels[i]))}
+  #
+  #   # Align the labels and cells.
+  #   labels = labels.each_index.map { |i|
+  #     aligned_cell(unsanitize(labels[i]), widths[i], alignments[i])
+  #   }
+  #   rows.map! { |row|
+  #     row.each_index.map { |i|
+  #       aligned_cell(unsanitize(row[i]), widths[i], alignments[i])
+  #     }
+  #   }
+  #
+  #   border = "\n|" + widths.map {|w| '=' * w}.join('|') + "|\n"
+  #   return (
+  #     border + [
+  #       '|' + labels.join('|') + '|',
+  #       rows.map {|row| '|' + row.join('|') + '|'}.join(border.tr('=', '-'))
+  #     ].join(border) + border
+  #   ).strip
+  #
+  # end
+
+  # Sanity checks for make_table.
+  private_class_method def self.validate(labels, data, align, is_rows)
+    if labels.class != Array
+      raise('labels must be an array')
+    end
+    if data.class != Array || data.any? {|datum| datum.class != Array}
+      raise('data must be a two-dimensional array')
+    end
+    if labels.empty?
+      raise('No column labels given')
+    end
+    if data.all? {|datum| datum.empty?}
+      raise('No cells given')
+    end
+    if labels.any? {|label| !label.respond_to?(:to_s)}
+      raise('One or more column labels cannot be made into a string')
+    end
+    if data.any? {|datum| datum.any? {|cell| !cell.respond_to?(:to_s)}}
+      raise('One or more cells cannot be made into a string')
+    end
+    if ![String, Array].include?(align.class)
+      raise('align must be a string or array')
+    end
+    if align.class == Array && align.any? {|val| val.class != String}
+      raise('One or more align values is not a string')
+    end
+    if !is_rows && data.length > labels.length
+      raise('Too many data columns given')
+    end
+    if is_rows && data.any? {|row| row.length > labels.length}
+      raise('One or more rows has too many cells')
+    end
+  end
+
+  # Convert some input to a string and replace any '|' characters  with
+  # a non-breaking equivalent,
+  private_class_method def self.sanitize(input)
+    bar = '&#124;'  # Non-breaking HTML vertical bar.
+    return input.to_s.gsub('|', bar)
+  end
+
+  # Replace non-breaking HTML characters with their plaintext counterparts.
+  private_class_method def self.unsanitize(input)
+    return input.gsub(/(&nbsp;)|(&#124;)/, '&nbsp;' => ' ', '&#124;' => '|')
+  end
+
+  # Generate the alignment line from a string or array.
+  # align must be a string or array of strings.
+  # n: number of labels in the table to be created.
+  private_class_method def self.parse_alignment(align, n)
+    align_map = {'l' => ':-', 'c' => ':-:', 'r' => '-:'}
+    alignments = align.map {|a| align_map[a]}
+    # If not enough values were given, center the remaining columns.
+    alignments.length < n && alignments += [':-:'] * (n - alignments.length)
+    return alignments.join('|')
+  end
+
+  # Align some text in a cell.
+  private_class_method def self.aligned_cell(text, width, align)
+    if align =~ /:-+:/  # Center alignment.
+      start = (width / 2) - (text.length / 2)
+    elsif align =~ /-+:/  # Right alignment.
+      start = width - text.length - 1
+    else  # Left alignment.
+      start = 1
+    end
+    return ' ' * start + text + ' ' * (width - start - text.length)
+  end
+
+  # Get the width for a column.
+  private_class_method def self.column_width(col)
+    # Pad each cell on either side and maintain a minimum 3 width of characters.
+    return [(!col.empty? ? col.map(&:length).max : 0) + 2, 3].max
+  end
+
+  # Add any missing empty values to a row.
+  private_class_method def self.fill(row, n)
+    row.length > n && raise('Sanity checks failed for fill')
+    return row.length < n ? row + ([''] * (n - row.length)) : row
+  end
+
+  #
+  # Used to get the string buffer (if needed).
+  #
+  def self.get_buffer
+    return @@buffer
+  end
+
+  #
+  # Used to set the string buffer (if needed).
+  #
+  def self.set_buffer( new_buffer )
+    @@buffer = new_buffer
+    return @@buffer
+  end
+
+  #
+  # Replaces a string in the buffer.
+  #
+  def self.replace( lookups, replacement )
+
+    #
+    # Check to see if the lookups are an array;
+    # If yes, do all the replacements.
+    # If no, then only do the single replacement.
+    #
+    if lookups.kind_of?(Array)
+      lookups.each do |lookup|
+        @@buffer.gsub!( lookup, replacement )
+      end
+    else
+      @@buffer.gsub!( lookups, replacement )
+    end
+
+    #
+    # Return the string buffer
+    #
+    return @@buffer
+  end
+
+  #
+  # Write the string out to file; and clear the string buffer.
+  #
+  def self.write
+    File.open( @@file, 'a+') { |file| file.write( @@buffer ) }
+    @@buffer = ""
+  end
+
+end

data/lib/idnio/program.rb

@@ -0,0 +1,153 @@
+require 'idnio/timer'
+require "json"
+#
+# Program Utility
+#
+module Program
+
+  def self.setConfig(config)
+
+  end
+
+  def self.line
+    $log.info "--------------------------------------------------------------"
+  end
+
+  #
+  # Utility method for printing messages.
+  #
+  def self.output( message )
+    Program.line
+    $log.info " #{message}"
+    Program.line
+  end
+
+  #
+  # Utility to start the program
+  #
+  def self.start( name, author, version, date )
+    Timer.start
+
+    $log.unknown "--------------------------------------------------------------"
+    $log.unknown " #{name}"
+    $log.unknown "--------------------------------------------------------------"
+    $log.unknown " Version: #{version}"
+    $log.unknown " Date: #{date}"
+    $log.unknown " Author: #{author}"
+    $log.unknown "--------------------------------------------------------------"
+
+    if $config.has_key?( 'tenant' )
+
+      $log.debug "Personal Access Token configuration detected.  Using settings from 'tenant'."
+
+      $url = $config['tenant']['url']
+      $tenant = $config['tenant']['url'].clone.gsub!(/.api|https:\/\//,"")
+
+      $log.debug "Calling IdentityNow to get a JWT OAuth token..."
+
+      response = IDNAPI.post_unauth( "#{$url}/oauth/token?grant_type=client_credentials&client_id=#{$config['tenant']['client-id']}&client_secret=#{$config['tenant']['client-secret']}" )
+
+      case response
+      when Net::HTTPSuccess
+
+        session = JSON.parse( response.body )
+
+        if !session.nil? && session.has_key?( 'access_token' )
+          $token = session['access_token']
+          $log.debug "Session token: #{$token}"
+        else
+          $log.fatal "Error: Unable to parse session token from response. (#{response.code})"
+          abort
+        end
+
+      else
+        $log.fatal "Error: Unable to retreive session token. Please check 'tenant' configuration and try again. (#{response.code})"
+        abort
+      end
+
+    elsif $config.has_key?( 'session' )
+
+      $log.debug "Session configuration detected.  Using settings from 'session'."
+
+      $tenant = $config['session']['baseUrl'].clone.gsub!(/.api|https:\/\//,"")
+      $url = $config['session']['baseUrl']
+      $token = $config['session']['accessToken']
+
+      $log.debug "Session token: #{$token}"
+
+    else
+
+      $log.fatal "Error: Configuration does not have details to retreive session. Please configure 'session' or 'tenant' settings."
+      abort
+
+    end
+
+    unless $tenant.nil?
+      $log.unknown " Tenant: #{$tenant}"
+      $log.unknown "--------------------------------------------------------------"
+    end
+
+  end
+
+  #
+  # Utility to end the program
+  #
+  def self.end( )
+    Timer.stop
+    output( "Process completed in #{Timer.elapsed}" )
+  end
+
+  #
+  # Utility method to write output files to a directory.
+  #
+  def self.write_file( directory, name, text )
+    name.gsub!(/[^0-9A-Za-z. -]/,"-")
+    FileUtils.mkdir_p directory
+    File.open( File.join( directory, name ), 'w+') { |file| file.write( text ) }
+  end
+
+  def self.get_filenames( directory )
+    output = Array.new
+    Dir.glob("#{directory}/*.json") do |file|
+      #tmp = file.slice! "#{directory}/"
+      output.push(file)
+    end
+    return output
+  end
+
+  def self.read_directory( directory )
+    output = Array.new
+    Dir.glob("#{directory}/*.json") do |file|
+      output.push(Program.read_file(file))
+    end
+    return output
+  end
+
+  #
+  # Utility method to read output files to a directory.
+  #
+  def self.read_file( directory, name )
+    return Program.read_file("#{directory}/#{name}")
+  end
+
+  def self.read_file( name )
+    output = ""
+    #name.gsub!(/[^0-9A-Za-z. -]/,"-")
+    f = File.open( name, 'r')
+    f.each_line do |line|
+      output += line
+    end
+    f.close
+    return output
+  end
+
+  def self.write_csv(directory, name, content)
+    if !File.file?("#{directory}/#{name}")
+      Program.write_file(directory, name, "")
+    end
+    CSV.open("#{directory}/#{name}", "ab") do |csv|
+      csv << content
+    end
+  end
+
+end