lib/wasserstand/level.rb in wasserstand-0.0.4 vs lib/wasserstand/level.rb in wasserstand-0.0.5

@@ -1,18 +1,41 @@
 module Wasserstand
   #
   # see http://www.pegelonline.wsv.de/gast/hilfe#hilfe_pegelparameter
   #
   class Level # Pegel
-    attr_reader :id
-    attr_accessor :name, :km, :measurements
+    class << self
+      def [](name)
+        Wasserstand.level_provider[name]
+      end
 
-    def initialize(id)
-      @id = id
+      def all
+        Wasserstand.level_provider.all
+      end
+
+      def find_by_name(regex)
+        Wasserstand.level_provider.find_by_name(regex)
+      end
+    end
+
+    attr_reader :name
+    attr_accessor :level_id, :km, :measurements
+    attr_writer :waterway
+
+    def initialize(name)
+      @name = name
       @measurements = []
     end
 
+    def waterway
+      if @waterway.respond_to?(:name)
+        @waterway
+      else
+        @waterway = Waterway[@waterway]
+      end
+    end
+
     def to_s
-      "#{@name} (km #{@km}): #{@measurements.last}"
+      name
     end
   end
 end