exaonruby 1.0.0

data/lib/exa/resources/webset.rb

@@ -0,0 +1,491 @@
+# frozen_string_literal: true
+
+# typed: strict
+
+module Exa
+  module Resources
+    class Webset < Base
+      # @return [String] Unique identifier
+      def id
+        get(:id)
+      end
+
+      # @return [String] Object type (always "webset")
+      def object
+        get(:object)
+      end
+
+      # @return [String] Status: idle, pending, running, paused
+      def status
+        get(:status)
+      end
+
+      # @return [String, nil] External identifier
+      def external_id
+        get(:external_id)
+      end
+
+      # @return [String, nil] Title of the webset
+      def title
+        get(:title)
+      end
+
+      # @return [Array<WebsetSearch>] Associated searches
+      def searches
+        @searches ||= (get(:searches, []) || []).map { |data| WebsetSearch.new(data) }
+      end
+
+      # @return [Array<WebsetImport>] Associated imports
+      def imports
+        @imports ||= (get(:imports, []) || []).map { |data| WebsetImport.new(data) }
+      end
+
+      # @return [Array<WebsetEnrichment>] Associated enrichments
+      def enrichments
+        @enrichments ||= (get(:enrichments, []) || []).map { |data| WebsetEnrichment.new(data) }
+      end
+
+      # @return [Array<WebsetMonitor>] Associated monitors
+      def monitors
+        @monitors ||= (get(:monitors, []) || []).map { |data| WebsetMonitor.new(data) }
+      end
+
+      # @return [Array<WebsetItem>] Items (when included)
+      def items
+        @items ||= (get(:items, []) || []).map { |data| WebsetItem.new(data) }
+      end
+
+      # @return [Array<Hash>] Exclude configurations
+      def excludes
+        get(:excludes, [])
+      end
+
+      # @return [Hash] Custom metadata
+      def metadata
+        get(:metadata, {})
+      end
+
+      # @return [Time, nil] Creation timestamp
+      def created_at
+        parse_time(get(:created_at))
+      end
+
+      # @return [Time, nil] Last update timestamp
+      def updated_at
+        parse_time(get(:updated_at))
+      end
+
+      # @return [Boolean] Whether the webset is idle
+      def idle?
+        status == "idle"
+      end
+
+      # @return [Boolean] Whether the webset is running
+      def running?
+        status == "running"
+      end
+
+      # @return [Boolean] Whether the webset is pending
+      def pending?
+        status == "pending"
+      end
+
+      # @return [Boolean] Whether the webset is paused
+      def paused?
+        status == "paused"
+      end
+
+      private
+
+      def parse_time(value)
+        return nil unless value
+
+        Time.parse(value)
+      rescue ArgumentError
+        nil
+      end
+
+      def inspectable_attributes
+        { id: id, status: status, title: title }
+      end
+    end
+
+    class WebsetSearch < Base
+      # @return [String] Unique identifier
+      def id
+        get(:id)
+      end
+
+      # @return [String] Object type
+      def object
+        get(:object)
+      end
+
+      # @return [String] Status: created, running, completed, canceled
+      def status
+        get(:status)
+      end
+
+      # @return [String] Parent webset ID
+      def webset_id
+        get(:webset_id)
+      end
+
+      # @return [String, nil] Search query
+      def query
+        get(:query)
+      end
+
+      # @return [Hash, nil] Entity type configuration
+      def entity
+        get(:entity)
+      end
+
+      # @return [String, nil] Entity type (company, person, research_paper)
+      def entity_type
+        dig(:entity, :type)
+      end
+
+      # @return [Array<Hash>] Search criteria
+      def criteria
+        get(:criteria, [])
+      end
+
+      # @return [Integer, nil] Target count of results
+      def count
+        get(:count)
+      end
+
+      # @return [Array<Hash>] Exclusion rules
+      def exclude
+        get(:exclude, [])
+      end
+
+      # @return [Array<Hash>] Scope configuration
+      def scope
+        get(:scope, [])
+      end
+
+      # @return [Hash, nil] Progress information
+      def progress
+        get(:progress)
+      end
+
+      # @return [Integer, nil] Number of items found
+      def found_count
+        dig(:progress, :found)
+      end
+
+      # @return [Integer, nil] Number of items analyzed
+      def analyzed_count
+        dig(:progress, :analyzed)
+      end
+
+      # @return [Integer, nil] Completion percentage (0-100)
+      def completion_percentage
+        dig(:progress, :completion)
+      end
+
+      # @return [Integer, nil] Estimated time remaining in seconds
+      def time_left
+        dig(:progress, :time_left)
+      end
+
+      # @return [Hash, nil] Recall estimation
+      def recall
+        get(:recall)
+      end
+
+      # @return [String, nil] Search behavior (override, append)
+      def behavior
+        get(:behavior)
+      end
+
+      # @return [Hash] Custom metadata
+      def metadata
+        get(:metadata, {})
+      end
+
+      # @return [Time, nil] Cancellation timestamp
+      def canceled_at
+        parse_time(get(:canceled_at))
+      end
+
+      # @return [String, nil] Cancellation reason
+      def canceled_reason
+        get(:canceled_reason)
+      end
+
+      # @return [Time, nil] Creation timestamp
+      def created_at
+        parse_time(get(:created_at))
+      end
+
+      # @return [Time, nil] Last update timestamp
+      def updated_at
+        parse_time(get(:updated_at))
+      end
+
+      # @return [Boolean] Whether search is running
+      def running?
+        status == "running"
+      end
+
+      # @return [Boolean] Whether search is completed
+      def completed?
+        status == "completed"
+      end
+
+      # @return [Boolean] Whether search was canceled
+      def canceled?
+        status == "canceled"
+      end
+
+      private
+
+      def parse_time(value)
+        return nil unless value
+
+        Time.parse(value)
+      rescue ArgumentError
+        nil
+      end
+    end
+
+    class WebsetImport < Base
+      # @return [String] Unique identifier
+      def id
+        get(:id)
+      end
+
+      # @return [String] Object type
+      def object
+        get(:object)
+      end
+
+      # @return [String] Status: pending, processing, completed, failed
+      def status
+        get(:status)
+      end
+
+      # @return [String, nil] Import format (csv)
+      def format
+        get(:format)
+      end
+
+      # @return [Hash, nil] Entity configuration
+      def entity
+        get(:entity)
+      end
+
+      # @return [String, nil] Title
+      def title
+        get(:title)
+      end
+
+      # @return [Integer, nil] Number of items
+      def count
+        get(:count)
+      end
+
+      # @return [Hash] Custom metadata
+      def metadata
+        get(:metadata, {})
+      end
+
+      # @return [String, nil] Failure reason if failed
+      def failed_reason
+        get(:failed_reason)
+      end
+
+      # @return [Time, nil] Failure timestamp
+      def failed_at
+        parse_time(get(:failed_at))
+      end
+
+      # @return [String, nil] Failure message
+      def failed_message
+        get(:failed_message)
+      end
+
+      # @return [Time, nil] Creation timestamp
+      def created_at
+        parse_time(get(:created_at))
+      end
+
+      # @return [Time, nil] Last update timestamp
+      def updated_at
+        parse_time(get(:updated_at))
+      end
+
+      # @return [Boolean] Whether import failed
+      def failed?
+        status == "failed"
+      end
+
+      private
+
+      def parse_time(value)
+        return nil unless value
+
+        Time.parse(value)
+      rescue ArgumentError
+        nil
+      end
+    end
+
+    class WebsetEnrichment < Base
+      # @return [String] Unique identifier
+      def id
+        get(:id)
+      end
+
+      # @return [String] Object type
+      def object
+        get(:object)
+      end
+
+      # @return [String] Status: pending, running, completed
+      def status
+        get(:status)
+      end
+
+      # @return [String] Parent webset ID
+      def webset_id
+        get(:webset_id)
+      end
+
+      # @return [String, nil] Title
+      def title
+        get(:title)
+      end
+
+      # @return [String, nil] Description of what to extract
+      def description
+        get(:description)
+      end
+
+      # @return [String, nil] Format: text, enum, number, boolean, date
+      def format
+        get(:format)
+      end
+
+      # @return [Array<Hash>] Options for enum format
+      def options
+        get(:options, [])
+      end
+
+      # @return [String, nil] Processing instructions
+      def instructions
+        get(:instructions)
+      end
+
+      # @return [Hash] Custom metadata
+      def metadata
+        get(:metadata, {})
+      end
+
+      # @return [Time, nil] Creation timestamp
+      def created_at
+        parse_time(get(:created_at))
+      end
+
+      # @return [Time, nil] Last update timestamp
+      def updated_at
+        parse_time(get(:updated_at))
+      end
+
+      private
+
+      def parse_time(value)
+        return nil unless value
+
+        Time.parse(value)
+      rescue ArgumentError
+        nil
+      end
+    end
+
+    class WebsetMonitor < Base
+      # @return [String] Unique identifier
+      def id
+        get(:id)
+      end
+
+      # @return [String] Object type
+      def object
+        get(:object)
+      end
+
+      # @return [String] Status: enabled, disabled
+      def status
+        get(:status)
+      end
+
+      # @return [String] Parent webset ID
+      def webset_id
+        get(:webset_id)
+      end
+
+      # @return [Hash, nil] Cadence configuration
+      def cadence
+        get(:cadence)
+      end
+
+      # @return [String, nil] Cron expression
+      def cron
+        dig(:cadence, :cron)
+      end
+
+      # @return [String, nil] Timezone
+      def timezone
+        dig(:cadence, :timezone)
+      end
+
+      # @return [Hash, nil] Behavior configuration
+      def behavior
+        get(:behavior)
+      end
+
+      # @return [Hash, nil] Last run information
+      def last_run
+        get(:last_run)
+      end
+
+      # @return [Time, nil] Next scheduled run
+      def next_run_at
+        parse_time(get(:next_run_at))
+      end
+
+      # @return [Hash] Custom metadata
+      def metadata
+        get(:metadata, {})
+      end
+
+      # @return [Time, nil] Creation timestamp
+      def created_at
+        parse_time(get(:created_at))
+      end
+
+      # @return [Time, nil] Last update timestamp
+      def updated_at
+        parse_time(get(:updated_at))
+      end
+
+      # @return [Boolean] Whether monitor is enabled
+      def enabled?
+        status == "enabled"
+      end
+
+      private
+
+      def parse_time(value)
+        return nil unless value
+
+        Time.parse(value)
+      rescue ArgumentError
+        nil
+      end
+    end
+  end
+end