app/models/segment.rb

   1 class Segment < ActiveRecord::Base
   2   require 'xml/libxml'
   3   set_table_name 'current_segments'
   4
   5   validates_presence_of :user_id, :timestamp
   6   validates_inclusion_of :visible, :in => [ true, false ]
   7   validates_numericality_of :node_a, :node_b
   8
   9   has_many :old_segments, :foreign_key => :id
  10   belongs_to :user
  11
  12   # using belongs_to :foreign_key = 'node_*', since if use has_one :foreign_key = 'id', segment preconditions? fails checking for segment id in node table
  13   belongs_to :from_node, :class_name => 'Node', :foreign_key => 'node_a'
  14   belongs_to :to_node, :class_name => 'Node', :foreign_key => 'node_b'
  15
  16   def self.from_xml(xml, create=false)
  17     begin
  18       p = XML::Parser.new
  19       p.string = xml
  20       doc = p.parse
  21
  22       segment = Segment.new
  23
  24       doc.find('//osm/segment').each do |pt|
  25         segment.node_a = pt['from'].to_i
  26         segment.node_b = pt['to'].to_i
  27
  28         unless create
  29           if pt['id'] != '0'
  30             segment.id = pt['id'].to_i
  31           end
  32         end
  33
  34         segment.visible = true
  35
  36         if create
  37           segment.timestamp = Time.now
  38         else
  39           if pt['timestamp']
  40             segment.timestamp = Time.parse(pt['timestamp'])
  41           end
  42         end
  43
  44         tags = []
  45
  46         pt.find('tag').each do |tag|
  47           tags << [tag['k'],tag['v']]
  48         end
  49
  50         tags = tags.collect { |k,v| "#{k}=#{v}" }.join(';')
  51         tags = '' if tags.nil?
  52
  53         segment.tags = tags
  54       end
  55     rescue
  56       segment = nil
  57     end
  58
  59     return segment
  60   end
  61
  62   def save_with_history
  63     begin
  64       Segment.transaction do
  65         self.timestamp = Time.now
  66         self.save!
  67         old_segment = OldSegment.from_segment(self)
  68         old_segment.save!
  69       end
  70
  71       return true
  72     rescue
  73       return nil
  74     end
  75   end
  76
  77   def to_xml
  78     doc = OSM::API.new.get_xml_doc
  79     doc.root << to_xml_node()
  80     return doc
  81   end
  82
  83   def to_xml_node(user_display_name_cache  = nil)
  84     el1 = XML::Node.new 'segment'
  85     el1['id'] = self.id.to_s
  86     el1['from'] = self.node_a.to_s
  87     el1['to'] = self.node_b.to_s
  88
  89     user_display_name_cache = {} if user_display_name_cache.nil?
  90
  91     if user_display_name_cache and user_display_name_cache[self.user_id]
  92       # use the cache if available
  93     elsif self.user.data_public?
  94       user_display_name_cache[self.user_id] = self.user.display_name
  95     else
  96       user_display_name_cache[self.user_id] = nil
  97     end
  98
  99     el1['user'] = user_display_name_cache[self.user_id] unless user_display_name_cache[self.user_id].nil?
 100
 101     Segment.split_tags(el1, self.tags)
 102     el1['visible'] = self.visible.to_s
 103     el1['timestamp'] = self.timestamp.xmlschema
 104     return el1
 105   end
 106
 107   def self.split_tags(el, tags)
 108     tags.split(';').each do |tag|
 109       parts = tag.split('=')
 110       key = ''
 111       val = ''
 112       key = parts[0].strip unless parts[0].nil?
 113       val = parts[1].strip unless parts[1].nil?
 114       if key != '' && val != ''
 115         el2 = XML::Node.new('tag')
 116         el2['k'] = key.to_s
 117         el2['v'] = val.to_s
 118         el << el2
 119       end
 120     end
 121   end
 122
 123   def preconditions_ok?
 124     from_node and from_node.visible and to_node and to_node.visible
 125   end
 126
 127 end