load("@inference_pip//:requirements.bzl", "requirement")
load("@rules_python//python:pip.bzl", "compile_pip_requirements")
load("@rules_python//python:py_binary.bzl", "py_binary")
load("@rules_python//python:py_library.bzl", "py_library")
load("@rules_python//python:py_test.bzl", "py_test")
load(":python_zip.bzl", "python_zip_file")

compile_pip_requirements(
  name = "requirements",
  src = "requirements.in",
  requirements_txt = "requirements_lock.txt",
)

exports_files(
  ["litellm_config.yaml"],
  visibility = ["//mrjunejune:__pkg__"],
)

py_binary(
  name = "sdk_probe",
  srcs = ["sdk_probe.py"],
  deps = [
    requirement("github-copilot-sdk"),
    requirement("litellm"),
  ],
)

py_binary(
  name = "litellm_proxy",
  srcs = ["litellm_proxy.py"],
  data = ["litellm_config.yaml"],
  deps = [
    requirement("fastapi"),
    requirement("litellm"),
    requirement("pyyaml"),
    requirement("uvicorn"),
  ],
  visibility = ["//mrjunejune:__pkg__"],
)

py_binary(
  name = "copilot_sidecar",
  srcs = ["copilot_sidecar.py"],
  args = ["$(rootpath @copilot_cli_linux_x86_64//:copilot)"],
  data = [
    "@copilot_cli_linux_x86_64//:copilot",
    "//mrjunejune/assistant:all_files",
  ],
  deps = [
    requirement("github-copilot-sdk"),
    ":public_knowledge",
  ],
  visibility = ["//mrjunejune:__pkg__"],
)

py_binary(
  name = "mock_sidecar",
  srcs = ["mock_sidecar.py"],
  data = ["mock_responses.json"],
  visibility = ["//mrjunejune:__pkg__"],
)

python_zip_file(
  name = "copilot_sidecar_zip",
  binary = ":copilot_sidecar",
  visibility = ["//mrjunejune:__pkg__"],
)

python_zip_file(
  name = "litellm_proxy_zip",
  binary = ":litellm_proxy",
  visibility = ["//mrjunejune:__pkg__"],
)

filegroup(
  name = "production_runtime_binaries",
  srcs = [
    ":copilot_sidecar_zip",
    ":litellm_proxy_zip",
  ],
  visibility = ["//mrjunejune:__pkg__"],
)

py_test(
  name = "sdk_import_test",
  srcs = ["sdk_import_test.py"],
  deps = [
    requirement("github-copilot-sdk"),
    requirement("litellm"),
  ],
)

py_test(
  name = "litellm_config_test",
  srcs = ["litellm_config_test.py"],
  data = ["litellm_config.yaml"],
  deps = [requirement("pyyaml")],
)

py_test(
  name = "litellm_gateway_test",
  srcs = [
    "litellm_gateway_test.py",
    "litellm_proxy.py",
  ],
  main = "litellm_gateway_test.py",
  deps = [
    requirement("fastapi"),
    requirement("httpx"),
    requirement("litellm"),
    requirement("pyyaml"),
    requirement("uvicorn"),
  ],
)

py_test(
  name = "mock_sidecar_test",
  srcs = [
    "mock_sidecar.py",
    "mock_sidecar_test.py",
  ],
  data = ["mock_responses.json"],
  main = "mock_sidecar_test.py",
)

py_test(
  name = "sdk_runtime_test",
  srcs = ["sdk_runtime_test.py"],
  args = ["$(rootpath @copilot_cli_linux_x86_64//:copilot)"],
  data = ["@copilot_cli_linux_x86_64//:copilot"],
  deps = [requirement("github-copilot-sdk")],
)

py_test(
  name = "sdk_provider_integration_test",
  srcs = ["sdk_provider_integration_test.py"],
  args = ["$(rootpath @copilot_cli_linux_x86_64//:copilot)"],
  data = ["@copilot_cli_linux_x86_64//:copilot"],
  deps = [requirement("github-copilot-sdk")],
  size = "medium",
  timeout = "moderate",
)

py_test(
  name = "copilot_sidecar_test",
  srcs = [
    "copilot_sidecar.py",
    "copilot_sidecar_test.py",
  ],
  main = "copilot_sidecar_test.py",
  data = ["//mrjunejune/assistant:all_files"],
  deps = [
    ":public_knowledge",
    requirement("github-copilot-sdk"),
  ],
)

py_library(
  name = "public_knowledge",
  srcs = ["public_knowledge.py"],
  visibility = [
    "//mrjunejune:__pkg__",
    "//mrjunejune/inference:__pkg__",
  ],
)

py_test(
  name = "public_knowledge_test",
  srcs = [
    "public_knowledge.py",
    "public_knowledge_test.py",
  ],
  main = "public_knowledge_test.py",
  data = ["//mrjunejune/assistant:all_files"],
)