qt-build-utils 0.9.1

// SPDX-FileCopyrightText: 2025 Klarälvdalens Datakonsult AB, a KDAB Group company <info@kdab.com>
// SPDX-FileContributor: Andrew Hayzen <andrew.hayzen@kdab.com>
//
// SPDX-License-Identifier: MIT OR Apache-2.0

use crate::{utils, Initializer, QtInstallation, QtTool};

use semver::Version;
use std::{
    ffi::{OsStr, OsString},
    path::{Path, PathBuf},
    process::Command,
};

/// A wrapper around the [rcc](https://doc.qt.io/qt-6/resources.html) tool
pub struct QtToolRcc {
    executable: PathBuf,
    qt_version: Version,
    custom_args: Vec<OsString>,
}

impl QtToolRcc {
    /// Construct a [QtToolRcc] from a given [QtInstallation]
    pub fn new(qt_installation: &dyn QtInstallation) -> Self {
        let executable = qt_installation
            .try_find_tool(QtTool::Rcc)
            .expect("Could not find rcc");

        // Ensure that the executable works
        utils::check_executable_help(&executable).unwrap();

        let qt_version = qt_installation.version();

        Self {
            executable,
            qt_version,
            custom_args: Vec::new(),
        }
    }

    /// Append custom arguments to the end of the rcc invocation.
    pub fn custom_args(mut self, args: impl IntoIterator<Item = impl AsRef<OsStr>>) -> Self {
        self.custom_args = args
            .into_iter()
            .map(|s| s.as_ref().to_os_string())
            .collect();
        self
    }

    /// Run [rcc](https://doc.qt.io/qt-6/resources.html) on a .qrc file and save the output into [cargo's OUT_DIR](https://doc.rust-lang.org/cargo/reference/environment-variables.html).
    /// The path to the generated C++ file is returned, which can then be passed to [cc::Build::files](https://docs.rs/cc/latest/cc/struct.Build.html#method.file).
    /// This function also returns a String that contains the name of the resource initializer
    /// function.
    /// The build system must ensure that if the .cpp file is built into a static library, either
    /// the `+whole-archive` flag is used, or the initializer function is called by the
    /// application.
    pub fn compile(&self, input_file: impl AsRef<Path>) -> Initializer {
        let input_path = input_file.as_ref();
        let output_folder = QtTool::Rcc.writable_path();
        std::fs::create_dir_all(&output_folder).expect("Could not create qrc dir");
        let mut output_path = output_folder.join(input_path.file_name().unwrap());
        output_path.set_extension("cpp");
        let name = input_path
            .file_name()
            .unwrap()
            .to_string_lossy()
            .replace('.', "_");

        let mut args = vec![
            OsString::from(input_path),
            OsString::from("-o"),
            OsString::from(&output_path),
            OsString::from("--name"),
            OsString::from(&name),
        ];
        args.extend(self.custom_args.iter().cloned());

        let cmd = Command::new(&self.executable)
            // Binaries should work without environment and this prevents
            // LD_LIBRARY_PATH from causing different Qt version clashes
            .env_clear()
            .args(args)
            .output()
            .unwrap_or_else(|_| panic!("rcc failed for {}", input_path.display()));

        if !cmd.status.success() {
            panic!(
                "rcc failed for {}:\n{}",
                input_path.display(),
                String::from_utf8_lossy(&cmd.stderr)
            );
        }

        let qt_6_5 = Version::new(6, 5, 0);
        let init_header = if self.qt_version >= qt_6_5 {
            // With Qt6.5 the Q_INIT_RESOURCE macro is in the QtResource header
            "QtCore/QtResource"
        } else {
            "QtCore/QDir"
        };
        Initializer {
            file: Some(output_path),
            init_call: Some(format!("Q_INIT_RESOURCE({name});")),
            init_declaration: Some(format!("#include <{init_header}>")),
        }
    }

    /// Run [rcc](https://doc.qt.io/qt-6/resources.html) on a .qrc file and return the paths of the sources
    pub fn list(&self, input_file: impl AsRef<Path>) -> Vec<PathBuf> {
        // Add the qrc file contents to the cargo rerun list
        let input_path = input_file.as_ref();
        let cmd_list = Command::new(&self.executable)
            .args(["--list", input_path.to_str().unwrap()])
            // Binaries should work without environment and this prevents
            // LD_LIBRARY_PATH from causing different Qt version clashes
            .env_clear()
            .output()
            .unwrap_or_else(|_| panic!("rcc --list failed for {}", input_path.display()));

        if !cmd_list.status.success() {
            panic!(
                "rcc --list failed for {}:\n{}",
                input_path.display(),
                String::from_utf8_lossy(&cmd_list.stderr)
            );
        }

        String::from_utf8_lossy(&cmd_list.stdout)
            .split_terminator('\n')
            .map(PathBuf::from)
            .collect()
    }
}