simple_tree 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/.rspec

@@ -0,0 +1 @@
+--color

data/CHANGELOG.md

@@ -0,0 +1,3 @@
+## v0.0.1
+
+* Initial release

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in simple_tree.gemspec
+gemspec

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require 'rspec/core/rake_task'
+
+
+RSpec::Core::RakeTask.new(:spec)
+task default: :spec

data/lib/simple_tree.rb

@@ -0,0 +1,63 @@
+require "simple_tree/version"
+require "simple_tree/railtie" if defined? Rails
+
+module SimpleTree
+  def self.included(base)
+    base.extend(ClassMethods)
+  end
+  module ClassMethods
+    def acts_as_simple_tree(options = {})
+      configuration = { :foreign_key => "parent_id", :order => nil, :counter_cache => nil }
+      configuration.update(options) if options.is_a?(Hash)
+
+      belongs_to :parent, :class_name => name, :foreign_key => configuration[:foreign_key], :counter_cache => configuration[:counter_cache]
+      has_many :children, :class_name => name, :foreign_key => configuration[:foreign_key], :order => configuration[:order], :dependent => :destroy
+
+      self.send(:include, SimpleTree::InstanceMethods)
+    end
+  end
+  module InstanceMethods
+
+    # Returns list of ancestors, starting from parent until root.
+    #
+    #   subchild1.ancestors # => [child1, root]
+    def ancestors
+      node, nodes = self,[]
+      nodes << node = node.parent while node.parent
+      nodes
+    end
+
+    # Returns the root node of the tree.
+    def root
+      node = self
+      node = node.parent while node.parent
+      node
+    end
+
+    # Returns all siblings of the current node.
+    #
+    #   subchild1.siblings # => [subchild2]
+    def siblings
+      self_and_siblings - [self]
+    end
+
+    # Returns all siblings and a reference to the current node.
+    #
+    #   subchild1.self_and_siblings # => [subchild1, subchild2]
+    def self_and_siblings
+      parent ? parent.children : []
+    end
+
+    def descendants
+      return [] if children.empty?
+      nodes = children
+      children.each {|c| nodes + c.descendants}
+      nodes
+    end
+
+    def move_to_child_of(parent)
+      self.parent = parent
+    end
+
+  end
+end

data/lib/simple_tree/railtie.rb

@@ -0,0 +1,9 @@
+module SimpleTree
+  class Railtie < Rails::Railtie
+    initializer 'simple_tree' do
+      ActiveSupport.on_load :active_record do
+        extend SimpleTree::ClassMethods
+      end
+    end
+  end
+end

data/lib/simple_tree/version.rb

@@ -0,0 +1,3 @@
+module SimpleTree
+  VERSION = "0.0.1"
+end

data/simple_tree.gemspec

@@ -0,0 +1,27 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "simple_tree/version"
+
+Gem::Specification.new do |s|
+  s.name        = "simple_tree"
+  s.version     = SimpleTree::VERSION
+  s.authors     = ["yinghai"]
+  s.email       = ["yinghai.zhao@mosaic.com"]
+  s.homepage    = ""
+  s.summary     = %q{simple tree gem}
+  s.description = %q{simple tree gem}
+
+  s.rubyforge_project = "simple_tree"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # specify any dependencies here; for example:
+  s.add_development_dependency "rspec"
+  s.add_development_dependency 'with_model'
+  s.add_development_dependency 'rails', '~> 3.0.9'
+  s.add_development_dependency 'sqlite3'
+  # s.add_runtime_dependency "rest-client"
+end

data/spec/simple_tree_spec.rb

@@ -0,0 +1,43 @@
+require 'spec_helper'
+
+
+describe SimpleTree do 
+  with_model :Test do
+    table do |t|
+      t.integer :parent_id
+      t.string :name
+    end
+    model do
+      include SimpleTree
+      acts_as_simple_tree
+    end
+  end
+
+  before do
+    @root = Test.create :name => 'root'
+    @c1 = Test.create :name => 'child1', :parent => @root
+    @c2 = Test.create :name => 'child1', :parent => @root
+    @gc1a = Test.create :name => 'child1', :parent => @c1
+    @gc2a = Test.create :name => 'child1', :parent => @c2
+  end
+  
+  describe "#children" do
+    it "should return all children" do
+      @root.children.should == [@c1,@c2]
+    end
+  end
+
+  describe "#ancestors" do
+    it "should return all ancestors" do
+      @gc1a.ancestors.should == [@c1,@root]
+    end
+  end
+  describe "#root" do
+    it "should return root for descendants" do
+      @gc1a.root.should == @root
+    end
+    it "should return itself for root" do
+      @root.root.should == @root
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,9 @@
+require 'simple_tree'
+require 'active_record'
+require 'with_model'
+
+ActiveRecord::Base.establish_connection(:adapter => "sqlite3", :database => ":memory:")
+
+RSpec.configure do |config|
+  config.extend WithModel
+end

metadata

@@ -0,0 +1,102 @@
+--- !ruby/object:Gem::Specification
+name: simple_tree
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- yinghai
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-03-14 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &70330458747320 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70330458747320
+- !ruby/object:Gem::Dependency
+  name: with_model
+  requirement: &70330458746900 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70330458746900
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: &70330458746400 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 3.0.9
+  type: :development
+  prerelease: false
+  version_requirements: *70330458746400
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: &70330458745980 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70330458745980
+description: simple tree gem
+email:
+- yinghai.zhao@mosaic.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rspec
+- CHANGELOG.md
+- Gemfile
+- Rakefile
+- lib/simple_tree.rb
+- lib/simple_tree/railtie.rb
+- lib/simple_tree/version.rb
+- simple_tree.gemspec
+- spec/simple_tree_spec.rb
+- spec/spec_helper.rb
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: simple_tree
+rubygems_version: 1.8.17
+signing_key: 
+specification_version: 3
+summary: simple tree gem
+test_files:
+- spec/simple_tree_spec.rb
+- spec/spec_helper.rb