Initial commit

lib/barong/seed.rb

@@ -0,0 +1,191 @@
+module Barong
+  class Seed
+    class ConfigError < RuntimeError; end
+
+    def initialize
+      @result = []
+    end
+
+    def seeds
+      YAML.safe_load(
+        ERB.new(
+          File.read(
+            Barong::App.config.seeds_file
+          )
+        ).result
+      )
+    end
+
+    def inspect
+      str = "Seeded users:\n"
+      str += @result.map do |user|
+        "Email: #{user[:email]}, password: #{user[:password]}"
+      end.join("\n")
+      return str
+    end
+
+    def logger
+      @logger ||= Logger.new(STDERR, progname: "db:seed")
+    end
+
+    def seed_provinces
+      logger.info "Seeding provinces"
+      seeds["provinces"].each_with_index do |province, index|
+        logger.info "---"
+        if Province.find_by(name: province["name"]).present?
+          logger.info "province '#{province['name']}' already exists"
+          next
+        end
+        province[:id] = index+1
+        Province.create!(province)
+      end
+    end
+
+    def seed_cities
+      logger.info "Seeding cities"
+      seeds["cities"].each_with_index do |city, index|
+        logger.info "---"
+        if City.find_by(name: city["name"]).present?
+          logger.info "province '#{city['name']}' already exists"
+          next
+        end
+        city[:id] = index+1
+        City.create!(city)
+      end
+    end
+
+    def seed_levels
+      logger.info "Seeding levels"
+      seeds["levels"].each_with_index do |level, index|
+        logger.info "---"
+        if Level.find_by(key: level["key"], value: level["value"]).present?
+          logger.info "Level '#{level['key']}:#{level['value']}' already exists"
+          next
+        end
+        level[:id] = index+1
+        Level.create!(level)
+      end
+    end
+
+    def seed_permissions
+      logger.info "Seeding permissions"
+      seeds["permissions"].each do |perm|
+        logger.info "---"
+        if Permission.find_by(role: perm["role"], verb: perm["verb"], path: perm["path"], action: perm["action"]).present?
+          logger.info "Permission for '#{perm['role']}' : '#{perm['verb']} to #{perm['path']}' already exists"
+          next
+        end
+
+        permission = Permission.new(perm)
+
+        unless permission.save
+          raise ConfigError.new("Can't create permission: #{permission.errors.full_messages.join('; ')}")
+        end
+      end
+    end
+
+    def seed_users
+      logger.info "Seeding users"
+      seeds["users"]&.each do |seed|
+        logger.info "---"
+
+        raise ConfigError.new("Email missing in users seed") if seed["email"].to_s.empty?
+        raise ConfigError.new("Level is missing for user #{seed["email"]}") unless seed["level"].is_a?(Integer)
+
+        # Skip existing users
+        if User.find_by(email: seed["email"]).present?
+          logger.info "User '#{seed['email']}' already exists"
+          @result.push(email: seed["email"])
+          next
+        end
+
+        user = User.new(seed)
+        user.password ||= SecureRandom.base64(30)
+
+        if user.save
+          logger.info "Created user for '#{user.email}'"
+
+          # Set correct level with labels
+          # levels = Level.where(id: 1..user.level)
+          levels = Level.where(id: 1..Level::LEVEL_ID_BOUNDS[user.level.to_s])
+          raise ConfigError.new("No enough levels found in database to grant the user to level #{user.level}") if levels.count < user.level
+
+          levels.find_each do |level|
+            user.labels.create(key: level.key, value: level.value, scope: 'private')
+          end
+
+          @result.push(email: user.email, password: user.password, level: user.level)
+        else
+          logger.error "Can't create user '#{user.email}': #{user.errors.full_messages.join('; ')}"
+        end
+      end
+    end
+
+    def seed_superadmin
+
+      logger.info "Seeding superadmin"
+      seeds["superadmin"]&.each do |seed|
+        logger.info "---"
+        raise ConfigError.new("Email missing in users seed") if seed["email"].to_s.empty?
+        raise ConfigError.new("Level is missing for user #{seed["email"]}") unless seed["level"].is_a?(Integer)
+
+        # Skip existing users
+        if User.find_by(email: seed["email"]).present?
+          logger.info "User '#{seed['email']}' already exists"
+          @result.push(email: seed["email"])
+          next
+        end
+
+        user = User.new(seed)
+        user.password ||= SecureRandom.base64(30)
+
+        if user.save
+          logger.info "Created user for '#{user.email}'"
+
+          # Set correct level with labels
+          # levels = Level.where(id: 1..user.level)
+          levels = Level.where(id: 1..Level::LEVEL_ID_BOUNDS[user.level.to_s])
+          raise ConfigError.new("No enough levels found in database to grant the user to level #{user.level}") if levels.count < user.level
+
+          levels.find_each do |level|
+            user.labels.create(key: level.key, value: level.value, scope: 'private')
+          end
+
+          @result.push(email: user.email, password: user.password, level: user.level)
+        else
+          logger.error "Can't create user '#{user.email}': #{user.errors.full_messages.join('; ')}"
+        end
+      end
+    end
+
+
+    def seed_restrictions
+      logger.info "Seeding restrictions"
+      return logger.info "Restrictions seed is empty!" if seeds["restrictions"].empty?
+
+      seeds["restrictions"].each do |seed|
+        logger.info "---"
+        if Restriction.find_by(category: seed["category"], scope: seed["scope"], value: seed["value"], state: seed["state"]).present?
+          logger.info "Restriction '#{seed['category']}' '#{seed['scope']} #{seed['value']}' #{seed['state']}' already exists"
+          next
+        end
+        raise ConfigError.new("category missing in restrictions seed") if seed["category"].nil?
+        raise ConfigError.new("scope is missing in restrictions seed") if seed["scope"].nil?
+        raise ConfigError.new("value is missing in restrictions seed") if seed["value"].nil?
+        raise ConfigError.new("state is missing in restrictions seed") if seed["state"].nil?
+
+        restriction = Restriction.new(
+          category: seed["category"],
+          scope: seed["scope"],
+          value: seed["value"],
+          state: seed["state"],
+          code: seed["code"]
+        )
+
+        unless restriction.save
+          raise ConfigError.new("Can't create restriction: #{restriction.errors.full_messages.join('; ')}")
+        end
+      end
+    end
+  end
+end