unity-solution-generator 0.1.1

Regenerates Unity .csproj/.sln files from asmdef/asmref layout without launching the Unity editor.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112

//! Test fixture helpers — port of `SolutionGeneratorTests` setup helpers.
//! `#![allow(dead_code)]` because each `tests/*.rs` is a separate compile unit
//! and any unused helper produces a per-binary warning. Tests as a whole use
//! all of these.
#![allow(dead_code)]

use std::collections::BTreeSet;
use std::fs;
use std::path::Path;

use tempfile::TempDir;

pub fn make_temp_root() -> TempDir {
    tempfile::tempdir().expect("tempdir")
}

pub fn write_file(root: &Path, rel: &str, content: &str) {
    let path = root.join(rel);
    if let Some(parent) = path.parent() {
        fs::create_dir_all(parent).unwrap();
    }
    fs::write(&path, content).unwrap();
}

pub fn read_file(root: &Path, rel: &str) -> String {
    fs::read_to_string(root.join(rel)).unwrap()
}

fn xml_unescape(value: &str) -> String {
    value
        .replace(""", "\"")
        .replace("'", "'")
        .replace("&lt;", "<")
        .replace("&gt;", ">")
        .replace("&amp;", "&")
}

/// Reads `<Compile Include="..."/>` patterns from a csproj and expands them
/// against the given root, mirroring `SolutionGeneratorTests.readCompileSet`.
pub fn read_compile_set(root: &Path, csproj_rel: &str) -> BTreeSet<String> {
    let content = read_file(root, csproj_rel);
    let mut out = BTreeSet::new();
    let mut i = 0;
    let needle = "<Compile Include=\"";
    let bytes = content.as_bytes();
    while i + needle.len() < bytes.len() {
        let Some(start) = content[i..].find(needle) else {
            break;
        };
        let pat_start = i + start + needle.len();
        let Some(end_off) = content[pat_start..].find('"') else {
            break;
        };
        let pattern = xml_unescape(&content[pat_start..pat_start + end_off]);
        out.extend(expand_pattern(&pattern, root));
        i = pat_start + end_off + 1;
    }
    out
}

fn expand_pattern(pattern: &str, root: &Path) -> Vec<String> {
    let mut stripped = pattern.to_string();
    while stripped.starts_with("../") {
        stripped = stripped[3..].to_string();
    }
    if let Some(dir) = stripped.strip_suffix("/*.cs") {
        return list_cs_files(root, dir);
    }
    if stripped.ends_with(".cs") {
        let path = stripped.replace('\\', "/");
        if root.join(&path).exists() {
            return vec![path];
        }
        return Vec::new();
    }
    Vec::new()
}

fn list_cs_files(root: &Path, rel: &str) -> Vec<String> {
    let dir = if rel.is_empty() {
        root.to_path_buf()
    } else {
        root.join(rel)
    };
    if !dir.exists() {
        return Vec::new();
    }
    let Ok(rd) = fs::read_dir(&dir) else {
        return Vec::new();
    };
    rd.filter_map(|e| e.ok())
        .filter(|e| e.file_name().to_string_lossy().ends_with(".cs"))
        .map(|e| {
            let name = e.file_name().to_string_lossy().into_owned();
            if rel.is_empty() {
                name
            } else {
                format!("{}/{}", rel, name)
            }
        })
        .collect()
}

pub fn assert_compile_set(root: &Path, csproj_rel: &str, expected: &[&str]) {
    let actual = read_compile_set(root, csproj_rel);
    let exp: BTreeSet<String> = expected.iter().map(|s| (*s).to_string()).collect();
    assert_eq!(
        actual, exp,
        "compile set mismatch in {}\n  actual:   {:?}\n  expected: {:?}",
        csproj_rel, actual, exp
    );
}