dan/batflip
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Initial commit with webserver logic.

Browse Source
This commit is contained in:
calcu1on 2024-12-22 15:21:55 -05:00
parent 5f14249746
commit 6447689921
5 changed files with 117 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
.gitignore vendored
View File

@ -18,4 +18,8 @@ Cargo.lock
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore # be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
# and can be added to the global gitignore or merged into this file. For a more nuclear # and can be added to the global gitignore or merged into this file. For a more nuclear
# option (not recommended) you can uncomment the following to ignore the entire idea folder. # option (not recommended) you can uncomment the following to ignore the entire idea folder.
#.idea/ #.idea/
# Added by cargo
/target

6
Cargo.toml Normal file
View File

@ -0,0 +1,6 @@
[package]
name = "hermes"
version = "0.1.0"
edition = "2021"
[dependencies]

5
public/200.json Normal file
View File

@ -0,0 +1,5 @@
{
"SomeKey": "Some Value goes here",
"SomeOther": "thisisAOnthervalue",
"Integer": 123
}

9
public/404.html Normal file
View File

@ -0,0 +1,9 @@
<!Doctype html>
<html>
<head>
<title>Page Not Found | Hermes</title>
</head>
<body>
<p>The requested page could not be found.</p>
</body>
</html>

92
src/main.rs Normal file
View File

@ -0,0 +1,92 @@
use std::{
fs,
io::{prelude::*, BufReader},
net::{TcpListener, TcpStream},
};
#[allow(dead_code)]
struct IncomingRequest {
req_type: String,
path: String
}
// Listen on port and return response.
fn main() {
let listener = TcpListener::bind("127.0.0.1:7878").unwrap();
for stream in listener.incoming() {
let stream = stream.unwrap();
handle_connection(stream);
}
}
#[allow(unused_assignments)]
fn handle_connection(mut stream: TcpStream) {
let buf_reader = BufReader::new(&stream);
let http_request: Vec<_> = buf_reader
.lines()
.map(|result| result.unwrap())
.take_while(|line| !line.is_empty())
.collect();
let request_line: String = http_request.first().expect("Not a string").to_string();
let req_vec: Vec<&str> = request_line.split(" ").collect();
let inc_request = IncomingRequest {
req_type: req_vec[0].to_string(),
path: req_vec[1].to_string(),
};
let api_key: String = get_api_key(http_request);
let mut response = String::new();
// If not a GET request, or missing API key, throw 404.
if inc_request.req_type != "GET" || api_key.is_empty() {
response = return_404();
}
else {
// Check that the API key is in fact fact valuid.
// Not really checking right now.
response = return_200();
}
// If not present, return 404.
// Perhaps move both of those conditions to a new function.
stream.write_all(response.as_bytes()).unwrap();
}
// get the apy key from the headers.
fn get_api_key(headers: Vec<String>) -> String {
let mut api_key = String::new();
for header in headers {
// split header, strip whitespace, check if exists.
let split_header: Vec<&str> = header.split(":").collect();
let header_key: String = split_header.first().expect("Nothing here").to_string();
if header_key == "x-api-key" {
api_key = split_header[1].to_string();
}
}
api_key
}
// Return a 404.
fn return_404() -> String {
let status_line = "HTTP/1.1 404 Not Found";
let contents = fs::read_to_string("./public/404.html").unwrap();
let length = contents.len();
let response =
format!("{status_line}\r\nContent-Length: {length}\r\n\r\n{contents}");
response
}
// Return a 200 with the json.
fn return_200() -> String {
let status_line = "HTTP/1.1 200 OK";
let contents = fs::read_to_string("./public/200.json").unwrap();
let length = contents.len();
let response =
format!("{status_line}\r\nContent-Length: {length}\r\n\r\n{contents}");
response
}