rbbt-rest 1.1.4 → 1.1.5

data/lib/rbbt/rest/entity/action_controller.rb

@@ -20,10 +20,15 @@ module EntityRESTHelpers
   def default_action_controller(entity, list_id = nil)
     action_controller = ActionController.new(entity, list_id)
 
-    if Array === entity
+    case
+    when Array === entity
       find_all_entity_list_action_templates(entity, true).each do |action|
         action_controller.add action, Misc.humanize(action, :format => :sentence), :reuse => true
       end
+    when TSV === entity
+      find_all_entity_map_action_templates(entity, true).each do |action|
+        action_controller.add action, Misc.humanize(action, :format => :sentence), :reuse => true
+      end
     else
       find_all_entity_action_templates(entity, true).each do |action|
         action_controller.add action, Misc.humanize(action, :format => :sentence)

data/lib/rbbt/rest/entity/entity_list_card.rb

@@ -2,7 +2,7 @@ class EntityListCard
 
   attr_accessor :list, :list_id
   attr_accessor :meta, :description
-  attr_accessor :list_container, :action_controller
+  attr_accessor :action_controller, :list_container
 
   def initialize(list, list_id)
     @list, @list_id = list, list_id

data/lib/rbbt/rest/entity/entity_map_card.rb

@@ -0,0 +1,15 @@
+class EntityMapCard
+
+  attr_accessor :map, :map_id
+  attr_accessor :action_controller
+
+  def initialize(map, map_id)
+    @map, @map_id = map, map_id
+  end
+end
+
+module EntityRESTHelpers
+  def entity_map_card_render(card)
+    partial_render('entity_partials/entity_map_card', :card => card)
+  end
+end

data/lib/rbbt/rest/entity/favourites.rb

@@ -71,7 +71,7 @@ module EntityRESTHelpers
     if (file = dir[entity_type]).exists?
       lists = Open.read(file).split("\n")
       lists << list
-      Open.write(file, lists * "\n")
+      Open.write(file, lists.uniq * "\n")
     else
       lists = [list]
       Open.write(file, lists * "\n")
@@ -94,4 +94,60 @@ module EntityRESTHelpers
     end
   end
 
+  #{{{ Entity Maps
+
+  def favourite_entity_maps
+    raise "You need to login to have favourites" unless authorized?
+
+    dir = Path.setup(File.join(settings.favourite_maps_dir, user))
+    favourites = {}
+    dir.find.glob('*').each do |type_dir|
+      type = File.basename(type_dir)
+      Path.setup(type_dir).glob('*').each do |file|
+        column = File.basename(file).gsub('--', '/')
+        maps = Open.read(file).split("\n")
+        favourites[type] ||= {}
+        favourites[type][column] = maps
+      end
+    end
+    favourites
+  end
+
+  def add_favourite_entity_map(entity_type, column, map)
+    raise "You need to login to have favourites" unless authorized?
+
+    column = column.gsub('/', '--')
+    dir = Path.setup(File.join(settings.favourite_maps_dir, user))
+
+    if (file = dir[entity_type][column]).exists?
+      maps = Open.read(file).split("\n")
+      maps << map
+      maps.uniq!
+      Open.write(file, maps.uniq * "\n")
+    else
+      maps = [map]
+      Open.write(file, maps * "\n")
+    end
+    ddd favourite_entity_maps
+  end
+
+  def remove_favourite_entity_map(entity_type, column, map)
+    raise "You need to login to have favourites" unless authorized?
+
+    column = column.gsub('/', '--')
+
+    dir = Path.setup(File.join(settings.favourite_maps_dir, user))
+
+    if (file = dir[entity_type][column]).exists?
+      maps = Open.read(file).split("\n")
+      maps -= [map]
+      if maps.any?
+        Open.write(file, maps * "\n")
+      else
+        FileUtils.rm file
+      end
+    end
+  end
+
+
 end

data/lib/rbbt/rest/entity/finder.rb

@@ -2,7 +2,7 @@ require 'rbbt/ner/finder'
 
 module EntityRESTHelpers
   def finder_find(term)
-    return [] unless settings.respond_to? :finder and not settings.finder.nil?
+    return [] unless settings.respond_to? :finder and not settings.finder.nil? and not settings.finder.instances.empty?
     results = settings.finder.find(term)
 
     results.uniq.collect{|r| 

data/lib/rbbt/rest/entity/helpers.rb

@@ -52,7 +52,7 @@ module EntityRESTHelpers
   end
 
   def page_type(path = nil)
-    path = request.path_info
+    path = request.path_info if path.nil?
 
     case
     when path.match(/^\/entity\//)
@@ -63,13 +63,17 @@ module EntityRESTHelpers
       return "entity_list"; 
     when path.match(/^\/entity_list_action\//)
       return "entity_list_action"; 
+    when path.match(/^\/entity_map\//)
+      return "entity_map"; 
+    when path.match(/^\/entity_map_action\//)
+      return "entity_map_action"; 
     else
       return nil;
     end
   end
 
  def page_entity(path = nil)
-   path = request.path_info
+   path = request.path_info if path.nil?
  
    case page_type
    when "entity"
@@ -82,7 +86,7 @@ module EntityRESTHelpers
  end
 
  def page_action(path = nil)
-   path = request.path_info
+   path = request.path_info if path.nil?
  
    case page_type
    when "entity_action", "entity_list_action"
@@ -92,9 +96,8 @@ module EntityRESTHelpers
    end
  end
  
-
  def page_entity_list(path = nil)
-   path = request.path_info
+   path = request.path_info if path.nil?
  
    case page_type
    when "entity_list"
@@ -105,17 +108,47 @@ module EntityRESTHelpers
      return nil
    end
  end
+
+ def page_entity_map(path = nil)
+   path = request.path_info if path.nil?
+ 
+   case page_type
+   when "entity_map"
+     return Entity::REST.restore_element(path.split("/")[4])
+   when "entity_map_action"
+     return Entity::REST.restore_element(path.split("/")[5])
+   else
+     return nil
+   end
+ end
  
  def page_entity_type(path = nil)
-   path = request.path_info
+   path = request.path_info if path.nil?
  
    case page_type
-   when "entity", "entity_list", "entity_action", "entity_list_action"
+   when "entity", "entity_list", "entity_action", "entity_list_action", "entity_map"
      return Entity::REST.restore_element(path.split("/")[2])
    else
      return nil
    end
  end
+
+ def page_entity_base_type(path = nil)
+   page_entity_type.split(":").first
+ end
+
+ def page_entity_map_column(path = nil)
+   path = request.path_info if path.nil?
+ 
+   case page_type
+   when  "entity_map"
+     return Entity::REST.restore_element(path.split("/")[3])
+   else
+     return nil
+   end
+ end
+
+
  #
  #
  #function page_entity_base_type(){

data/lib/rbbt/rest/entity/list.rb

@@ -2,9 +2,17 @@ require 'rbbt/entity'
 require 'cgi'
 
 module Entity
+
+  class << self
+    attr_accessor :entity_list_cache
+  end
+
+  self.entity_list_cache     = "var/entity_list"
+  
   module List
 
     def self.list_file(entity_type, id, user = nil)
+      id = Entity::REST.clean_element(id)
       id = Misc.sanitize_filename(id)
 
       entity_type = entity_type.split(":").first

data/lib/rbbt/rest/entity/locate.rb

@@ -278,4 +278,118 @@ module EntityRESTHelpers
 
     raise "Template not found for list #{ action } (#{list.annotation_types * ", "})"
   end
+
+  #{{{ ENTITY MAP
+
+  def locate_entity_map_template_from_resource(resource, type)
+    if type == "Default" 
+      path = resource.entity_map["Default.haml"]
+      if path.exists?
+        return path
+      else
+        return nil
+      end
+    end
+
+    path = resource.entity_map[type.to_s + ".haml"]
+    return path if path.exists?
+
+    nil
+  end   
+
+  def locate_entity_map_template(type, column)
+    entity_resources.each do |resource|
+      path = locate_entity_map_template_from_resource(resource, type)
+      return path if path and path.exists?
+    end
+
+    entity_resources.each do |resource|
+      path = locate_entity_map_template_from_resource(resource, "Default")
+      return path if path and path.exists?
+    end
+
+    raise "Template not found for list (#{list.annotation_types * ", "})"
+  end
+
+
+  #{{{ ENTITY MAP ACTION
+ 
+  def find_all_entity_map_action_templates_from_resource(resource, map)
+    field = map.key_field
+
+    if map.entity_templates[field] 
+      annotation_types = map.entity_templates[field].annotation_types
+    else
+      annotation_types = [Entity.formats[field]].compact
+    end
+
+    annotation_types += ["Default"]
+
+    annotation_types.collect do |annotation|
+      resource.entity_map[annotation].glob('*.haml')
+    end.compact.flatten
+  end   
+
+  def find_all_entity_map_action_templates(map, check = false)
+    paths = []
+
+    entity_resources.each do |resource|
+      paths.concat find_all_entity_map_action_templates_from_resource(resource, map)
+    end
+
+    if check
+      paths = paths.reject do |path|
+        check_file = path.sub(/\.haml$/, '.check')
+        case
+        when (path.basename == "edit.haml" or path.basename == 'new.haml')
+          true
+        when File.exists?(check_file)
+          begin
+            Log.debug("Checking action template: #{path}")
+            code = File.read(check_file)
+            accept = eval code
+            not accept
+          rescue
+            Log.debug("Error Checking action template #{path}: #{$!.message}")
+            true
+          end
+        else
+          false
+        end
+      end
+    end
+
+    paths.collect{|file| file.basename.sub('.haml', '') }.uniq
+  end
+
+ 
+  def locate_entity_map_action_template_from_resource(resource, map, action)
+    field = map.key_field
+
+    if map.entity_templates[field] 
+      annotation_types = map.entity_templates[field].annotation_types
+    else
+      annotation_types = [Entity.formats[field]].compact
+    end
+
+    annotation_types += ["Default"]
+
+    annotation_types.each do |annotation|
+      path = resource.entity_map[annotation][action.to_s + ".haml"]
+      return path if path.exists?
+    end
+
+    nil
+  end   
+
+  def locate_entity_map_action_template(map, action)
+
+    entity_resources.each do |resource|
+      path = locate_entity_map_action_template_from_resource(resource, map, action)
+      return path if path and path.exists?
+    end
+
+    raise "Template not found for map #{ action } (#{map.key_field * ", "})"
+  end
+
 end

data/lib/rbbt/rest/entity/map.rb

@@ -0,0 +1,99 @@
+require 'rbbt/entity'
+require 'cgi'
+
+module Entity
+
+  class << self
+    attr_accessor :entity_map_cache
+  end
+
+  self.entity_map_cache     = "var/entity_map"
+  
+
+  module Map
+    def self.map_file(entity_type, column, id, user = nil)
+      id = Misc.sanitize_filename(Entity::REST.clean_element(id))
+      column = Entity::REST.clean_element(column)
+
+      entity_type = entity_type.split(":").first
+
+      raise "Ilegal map id: #{ id }" unless Misc.path_relative_to Entity.entity_map_cache, File.join(Entity.entity_map_cache, id)
+
+      path = if user.nil?
+               Dir.glob(File.join(Entity.entity_map_cache, entity_type.to_s, column, Regexp.quote(id))).first || File.join(Entity.entity_map_cache, entity_type.to_s, column, id)
+             else
+               Dir.glob(File.join(Entity.entity_map_cache, user.to_s, entity_type.to_s, column, Regexp.quote(id))).first || File.join(Entity.entity_map_cache, user.to_s, entity_type.to_s, column, id)
+             end
+
+      path
+    end
+
+    def self.map_files(user = nil)
+      path = user.nil? ?
+        File.join(Entity.entity_map_cache, '*', '*') :
+        File.join(Entity.entity_map_cache, user, '*', '*') 
+
+      maps = {}
+      Dir.glob(path).each do |file|
+        next if File.directory? file
+
+        file = File.expand_path(file)
+        raise "Ilegal path: #{ file }. Not relative to #{File.expand_path(Entity.entity_map_cache)}" unless 
+          Misc.path_relative_to(File.expand_path(Entity.entity_map_cache), file)
+
+        if user.nil?
+          entity_type, column, map = file.split("/")[-2..-1]
+        else
+          user, entity_type, column, map = file.split("/")[-3..-1]
+        end
+
+        maps[entity_type] ||= []
+        maps[entity_type][column] ||= []
+        maps[entity_type][column] << map
+      end
+
+      maps
+    end
+
+    def self.load_map(entity_type, column, id, user = nil)
+      path = map_file(entity_type, column, id, user)
+      path = map_file(entity_type, column, id, :public) unless path != nil and File.exists? path
+      path = map_file(entity_type, column, id) unless path != nil and File.exists? path
+
+      begin
+        RbbtRESTHelpers.load_tsv(path).first
+      rescue
+        Log.error("Error loading map #{ path }: #{$!.message}")
+        nil
+      end
+    end
+
+    def self.save_map(entity_type, column, id, map, user = nil)
+      path = map_file(entity_type, column, id, user)
+
+      Misc.lock path do
+        begin
+          RbbtRESTHelpers.save_tsv(map, path)
+        rescue
+          FileUtils.rm path if path and File.exists? path
+          raise $!
+        end
+      end
+    end
+
+    def self.delete_map(entity_type, column, id, user)
+      path = map_file(entity_type, column, id, user)
+
+      "This map does not belong to #{ user }: #{[entity_type, column, id] * ": "}" unless File.exists? path
+
+      Misc.lock path do
+        begin
+          FileUtils.rm path if File.exists? path
+        rescue
+          raise $!
+        end
+      end
+    end
+  end
+end
+