pristine.rb in cbac-0.6.3

- old
+ new
@@ -1,138 +1,138 @@
-require File.expand_path(File.join(File.dirname(__FILE__), 'pristine_file'))
-require File.expand_path(File.join(File.dirname(__FILE__), 'pristine_permission'))
-
-module Cbac
-  module CbacPristine
-    #creates a yml file containing all generic roles from the specified pristine file objects
-    def create_generic_role_fixtures_file(pristine_files, fixtures_file_name)
-      roles = []
-
-      pristine_files.each do |pristine_file|
-        #if the pristine file wasn't parsed yet, we'll do it here
-        pristine_file.parse(false) if pristine_file.permissions.empty?
-        pristine_file.generic_roles.each do |generic_role|
-          # we only want the unique generic roles, because the yml file cannot have duplicates
-          has_role = false
-          roles.each do |role|
-            if role.name == generic_role.name
-              has_role = true
-            end
-          end
-          roles.push(generic_role) unless has_role
-        end
-      end
-      create_fixtures_file(roles, fixtures_file_name)
-    end
-
-    # creates a yml file containing all cbac_permissions from the specified pristine file objects
-    def create_permissions_fixtures_file(pristine_files, fixtures_file_name)
-      permissions = []
-
-      pristine_files.each do |pristine_file|
-        pristine_file.parse(false) if pristine_file.permissions.empty?
-        pristine_file.permission_set.each do |line|
-          permissions.push(line)
-        end
-      end
-      create_fixtures_file(permissions, fixtures_file_name)
-    end
-
-    # turns the fixtures into yml and writes them to a file with specified name.
-    def create_fixtures_file(fixtures, fixtures_file_name)
-      File.delete(fixtures_file_name) if File.exists?(fixtures_file_name)
-      f = File.new(fixtures_file_name, "w")
-      flock(f, File::LOCK_EX) do |f|
-        fixtures.each_with_index do |fixture, index|
-          f.write(fixture.to_yml_fixture(index + 1))
-        end
-      end
-    end
-
-    # set all cbac permissions and generic roles to the state in the specified pristine file objects
-    def set_pristine_state(pristine_files, clear_tables)
-      clear_cbac_tables if clear_tables
-      pristine_files.each do |pristine_file|
-        pristine_file.parse if pristine_file.permissions.empty?
-        pristine_file.permissions.each do |permission|
-          permission.accept
-        end
-      end
-    end
-
-    # stage all unknown cbac_permissions
-    def stage_permissions(pristine_files)
-
-      pristine_files.each do |pristine_file|
-        pristine_file.parse(true) if pristine_file.permissions.empty?
-        pristine_file.permissions.each do |permission|
-          permission.stage
-        end
-      end
-    end
-
-    def clear_cbac_tables
-      Cbac::GenericRole.delete_all
-      Cbac::Membership.delete_all
-      Cbac::Permission.delete_all
-      Cbac::KnownPermission.delete_all
-      Cbac::CbacPristine::PristineFile.delete_all
-      Cbac::CbacPristine::PristinePermission.delete_all
-      Cbac::CbacPristine::PristineRole.delete_all
-    end
-
-    def delete_generic_known_permissions
-      known_permissions = Cbac::KnownPermission.find(:all, :conditions => {:permission_type => Cbac::KnownPermission.PERMISSION_TYPES[:generic]})
-      known_permissions.each { |p| p.destroy }
-    end
-
-    def delete_generic_permissions
-      permissions = Cbac::Permission.find(:all, :conditions => {:context_role => nil})
-      # for backwards compatibility, generic_role name was administrators instead of administrator
-      # SMELL: administrator role *only* identified by name
-      (permissions.select { |perm| perm.generic_role.name != "administrator" and perm.generic_role.name != "administrators" }).each { |p| p.destroy }
-    end
-
-    def delete_non_generic_staged_permissions
-      PristinePermission.delete_non_generic_permissions
-    end
-
-    def delete_generic_staged_permissions
-      PristinePermission.delete_generic_permissions
-    end
-
-    def database_contains_cbac_data?
-      (Cbac::GenericRole.count != 0 or Cbac::Membership.count != 0 or Cbac::Permission.count != 0 or Cbac::KnownPermission.count != 0 or Cbac::CbacPristine::PristinePermission.count != 0 or Cbac::CbacPristine::PristineRole.count != 0)
-    end
-
-    def find_or_create_generic_pristine_file(file_name)
-      pristine_file = GenericPristineFile.find_by_file_name(file_name)
-      pristine_file.present? ? pristine_file : GenericPristineFile.create(:file_name => file_name)
-    end
-
-    def find_or_create_pristine_file(file_name)
-      pristine_file = PristineFile.find_by_file_name(file_name)
-      pristine_file.present? ? pristine_file : PristineFile.create(:file_name => file_name)
-    end
-
-    def number_of_generic_staged_permissions
-      PristinePermission.count_generic_permissions
-    end
-
-    def number_of_non_generic_staged_permissions
-      PristinePermission.count_non_generic_permissions
-    end
-
-    def flock(file, mode)
-      success = file.flock(mode)
-      if success
-        begin
-          yield file
-        ensure
-          file.flock(File::LOCK_UN)
-        end
-      end
-      return success
-    end
-
-  end
-end
+require File.expand_path(File.join(File.dirname(__FILE__), 'pristine_file'))
+require File.expand_path(File.join(File.dirname(__FILE__), 'pristine_permission'))
+
+module Cbac
+  module CbacPristine
+    #creates a yml file containing all generic roles from the specified pristine file objects
+    def create_generic_role_fixtures_file(pristine_files, fixtures_file_name)
+      roles = []
+
+      pristine_files.each do |pristine_file|
+        #if the pristine file wasn't parsed yet, we'll do it here
+        pristine_file.parse(false) if pristine_file.permissions.nil? || pristine_file.permissions.empty?
+        pristine_file.generic_roles.each do |generic_role|
+          # we only want the unique generic roles, because the yml file cannot have duplicates
+          has_role = false
+          roles.each do |role|
+            if role.name == generic_role.name
+              has_role = true
+            end
+          end
+          roles.push(generic_role) unless has_role
+        end
+      end
+      create_fixtures_file(roles, fixtures_file_name)
+    end
+
+    # creates a yml file containing all cbac_permissions from the specified pristine file objects
+    def create_permissions_fixtures_file(pristine_files, fixtures_file_name)
+      permissions = []
+
+      pristine_files.each do |pristine_file|
+        pristine_file.parse(false) if pristine_file.permissions.nil? || pristine_file.permissions.empty?
+        pristine_file.permission_set.each do |line|
+          permissions.push(line)
+        end
+      end
+      create_fixtures_file(permissions, fixtures_file_name)
+    end
+
+    # turns the fixtures into yml and writes them to a file with specified name.
+    def create_fixtures_file(fixtures, fixtures_file_name)
+      File.delete(fixtures_file_name) if File.exists?(fixtures_file_name)
+      f = File.new(fixtures_file_name, "w")
+      flock(f, File::LOCK_EX) do |f|
+        fixtures.each_with_index do |fixture, index|
+          f.write(fixture.to_yml_fixture(index + 1))
+        end
+      end
+    end
+
+    # set all cbac permissions and generic roles to the state in the specified pristine file objects
+    def set_pristine_state(pristine_files, clear_tables)
+      clear_cbac_tables if clear_tables
+      pristine_files.each do |pristine_file|
+        pristine_file.parse if pristine_file.permissions.nil? || pristine_file.permissions.empty?
+        pristine_file.permissions.each do |permission|
+          permission.accept
+        end
+      end
+    end
+
+    # stage all unknown cbac_permissions
+    def stage_permissions(pristine_files)
+
+      pristine_files.each do |pristine_file|
+        pristine_file.parse(true) if pristine_file.permissions.nil? || pristine_file.permissions.empty?
+        pristine_file.permissions.each do |permission|
+          permission.stage
+        end
+      end
+    end
+
+    def clear_cbac_tables
+      Cbac::GenericRole.delete_all
+      Cbac::Membership.delete_all
+      Cbac::Permission.delete_all
+      Cbac::KnownPermission.delete_all
+      Cbac::CbacPristine::PristineFile.delete_all
+      Cbac::CbacPristine::PristinePermission.delete_all
+      Cbac::CbacPristine::PristineRole.delete_all
+    end
+
+    def delete_generic_known_permissions
+      known_permissions = Cbac::KnownPermission.find(:all, :conditions => {:permission_type => Cbac::KnownPermission.PERMISSION_TYPES[:generic]})
+      known_permissions.each { |p| p.destroy }
+    end
+
+    def delete_generic_permissions
+      permissions = Cbac::Permission.find(:all, :conditions => {:context_role => nil})
+      # for backwards compatibility, generic_role name was administrators instead of administrator
+      # SMELL: administrator role *only* identified by name
+      (permissions.select { |perm| perm.generic_role.name != "administrator" and perm.generic_role.name != "administrators" }).each { |p| p.destroy }
+    end
+
+    def delete_non_generic_staged_permissions
+      PristinePermission.delete_non_generic_permissions
+    end
+
+    def delete_generic_staged_permissions
+      PristinePermission.delete_generic_permissions
+    end
+
+    def database_contains_cbac_data?
+      (Cbac::GenericRole.count != 0 or Cbac::Membership.count != 0 or Cbac::Permission.count != 0 or Cbac::KnownPermission.count != 0 or Cbac::CbacPristine::PristinePermission.count != 0 or Cbac::CbacPristine::PristineRole.count != 0)
+    end
+
+    def find_or_create_generic_pristine_file(file_name)
+      pristine_file = GenericPristineFile.find_by_file_name(file_name)
+      pristine_file.present? ? pristine_file : GenericPristineFile.create(:file_name => file_name)
+    end
+
+    def find_or_create_pristine_file(file_name)
+      pristine_file = PristineFile.find_by_file_name(file_name)
+      pristine_file.present? ? pristine_file : PristineFile.create(:file_name => file_name)
+    end
+
+    def number_of_generic_staged_permissions
+      PristinePermission.count_generic_permissions
+    end
+
+    def number_of_non_generic_staged_permissions
+      PristinePermission.count_non_generic_permissions
+    end
+
+    def flock(file, mode)
+      success = file.flock(mode)
+      if success
+        begin
+          yield file
+        ensure
+          file.flock(File::LOCK_UN)
+        end
+      end
+      return success
+    end
+
+  end
+end