wolf_core 1.1.16 → 1.1.17

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f6517e8c433534aeeff5bb76afd2b66ebfae48b55015f8c7e400d52dc6a9350f
-  data.tar.gz: e4a67c85e97cfd59aacf54dd6245b549c88601a7fa3ce986b13cc7acf859b2b3
+  metadata.gz: 45745f516e8eb69746381a70a65e22f580f3dda4b3600b6eda99eda75bf0f057
+  data.tar.gz: 1448150fb318029c96b9406852ac0326fd4f433af9226308441583a3511083d5
 SHA512:
-  metadata.gz: 4a416bf327987838e7e4f4fa0ac706bf556d4b9c80d7d1126a14b2ab50cfa94e4e050e3f8a542d17407900fcdf55a3bef5f9cd8fe4fb09c0ed4c693abdfe7768
-  data.tar.gz: eab0636707b283a7a6132b70f1f9a03ff225755da0788cb620beadd20bc3de57878c0823bfd0347a900110d2dfa63226f9a9cb567d179842d50ee05b4c007679
+  metadata.gz: bd293374a86ee9166d0b820a0392493efa6828597c3e3a7d74e8337141782142e8303cb82d525e17cefe7d1e6dc15f479e15b17af295920339bb5995e4e78037
+  data.tar.gz: 32c9a77c6ba34ba5ffc8e5ea81a1294d65f7a13314489d6d8ee26125105dcd9a78a02117fe53cf5363bff0d1f6359c742f66eabd4d6d5aab1aeb4ba9ce123a07

data/lib/wolf_core/infrastructure/job_scheduler_data_source.rb

@@ -0,0 +1,126 @@
+# frozen_string_literal: true
+
+require "time"
+require "json"
+
+module WolfCore
+  class JobSchedulerDataSource
+    attr_reader :default_group_name, :default_timezone
+
+    def initialize(region: "us-east-1", client: nil, default_group_name: "default", default_timezone: "UTC")
+      @client = client || Aws::Scheduler::Client.new(region: region)
+      @default_group_name = default_group_name
+      @default_timezone = default_timezone
+    end
+
+    def schedule_at(job_id:, run_at:, target:, config: {})
+      Result.try do
+        expr = at_expression(run_at)
+        normalized_target = normalize_target(target)
+        response = schedule_with_expression(job_id: job_id, expr: expr, target: normalized_target, config: config)
+        Result.success(data: { response: response })
+      end
+    end
+
+      def schedule_recurring(job_id:, schedule:, target:, config: {})
+        Result.try do
+          expr = expression_from_schedule!(schedule)
+          normalized_target = normalize_target(target)
+          response = schedule_with_expression(job_id: job_id, expr: expr, target: normalized_target,
+                                              config: config)
+          Result.success(data: { response: response })
+        end
+      end
+
+    def cancel_schedule(job_id:, options: {})
+      Result.try do
+        response = @client.delete_schedule(
+          name: job_id,
+          group_name: options[:group_name] || default_group_name
+        )
+        Result.success(data: { response: response })
+      end
+    end
+
+    def get_schedule(job_id:, options: {})
+      Result.try do
+        response = @client.get_schedule(
+          name: job_id,
+          group_name: options[:group_name] || default_group_name
+        )
+        Result.success(data: { schedule: response })
+      end
+    end
+
+    private
+
+    def create_schedule(name:, schedule_expression:, target:, options: {})
+      @client.create_schedule(
+        name: name,
+        group_name: options[:group_name],
+        schedule_expression: schedule_expression,
+        schedule_expression_timezone: options[:timezone],
+        flexible_time_window: { mode: "OFF" },
+        target: target,
+        description: options[:description],
+        state: "ENABLED",
+        client_token: options[:client_token]
+      )
+    end
+
+    def schedule_with_expression(job_id:, expr:, target:, config: {})
+      options = build_options(config)
+      create_schedule(name: job_id, schedule_expression: expr, target: target, options: options)
+    end
+
+    def normalize_target(target)
+      arn = target[:arn] || target[:target_arn]
+      role = target[:role_arn]
+      payload = target[:payload]
+      raise ArgumentError, "target[:arn] o target[:target_arn] es requerido" unless arn
+      raise ArgumentError, "target[:role_arn] es requerido" unless role
+
+      build_target(target_arn: arn, role_arn: role, payload: payload)
+    end
+
+    def build_options(config)
+      {
+        description: config[:description],
+        group_name: config[:group_name] || default_group_name,
+        timezone: config[:timezone] || default_timezone,
+        client_token: config[:client_token]
+      }
+    end
+
+    def expression_from_schedule!(schedule)
+      expr = schedule[:cron_expression] || schedule[:cron] || schedule[:rate_expression] || schedule[:rate]
+      raise ArgumentError, "Se requiere schedule[:cron] o schedule[:rate]" unless expr
+
+      expr
+    end
+
+    def build_target(target_arn:, role_arn:, payload: nil)
+      input = case payload
+              when nil then nil
+              when String then payload
+              else
+                payload.respond_to?(:to_json) ? payload.to_json : payload.to_s
+              end
+
+      target = { arn: target_arn, role_arn: role_arn }
+      target[:input] = input if input
+      target
+    end
+
+    def at_expression(run_at)
+      time = case run_at
+             when Time then run_at.utc
+             when DateTime then run_at.to_time.utc
+             when String then Time.parse(run_at).utc
+             else
+               raise ArgumentError, "run_at debe ser Time, DateTime o String ISO8601"
+             end
+      "at(#{time.iso8601})"
+    end
+  end
+end

data/lib/wolf_core/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module WolfCore
-  VERSION = "1.1.16"
+  VERSION = "1.1.17"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: wolf_core
 version: !ruby/object:Gem::Version
-  version: 1.1.16
+  version: 1.1.17
 platform: ruby
 authors:
 - Javier Roncallo
@@ -143,6 +143,7 @@ files:
 - lib/wolf_core/infrastructure/in_memory_storage_data_source.rb
 - lib/wolf_core/infrastructure/in_memory_storage_operations.rb
 - lib/wolf_core/infrastructure/instance_application_serializer.rb
+- lib/wolf_core/infrastructure/job_scheduler_data_source.rb
 - lib/wolf_core/infrastructure/lambda_function_data_source.rb
 - lib/wolf_core/infrastructure/lambda_function_operations.rb
 - lib/wolf_core/infrastructure/no_sql_db_data_source.rb