[env.development]
_MK_WASMPACK_MOD = "--dev"

[env.production]
_MK_WASMPACK_MOD = "--release"

[tasks.release]
dependencies = [
    "build_inference_models",
    "build_extension"
]

[tasks.dev]
dependencies = [
    "build_inference_models",
    "extension_npm_setup"
]
cwd = "extension"
command = "pnpm"
args = ["dev"]

[tasks.build_extension]
dependencies = [
    "build_extension_mozf",
    "zip_extension_mozf",
    "build_extension_chro",
    "zip_extension_chro"
]

[tasks.zip_extension_chro]
cwd = "extension"
command = "pnpm"
args = ["zip:chrome"]

[tasks.zip_extension_mozf]
cwd = "extension"
command = "pnpm"
args = ["zip:firefox"]

[tasks.build_extension_chro]
dependencies = [
    "extension_npm_setup"
]
cwd = "extension"
command = "pnpm"
args = ["build:chrome"]

[tasks.build_extension_mozf]
dependencies = [
    "extension_npm_setup"
]
cwd = "extension"
command = "pnpm"
args = ["build:firefox"]

[tasks.extension_npm_setup]
cwd = "extension"
command = "pnpm"
args = ["install"]

[tasks.build_inference_models]
dependencies = [
    "build_inference_models_ai_text",
    "build_inference_models_ai_text_lite"
]

[tasks.build_inference_models_ai_text]
env = {_MK_MODEL_AI_TEXT = "onnx-community/modernbert-ai-detection-raid-mage-ONNX",_MK_MODEL_AI_TEXT_ATTENTION = "yes",_MK_MODEL_AI_TEXT_TYPES = "no"}
cwd = "wasm/inference-models"
install_crate = "wasm-pack"
command = "wasm-pack"
args  = [
    "build",
    "${_MK_WASMPACK_MOD}",
    "--reference-types",
    "--target", "web",
    "--out-dir", "pkg/inference-ai-text",
    "--out-name", "inference-ai-text",
    "--features", "ai_text"
]

[tasks.build_inference_models_ai_text_lite]
env = {_MK_MODEL_AI_TEXT = "onnx-community/e5-small-lora-ai-generated-detector-ONNX",_MK_MODEL_AI_TEXT_ATTENTION = "yes",_MK_MODEL_AI_TEXT_TYPES = "yes"}
cwd = "wasm/inference-models"
install_crate = "wasm-pack"
command = "wasm-pack"
args  = [
    "build",
    "${_MK_WASMPACK_MOD}",
    "--reference-types",
    "--target", "web",
    "--out-dir", "pkg/inference-ai-text-lite",
    "--out-name", "inference-ai-text-lite",
    "--features", "ai_text"
]

[tasks.export_onnx]
dependencies = ["export_onnx_ai_text"]

[tasks.export_onnx_ai_text]
dependencies = ["exportenv"]
command = ".export_venv/bin/optimum-cli"
args = [
    "export", "onnx",
    "--model", "GeorgeDrayson/modernbert-ai-detection-raid-mage",
    "--task", "text-classification",
    "--dtype", "fp32",
    "--opset", "18",
    "--optimize", "O1",
    "--monolith",
    "--no-post-process",
    "wasm/inference-models/models/ai_text"
]

[tasks.exportenv]
dependencies = [
    "exportenv_create",
    "exportenv_setup"
]

[tasks.exportenv_create]
command = "python3"
args = ["-m", "venv", ".export_venv"]

[tasks.exportenv_setup]
command = ".export_venv/bin/pip"
args = ["install", "--upgrade", "-r", "exportenv_requirements.txt"]