ethanaobrien/ew
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add ability to specify protocol with command line argument

Browse source
This commit is contained in:
Ethan O'Brien 2024-05-04 14:44:26 -05:00
parent 32b5f5b68c
commit 1cec98a01e
2 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/main.rs
View file

@ -3,6 +3,8 @@ mod router;
mod sql;
use json::object;
use actix_web::{
App,
HttpServer,
get,
HttpResponse,
HttpRequest,
@ -28,7 +30,6 @@ fn unhandled(req: HttpRequest, body: String) -> HttpResponse {
}
async fn request(req: HttpRequest, body: String) -> HttpResponse {
//let origbody = body.clone();
if req.method() == "POST" {
match req.path() {
"/v1.0/auth/initialize" => router::gree::initialize(req, body),
@ -138,12 +139,8 @@ async fn js(_req: HttpRequest) -> HttpResponse {
.body(include_str!("../webui/dist/index.js"))
}
#[actix_web::main]
async fn main() -> std::io::Result<()> {
use actix_web::{App, HttpServer};
let rv = HttpServer::new(|| App::new()
.wrap_fn(|req, srv| {
println!("Request: {}", req.path());

9
src/router/gree.rs
View file

@ -2,6 +2,7 @@ use crate::router::global;
use actix_web::{HttpResponse, HttpRequest, http::header::HeaderValue, http::header::ContentType, http::header::HeaderMap};
use base64::{Engine as _, engine::general_purpose};
use std::collections::HashMap;
use std::env;
use sha1::Sha1;
use substring::Substring;
use json::{object, JsonValue};
@ -335,6 +336,12 @@ pub fn migration_password_register(req: HttpRequest, body: String) -> HttpRespon
send(req, resp)
}
fn get_protocol() -> String {
if env::args().nth(1).unwrap_or(String::new()) == String::from("https") {
return String::from("https");
}
String::from("http")
}
fn gree_authorize(req: &HttpRequest) -> String {
type HmacSha1 = Hmac<Sha1>;
@ -357,7 +364,7 @@ fn gree_authorize(req: &HttpRequest) -> String {
}
let hostname = req.headers().get("host").unwrap_or(&blank_header).to_str().unwrap_or("");
let current_url = format!("http://{}{}", hostname, req.path());
let current_url = format!("{}://{}{}", get_protocol(), hostname, req.path());
let uri = req.uri().to_string();
let extra = if uri.contains("?") {
format!("&{}", uri.split('?').nth(1).unwrap_or(""))