# Copyright 2023 Google LLC
#
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import("../../gn/skia.gni")
import("../third_party.gni")

# The following shells out to bazelisk to compile static library dependencies that bridge the
# vello_cpp crate Rust code and its C++ FFI symbols. This produces a single static library target
# that is reachable via the ":vello" target.
#
# Pre-compiled shader dependencies:
#
#   The vello_shaders crate bundles compute shaders that are pre-compiled into a target format based
#   on the GPU backend. This is driven by build arguments defined in //gn/skia.gni.

if (skia_enable_vello_shaders) {
  bazel_args = []
  if (is_mac && target_cpu == "arm64") {
    # TODO: Normally the target toolchain would be specified with `--platforms` but that doesn't
    # work. When building and running on an arm64 mac, setting `--host_platform` seems to do the
    # right thing but may not be the right build configuration in the long run.
    bazel_args += [ "--host_platform=//bazel/platform:mac_arm64_hermetic" ]
  }
  if (!is_debug) {
    bazel_args += [ "--compilation_mode=opt" ]
  }
  if (skia_use_metal) {
    bazel_args += [ "--define=VELLO_MSL_SHADERS=true" ]
  }
  if (skia_use_dawn) {
    bazel_args += [ "--define=VELLO_WGSL_SHADERS=true" ]
  }
  action("compile_vello_ffi") {
    script = "../../gn/bazel_build.py"
    sources = [
      "BUILD.bazel",
      "src/lib.rs",
    ]
    outputs = [ "$root_out_dir/libvello_ffi.a" ]
    args = [
             "//third_party/vello:vello_ffi",
             "../../bazel-bin/third_party/vello/libvello_ffi.a",
           ] + bazel_args
  }

  action("compile_vello_rust") {
    script = "../../gn/bazel_build.py"
    sources = [
      "BUILD.bazel",
      "src/encoding.rs",
      "src/lib.rs",
      "src/shaders.rs",
    ]
    outputs = [ "$root_out_dir/libvello_rust.a" ]
    args = [
             "//third_party/vello:vello_rust",
             "../../bazel-bin/third_party/vello/libvello_rust.a",
           ] + bazel_args
  }
}

third_party("vello") {
  public_include_dirs = [ "../../bazel-bin/" ]
  libs = [
    "$root_out_dir/libvello_ffi.a",
    "$root_out_dir/libvello_rust.a",
  ]
  deps = [
    ":compile_vello_ffi",
    ":compile_vello_rust",
  ]
}