command-with-timestamps.rb

module CommandPlugins
  module ActiveRecordTimestamps
    class HashWithTimestamps
      def initialize(input)
        @input = input
      end

      def [](*args)
        tuple = @input[*args]

        now = Time.zone.now

        if tuple[:created_at]
          tuple[:updated_at] = now
        else
          tuple[:created_at] = now
          tuple[:updated_at] = now
        end
        
        tuple
      end
    end

    def self.included(klass)
      super

      klass.input HashWithTimestamps.new(klass.input)
    end
  end
end

ROM.plugins do
  adapter :sql do
    register :timestamps, CommandPlugins::ActiveRecordTimestamps, type: :command
  end
end

class CreateAccount < ROM::Commands::Create[:sql]
  register_as :create
  relation :accounts
  result :one

  use :timestamps
end