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