defmodule ExTorch.Vision.MixProject do
  use Mix.Project

  def project do
    [
      app: :extorch_vision,
      version: "0.1.0",
      elixir: "~> 1.16",
      start_permanent: Mix.env() == :prod,
      deps: deps(),
      description: description(),
      package: package(),
      docs: [main: "ExTorch.Vision", extras: ["README.md"]]
    ]
  end

  def application do
    [
      extra_applications: [:logger]
    ]
  end

  defp deps do
    [
      extorch_dep(),
      {:ex_doc, "~> 0.35", only: :dev, runtime: false}
    ]
  end

  defp description do
    "TorchVision ops for ExTorch — NMS, ROI Align, deformable convolution, " <>
      "and image I/O on the BEAM. Builds libtorchvision from source, " <>
      "no C++ code in this package."
  end

  defp package do
    [
      name: "extorch_vision",
      files: ~w(lib mix.exs README* LICENSE* CHANGELOG*),
      licenses: ["MIT"],
      links: %{
        "GitHub" => "https://github.com/andfoy/extorch_vision",
        "ExTorch" => "https://github.com/andfoy/extorch"
      }
    ]
  end

  # Use a local path for development, Hex for CI/release.
  # Set EXTORCH_PATH=../extorch for local dev,
  # or unset it to pull from Hex.
  defp extorch_dep do
    case System.get_env("EXTORCH_PATH") do
      nil -> {:extorch, "~> 0.4.0"}
      path -> {:extorch, path: path}
    end
  end
end