securial 2.1.3 → 2.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6685ddf1efdce3df66dac73131e8cb8cb38c1c02841d469eeea090a32f5bf12d
-  data.tar.gz: 5e225dd6071383dda5b76636edf65549debabae162db4d434e67cb259fe4f8f5
+  metadata.gz: 380f3eb7c41909962f49e0d860ac4ebfb8948b4477557ca0d4fc3a3b599b78c6
+  data.tar.gz: fe9c3e04c95f70836d453e36f4bbaf67b428a397c05a7516ca42d152a8368f9d
 SHA512:
-  metadata.gz: 5fb0b09611232bd05c972b5e122f7afd382d2666c87fde018ef4b10878c02db60b0d4ecbc6c0fade302948b0a9969b0dd675c56e7d176f0a535c5bdafbff311c
-  data.tar.gz: f332fd17e74bdec705ac54045a73b96ec4f4c738d37407dccb2858d2cd5cfe447c4ec34fb9910373c7c55d22c5d9cd2e5421543d3226d6f7a22257201dd63771
+  metadata.gz: 2a144e1b69790b12456fb2db21218f3ae77cf72c148a92af38d7934ab21740af566b4bd2155738d90d490da4004f12531a447296c674565b54774f667c33719b
+  data.tar.gz: 50404735621bed58aa44b3a98a5387a85093ea9d0a68050daa39e21fcad42fab30efbcf953b63c118b5d77dec4c640c47bc0e84a1c1382225030670f9aa18612

data/lib/securial/cli.rb

@@ -197,22 +197,55 @@ module Securial
     # @param app_name [String] name of the Rails application
     # @return [void]
     #
-    def update_database_yml_host(app_name)
+    def update_database_yml_host(app_name) # rubocop:disable Metrics/MethodLength
       db_config_path = File.join(app_name, "config", "database.yml")
+
+      # Step 1: Parse to check adapter
       raw = File.read(db_config_path)
       rendered = ERB.new(raw).result
       config = YAML.safe_load(rendered, aliases: true) || {}
 
-      config["default"] ||= {}
-      return unless config["default"]["adapter"].is_a?(String) && !config["default"]["adapter"].empty?
-      return unless ["mysql2", "postgresql"].include?(config["default"]["adapter"])
-      config["default"]["host"] = "<%= ENV.fetch(\"DB_HOST\", \"localhost\") %>"
-      config["default"]["username"] = "<%= ENV.fetch(\"DB_USERNAME\") { \"postgres\" } %>"
-      config["default"]["password"] = "<%= ENV.fetch(\"DB_PASSWORD\") { \"postgres\" } %>"
+      adapter = config.dig("default", "adapter")
+      return unless adapter.is_a?(String) && %w[postgresql mysql2].include?(adapter)
+
+      # Step 2: Modify the raw file line-by-line to preserve anchors and formatting
+      lines = File.readlines(db_config_path)
+      inside_default = false
+      updated_lines = []
+
+      lines.each_with_index do |line, index|
+        if line =~ /^default:/
+          inside_default = true
+          updated_lines << line
+          next
+        end
+
+        if inside_default
+          # Break out of default block when another top-level key appears
+          if line =~ /^\S/ && line !~ /^\s/
+            inside_default = false
+          end
+
+          # Skip existing host/username/password lines
+          next if line =~ /^\s*(host|username|password):/
+        end
+
+        updated_lines << line
+      end
+
+      # Find index of `default:` line to insert after
+      insert_index = updated_lines.find_index { |l| l =~ /^default:/ }
+      return unless insert_index
+
+      injection = [
+        "  host: <%= ENV.fetch(\"DB_HOST\", \"localhost\") %>",
+        "  username: <%= ENV.fetch(\"DB_USERNAME\") { \"postgres\" } %>",
+        "  password: <%= ENV.fetch(\"DB_PASSWORD\") { \"postgres\" } %>",
+      ]
+
+      updated_lines.insert(insert_index + 1, *injection)
 
-      # Dump YAML manually to preserve formatting
-      yaml = config.to_yaml.gsub(/['"](<%= .*? %>)['"]/, '\1') # Unquote the ERB
-      File.write(db_config_path, yaml)
+      File.write(db_config_path, updated_lines.join)
     end
 
     # Adds the Securial gem to the application's Gemfile.

data/lib/securial/version.rb

@@ -6,5 +6,5 @@ module Securial
   #
   # @see https://semver.org/ Semantic Versioning 2.0.0
   # @return [String] the current version in the format "major.minor.patch"
-  VERSION = "2.1.3".freeze
+  VERSION = "2.1.5".freeze
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: securial
 version: !ruby/object:Gem::Version
-  version: 2.1.3
+  version: 2.1.5
 platform: ruby
 authors:
 - Aly Badawy