yan/archived-tesla-auth
1
0
Fork 0
mirror of https://github.com/adriankumpf/tesla_auth.git synced 2026-02-27 09:54:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Extract functions

Browse Source
This commit is contained in:
Adrian Kumpf
2021-10-02 23:26:44 +02:00
parent 04ead6c686
commit b39e58f4bb
1 changed files with 55 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main.rs
+55 -47
View File
@@ -55,17 +55,7 @@ struct Args {
fn main() -> anyhow::Result<()> {
let args: Args = argh::from_env();
let level_filter = if args.debug {
LevelFilter::Debug
} else {
LevelFilter::Error
};
SimpleLogger::new()
.with_level(LevelFilter::Off)
.with_module_level("reqwest", level_filter)
.with_module_level("tesla_auth", level_filter)
.init()?;
init_logger(args.debug)?;
let event_loop = EventLoop::<CustomEvent>::with_user_event();
let event_proxy = event_loop.create_proxy();
@@ -105,44 +95,12 @@ fn main() -> anyhow::Result<()> {
Event::UserEvent(CustomEvent::Failure(e)) => {
error!("{}", e);
let script = r#"
const html = `
<h4 style="text-align: center;">An error occured. Please try again ...</h4>
<p style="text-align: center;color:red;margin-bottom:20px;">{msg}</p>
`;
document.querySelector("h1.h1").outerHTML = html;
"#
.replace("{msg}", &e.to_string());
webview.evaluate_script(&script).unwrap();
webview.evaluate_script(&render_error_view(e)).unwrap();
}
Event::UserEvent(CustomEvent::Tokens(tokens)) => {
println!("{}", tokens);
let script = r#"
const html = `
<h4 style="text-align: center;">Access Token</h4>
<textarea readonly onclick="this.setSelectionRange(0, this.value.length)"
cols="140" rows="12" style="resize:none;padding:4px;font-size:0.9em;"
>{access_token}</textarea>
<h4 style="text-align: center;">Refresh Token</h4>
<textarea readonly onclick="this.setSelectionRange(0, this.value.length)"
cols="140" rows="12" style="resize:none;padding:4px;font-size:0.9em;"
>{refresh_token}</textarea>
<small style="margin-top:12px;margin-bottom:20px;text-align:center;color:seagreen;">
Valid for {expires_in}
</small>
`;
document.querySelector("h1.h1").outerHTML = html;
"#
.replace("{access_token}", tokens.access.secret())
.replace("{refresh_token}", tokens.refresh.secret())
.replace("{expires_in}", &format!("{}", tokens.expires_in));
webview.evaluate_script(&script).unwrap();
Event::UserEvent(CustomEvent::Tokens(t)) => {
println!("{}", t);
webview.evaluate_script(&render_tokens_view(t)).unwrap();
}
_ => (),
@@ -150,6 +108,22 @@ fn main() -> anyhow::Result<()> {
});
}
fn init_logger(debug: bool) -> anyhow::Result<()> {
let level_filter = if debug {
LevelFilter::Debug
} else {
LevelFilter::Error
};
SimpleLogger::new()
.with_level(LevelFilter::Off)
.with_module_level("reqwest", level_filter)
.with_module_level("tesla_auth", level_filter)
.init()?;
Ok(())
}
fn build_menu() -> (MenuBar, MenuId) {
let mut menu_bar = MenuBar::new();
let mut menu = MenuBar::new();
@@ -212,3 +186,37 @@ fn parse_url(params: Value) -> anyhow::Result<Url> {
_ => Err(anyhow!("Invalid url param!")),
}
}
fn render_error_view(error: anyhow::Error) -> String {
r#"
const html = `
<h4 style="text-align: center;">An error occured. Please try again ...</h4>
<p style="text-align: center;color:red;margin-bottom:20px;">{msg}</p>
`;
document.querySelector("h1.h1").outerHTML = html;
"#
.replace("{msg}", &error.to_string())
}
fn render_tokens_view(tokens: auth::Tokens) -> String {
r#"
const html = `
<h4 style="text-align: center;">Access Token</h4>
<textarea readonly onclick="this.setSelectionRange(0, this.value.length)"
cols="140" rows="12" style="resize:none;padding:4px;font-size:0.9em;"
>{access_token}</textarea>
<h4 style="text-align: center;">Refresh Token</h4>
<textarea readonly onclick="this.setSelectionRange(0, this.value.length)"
cols="140" rows="12" style="resize:none;padding:4px;font-size:0.9em;"
>{refresh_token}</textarea>
<small style="margin-top:12px;margin-bottom:20px;text-align:center;color:seagreen;">
Valid for {expires_in}
</small>
`;
document.querySelector("h1.h1").outerHTML = html;
"#
.replace("{access_token}", tokens.access.secret())
.replace("{refresh_token}", tokens.refresh.secret())
.replace("{expires_in}", &format!("{}", tokens.expires_in))
}