Updated COLUMN_PATTERN to handle optional metadata (e.g., constraints or descriptions) enclosed in parentheses.

lib/annotate_rb/model_annotator/annotation_diff_generator.rb

@@ -5,7 +5,13 @@ module ModelAnnotator
     # Compares the current file content and new annotation block and generates the column annotation differences
     class AnnotationDiffGenerator
       HEADER_PATTERN = /(^# Table name:.*?\n(#.*\r?\n)*\r?)/
-      COLUMN_PATTERN = /^#[\t ]+[\w*.`\[\]():]+[\t ]+.+$/
+      # Example matches:
+      #   - "#  id              :uuid             not null, primary key"
+      #   - "#  title(length 255) :string          not null"
+      #   - "#  status(a/b/c)    :string           not null"
+      #   - "#  created_at       :datetime         not null"
+      #   - "#  updated_at       :datetime         not null"
+      COLUMN_PATTERN = /^#[\t ]+[\w*.`\[\]():]+(?:\(.*?\))?[\t ]+.+$/
 
       class << self
         def call(file_content, annotation_block)

spec/lib/annotate_rb/model_annotator/annotation_diff_generator_spec.rb

@@ -142,5 +142,55 @@ class User < ApplicationRecord
         expect(subject.changed?).to eq(true)
       end
     end
+
+    context "when a new column with metadata in parentheses is added" do
+      let(:annotation_block) do
+        <<~SCHEMA
+          # == Schema Information
+          #
+          # Table name: users
+          #
+          #  id                               :bigint           not null, primary key
+          #  name([sensitivity: medium])      :string(50)       not null
+          #  status(active/pending/inactive)  :string           not null
+          #
+        SCHEMA
+      end
+      let(:file_content) do
+        <<~FILE
+          # == Schema Information
+          #
+          # Table name: users
+          #
+          #  id                               :bigint           not null, primary key
+          #  name([sensitivity: medium])      :string(50)       not null
+          #
+          class User < ApplicationRecord
+          end
+        FILE
+      end
+
+      let(:current_columns) do
+        [
+          "#  id                               :bigint           not null, primary key",
+          "#  name([sensitivity: medium])      :string(50)       not null",
+          "# Table name: users"
+        ]
+      end
+      let(:new_columns) do
+        [
+          "#  id                               :bigint           not null, primary key",
+          "#  name([sensitivity: medium])      :string(50)       not null",
+          "#  status(active/pending/inactive)  :string           not null",
+          "# Table name: users"
+        ]
+      end
+
+      it "returns an AnnotationDiff object with the expected old and new columns" do
+        test_columns_match_expected
+
+        expect(subject.changed?).to eq(true)
+      end
+    end
   end
 end