rbox 0.1.7

Rust library for interacting with the local and export data of Pioneers Rekordbox DJ software
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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136

// Copyright (C) 2026 Dylan Jones
// SPDX-License-Identifier: GPL-3.0-only

#![allow(dead_code)]

use std::env;
use std::fs;
use std::path::{Path, PathBuf};
use tempfile::{tempdir_in, NamedTempFile, TempDir};

/// Copy a directory tree from src to dst.
fn copy_tree(src: impl AsRef<Path>, dst: impl AsRef<Path>) -> std::io::Result<()> {
    for entry in fs::read_dir(src)? {
        let entry = entry?;
        let ty = entry.file_type()?;
        if ty.is_dir() {
            copy_tree(entry.path(), dst.as_ref().join(entry.file_name()))?;
        } else {
            fs::copy(entry.path(), dst.as_ref().join(entry.file_name()))?;
        }
    }
    Ok(())
}

/// Get the root directory of the testdata.
pub fn testdata_dir() -> PathBuf {
    let current_dir = env::current_dir().expect("Failed to get current directory");
    let root = current_dir.parent().unwrap().join(".testdata");
    root
}

/// Get the root directory of the demo rekordbox library testdata.
pub fn testdata_demo_dir(v: usize) -> PathBuf {
    let root = testdata_dir();
    root.join(format!("RBv{}", v)).join("demo")
}

/// Get the options.json file path of the demo rekordbox library testdata.
pub fn options_path(v: usize) -> PathBuf {
    let root = testdata_demo_dir(v);
    root.join("rekordboxAgent")
        .join("storage")
        .join("options.json")
}

/// Get the root directory of the MySettings testdata.
pub fn settings_dir() -> PathBuf {
    let root = testdata_dir();
    root.join("mysettings")
}

pub fn anlz_dir(v: usize) -> PathBuf {
    let root = testdata_demo_dir(v);
    root.join("rekordbox")
        .join("share")
        .join("PIONEER")
        .join("USBANLZ")
}

pub fn export_dir(v: usize) -> PathBuf {
    let root = testdata_demo_dir(v);
    root.join("export").join("PIONEER")
}

/// Get the root directory of the temp files and create it if it doesn't exist.
pub fn get_temp_root_dir() -> PathBuf {
    let tmp_root = testdata_dir();
    if !tmp_root.exists() {
        fs::create_dir_all(&tmp_root).expect("Failed to create temp root directory");
    }
    tmp_root
}

/// Create a copy of a fixture path as a temp file.
fn create_temp_file(fixture_path: &PathBuf) -> NamedTempFile {
    let tmp_root = get_temp_root_dir();
    let tmp = NamedTempFile::new_in(tmp_root).expect("Failed to create temp file");
    fs::copy(&fixture_path, tmp.path()).expect("Failed to copy file");
    tmp
}

/// Create a new temp directory.
pub fn create_temp_dir() -> TempDir {
    let tmp_root = get_temp_root_dir();
    fs::create_dir_all(&tmp_root).expect("Failed to create temp root directory");
    tempdir_in(tmp_root).expect("Failed to create temp directory")
}

/// Setup code for the master.db tests
pub fn setup_master_db_path(v: usize) -> NamedTempFile {
    let fixture_path = testdata_demo_dir(v).join("rekordbox").join("master.db");
    let tmp = create_temp_file(&fixture_path);
    tmp
}

/// Setup code for the master.db tests that need all additional files.
pub fn setup_master_db_dir_path(v: usize) -> TempDir {
    let rb_dir = testdata_demo_dir(v).join("rekordbox");

    // Recreate the temp directory with all files
    let tmp_dir = create_temp_dir();
    let dir = tmp_dir.path().join("rekordbox");
    fs::create_dir_all(&dir).expect("Failed to create temp directory");
    // Copy the database file
    let src = rb_dir.join("master.db");
    fs::copy(&src, dir.join("master.db")).expect("Failed to copy database file");
    // Copy the playlist xml file
    let src = rb_dir.join("masterPlaylists6.xml");
    fs::copy(&src, dir.join("masterPlaylists6.xml")).expect("Failed to copy xml file");

    tmp_dir
}

/// Setup code for the one-library tests
pub fn setup_one_library_path() -> NamedTempFile {
    let fixture_path = export_dir(7).join("rekordbox").join("exportLibrary.db");
    let tmp = create_temp_file(&fixture_path);
    tmp
}

/// Setup code for the anlz files
pub fn setup_anlz_files() -> TempDir {
    let root = anlz_dir(6);
    let anlz_dir = root.join("e35").join("fa187-3f34-47e2-9880-2b33cb8d1304");
    // Recreate the temp directory with all files
    let tmp_dir = create_temp_dir();
    copy_tree(&anlz_dir, tmp_dir.path()).expect("Failed to copy files");

    tmp_dir
}

/// Setup code for the database tests.
pub fn setup_rekordbox_xml_path() -> NamedTempFile {
    let fixture_path = testdata_demo_dir(6).join("database.xml");
    create_temp_file(&fixture_path)
}