reivt 1.1.0

data/lib/revit/cli.rb

@@ -0,0 +1,158 @@
+# rubocop: disable Metrics/AbcSize, Metrics/MethodLength, Metrics/ClassLength
+require 'paint'
+require 'thor'
+require 'tty-spinner'
+
+#
+# An extension of our main module
+#
+# @author [brwnrclse]
+#
+module Revit
+  #
+  # Rev's cli
+  #
+  # @author [brwnrclse]
+  #
+  class CLI < Thor
+    namespace :revit
+
+    map %w(--version -v) => :__print_version
+    desc '--version, -v', 'print the version'
+    def __print_version
+      puts VERSION
+    end
+
+    method_option :description, type: :string, aliases: '-d', lazy_default: '',
+                                desc: 'A short summary of the changes'
+    method_option :list_files, type: :array, aliases: '-l',
+                               desc: 'A list of files to add to the rev'
+    method_option :title, type: :string, aliases: '-t', lazy_default: '',
+                          desc: 'A title for the rev'
+    desc 'create PATH', 'Make a new rev object from the parameters'
+    long_desc <<-LONGDESC
+      `Initiates a flow for creating a new rev. To bypass the prompts users can use the provided flags.
+      If using git the title, description and files will be retrieved from git via the git repo name, latest commit msg and latest commit file list respectively.
+
+      `revit create ./Bucket/repo` # Create a new rev from the provided path
+      `revit create ./Bucket/repo --title, -t "Test Repo"` # Create a new revo with a title of Test Repo
+      `revit create ./Bucket/repo --description, -d "A test repo."` # Create a new revo with a description of "a test repo"
+      `revit create ./Bucket/repo --list_files, -l ~/Bucket/oneoffs/help.js` # Create a new rev with a list of files, ignoring the current folder
+    LONGDESC
+    def create(path)
+      doc_set = Set.new
+      doc_ids = Set.new
+      paths = path.split
+      paths += options[:list_files] if options[:list_files]
+      rev_id = ''
+      spinner = TTY::Spinner.new('[:spinner] :msg', format: :bouncing_ball)
+
+      spinner.update(msg: 'Creating a rev...')
+      spinner.run do
+        rev_id = RevAPI.create_rev(options[:title], options[:description])
+      end
+      spinner.success(Paint['rev created', :green])
+
+      paths.each do |x|
+        type = File.exist?(x) ? File.ftype(x) : 'other'
+
+        if type == 'file'
+          spinner.update(msg: "#{Paint['Adding', nil, '#e3b505']} #{x}")
+          spinner.run do
+            doc_set.add(Util.doc_from_path(x))
+          end
+          spinner.success(Paint['document created', :green])
+        elsif type == 'directory'
+          spinner.update(msg: "#{Paint['Adding', nil, '#e3b505']} docs from " \
+                         "#{Paint[path, '#2de1fc']}")
+
+          if Dir.entries(x).include?('.git')
+            spinner.run do
+              doc_set.merge(Util.docs_from_repo(x, spinner))
+            end
+            spinner.success(Paint['repo docs created', :green])
+          else
+            spinner.run do
+              doc_set.merge(Util.docs_from_dir(x))
+            end
+            spinner.success(Paint['dir docs created', :green])
+          end
+        else
+          spinner.error("#{Paint['Unsupported file type:']} #{path}")
+        end
+      end
+
+      spinner.update(msg: 'Uploading docs to rev api')
+      spinner.run do
+        doc_set.each do |doc|
+          doc_id = RevAPI.create_doc(doc.blob, doc.content_type, doc.doc_name,
+                                     doc.has_diff, rev_id)
+          doc_ids.add(doc_id)
+        end
+      end
+      spinner.success(Paint['docs uploaded to api', :green])
+
+      puts "\nLogin at #{Paint['wver.vaemoi.co/', :green]} to start your rev!\n"
+    rescue Errno::ECONNRESET, Errno::EINVAL, EOFError, Net::HTTPBadResponse,
+           Net::HTTPHeaderSyntaxError, Net::OpenTimeout, Net::ProtocolError,
+           Revit::BaemptyException, Revit::GraphQLDataException,
+           Revit::GraphQLValidationException => e
+
+      Revit::DEVLOGGER.error(e.message)
+      spinner.error(e.message)
+
+      doc_ids.each do |id|
+        RevAPI.delete_doc(id)
+      end
+
+      RevAPI.delete_rev(rev_id)
+    end
+
+    desc 'login', 'Get an auth token for accessing the rev api'
+    long_desc <<-LONGDESC
+      Sends a request for a valid auth token (jwt) to use for accessing the API.
+      Refresh is handled automatically as along as the user keeps on using rev otherwise it'll expire after about a week.
+
+      `rev login --username brwnrclse`
+
+      `rev login -u brwnrclse`
+    LONGDESC
+    def login
+      puts "Login here for your token:\n\n"
+      puts "#{Paint[Auth.auth_code_url, :green]}\n\n"
+
+      auth_code = Thor::Shell::Basic.new.ask('Enter your auth code => ')
+      spinner = TTY::Spinner.new('[:spinner] :msg', format: :bouncing_ball)
+
+      spinner.update(msg: 'Logging in')
+      spinner.run do
+        begin
+          auth_token = Auth.auth_token(auth_code)
+          user_id = RevAPI.signin_user(auth_token[:auth0_id])
+
+          if user_id.nil?
+            spinner.update(msg: 'User not found! Creating...')
+            user_id = RevAPI.create_user(auth_token[:auth0_id])
+            spinner.success(Paint['User created', :green])
+          end
+
+          Revit::REVIT_STORE.transaction do |store|
+            store[:access_token] = auth_token[:access_token].strip
+            store[:expires] = auth_token[:expires]
+            store[:auth0_id] = auth_token[:auth0_id].strip
+            store[:user_id] = user_id
+          end
+          spinner.success(Paint['Login successful :)', :green])
+        rescue Errno::ECONNRESET, Errno::EINVAL, EOFError,
+               Net::HTTPBadResponse, Net::HTTPHeaderSyntaxError,
+               Net::OpenTimeout, Net::ProtocolError,
+               Revit::GraphQLDataException,
+               Revit::GraphQLValidationException => e
+
+          Revit::DEVLOGGER.error(e.message)
+          spinner.error(Paint[e.message, :red])
+        end
+      end
+    end
+  end
+end

data/lib/revit/document.rb

@@ -0,0 +1,67 @@
+# An extension of our main module
+#
+# @author [brwnrclse]
+#
+module Revit
+  # Blueprint for Document objects
+  #
+  # @author [brwnrclse]
+  #
+  # @!attribute [rw] blob
+  #   @return [String] The contents of the file
+  #
+  # @!attribute [rw] content_type
+  #   @return [String] The type of content within the file
+  #
+  # @!attribute [rw] doc_name
+  #   @return [Array<DocumentClass>] The name of the file
+  #
+  # @!attribute [rw] has_diff
+  #   @return [Boolean] Whether or not the file will have a diff
+  #
+  class Document
+    attr_accessor(:blob, :content_type, :doc_name, :name, :has_diff)
+
+    def initialize(blob, content_type, doc_name, has_diff = false)
+      @blob = blob
+      @content_type = content_type
+      @has_diff = has_diff
+      @doc_name = doc_name
+    end
+
+    # Allows Documents objects to be compared using .eql? using their contents
+    #       and file name.
+    #
+    # @param other [Rev::Document] The Document to compare to
+    #
+    # @return [Boolean] True if equal
+    #                False if not equa
+    #
+    def eql?(other)
+      self == other
+    end
+
+    # Allows Documents objects to be compared using hashes created from a
+    #       logical and (&) between the files contents and name.
+    #
+    # @param other [Rev::Document] The Document to compare to
+    #
+    # @return [Fixnum] The & between file content and name
+    #
+    def hash
+      @blob.hash & @doc_name.hash
+    end
+
+    # Allows Documents objects to be compared using == using their contents
+    #       and file name.
+    #
+    # @param other [Rev::Document] The Document to compare to
+    #
+    # @return [Boolean] True if equal
+    #                   False if not equa
+    #
+    def ==(other)
+      @blob == other.blob && @doc_name == other.doc_name
+    end
+  end
+end

data/lib/revit/exception.rb

@@ -0,0 +1,35 @@
+# An extension of our main module
+#
+# @author [brwnrclse]
+#
+module Revit
+  # Custom errors for dealing with repos in revit
+  #
+  # @author [bwrnrclse]
+  #
+  class BaemptyException < IOError
+    def initialize(msg = '')
+      super(msg)
+    end
+  end
+
+  # Custom errors for dealing with graphql in revit
+  #
+  # @author [bwrnrclse]
+  #
+  class GraphQLDataException < IOError
+    def initialize(msg = '')
+      super(msg)
+    end
+  end
+
+  # Custom errors for dealing with graphql in revit
+  #
+  # @author [bwrnrclse]
+  #
+  class GraphQLValidationException < IOError
+    def initialize(msg = '')
+      super(msg)
+    end
+  end
+end