castic 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 802ec01397ff02f562d4f962018aa246549b6cbc
-  data.tar.gz: 61d64a07cc33dda23eb69d621cf3e2de404bb737
+  metadata.gz: 71a3fd9478158ff682fc04943f29e3b307a702e9
+  data.tar.gz: 9fbcb2ed1d6895796e357ed6c7f3873a76efac78
 SHA512:
-  metadata.gz: 80b0e4a52ffd0b531e2f9368280b1e5d065002421fb80df3a3f5964220cb0e2be806b1453a621a955fd4631b9604462895fa6b97c90dae8306aec923877f6c14
-  data.tar.gz: 18089ba17418af4a5057a8027bc951baae417e134dff02ef59fcf5c975db33aa82c63af1e614bc4ee219e1e223ea4df032bc41dbccae245b77822611c5370ab3
+  metadata.gz: b101499421df6885294bd5301228ebbc5dda253b68632dd90531db30659718c5e2d3c66f8af80014e82c6338991ae81105c6d0bb9476e026a22160ffe6bb33c7
+  data.tar.gz: 01cdc4de592269b46b4c2646bfbddad5d891fec80b5d671f98913b3be89660e597ab3a27c954a5834c9023126f42fc1e91d0b85be80a1943647975af988bdcce

data/lib/castic.rb

@@ -2,14 +2,16 @@ require 'parser/ruby18'
 require_relative 'extends/parser'
 
 class Castic
-  def initialize(cask)
-    @file = cask
-    @source = IO.read cask
+  def initialize(file, type = :Cask)
+    @type = type
+    @file = file
+    @source = IO.read file
     @parsed = Parser::Ruby18.parse @source
     @tree = @parsed.to_tree
   end
 
   attr_reader :tree
+  attr :type
 
   def expected_name
     @file.to_s
@@ -21,14 +23,29 @@ class Castic
       .sub /\.rb$/, ''
   end
 
+  def class_index
+    if @type == :Formula
+      return @tree[1].index [nil, @type]
+    end
+
+    @tree.index [nil, @type]
+  end
+
   def name
-    k = @tree.index [nil, :Cask]
-    @tree[k-1][1].to_s
+    if @type == :Formula
+      @tree[1]
+    else
+      @tree
+    end[class_index-1][1].to_s
   end
 
   def props(key = nil)
-    k = @tree.index [nil, :Cask]
-    body = @tree[k+1]
+    body = if @type == :Formula
+      @tree[1][class_index+1]
+    else
+      @tree[class_index+1]
+    end
+
     if key
       body.select { |n|
         n[1] == key.to_sym

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: castic
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Felix Saparelli