annotato 0.1.15 → 0.1.16

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c9018e17dc7b82a75e12b7e43cf5845abf9d397512cfc987aaf40dc4c70af045
-  data.tar.gz: ba4b35b5f17dbeebe540e32bc5bbfa4b124667915072bc73d6690429b5e3777f
+  metadata.gz: 53d8c10c79ec7a147fde40ed706fa11b77e62a65f1afbb87a204b8f2153d6f29
+  data.tar.gz: 488da6fb4970e359d266cc6e817502a72cdfc29a2d7ab59f10219c1e73978ef6
 SHA512:
-  metadata.gz: 2d573afafb8041a325f75eca8cf1ee9396520e380d28cd5a293b375e17b60f9bb872525c99a7cf0df7a54da402d5a89cadb36ec5cefbdd6906340f8bcae49a40
-  data.tar.gz: 0753e255496f9f03ec1d91ab54317ff435d2133d9e0dee7fa285c30bdef6a8692abe4a79a7d2412784436ec2696bed060feb30c5c0efde384ffcc75ce9c08273
+  metadata.gz: 6dd785f4e27b78b9a2084d2f846e531aba53774be945e5534386a179165c9509028bb8010f97fbab57c31693ed33aeda2fc2ec7f3ee807b711f7ad4086270048
+  data.tar.gz: 52d70e99d63296a518007970d09f20c209b15b5bb993b1c4f114c0912ba143c28d79e091491b74c7585898fef9c0c1140486db61df1a9782deefe935849720aa

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    annotato (0.1.15)
+    annotato (0.1.16)
       rails (>= 6.0)
 
 GEM

data/README.md

@@ -16,6 +16,7 @@
   * Enums with values
   * Indexes (only if present)
   * Triggers (only if present)
+  * Check Constraints (only if present)
 * Skips unchanged annotations
 * Smart formatting with aligned columns
 

data/lib/annotato/annotation_builder.rb

@@ -1,10 +1,12 @@
 # frozen_string_literal: true
 
+require_relative "wrap_helper"
 require_relative "column_formatter"
 require_relative "index_formatter"
 require_relative "trigger_formatter"
 require_relative "line_formatter"
 require_relative "enum_formatter"
+require_relative "check_constraint_formatter"
 
 module Annotato
   class AnnotationBuilder
@@ -41,6 +43,13 @@ module Annotato
         lines += triggers
       end
 
+      check_constraints = CheckConstraintFormatter.format(conn, table_name)
+      unless check_constraints.empty?
+        lines << ""
+        lines << "Check Constraints:"
+        lines += check_constraints
+      end
+
       LineFormatter.format(lines)
     end
   end

data/lib/annotato/check_constraint_formatter.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module Annotato
+  class CheckConstraintFormatter
+    extend WrapHelper
+
+    def self.format(conn, table_name)
+      conn.check_constraints(table_name).map do |chk|
+        expr_clause = ""
+        if chk.expression
+          expr_clause = "\n" + wrap_sql(chk.expression, first_prefix: "#    (", cont_prefix: "#    ")
+        end
+        "#  #{chk.name}#{expr_clause}"
+      end
+    end
+  end
+end

data/lib/annotato/index_formatter.rb

@@ -2,7 +2,7 @@
 
 module Annotato
   class IndexFormatter
-    MAX_LINE = 100
+    extend WrapHelper
 
     def self.format(conn, table_name)
       conn.indexes(table_name).map do |idx|
@@ -11,75 +11,11 @@ module Annotato
 
         where_clause = ""
         if idx.where
-          where_clause = "\n" + wrap_where(idx.where, max_len: MAX_LINE)
+          where_clause = "\n" + wrap_sql(idx.where, first_prefix: "#    where (", cont_prefix: "#          ")
         end
 
         "#  #{idx.name} (#{cols_list})#{unique_clause}#{where_clause}"
       end
     end
-
-    # Produces one or more lines, each already prefixed with "#    "
-    # First line: "#    where (..."
-    # Next lines:  "#          ..." (aligned under the "(")
-    def self.wrap_where(where_sql, max_len:)
-      first_prefix = "#    where ("
-      cont_prefix  = "#          " # aligns under the "(" after "where "
-
-      text = where_sql.to_s
-      lines = []
-
-      # If it already fits, keep it one-liner
-      if (first_prefix.length + text.length + 1) <= max_len
-        return "#{first_prefix}#{text})"
-      end
-
-      remaining = text.dup
-      current_prefix = first_prefix
-
-      while remaining.length > 0
-        available = max_len - current_prefix.length - 1 # -1 for closing ")" on last line (or just safety)
-        available = 20 if available < 20
-
-        if remaining.length <= available
-          lines << "#{current_prefix}#{remaining}"
-          remaining = ""
-          break
-        end
-
-        # Prefer breaking on logical operators within the window
-        window = remaining[0, available]
-        cut =
-          window.rindex(" AND ") ||
-          window.rindex(" OR ")  ||
-          window.rindex(", ")    ||
-          window.rindex(") ")    ||
-          window.rindex(" ")     # last resort: whitespace
-
-        # If we found a breakpoint, include it in the line (so operators stay visible)
-        if cut
-          # If we’re cutting on AND/OR/comma, keep that token at the end of the line when possible
-          if window[cut, 5] == " AND " || window[cut, 4] == " OR "
-            cut += (window[cut, 5] == " AND " ? 5 : 4)
-          elsif window[cut, 2] == ", "
-            cut += 2
-          elsif window[cut, 2] == ") "
-            cut += 2
-          else
-            cut += 1
-          end
-        else
-          cut = available
-        end
-
-        lines << "#{current_prefix}#{remaining[0, cut].rstrip}"
-        remaining = remaining[cut..].to_s.lstrip
-        current_prefix = cont_prefix
-      end
-
-      # Close the paren on the last line
-      lines[-1] = "#{lines[-1]})"
-
-      lines.join("\n")
-    end
   end
 end

data/lib/annotato/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Annotato
-  VERSION = "0.1.15"
+  VERSION = "0.1.16"
 end

data/lib/annotato/wrap_helper.rb

@@ -0,0 +1,68 @@
+# frozen_string_literal: true
+
+module Annotato
+  module WrapHelper
+    MAX_LINE = 100
+
+    # Produces one or more lines, each already prefixed with "#    "
+    # First line: "#    where (..."
+    # Next lines: "#          ..." (aligned under the "(")
+    def wrap_sql(sql, first_prefix:, cont_prefix:, max_len: MAX_LINE)
+      text = sql.to_s
+      lines = []
+
+      # If it already fits, keep it one-liner
+      if (first_prefix.length + text.length + 1) <= max_len
+        return "#{first_prefix}#{text})"
+      end
+
+      remaining = text.dup
+      current_prefix = first_prefix
+
+      while remaining.length > 0
+        available = max_len - current_prefix.length - 1 # -1 for closing ")" on last line (or just safety)
+        available = 20 if available < 20
+
+        if remaining.length <= available
+          lines << "#{current_prefix}#{remaining}"
+          remaining = ""
+          break
+        end
+
+        # Prefer breaking on logical operators within the window
+        window = remaining[0, available]
+        cut =
+          window.rindex(" AND ") ||
+          window.rindex(" OR ")  ||
+          window.rindex(", ")    ||
+          window.rindex(") ")    ||
+          window.rindex(" ")     # last resort: whitespace
+
+        # If we found a breakpoint, include it in the line (so operators stay visible)
+        if cut
+          # If we’re cutting on AND/OR/comma, keep that token at the end of the line when possible
+          if window[cut, 5] == " AND " || window[cut, 4] == " OR "
+            cut += (window[cut, 5] == " AND " ? 5 : 4)
+          elsif window[cut, 2] == ", "
+            cut += 2
+          elsif window[cut, 2] == ") "
+            cut += 2
+          else
+            cut += 1
+          end
+        else
+          cut = available
+        end
+
+        lines << "#{current_prefix}#{remaining[0, cut].rstrip}"
+        remaining = remaining[cut..].to_s.lstrip
+        current_prefix = cont_prefix
+      end
+
+      # Close the paren on the last line
+      lines[-1] = "#{lines[-1]})"
+
+      lines.join("\n")
+    end
+  end
+end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: annotato
 version: !ruby/object:Gem::Version
-  version: 0.1.15
+  version: 0.1.16
 platform: ruby
 authors:
 - Serhii Bodnaruk
 bindir: exe
 cert_chain: []
-date: 2025-12-19 00:00:00.000000000 Z
+date: 2026-01-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -45,6 +45,7 @@ files:
 - bin/setup
 - lib/annotato.rb
 - lib/annotato/annotation_builder.rb
+- lib/annotato/check_constraint_formatter.rb
 - lib/annotato/column_formatter.rb
 - lib/annotato/enum_formatter.rb
 - lib/annotato/index_formatter.rb
@@ -53,6 +54,7 @@ files:
 - lib/annotato/railtie.rb
 - lib/annotato/trigger_formatter.rb
 - lib/annotato/version.rb
+- lib/annotato/wrap_helper.rb
 - lib/tasks/annotato.rake
 - sig/annotato.rbs
 homepage: https://github.com/boserh/annotato