trains 0.0.5 → 0.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e4c83b37c56def34baa1bd60260cecec154d1da870f2b43471994f60c2a7397e
-  data.tar.gz: 3075d23e3f460a9787a0c1184c05ece0bbe21ca8d676bfa8828bb8b5cd91c1ac
+  metadata.gz: 2b1af8d7a53681c21f561432534383da1d83bf4615bc8e85bdfa6c8ce7b210aa
+  data.tar.gz: aa4798c33b271c590c5b966078eb7a6f9fd603a069842ce454cadb14de8cc08d
 SHA512:
-  metadata.gz: 47357749dc2a2effab891336c1c0d2e338474288b720dd01eac00173adba3b9ce41f5bbd06bafec55b2319e25b2faa128aab9805c4251009de0ac3617661ac2c
-  data.tar.gz: 7dd8d5c09dd6a6e84da47a74f805dc9b26cfad3943f98d433668bb2f8d4adb1d4f08a5607355e1c2e9a25cee91e59ff4b3016f087dad05f9a86e57fbfe01c7cc
+  metadata.gz: f6faddb254499e05f02a13def98dc7ab1f7df69d2ff694617cbd24410e2cb23266c7fd4c067cb2e4e2dd4fd7e7552f7d5ba79c995f2f951ff452150b8d800222
+  data.tar.gz: 95748dab8e2f16aed324d9c55a8d14c0eeec265361d49794bf7d81f21f6b42b8a25ba7426f719fea673efc982864ff49c745d3cdeb6888780b63a3bdf81038b9

data/lib/trains/dto/callback.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+module Trains
+  module DTO
+    Callback = Data.define(:method, :arguments)
+  end
+end

data/lib/trains/dto/controller.rb

@@ -2,6 +2,6 @@ require "set"
 
 module Trains
   module DTO
-    Controller = Data.define(:name, :method_list)
+    Controller = Data.define(:name, :method_list, :callbacks)
   end
 end

data/lib/trains/utils/args.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+module Trains
+  module Utils
+    # utility module to deal with parsing of arguments
+    module Args
+      def parse_args(node)
+        case node.type
+        when :hash
+          parse_hash(node)
+        else
+          parse_value(node)
+        end
+      end
+
+      def parse_hash(node)
+        options = {}
+        return options unless node.type == :hash
+
+        node.each_pair { |key, value| options[key.value] = parse_value(value) }
+      rescue StandardError => e
+        puts node.parent
+      ensure
+        return options
+      end
+
+      def parse_value(node)
+        case node.type
+        when :hash
+          parse_hash(node)
+        when :array
+          node.values.map { |value| parse_value(value) }
+        when :send
+          if node.method_name == :redirect
+            { redirect: node.arguments.first.value }
+          end
+        else
+          node.value
+        end
+      end
+    end
+  end
+end

data/lib/trains/version.rb

@@ -1,3 +1,3 @@
 module Trains
-  VERSION = '0.0.5'
+  VERSION = '0.0.6'
 end

data/lib/trains/visitor/controller.rb

@@ -1,14 +1,24 @@
-require_relative "../dto/controller"
-require_relative "../dto/method"
+# frozen_string_literal: true
+
+require_relative '../dto/callback'
+require_relative '../dto/controller'
+require_relative '../dto/method'
 
 module Trains
   module Visitor
     # Visitor that parses controllers and returns a DTO::Controller object
     class Controller < Base
+      include Utils::Args
+
+      TRACKED_CALLBACKS = [
+        :http_basic_authenticate_with
+      ].freeze
+
       def initialize
         @method_list = []
         @methods = {}
         @class_name = nil
+        @callbacks = []
       end
 
       def on_class(node)
@@ -17,15 +27,25 @@ module Trains
         return unless controller?(parent_class)
 
         @class_name = class_name
+        find_callbacks(node)
         parse_body(node.body) unless node.body.nil?
       end
 
       def result
-        DTO::Controller.new(name: @class_name, method_list: @method_list.uniq)
+        DTO::Controller.new(name: @class_name, method_list: @method_list.uniq, callbacks: @callbacks)
       end
 
       private
 
+      def find_callbacks(klass_node)
+        klass_node.each_descendant(:send) do |node|
+          if node.receiver.nil? && TRACKED_CALLBACKS.include?(node.method_name)
+            @callbacks << DTO::Callback.new(method: node.method_name,
+                                            arguments: node.arguments.map { |arg| parse_args(arg) })
+          end
+        end
+      end
+
       def parse_body(body)
         body.each_child_node do |child|
           @method_list << parse_method(child) if child.type == :def

data/lib/trains/visitor/route.rb

@@ -1,8 +1,11 @@
-require_relative "../dto/route"
+require_relative '../dto/route'
 
 module Trains
   module Visitor
+    # Visitor for Parsing Rails routes
     class Route < Base
+      include Utils::Args
+
       def_node_matcher :route_parent?, <<~PATTERN
         (block (send (send (send
           (const nil? :Rails) :application) :routes) :draw)
@@ -13,7 +16,7 @@ module Trains
         (send nil? %1 ...)
       PATTERN
 
-      ALLOWED_VERBS = %i[get put post update delete resources scope]
+      ALLOWED_VERBS = %i[get put post update delete resources scope].freeze
 
       def initialize
         @route_list = []
@@ -46,33 +49,7 @@ module Trains
             node.arguments[0].value
           end
         options = parse_hash(node.arguments[1])
-        DTO::Route.new(method: method, param: param, options: options)
-      end
-
-      def parse_hash(node)
-        options = {}
-        return options unless node.type == :hash
-
-        node.each_pair { |key, value| options[key.value] = parse_value(value) }
-      rescue StandardError => e
-        puts node.parent
-      ensure
-        return options
-      end
-
-      def parse_value(node)
-        case node.type
-        when :hash
-          parse_hash(node)
-        when :array
-          node.values.map { |value| parse_value(value) }
-        when :send
-          if node.method_name == :redirect
-            { redirect: node.arguments.first.value }
-          end
-        else
-          node.value
-        end
+        DTO::Route.new(method:, param:, options:)
       end
     end
   end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: trains
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.0.6
 platform: ruby
 authors:
 - Syed Faraaz Ahmad
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-04-21 00:00:00.000000000 Z
+date: 2023-05-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -85,6 +85,7 @@ files:
 - Rakefile
 - lib/trains.rb
 - lib/trains/dto/app.rb
+- lib/trains/dto/callback.rb
 - lib/trains/dto/controller.rb
 - lib/trains/dto/field.rb
 - lib/trains/dto/method.rb
@@ -92,6 +93,7 @@ files:
 - lib/trains/dto/model.rb
 - lib/trains/dto/route.rb
 - lib/trains/scanner.rb
+- lib/trains/utils/args.rb
 - lib/trains/utils/logger.rb
 - lib/trains/utils/migration_tailor.rb
 - lib/trains/utils/rails_dir.rb