ki 0.0.2 → 0.1.0

data/lib/modules/callbacks.rb

@@ -0,0 +1,27 @@
+module Ki
+  module Callbacks
+    def before_find
+    end
+
+    def after_find
+    end
+
+    def before_create
+    end
+
+    def after_create
+    end
+
+    def before_update
+    end
+
+    def after_update
+    end
+
+    def before_delete
+    end
+
+    def after_delete
+    end
+  end
+end

data/lib/modules/query_interface.rb

@@ -0,0 +1,29 @@
+module Ki
+  module QueryInterface 
+    def count
+      Db.instance.db[class_name].count
+    end
+
+    def find id
+      Db.instance.find class_name, id
+    end
+
+    def create hash
+      Db.instance.create class_name, hash
+    end
+
+    def update hash
+      Db.instance.update class_name, hash
+    end
+
+    def delete id
+      Db.instance.delete class_name, id
+    end
+
+    private
+
+    def class_name
+      self.to_s.downcase
+    end
+  end
+end

data/lib/modules/restrictions.rb

@@ -0,0 +1,62 @@
+module Ki
+  module Restrictions
+    def forbidden_actions
+      []
+    end
+
+    def forbid *actions
+      send :define_method, :forbidden_actions do
+        actions
+      end
+
+      eigen_class = class << self; self; end
+      eigen_class.send(:define_method, :forbidden_actions) do
+        actions
+      end
+    end
+
+    def allowed_formats
+      [:json, :html]
+    end
+
+    def respond_to *formats
+      send :define_method, :allowed_formats do
+        formats
+      end
+
+      eigen_class = class << self; self; end
+      eigen_class.send(:define_method, :allowed_formats) do
+        formats
+      end
+    end
+
+    def required_attributes
+      []
+    end
+
+    def requires *attributes
+      send :define_method, :required_attributes do
+        attributes
+      end
+
+      eigen_class = class << self; self; end
+      eigen_class.send(:define_method, :required_attributes) do
+        attributes
+      end
+    end
+  end
+
+  module InitRestrictions
+    def forbidden_actions
+      []
+    end
+
+    def allowed_formats
+      [:json, :html]
+    end
+
+    def required_attributes
+      []
+    end
+  end
+end

data/lib/req.rb

@@ -0,0 +1,30 @@
+module Ki
+  class Req < Rack::Request
+    attr_accessor :code, :klass, :format
+
+    def initialize env
+      super env
+    end
+
+    def serve
+      unless static_path_requested?
+        route
+        Resp.new(self).finish
+      else
+        StaticFile.new(self).finish
+      end
+    end
+
+    def route
+      @klass = Util.base(path).to_class
+      @format = Util.format(path)
+      @code = klass || path == '/' ? 200 : 404
+    end
+
+    private
+
+    def static_path_requested?
+      path.start_with? '/public/'
+    end
+  end
+end

data/lib/resp.rb

@@ -0,0 +1,50 @@
+require 'json'
+
+module Ki
+  class Resp < Rack::Response
+    include Helpers
+
+    attr_accessor :req
+
+    def initialize req
+      @req = req
+      headers = {
+        'Content-Type' => respond_to(@req.format)
+      }
+
+      klass = @req.klass
+      if klass.allowed_formats.include? @req.format
+        if @req.code == 404
+          body = @req.format == :html ? partial('404') : { :error => 404 }
+        else
+          if @req.path == '/'
+            body = @req.format == :html ? partial('index') : { :index => 200 }
+          else
+            klass_view = Util.root_view_path(klass.to_s.downcase)
+            if @req.format == :html
+              body = File.exists?(klass_view) ? partial(klass.to_s.downcase) : partial('index')
+            else
+              begin
+                body = klass.new(@req).to_json
+              rescue Exception => e
+                @req.code = 400
+                body = { :error => e.to_s }.to_json
+              end
+            end
+          end
+        end
+      else
+        @req.code = 406
+        body = @req.format == :html ? partial('406') : { :error => 'format not allowed' }.to_json
+      end
+
+      super [body], @req.code, headers
+    end
+
+    private
+
+    def respond_to resp
+      resp == :html ? 'text/html' : 'application/json'
+    end
+  end
+end

data/lib/static_file.rb

@@ -0,0 +1,14 @@
+module Ki
+  class StaticFile < Rack::File
+    attr_reader :env
+
+    def initialize req
+      super Util.root
+      @env = req.env
+    end
+
+    def finish
+      call(@env)
+    end
+  end
+end

data/lib/util.rb

@@ -1,3 +1,9 @@
+class NilClass
+  def allowed_formats
+    [:json, :html]
+  end
+end
+
 class String
   def to_class
     chain = self.split "::"
@@ -13,22 +19,6 @@ end
 
 module Ki
   class Util
-    # the order in which files are loaded is important
-    def self.static_files
-      static_files = Dir.glob(File.join(Ki.root, 'lib/public/*'))
-      static_files << Dir.glob(File.join(Dir.pwd, 'public/*'))
-      static_files.flatten!
-    end
-
-    def self.partial s, hash={}
-      path = File.join(Dir.pwd, 'views', s) + '.haml'
-      unless File.exists? path
-        path = File.join(Ki.root, 'lib', 'views', s) + '.haml'
-      end
-      contents = File.read(path)
-      Haml::Engine.new(contents).render(Object.new(), hash)
-    end
-
     def self.base s
       if s == '/'
         return ''
@@ -52,19 +42,8 @@ module Ki
       end
     end
 
-    def self.config_db
-      config_path = File.join(Dir.pwd, 'config.yml')
-      yaml = {}
-      if File.exists? config_path
-        yaml = YAML.load_file(config_path)
-      else
-        yaml = {
-          'host' => '127.0.0.1',
-          'port' => 27017,
-          'db_name' => 'ki'
-        }
-      end
-      Db.instance.config yaml['host'], yaml['port'], yaml['db_name']
+    def self.root_view_path name
+      File.join(root, 'views', name) + '.haml'
     end
 
     def self.format s
@@ -74,10 +53,22 @@ module Ki
 
       base = s.split('/')[1]
       if base.include? '.'
-        base.split('.')[1].split('?')[0].to_sym
+        base.split('.')[1].split('?')[0].to_sym rescue :html
       else
         :html
       end
     end
+
+    def self.src_path
+      File.join(File.expand_path(File.join(File.dirname(__FILE__), '..', 'lib')))
+    end
+
+    def self.root
+      Dir.getwd
+    end
+
+    def self.app_name
+      File.basename(root)
+    end
   end
 end

data/lib/views/404.haml

@@ -0,0 +1,7 @@
+!!!
+%html
+  %head
+    %title 404 - Not Found
+    %meta{"http-equiv" => "Content-Type", :content => "text/html; charset=utf-8"}
+  %body
+    %center 404 - Not Found

data/lib/views/406.haml

@@ -0,0 +1,7 @@
+!!!
+%html
+  %head
+    %title 406 - Not Acceptable Format
+    %meta{"http-equiv" => "Content-Type", :content => "text/html; charset=utf-8"}
+  %body
+    %center 406 - Not Acceptable Format

data/lib/views/index.haml

@@ -1,19 +1,7 @@
-!!! 5
-%html{:lang => "en"}
+!!!
+%html
   %head
-    %meta{:charset => "utf-8"}
-    %title= "#{Ki::Ki.app_name} - #{req.klass.to_s}"
-    %meta{:content => "width=device-width, initial-scale=1.0", :name => "viewport"}
-    %meta{:content => "", :name => "description"}
-    %meta{:content => "", :name => "author"}
-
-    %link{:href => "/public/bootstrap.css", :rel => "stylesheet"}
-    %link{:href => "/public/ki.css", :rel => "stylesheet"}
-    %link{:href => "/public/custom.css", :rel => "stylesheet"}
+    %title 200 - Hello World
+    %meta{"http-equiv" => "Content-Type", :content => "text/html; charset=utf-8"}
   %body
-
-    = Ki::Util.partial 'container', locals
-
-    %script{:src => "/public/jquery-1.8.2.min.js"}
-    %script{:src => "/public/bootstrap.js"}
-    %script{:src => "/public/custom.js"}
+    %center 200 - Hello World

data/spec/db_spec.rb

@@ -0,0 +1,71 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe Ki::Db do
+  before :each do
+    @db = Ki::Db.instance
+    @db.remove_collections
+  end
+
+  it "should start with an empty database" do
+    @db.collection_names.empty?.should be_true
+  end
+  
+  it "should create a collection" do
+    @db.collection_names.empty?.should be_true
+    @db.create 'zoidberg', { :foo => 'bar' }
+    @db.collection_names.empty?.should be_false
+    @db.collection_names.include?('zoidberg').should be_true
+  end
+
+  it "should create an model" do
+    @db.db['zoidberg'].count.should == 0
+    @db.create 'zoidberg', { :foo => [1,2] }
+    @db.db['zoidberg'].count.should == 1
+  end
+
+  it "should return doc id when creating a model" do
+    post_id = @db.create 'zoidberg', { :foo => [1,2] }
+    post_id.class.should == String
+    bson_id = BSON::ObjectId.from_string(post_id)
+    @db.db['zoidberg'].find('_id' => bson_id).first['foo'].should == [1,2]
+  end
+
+  it "should find a specific document" do
+    post_id = @db.create 'zoidberg', { :foo => [1,2] }
+    @db.find('zoidberg', post_id)['foo'].should == [1,2]
+  end
+
+  it "should update a document" do
+    post_id = @db.create 'zoidberg', { :foo => [1,2], :bar => 'highj' }
+    @db.update 'zoidberg', { '_id' => post_id, :foo => 3.1415 }
+    doc = @db.find('zoidberg', post_id)
+    doc['foo'].should == 3.1415
+    doc['bar'].should == nil
+  end
+
+  it "should delete a document" do
+    post_id = @db.create 'zoidberg', { :foo => [1,2], :bar => 'highj' }
+    @db.db['zoidberg'].count.should == 1
+    @db.delete('zoidberg', post_id)
+    @db.db['zoidberg'].count.should == 0
+  end
+
+  it "should find all documents in collection" do
+    @db.create 'zoidberg', { :foo => [1,2], :bar => 'highj' }
+    @db.create 'zoidberg', { :foo => [3,4], :bar => 'lowj' }
+    @db.find_all('zoidberg').count.should == @db.db['zoidberg'].count
+  end
+
+  it "should find by an attribute" do
+    @db.create 'zoidberg', { :foo => [1,2], :bar => 'highj' }
+    @db.create 'zoidberg', { :foo => [3,4], :bar => 'lowj' }
+    @db.create 'zoidberg', { :foo => [5,6], :bar => 'lowj' }
+    r = @db.find_by 'zoidberg', { :bar => 'lowj' }
+    r.class.should == Array
+    r.length.should == 2
+
+    r = @db.find_by 'zoidberg', { :bar => 'lowj', :foo => [3,4] }
+    r.class.should == Array
+    r.length.should == 1
+  end
+end

data/spec/ki_spec.rb

@@ -1,7 +1,7 @@
 require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
 
-describe "Ki" do
-  it "works" do
+describe Ki do
+  it "should work" do
     1.should == 1
   end
 end

data/spec/model_spec.rb

@@ -0,0 +1,82 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe Ki::Model do
+  before :each do
+    class Zoidberg < Ki::Model
+    end
+    @db = Ki::Db.instance
+    @db.remove_collections
+  end
+
+  it "should correctly extract class_name" do
+    # Zoidberg.new(@get).class_name.should == "zoidberg"
+  end
+
+  it "should create a new item in the databse" do
+    post = ReqFactory.new(:post_homer)
+    @db.db['zoidberg'].count.should == 0
+    hash = Zoidberg.new(post).hash
+    @db.db['zoidberg'].count.should == 1
+    @db.find('zoidberg', hash['_id'])['user'].should == 'homer'
+  end
+
+  it "should get a model depending on id" do
+    id = @db.create 'zoidberg', { :name => 'homer' }
+
+    get = ReqFactory.new(:get, {'_id' => id})
+    Zoidberg.new(get).hash['name'].should == 'homer'
+  end
+
+  it "should raise error if no params are given when getting model" do
+    expect {
+      Zoidberg.new(ReqFactory.new(:get))
+    }.to raise_error
+  end
+
+  it "should raise error if no id is given when getting model" do
+    expect {
+      Zoidberg.new(ReqFactory.new(:get_with_params, { :name => 'homer' }))
+    }.to raise_error
+  end
+
+  it "should only accept valid object ids" do
+    expect {
+      Zoidberg.new(ReqFactory.new(:get_with_params, { :_id => 'homer' }))
+    }.to raise_error
+  end
+
+  it "should update a document" do
+    id = @db.create 'zoidberg', { :name => 'homer' }
+    @db.find('zoidberg', id)['name'].should == 'homer'
+
+    Zoidberg.new(ReqFactory.new(:put, {'_id' => id, :name => 'bart'}))
+    @db.find('zoidberg', id)['name'].should == 'bart'
+  end
+
+  it "should raise error if id is missing when updating" do
+    expect {
+      Zoidberg.new(ReqFactory.new(:put, {:name => 'bart'}))
+    }.to raise_error
+  end
+
+  it "should raise error if id is invalid" do
+    expect {
+      Zoidberg.new(ReqFactory.new(:put, {'_id' => 'f*ck', :name => 'bart'}))
+    }.to raise_error
+  end
+
+  it "should delete a document" do
+    expect {
+      Zoidberg.new(ReqFactory.new(:delete))
+    }.to raise_error
+
+    expect {
+      Zoidberg.new(ReqFactory.new(:delete, {'_id' => 'f*ck', :name => 'bart'}))
+    }.to raise_error
+
+    id = @db.create 'zoidberg', { :name => 'homer' }
+    expect {
+      Zoidberg.new(ReqFactory.new(:delete, {'_id' => id}))
+    }.to change(@db.db['zoidberg'], :count).by(-1)
+  end
+end