xzam 0.1.0

Shazam implementation
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

use clap::{Parser, Subcommand, command};
use xzam::{score_sample, store_audio};

// Clap's configuration
#[derive(Parser)]
#[command(name = "kvs", version, about, author)]
struct Args {
    /// Sub commands for kvs
    #[clap(subcommand)]
    cmd: Subcommands,
}

#[derive(Subcommand)]
enum Subcommands {
    // Add audio to db
    StoreAudio {
        // The name of the audio, use track_id style eg jcole_amara_23
        track_id: String,
        // the path relative to test_data eg amapiano.wav
        track_path: String,
    },
    // Check score of sample
    FindSample {
        // the path relative to test_data eg amapiano_sample.wav
        track_path: String,
    },
}

fn main() {
    let env_vars = Args::parse();

    match env_vars.cmd {
        Subcommands::StoreAudio {
            track_id,
            track_path,
        } => {
            let res = store_audio(&track_id, &track_path);
            match res {
                Ok(()) => {
                    println!("Track - {track_id} saved successfully");
                }
                Err(e) => {
                    eprintln!("{:?}", e);
                }
            }
        }
        Subcommands::FindSample { track_path } => {
            let res = score_sample(&track_path);
            match res {
                Ok((track_id, score, delta)) => {
                    println!(
                        "Winning audio has track_id: {} with score: {}; delta suggests offset: {}",
                        track_id, score, delta
                    );
                }
                Err(e) => {
                    eprintln!("Unable to score {:?}", e);
                }
            }
        }
    }
}