]> git.openstreetmap.org Git - rails.git/blob - app/models/segment.rb
Fixed precondition typo and segment_controller foreign key
[rails.git] / 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