# Oban Pro v1.6.10 - Nested Sub-Workflow Context Bug Reproducer
#
# Bug: When using `add_cascade` with fan-out tuples inside a sub-workflow,
# the `sup_workflow_id` of the fan-out jobs incorrectly points to the ROOT
# workflow instead of the immediate parent sub-workflow. This causes
# `Workflow.get_context/1` to return the wrong context.
#
# Expected: Fan-out jobs should have `sup_workflow_id` pointing to their
# immediate parent sub-workflow, so context lookup finds the correct context.
#
# Actual: Fan-out jobs have `sup_workflow_id` pointing to the root workflow,
# causing context lookup to skip the intermediate sub-workflow's context.

defmodule ReproducerWorkflows do
  @moduledoc """
  Minimal reproduction of nested sub-workflow context bug.
  
  Run with:
    mix run oban_pro_nested_context_bug.ex
  
  Or in iex:
    iex> ReproducerWorkflows.run()
  """

  use Oban.Pro.Decorator

  # ============================================================================
  # Inner Workflow (simulates AmazonDsp)
  # ============================================================================

  def inner_workflow do
    items = [1, 2]  # Fan-out over 2 items

    Oban.Pro.Workflow.new()
    |> Oban.Pro.Workflow.put_context(%{
      inner_secret: "I should be visible to fan-out jobs",
      oauth_connection_id: 123
    })
    |> Oban.Pro.Workflow.add_cascade(
      :step_one,
      &step_one/1,
      queue: :default
    )
    |> Oban.Pro.Workflow.add_cascade(
      :fan_out_step,
      {items, &fan_out_step/2},
      queue: :default,
      deps: [:step_one]
    )
    |> Oban.Pro.Workflow.add_cascade(
      :final_step,
      &final_step/1,
      queue: :default,
      deps: [:fan_out_step]
    )
  end

  # ============================================================================
  # Outer Workflow (simulates PublishCampaign)
  # ============================================================================

  def outer_workflow do
    Oban.Pro.Workflow.new()
    |> Oban.Pro.Workflow.put_context(%{
      outer_only: "I am from the outer workflow",
      campaign_id: 999
    })
    |> Oban.Pro.Workflow.add_workflow(:inner, inner_workflow())
    |> Oban.Pro.Workflow.add_cascade(
      :notify,
      &notify/1,
      queue: :default,
      deps: [:inner]
    )
  end

  # ============================================================================
  # Cascade Functions
  # ============================================================================

  @job queue: :default, recorded: true
  def step_one(ctx) do
    IO.puts("[step_one] Context keys: #{inspect(Map.keys(ctx))}")
    {:ok, %{step_one_result: "done"}}
  end

  @job queue: :default, recorded: true
  def fan_out_step(item, ctx) do
    # BUG: ctx will contain outer_only and campaign_id (from outer workflow)
    #      instead of inner_secret and oauth_connection_id (from inner workflow)
    
    IO.puts("[fan_out_step item=#{item}] Context keys: #{inspect(Map.keys(ctx))}")
    IO.puts("[fan_out_step item=#{item}] Has inner_secret? #{Map.has_key?(ctx, :inner_secret)}")
    IO.puts("[fan_out_step item=#{item}] Has outer_only? #{Map.has_key?(ctx, :outer_only)}")
    
    if Map.has_key?(ctx, :outer_only) and not Map.has_key?(ctx, :inner_secret) do
      IO.puts("\n!!! BUG REPRODUCED: fan_out_step received outer context instead of inner context !!!\n")
    end
    
    {:ok, %{item: item, processed: true}}
  end

  @job queue: :default, recorded: true
  def final_step(ctx) do
    IO.puts("[final_step] Context keys: #{inspect(Map.keys(ctx))}")
    {:ok, %{finalized: true}}
  end

  @job queue: :default, recorded: true
  def notify(ctx) do
    IO.puts("[notify] Context keys: #{inspect(Map.keys(ctx))}")
    {:ok, %{notified: true}}
  end

  # ============================================================================
  # Runner
  # ============================================================================

  def run do
    # Insert the workflow
    {:ok, jobs} = Oban.insert_all(outer_workflow())
    
    IO.puts("\n=== Inserted #{length(jobs)} jobs ===\n")
    
    # Find and display the fan-out jobs
    fan_out_jobs = Enum.filter(jobs, fn job ->
      job.args["fun"] == "fan_out_step"
    end)
    
    IO.puts("=== Fan-out job analysis ===\n")
    
    for job <- fan_out_jobs do
      IO.puts("Job #{job.id} (#{job.args["fun"]}):")
      IO.puts("  workflow_id:     #{job.meta["workflow_id"]}")
      IO.puts("  sup_workflow_id: #{job.meta["sup_workflow_id"]}")
      IO.puts("  sub_name:        #{job.meta["sub_name"]}")
      IO.puts("")
    end
    
    # Find context jobs to show which workflow_id has which context
    context_jobs = Enum.filter(jobs, fn job ->
      job.meta["context"] == true
    end)
    
    IO.puts("=== Context job analysis ===\n")
    
    for job <- context_jobs do
      # Decode the return value to show the context
      {:ok, decoded} = decode_return(job.meta["return"])
      
      IO.puts("Context job #{job.id}:")
      IO.puts("  workflow_id: #{job.meta["workflow_id"]}")
      IO.puts("  sub_name:    #{job.meta["sub_name"] || "(root)"}")
      IO.puts("  context:     #{inspect(decoded)}")
      IO.puts("")
    end
    
    IO.puts("""
    =============================================================================
    EXPECTED BEHAVIOR:
    - Fan-out jobs should have sup_workflow_id pointing to the "inner" sub-workflow
    - This would allow them to find the context with inner_secret and oauth_connection_id
    
    ACTUAL BEHAVIOR:
    - Fan-out jobs have sup_workflow_id pointing to the ROOT workflow
    - They find the outer context with outer_only and campaign_id instead
    =============================================================================
    """)
    
    {:ok, jobs}
  end
  
  defp decode_return(encoded) do
    # Oban Pro uses Base64-encoded ETF for recorded values
    encoded
    |> Base.decode64!()
    |> :erlang.binary_to_term()
    |> then(&{:ok, &1})
  rescue
    _ -> {:error, :decode_failed}
  end
end

# ============================================================================
# Instructions for running
# ============================================================================

IO.puts("""
================================================================================
Oban Pro v1.6.10 - Nested Sub-Workflow Context Bug Reproducer
================================================================================

To run this reproducer:

1. Add this file to your project
2. Run: mix run oban_pro_nested_context_bug.ex
   Or in iex: ReproducerWorkflows.run()

3. Observe that fan-out jobs have incorrect sup_workflow_id

To verify the bug at runtime:
1. Let the workflow execute
2. Check the logs - fan_out_step will report having outer_only instead of inner_secret

Environment:
- oban_pro ~> 1.6.10
- oban ~> 2.18 (or compatible version)
================================================================================
""")