lib/fx/definition.rb in fx-0.8.0 vs lib/fx/definition.rb in fx-0.9.0

@@ -1,19 +1,29 @@
 module Fx
   # @api private
   class Definition
-    def initialize(name:, version:, type: "function")
+    FUNCTION = "function".freeze
+    TRIGGER = "trigger".freeze
+
+    def self.function(name:, version:)
+      new(name: name, version: version, type: FUNCTION)
+    end
+
+    def self.trigger(name:, version:)
+      new(name: name, version: version, type: TRIGGER)
+    end
+
+    def initialize(name:, version:, type:)
       @name = name
       @version = version.to_i
       @type = type
     end
 
     def to_sql
-      File.read(find_file || full_path).tap do |content|
-        if content.empty?
-          raise "Define #{@type} in #{path} before migrating."
-        end
-      end
+      content = File.read(find_file || full_path)
+      raise "Define #{@type} in #{path} before migrating." if content.empty?
+
+      content
     end
 
     def full_path
       Rails.root.join(path)
     end