Syntax changes

extensions/xssrays/models/xssraysdetail.rb

@@ -9,25 +9,20 @@ module Models
   #
   # Store the rays details, basically verified XSS vulnerabilities
   #
-  class Xssraysdetail
-  
-    include DataMapper::Resource
-    
-    storage_names[:default] = 'extension_xssrays_details'
-    
-    property :id, Serial
+  class Xssraysdetail < ActiveRecord::Base
+    attribute :id, :Serial
 
     # The hooked browser id
-    property :hooked_browser_id, Text, :lazy => false
+    attribute :hooked_browser_id, :Text, :lazy => false
 
     # The XssRays vector name for the vulnerability
-    property :vector_name, Text, :lazy => true
+    attribute :vector_name, :Text, :lazy => true
 
     # The XssRays vector method (GET or POST) for the vulnerability
-    property :vector_method, Text, :lazy => true
+    attribute :vector_method, :Text, :lazy => true
 
     # The XssRays Proof of Concept for the vulnerability
-    property :vector_poc, Text, :lazy => true
+    attribute :vector_poc, :Text, :lazy => true
 
     belongs_to :xssraysscan
   end

extensions/xssrays/models/xssraysscan.rb

@@ -9,28 +9,23 @@ module Models
   #
   # Store the XssRays scans started and finished, with relative ID
   #
-  class Xssraysscan
+  class Xssraysscan < ActiveRecord::Base
 
-    include DataMapper::Resource
+    attribute :id, :Serial
 
-    storage_names[:default] = 'extension_xssrays_scans'
+    attribute :hooked_browser_id, :Text, :lazy => false
 
-    property :id, Serial
+    attribute :scan_start, :DateTime, :lazy => true
+    attribute :scan_finish, :DateTime, :lazy => true
 
-    property :hooked_browser_id, Text, :lazy => false
+    attribute :domain, :Text, :lazy => true
+    attribute :cross_domain, :Text, :lazy => true
+    attribute :clean_timeout, :Integer, :lazy => false
 
-    property :scan_start, DateTime, :lazy => true
-    property :scan_finish, DateTime, :lazy => true
-
-    property :domain, Text, :lazy => true
-    property :cross_domain, Text, :lazy => true
-    property :clean_timeout, Integer, :lazy => false
-
-    property :is_started, Boolean, :lazy => false, :default => false
-    property :is_finished, Boolean, :lazy => false, :default => false
-
-    has n, :extension_xssrays_details, 'Xssraysdetail'
+    attribute :is_started, :Boolean, :lazy => false, :default => false
+    attribute :is_finished, :Boolean, :lazy => false, :default => false
 
+    belongs_to :extension_xssrays_details
   end
 
 end