treeify 0.02 → 0.03

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4c05d601b25f5e493f430b1ae5354410f237ae90
-  data.tar.gz: 28c555a605c4b17e542127398ea4feee6da00216
+  metadata.gz: 422fe3f0d572c4b67db08dd518122b2d0a04262f
+  data.tar.gz: bb8044d5cc0267c3f80ad7709f337bd04b826218
 SHA512:
-  metadata.gz: eabf4eca128f87d7c3034021e40ddbb9ba752ee826be1a73a53552079a4dbac4516cd2df0cc60c96688f4de0030fe7002f1e3abd2b7211e17cf78d4bc318e786
-  data.tar.gz: f2653463b3f9d161136447b3f827ff9105a0cb90e1cde341defe34c93028214323791c8ec272c87cd9e771e6c45b9e0d75e5c08a94fafc1ab262a86386f77208
+  metadata.gz: 7a88024c53a98ffd0ca89ac1d7ce643425ded66d3fda586b3655f94ab10367f82ea47c32bd8f6aac2726b46e60c461aa4ed120cacdf491cee3f83471077333be
+  data.tar.gz: bd3020aa3d4f182547acd64ba07cd91e54f587ad1fae9655b4dea19f938b85fd16d8ed76c0a70172b4abc884c6a2a78d49152f79c9cd699353186ca9442dcfb0

data/lib/treeify.rb

@@ -7,6 +7,12 @@ module Treeify
   extend ActiveSupport::Concern 
 
   included do
+    has_many :children,
+             class_name: self,
+             foreign_key: "parent_id"
+    belongs_to :parent,
+                class_name: self,
+                foreign_key: "parent_id"
     class_attribute :cols
     scope :roots, -> { where(parent_id: nil) }
     scope :tree_for, ->(instance) { where("#{table_name}.id IN (#{tree_sql_for(instance)})").order("#{table_name}.id") }
@@ -43,7 +49,19 @@ module Treeify
     end
 
     def tree_sql_for_ancestors(instance)
-      "#{tree_sql(instance)}
+      "WITH RECURSIVE cte (id, path)  AS (
+         SELECT  id,
+           array[id] AS path
+         FROM    #{table_name}
+         WHERE   id = #{instance.id}
+         
+         UNION ALL
+
+         SELECT  #{table_name}.id,
+            cte.path || #{table_name}.id
+         FROM    #{table_name}
+         JOIN cte ON #{table_name}.parent_id = cte.id
+       )
       SELECT cte.id FROM cte WHERE cte.id != #{instance.id}"
     end
   end
@@ -59,4 +77,38 @@ module Treeify
   def self_and_descendents
     self.class.tree_for(self)
   end
+
+  def is_root?
+    self.parent_id != nil
+  end
+
+  def siblings
+    self.class.where(parent_id: self.parent_id) - [self]
+  end
+
+  def descendent_tree
+    # give build_tree an array of hashes with the AR objects serialized into a hash
+    build_tree(descendents.to_a.map(&:serializable_hash))
+  end
+
+  def build_tree(data)
+    # turn our AoH into a hash where we've mapped the ID column
+    # to the rest of the hash + a comments array for nested comments
+    nested_hash = Hash[data.map{|e| [e['id'], e.merge('children' => [])]}]
+
+    # if we have a parent ID, grab all the comments
+    # associated with that parent and push them into the comments array
+    nested_hash.each do |id, item|
+      parent = nested_hash[item['parent_id']]
+      parent['children'] << item if parent
+    end
+
+    # return the values of our nested hash, ie our actual comment hash data
+    # reject any descendents whose parent ID already exists in the main hash so we don't
+    # get orphaned descendents listed as their own comment
+     
+    nested_hash.reject{|id, item| 
+      nested_hash.has_key? item['parent_id']
+    }.values
+  end
 end 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: treeify
 version: !ruby/object:Gem::Version
-  version: '0.02'
+  version: '0.03'
 platform: ruby
 authors:
 - Devin Austin
@@ -9,7 +9,77 @@ autorequire:
 bindir: bin
 cert_chain: []
 date: 2014-06-11 00:00:00.000000000 Z
-dependencies: []
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.17.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.17.1
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 4.1.6
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 4.1.6
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 3.1.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 3.1.0
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 10.3.2
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 10.3.2
+- !ruby/object:Gem::Dependency
+  name: database_cleaner
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.3.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.3.0
 description: Simple trees for ActiveRecord
 email: devin.austin@gmail.com
 executables: []
@@ -27,17 +97,17 @@ require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
-  - - '>='
+  - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
-  - - '>='
+  - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.0.7
+rubygems_version: 2.2.2
 signing_key: 
 specification_version: 4
 summary: Simple trees for ActiveRecord using PostgreSQL