app/models/version.rb in cms_scanner-0.0.13 vs app/models/version.rb in cms_scanner-0.0.14

@@ -5,13 +5,39 @@
 
     attr_reader :number
 
     def initialize(number, opts = {})
       @number = number.to_s
+      @number = "0#{number}" if @number[0, 1] == '.'
+
       parse_finding_options(opts)
     end
 
+    # @param [ Version, String ] other
     def ==(other)
-      number == other.number
+      (self <=> other) == 0
+    end
+
+    # @param [ Version, String ] other
+    def <(other)
+      (self <=> other) == -1
+    end
+
+    # @param [ Version, String ] other
+    def >(other)
+      (self <=> other) == 1
+    end
+
+    # @param [ Version, String ] other
+    def <=>(other)
+      other = self.class.new(other) unless other.is_a?(self.class) # handle potential '.1' version
+
+      Gem::Version.new(number) <=> Gem::Version.new(other.number)
+    rescue
+      false
+    end
+
+    def to_s
+      number
     end
   end
 end