HandledScript Examples

class CreateTrackingsFromShipments < BaseAutomationService
  def execute
    @errors = ActiveModel::Errors.new(self)
    
    # Get time zone from variables or default to EST
    time_zone = @variables.dig("time_zone") || "America/New_York"
    
    # Setup date range
    setup_time_range(time_zone)
    
    # Initialize clients
    initialize_clients
    
    # Process shipments
    process_shipments
  end

  private

  def setup_time_range(time_zone)
    @start_time = if @variables.dig("start_time")
      DateTime.parse(@variables.dig("start_time")).in_time_zone(time_zone).strftime("%Y-%m-%dT%H:%M:%S")
    elsif @automation_workflow_execution.automation_workflow.last_successful_execution_at
      @automation_workflow_execution.automation_workflow.last_successful_execution_at.in_time_zone(time_zone).strftime("%Y-%m-%dT%H:%M:%S")
    else
      3.days.ago.in_time_zone(time_zone).strftime("%Y-%m-%dT%H:%M:%S")
    end

    @end_time = if @variables.dig("end_time")
      DateTime.parse(@variables.dig("end_time")).in_time_zone(time_zone).strftime("%Y-%m-%dT%H:%M:%S")
    else
      Time.now.in_time_zone(time_zone).strftime("%Y-%m-%dT%H:%M:%S")
    end
    
    Rails.logger.info("Time range determined: #{@start_time} to #{@end_time}")
  end

  def initialize_clients
    @shiphero_client = CommerceAutomationNodes::ShipheroNode.new(credentials: @credentials)
    @slack_client = CommerceAutomationNodes::SlackNode.new(credentials: @credentials)
    
    Rails.logger.info("Clients initialized successfully")
  end

  def process_shipments
    # Get shipments from Shiphero
    shipments_response = @shiphero_client.get_shipments(
      date_from: @start_time,
      date_to: @end_time
    )&.parsed_body
    
    Rails.logger.info("Shipments retrieved from Shiphero")
    
    # Extract shipments from response
    shipments = shipments_response.dig("data", "shipments", "data", "edges")&.map { |edge| edge["node"] } || []
    
    Rails.logger.info("Processing #{shipments.size} shipments")

    successful_count = 0
    failed_shipments = []

    # Process each shipment
    shipments.each do |shipment|
      begin
        process_single_shipment(shipment)
        successful_count += 1
      rescue => e
        Rails.logger.error("Error processing shipment: #{e.message}")
        failed_shipments << {
          id: shipment.dig("id"),
          error: e.message
        }
      end
    end

    # Send summary to Slack
    send_summary_to_slack(successful_count, failed_shipments, shipments.size)
  end

  def process_single_shipment(shipment)
    created_trackings = []
    
    # Process each shipping label
    shipping_labels = shipment.dig("shipping_labels") || []
    
    shipping_labels.each do |label|
      Rails.logger.info(label)
      # Find or create tracking record for each label
      tracking = Tracking.find_or_create_by(tracking_number: label.dig("tracking_number")) do |t|
        t.account_id = @automation_workflow_execution.account_id
        t.wms_order_id = shipment.dig("order_id")
        t.commerce_platform_order_name = shipment.dig("order", "order_number")
        t.carrier_id = Carrier.find_by(carrier_code: label.dig("carrier")&.downcase)&.id
        t.carrier_service_level = label.dig("shipping_name")
        t.notes = "Created via automation on #{Time.now}"
        t.fulfillment_center_id = @variables.dig("fulfillment_center_id")
        t.tracking_type = "FULFILLMENT_PACKAGE"
      end
      
      created_trackings << tracking
      Rails.logger.info("Found or created tracking record #{tracking.id} for shipment #{shipment.dig('id')} with tracking number #{label.dig('tracking_number')}")
    end

    created_trackings
  end

  def send_summary_to_slack(successful_count, failed_shipments, total_shipments)
    message = build_slack_message(successful_count, failed_shipments, total_shipments)
    
    @slack_client.post_message(
      channel: @variables.dig("slack_channel").to_s,
      text: "Tracking Creation Summary",
      blocks: [
        {
          type: "section",
          text: {
            type: "mrkdwn",
            text: message
          }
        }
      ]
    )
  end

  def build_slack_message(successful_count, failed_shipments, total_shipments)
    message = ["*Tracking Creation Summary*
"]
    message << "Time Range: #{@start_time} to #{@end_time}"
    message << "Total Shipments Processed: #{total_shipments}"
    message << "Successfully Created: #{successful_count}"
    message << "Failed: #{failed_shipments.size}"

    if failed_shipments.any?
      message << "
*Failed Shipments:*"
      failed_shipments.each do |failure|
        message << "• Shipment #{failure[:id]}: #{failure[:error]}"
      end
    end

    message.join("
")
  end
end