gdsh 0.4.0

data/lib/gdsh/error.rb

@@ -0,0 +1,7 @@
+require 'colorize'
+
+module DriveError
+  def drive_error_string
+    puts "An error occurred: #{result.data['error']['message']}".colorize(:red)
+  end
+end

data/lib/gdsh/get_file.rb

@@ -0,0 +1,65 @@
+require_relative 'commands'
+require_relative 'query_revision'
+
+module Commands
+  ##
+  # Downloads Files/Revisions
+  #
+  class GetFile < QueryRevision
+    def self.command_name
+      'get_txt'
+    end
+
+    def self.parameters
+      '(<file_id>[,<revision_number>])'
+    end
+
+    def self.function
+      'Downloads a specific revision of a file if <revision_number> is ' \
+      'specified; downloads all revisions otherwise.'
+    end
+
+    def initialize(client, params)
+      super(client, params)
+      @revision = (params.length == 3) ? params[2] : nil
+      @file_id = @params[1]
+    end
+
+    def puts_downloading_banner(url)
+      puts 'Downloading '.colorize(:cyan) + "#{url} ...".colorize(:light_yellow)
+    end
+
+    def download(url)
+      return unless @client
+
+      puts_downloading_banner(url)
+      result = @client.execute(uri: url)
+      if result.status == 200
+        result.body
+      else
+        puts drive_error_string
+      end
+    end
+
+    def download_revision_as_txt(rev)
+      download(txt_link(rev))
+    end
+
+    def write_to_file(revision)
+      filename = generate_filename_from_revision(revision)
+      File.write(filename, download_revision_as_txt(revision))
+    end
+
+    def generate_filename_from_revision(revision)
+      @file_id + '_rev_' + revision + '.txt'
+    end
+
+    def execute
+      if @revision.nil?
+        revisions.each { |r| write_to_file(r['id']) }
+      else
+        write_to_file(@revision)
+      end
+    end
+  end
+end

data/lib/gdsh/help.rb

@@ -0,0 +1,20 @@
+require_relative 'commands'
+
+module Commands
+  ##
+  # Help command
+  #
+  class Help < Command
+    def self.command_name
+      'help'
+    end
+
+    def self.function
+      'Returns this usage information.'
+    end
+
+    def execute
+      Commands.usage
+    end
+  end
+end

data/lib/gdsh/list_files.rb

@@ -0,0 +1,78 @@
+require_relative 'commands'
+
+module Commands
+  ##
+  # List all files accessible by the application.
+  #
+  class ListFiles < Command
+    def self.command_name
+      'ls'
+    end
+
+    def self.function
+      'List files accessible by this application.'
+    end
+
+    def retrieve_file_by_page_token(pagetoken)
+      parameters = pagetoken.to_s.empty? ? {} : { pageToken: pagetoken }
+      drive = @client.discovered_api('drive', 'v2')
+      @client.execute(
+        api_method: drive.files.list,
+        parameters: parameters)
+    end
+
+    def puts_banner
+      puts 'Retrieving list of files accessible...'.colorize(:green)
+    end
+
+    def filelist
+      puts_banner
+
+      result = []
+      page_token = nil
+
+      loop do
+        api_result = retrieve_file_by_page_token(page_token)
+
+        if api_result.status == 200
+          files = api_result.data
+          result.concat(files.items)
+          page_token = files.next_page_token
+        else
+          drive_error_string
+          page_token = nil
+        end
+
+        break if page_token.to_s.empty?
+      end
+
+      result
+    end
+
+    def title_label
+      'Title: '.colorize(:light_magenta)
+    end
+
+    def id_label
+      'id: '.colorize(:light_magenta)
+    end
+
+    def created_at_label
+      'created at: '.colorize(:light_magenta)
+    end
+
+    def puts_file_info(f)
+      puts title_label + "#{f['title']}"
+      puts id_label + "#{f['id']}"
+      puts created_at_label + "#{f['createdDate']}"
+      puts ''
+    end
+
+    def execute
+      filelist.each do |f|
+        next if f['labels']['trashed']
+        puts_file_info(f)
+      end
+    end
+  end
+end

data/lib/gdsh/mime.rb

@@ -0,0 +1,52 @@
+##
+# MIME types supported by Google Drive.
+#
+module Mime
+  def html
+    'text/html'
+  end
+
+  def txt
+    'text/plain'
+  end
+
+  def rtf
+    'application/rtf'
+  end
+
+  def odt
+    'application/vnd.oasis.opendocument.text'
+  end
+
+  def pdf
+    'application/pdf'
+  end
+
+  def docx
+    'application/vnd.openxmlformats-officedocument.wordprocessingml.document'
+  end
+
+  def xlsx
+    'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
+  end
+
+  def ods
+    'application/x-vnd.oasis.opendocument.spreadsheet'
+  end
+
+  def jpeg
+    'image/jpeg'
+  end
+
+  def png
+    'image/png'
+  end
+
+  def svg
+    'image/svg+xml'
+  end
+
+  def pptx
+    'application/vnd.openxmlformats-officedocument.presentationml.presentation'
+  end
+end

data/lib/gdsh/query_revision.rb

@@ -0,0 +1,106 @@
+require_relative 'commands'
+require_relative 'mime'
+
+module Commands
+  ##
+  # Queries all revisions of a file.
+  #
+  class QueryRevision < Command
+    include Mime
+
+    def self.command_name
+      'query'
+    end
+
+    def self.parameters
+      '(<file_id>)'
+    end
+
+    def self.function
+      'Queries all revisions of a file.'
+    end
+
+    def initialize(client, params)
+      super(client, params)
+      @file_id = params[1]
+    end
+
+    def revisions
+      drive = @client.discovered_api('drive', 'v2')
+      api_result = @client.execute(
+        api_method: drive.revisions.list,
+        parameters: { 'fileId' => @file_id })
+      if api_result.status == 200
+        api_result.data.items
+      else
+        puts drive_error_string
+      end
+    end
+
+    def puts_banner
+      puts 'Revisions'.colorize(:magenta)
+      puts '---------'.colorize(:magenta)
+    end
+
+    def revision_id_label
+      'Revision id: '.colorize(:magenta)
+    end
+
+    def modified_date_label
+      'Modified: '.colorize(:magenta)
+    end
+
+    def modifying_user_label
+      'Modifying User: '.colorize(:magenta)
+    end
+
+    def pdf_link_label
+      'Download pdf: '.colorize(:magenta)
+    end
+
+    def docx_link_label
+      'Download docx: '.colorize(:magenta)
+    end
+
+    def txt_link_label
+      'Download txt: '.colorize(:magenta)
+    end
+
+    def puts_download_links(revision)
+      puts pdf_link_label + "#{revision['exportLinks'][pdf]}"
+      puts docx_link_label + "#{revision['exportLinks'][docx]}"
+      puts txt_link_label + "#{revision['exportLinks'][txt]}"
+    end
+
+    def puts_revision_info(revision)
+      puts revision_id_label + "#{revision['id']}"
+      puts modified_date_label + "#{revision['modifiedDate']}"
+      puts modifying_user_label + "#{revision['lastModifyingUserName']}"
+      puts_download_links
+      puts ''
+    end
+
+    def execute
+      return if revisions.nil?
+
+      puts_banner
+      revisions.each do |r|
+        puts_revision_info(r)
+      end
+    end
+
+    def txt_link(revision)
+      rev = revisions.select { |r| r['id'] == revision }
+      # should only have one element if revision exists
+      rev.first['exportLinks'][txt] unless rev.empty?
+    end
+
+    def modifying_users
+      modifying_hash = {}
+      revisions.each do |r|
+        modifying_hash[r['id']] = r['lastModifyingUserName']
+      end
+      modifying_hash
+    end
+  end
+end

data/lib/gdsh/quit.rb

@@ -0,0 +1,23 @@
+require_relative 'commands'
+
+module Commands
+  ##
+  # Quit command.
+  #
+  class Quit < Command
+    def self.command_name
+      'quit'
+    end
+
+    def self.function
+      'Exit the application.'
+    end
+
+    def self.terminal?
+      true
+    end
+
+    def execute
+    end
+  end
+end

data/lib/gdsh/remove.rb

@@ -0,0 +1,34 @@
+require_relative 'commands'
+
+module Commands
+  class Remove < Command
+    def self.command_name
+      'rm'
+    end
+
+    def self.parameters
+      '(<file_id>)'
+    end
+
+    def self.function
+      'Removes the file.'
+    end
+
+    def initialize(client, params)
+      super(client, params)
+      @file_id = @params[1]
+    end
+
+    def execute
+      drive = @client.discovered_api('drive', 'v2')
+      result = @client.execute(
+        api_method: drive.files.trash,
+        parameters: { fileId: @file_id })
+      if result.status != 200
+        puts drive_error_string
+      else
+        puts 'Deleted.'.colorize(:green)
+      end
+    end
+  end
+end

data/lib/gdsh/revision_diff.rb

@@ -0,0 +1,87 @@
+require 'differ'
+
+require_relative 'commands'
+require_relative 'get_file'
+
+module Commands
+  ##
+  # Summarizes changes between revisions
+  #
+  class RevisionDiff < GetFile
+    def self.command_name
+      'diff'
+    end
+
+    def self.parameters
+      '(<file_id>[, <rev_1>, <rev_2>])'
+    end
+
+    def self.function
+      'Compares and summarizes the changes between two revisions. If no ' \
+      'revision numbers are provided, a consolidated summary is returned.'
+    end
+
+    def initialize(client, params)
+      super(client, params)
+      init_revisions
+    end
+
+    def init_revisions
+      @low_rev = (@params.length == 4) ? @params[2] : nil
+      @high_rev = (@params.length == 4) ? @params[3] : nil
+      @all = @low_rev.nil? && @high_rev.nil?
+      return if @all || order_is_correct
+      @low_rev, @high_rev = @high_rev, @low_rev
+    end
+
+    def order_is_correct
+      @high_rev.to_i > @low_rev.to_i
+    end
+
+    def consecutive_revisions
+      return unless @all
+      keys = modifying_users.keys.sort_by { |x| x.to_i }
+      len = keys.length
+      keys.first(len - 1).zip(keys.last(len - 1))
+    end
+
+    def compare_two_revs(low, high)
+      first = download_revision_as_txt(low)
+      second = download_revision_as_txt(high)
+      Differ.diff_by_word(first, second)
+    end
+
+    def print_summary_of_changes(changes)
+      puts "#{changes.change_count} words changed, ".colorize(:green) +
+      "#{changes.insert_count} inserts, ".colorize(:green) +
+      "#{changes.delete_count} deletes.".colorize(:green)
+    end
+
+    def compare_and_print_change_count(low, high)
+      changes = compare_two_revs(low, high)
+      print_summary_of_changes(changes)
+    end
+
+    def puts_diff_note
+      puts "Note: 'ab' -> 'ac' counts as both an insert and".colorize(:green) +
+      ' a delete but counts as only one change.'.colorize(:green)
+    end
+
+    def diff_author(pair)
+      "From rev #{pair[0]} to rev #{pair[1]} ".colorize(:green) +
+      "modified by #{modifying_users[pair[0]]}".colorize(:green)
+    end
+
+    def execute
+      puts_diff_note
+      if @all
+        consecutive_revisions.each do |pair|
+          puts diff_author(pair)
+          compare_and_print_change_count(pair[0], pair[1])
+        end
+      else
+        compare_and_print_change_count(@low_rev, @high_rev)
+      end
+    end
+  end
+end