cargo-dbapp 0.1.1

builds and bundles the project as databricks app
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

mod args;
mod cargo;

use crate::{
    args::{Cmd, parse_args},
    cargo::BuildCommand,
};
use anyhow::{Context, Result};
use cargo_metadata::MetadataCommand;
use std::fs;

const TARGET: &str = "x86_64-unknown-linux-musl";
const LINKER: &str = "rust-lld";

const APP_YAML_CONTENT: &str = "command: ['sh', 'start.sh']\n";
const START_SH_CONTENT: &str = r#"#!/bin/sh
DIR="$(cd "$(dirname "$0")" && pwd)"
chmod +x "$BIN/bin"
exec "$BIN/bin"
"#;

fn main() -> Result<()> {
    let args = parse_args();

    return match args.cmd {
        Cmd::Build => build(),
    };
}

fn build() -> Result<()> {
    // Getting metadata
    let metadata = MetadataCommand::new().no_deps().exec()?;
    let target_path = &metadata.target_directory;
    let name = &*metadata
        .root_package()
        .context("run inside a package")?
        .name;

    // Building the binary
    let build_path = target_path.join(TARGET).join("debug").join(name);
    BuildCommand::new()
        .release()
        .target_with_linker(TARGET, LINKER)
        .exec()?;

    // Creating the databricks directory
    let databricks_dir = target_path.join("databricks");
    fs::remove_dir_all(&databricks_dir)?;
    fs::create_dir_all(&databricks_dir)?;

    // Clopying the built binary to the databricks folder
    fs::copy(build_path, databricks_dir.join("bin"))?;

    // Writing the app.yaml and build.sh
    fs::write(databricks_dir.join("app.yaml"), APP_YAML_CONTENT)?;
    fs::write(databricks_dir.join("start.sh"), START_SH_CONTENT)?;

    return Ok(());
}