cargo-esp-build 0.1.0

A cargo-build extension for esp8266 chips
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


use clap::Parser;
use std::{
  io,
  env,
  sync::LazyLock,
  process::Command,
};


#[derive(Parser)]
struct App {
  #[arg(long)]
  release: bool,
  #[arg(long)]
  flash: bool,
  #[arg(long)]
  dry_run: bool,
}

static TARGET: &str="xtensa-esp8266-none-elf";
static RUSTFLAGS: LazyLock<String>=LazyLock::new(|| match env::var("RUSTFLAGS") {
  Err(_)=> "-C link-arg=-nostartfiles -C link-arg=-Wl,-Tlink.x".to_owned(),
  Ok(var)=> var
});

fn main()-> io::Result<()> {
  let args=env::args().collect::<Vec<_>>();
  let trailing_idx=idx_of(&args,"--");
  let args=match trailing_idx {
    Some(trailing)=> &args[1..trailing],
    _=> &args[1..]
  };

  let app=App::parse_from(args);
  let mut process=Command::new("rustup");

  process.env("RUSTFLAGS",&*RUSTFLAGS);
  process.args(["run","esp","cargo"]);

  if app.flash {
    process.args(["espflash","flash"]);
  } else {
    process.arg("build");
  }

  if app.release {
    process.arg("--release");
  }

  process.args(["--target",TARGET]);
  if let Some(trailing_idx)=trailing_idx {
    if trailing_idx+1<args.len() {
      process.args(&args[trailing_idx+1..]);
    }
  }

  if app.dry_run {
    println!("{:#?}",process);
    return Ok(());
  }

  let status=process.status()?;
  if !status.success() {
    let code=status.code().unwrap_or(1);
    return Err(io::Error::from_raw_os_error(code));
  }

  Ok(())
}

fn idx_of<T,U: PartialEq<T>>(slice: &[T],element: U)-> Option<usize> {
  let (idx,_)=slice.iter()
  .enumerate()
  .find(|(_,entity)| element.eq(entity))?;

  Some(idx)
}