auth (maybe?)

This commit is contained in:
Alex Janka 2023-12-25 21:22:08 +11:00
parent 871ba961cc
commit ed2fee6571
6 changed files with 2200 additions and 3 deletions

2095
Cargo.lock generated

File diff suppressed because it is too large Load diff

View file

@ -24,3 +24,9 @@ assets = [
]
[dependencies]
clap = { version = "4.0", features = ["derive"] }
ron = "0.8"
serde = { version = "1.0", features = ["derive"] }
tokio = { version = "1.35.1", features = ["full"] }
teslatte = "0.1.11"
thiserror = "1.0"

2
debian/service vendored
View file

@ -8,7 +8,7 @@ Type=simple
Restart=always
RestartSec=1
User=tesla
ExecStart=/usr/bin/tesla-charge-controller
ExecStart=/usr/bin/tesla-charge-controller watch
[Install]
WantedBy=multi-user.target

2
rust-toolchain.toml Normal file
View file

@ -0,0 +1,2 @@
[toolchain]
channel = "nightly"

7
src/config.rs Normal file
View file

@ -0,0 +1,7 @@
use serde::{Deserialize, Serialize};
#[derive(Serialize, Deserialize, Clone, Debug)]
pub struct Config {}
#[derive(Serialize, Deserialize, Clone, Debug)]
pub struct Auth {}

View file

@ -1,3 +1,90 @@
fn main() {
println!("Hello, world!");
use std::{io::BufRead, path::PathBuf};
use clap::{Parser, Subcommand};
use teslatte::{auth::AccessToken, OwnerApi};
use thiserror::Error;
mod config;
#[derive(Parser, Debug, Clone)]
#[clap(author, version, about, long_about = None)]
struct Args {
#[command(subcommand)]
command: Commands,
#[clap(long, default_value = "/etc/tesla-charge-controller")]
config_dir: PathBuf,
}
#[derive(Subcommand, Debug, Clone)]
enum Commands {
/// Run charge controller server
Watch,
/// Authenticate with Tesla login
Auth {},
}
fn press_y_to_continue() -> bool {
println!("Continue? [y/N]");
let mut line = String::new();
let stdin = std::io::stdin();
stdin.lock().read_line(&mut line).unwrap();
if line.to_uppercase() == "Y\n" {
true
} else {
println!("Exiting now!");
false
}
}
#[tokio::main]
async fn main() {
let args = Args::parse();
let auth_path = args.config_dir.join("auth");
let config_path = args.config_dir.join("config");
match args.command {
Commands::Auth {} => {
if auth_path.exists() {
println!("Auth file already exists");
if !press_y_to_continue() {
return;
}
}
if let Err(e) = log_in(auth_path).await {
println!("Error: {e:#?}")
}
}
Commands::Watch => match get_auth(auth_path) {
Ok(api) => {
println!("got products: {:#?}", api.products().await)
}
Err(e) => println!("error getting auth: {e:#?}"),
},
}
}
#[derive(Error, Debug)]
pub enum TeslaError {
#[error("teslatte error")]
Teslatte(#[from] teslatte::error::TeslatteError),
#[error("ron error")]
Ron(#[from] ron::Error),
#[error("stdio error")]
StdIo(#[from] std::io::Error),
#[error("ron - spanned error")]
RonSpanned(#[from] ron::error::SpannedError),
#[error("unknown error")]
Unknown,
}
fn get_auth(auth_path: PathBuf) -> Result<OwnerApi, TeslaError> {
let key: AccessToken = ron::from_str(&std::fs::read_to_string(auth_path)?)?;
Ok(OwnerApi::new(key, None))
}
async fn log_in(auth_path: PathBuf) -> Result<(), TeslaError> {
let v = OwnerApi::from_interactive_url().await?;
std::fs::write(auth_path, ron::ser::to_string(&v.access_token)?)?;
println!("Auth successfully saved");
Ok(())
}