app/models/segment.rb

   1 class Segment < ActiveRecord::Base
   2   require 'xml/libxml'
   3   set_table_name 'current_segments'
   4
   5   validates_numericality_of :node_a
   6   validates_numericality_of :node_b
   7   # FIXME validate a nd b exist and are visible
   8
   9   has_many :old_segments, :foreign_key => :id
  10   belongs_to :user
  11
  12
  13   def self.from_xml(xml, create=false)
  14     p = XML::Parser.new
  15     p.string = xml
  16     doc = p.parse
  17
  18     segment = Segment.new
  19
  20     doc.find('//osm/segment').each do |pt|
  21
  22       segment.node_a = pt['from'].to_i
  23       segment.node_b = pt['to'].to_i
  24
  25       if pt['id'] != '0'
  26         segment.id = pt['id'].to_i
  27       end
  28
  29       segment.visible = true
  30
  31       if create
  32         segment.timestamp = Time.now
  33       else
  34         if pt['timestamp']
  35           segment.timestamp = Time.parse(pt['timestamp'])
  36         end
  37       end
  38
  39       tags = []
  40
  41       pt.find('tag').each do |tag|
  42         tags << [tag['k'],tag['v']]
  43       end
  44
  45       tags = tags.collect { |k,v| "#{k}=#{v}" }.join(';')
  46       tags = '' if tags.nil?
  47
  48       segment.tags = tags
  49
  50     end
  51     return segment
  52   end
  53
  54   def save_with_history
  55     begin
  56       Segment.transaction do
  57         self.save
  58         old_segment = OldSegment.from_segment(self)
  59         old_segment.save
  60       end
  61       return true
  62     rescue Exception => ex
  63       return nil
  64     end
  65   end
  66
  67   def to_xml
  68     doc = XML::Document.new
  69     doc.encoding = 'UTF-8'
  70     root = XML::Node.new 'osm'
  71     root['version'] = API_VERSION
  72     root['generator'] = 'OpenStreetMap server'
  73     doc.root = root
  74     root << to_xml_node()
  75     return doc
  76   end
  77
  78   def to_xml_node
  79     el1 = XML::Node.new 'segment'
  80     el1['id'] = self.id.to_s
  81     el1['from'] = self.node_a.to_s
  82     el1['to'] = self.node_b.to_s
  83     Segment.split_tags(el1, self.tags)
  84     el1['visible'] = self.visible.to_s
  85     el1['timestamp'] = self.timestamp.xmlschema
  86     return el1
  87   end
  88
  89   def self.split_tags(el, tags)
  90     tags.split(';').each do |tag|
  91       parts = tag.split('=')
  92       key = ''
  93       val = ''
  94       key = parts[0].strip unless parts[0].nil?
  95       val = parts[1].strip unless parts[1].nil?
  96       if key != '' && val != ''
  97         el2 = XML::Node.new('tag')
  98         el2['k'] = key.to_s
  99         el2['v'] = val.to_s
 100         el << el2
 101       end
 102     end
 103   end
 104
 105
 106 end