ccxt_client

defmodule CCXT.WS.Generator do
  @moduledoc """
  Macro that generates WebSocket subscription modules from spec files.

  This module generates `watch_*_subscription/2` functions that build
  subscription messages for WebSocket channels. The actual WebSocket
  connection management is handled by a separate GenServer adapter (W6).

  ## Usage

      defmodule CCXT.Bybit.WS do
        use CCXT.WS.Generator, spec: "bybit"
      end

  This generates:
  - Subscription builders for all `watch_*` methods in the spec
  - Introspection functions for pattern and channel info
  - Pure functions returning subscription message maps

  ## Generated Functions

  For each `watch_*` method in `spec.ws.has`:

      watch_ticker_subscription(symbol, opts \\\\ [])
      #=> {:ok, %{channel: "tickers.BTCUSDT", message: %{"op" => "subscribe", ...}}}

      watch_order_book_subscription(symbol, opts \\\\ [])
      watch_trades_subscription(symbol, opts \\\\ [])
      watch_balance_subscription(opts \\\\ [])  # auth required
      watch_orders_subscription(symbol, opts \\\\ [])  # auth required

  ## Return Format

  All subscription functions return:

      {:ok, %{
        channel: String.t(),
        message: map(),
        method: atom(),
        auth_required: boolean()
      }}

  The consumer (GenServer in W6) uses the message for the WebSocket send
  and the channel for routing incoming messages.

  ## Introspection

  - `__ccxt_ws_spec__/0` - WS portion of the spec
  - `__ccxt_ws_pattern__/0` - Subscription pattern atom
  - `__ccxt_ws_channels__/0` - Channel template map

  """

  alias CCXT.Generator.SpecLoader
  alias CCXT.WS.Generator.Adapter
  alias CCXT.WS.Generator.Functions

  @doc false
  # Derives the REST module from the WS module name
  # CCXT.Bybit.WS -> CCXT.Bybit
  @spec derive_rest_module(module()) :: module()
  def derive_rest_module(ws_module) do
    ws_module
    |> Module.split()
    |> List.delete_at(-1)
    |> Module.concat()
  end

  @doc """
  Generates a WebSocket module from a spec file.

  ## Options

  - `:spec` - The exchange ID (e.g., "bybit"). Loads from `priv/specs/{spec}.exs`
  - `:spec_path` - Full path to spec file (alternative to `:spec`)

  ## Example

      defmodule CCXT.Bybit.WS do
        use CCXT.WS.Generator, spec: "bybit"
      end

  """
  @spec __using__(keyword()) :: Macro.t()
  defmacro __using__(opts) do
    spec_id = Keyword.get(opts, :spec)
    spec_path = Keyword.get(opts, :spec_path)

    quote do
      alias CCXT.WS.Generator

      require Generator

      Generator.__generate__(unquote(spec_id), unquote(spec_path))
    end
  end

  @doc false
  # Builds symbol_context from spec and injects it into ws_config.
  # symbol_context enables exchange-aware symbol formatting in WS patterns
  # (e.g., "BTC/USDT" → "BTC-USDT" for OKX instead of naive "BTCUSDT").
  @spec inject_symbol_context(map(), map()) :: map()
  def inject_symbol_context(ws_config, spec) do
    symbol_context = %{
      symbol_patterns: Map.get(spec, :symbol_patterns),
      symbol_format: Map.get(spec, :symbol_format),
      symbol_formats: Map.get(spec, :symbol_formats),
      currency_aliases: Map.get(spec, :currency_aliases, %{})
    }

    has_symbol_data =
      symbol_context.symbol_patterns ||
        symbol_context.symbol_format ||
        symbol_context.symbol_formats

    if has_symbol_data do
      Map.put(ws_config, :symbol_context, symbol_context)
    else
      ws_config
    end
  end

  @doc false
  @spec __generate__(String.t() | nil, String.t() | nil) :: Macro.t()
  defmacro __generate__(spec_id, spec_path) do
    # Resolve spec path at compile time
    resolved_path = SpecLoader.resolve_spec_path(spec_id, spec_path)

    # Load spec at compile time (use same loader as REST)
    spec = SpecLoader.load_and_validate_spec!(resolved_path)

    # Extract WS config (use Map.get for struct access)
    ws_config = Map.get(spec, :ws) || %{}

    # Skip generation if no WS support (check BEFORE injecting symbol_context,
    # which would make an empty map non-empty and bypass this check)
    if ws_config == %{} do
      quote do
        @moduledoc "WebSocket not supported for this exchange."
        def __ccxt_ws_spec__, do: nil
        def __ccxt_ws_pattern__, do: nil
        def __ccxt_ws_channels__, do: nil
      end
    else
      # Inject symbol context for exchange-aware symbol formatting
      ws_config = inject_symbol_context(ws_config, spec)

      # Generate moduledoc
      moduledoc = Functions.generate_moduledoc(spec)

      # Derive REST module from WS module name (CCXT.Bybit.WS -> CCXT.Bybit)
      rest_module = derive_rest_module(__CALLER__.module)

      # Generate adapter AST (will be injected as nested module)
      spec_id = Map.get(spec, :id)

      # Read pipeline config for conditional normalization/validation generation.
      # Uses get_env because compile_env can only be called in module body.
      app = Mix.Project.config()[:app]

      # credo:disable-for-lines:3 Credo.Check.Refactor.Apply
      pipeline_default =
        if Code.ensure_loaded?(CCXT.Pipeline) and function_exported?(CCXT.Pipeline, :default, 0),
          do: apply(CCXT.Pipeline, :default, []),
          else: []

      pipeline = Application.get_env(app, :pipeline, pipeline_default)

      adapter_ast = Adapter.generate_adapter(__CALLER__.module, rest_module, ws_config, spec_id, pipeline)

      quote do
        # Track spec file for recompilation
        @external_resource unquote(resolved_path)

        # Suppress Dialyzer for generated functions
        @dialyzer [:no_return, :no_match]

        @moduledoc unquote(moduledoc)

        # Store WS spec as module attribute
        @ws_spec unquote(Macro.escape(ws_config))

        # Generate introspection functions
        unquote(Functions.generate_introspection(ws_config))

        # Generate watch_* subscription functions (timeframes passed separately for OHLCV translation)
        unquote(Functions.generate_watch_functions(ws_config, Map.get(spec, :timeframes, %{})))

        # Generate the Adapter nested module
        defmodule Adapter do
          @moduledoc false
          unquote(adapter_ast)
        end
      end
    end
  end
end