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
+
+use actix_web::web;
+use reqwest::Client;
+
+#[derive(Clone)]
+pub struct Data {
+ pub client: Client,
+}
+
+pub type AppData = web::Data<Data>;
+
+impl Data {
+ pub fn new() -> AppData {
+ AppData::new(Self {
+ client: Client::new(),
+ })
+ }
+}
+
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
+85
+86
+87
+88
+89
+90
+91
+92
+93
+94
+95
+96
+97
+
+use std::env;
+
+use actix_web::{
+ error::InternalError, http::StatusCode, middleware as actix_middleware, web::JsonConfig, App,
+ HttpServer,
+};
+use lazy_static::lazy_static;
+use log::info;
+
+mod data;
+mod meta;
+mod proxy;
+mod routes;
+mod settings;
+
+pub use data::AppData;
+pub use data::Data;
+pub use routes::ROUTES as V1_API_ROUTES;
+pub use settings::Settings;
+
+lazy_static! {
+ pub static ref SETTINGS: Settings = Settings::new().unwrap();
+}
+
+pub const CACHE_AGE: u32 = 604800;
+
+pub const GIT_COMMIT_HASH: &str = env!("GIT_HASH");
+pub const VERSION: &str = env!("CARGO_PKG_VERSION");
+pub const PKG_NAME: &str = env!("CARGO_PKG_NAME");
+pub const PKG_DESCRIPTION: &str = env!("CARGO_PKG_DESCRIPTION");
+pub const PKG_HOMEPAGE: &str = env!("CARGO_PKG_HOMEPAGE");
+
+#[cfg(not(tarpaulin_include))]
+#[actix_web::main]
+async fn main() -> std::io::Result<()> {
+ env::set_var("RUST_LOG", "info");
+
+ pretty_env_logger::init();
+
+ info!(
+ "{}: {}.\nFor more information, see: {}\nBuild info:\nVersion: {} commit: {}",
+ PKG_NAME, PKG_DESCRIPTION, PKG_HOMEPAGE, VERSION, GIT_COMMIT_HASH
+ );
+
+ println!("Starting server on: http://{}", SETTINGS.server.get_ip());
+
+ let data = Data::new();
+
+ HttpServer::new(move || {
+ App::new()
+ .wrap(actix_middleware::Logger::default())
+ .wrap(actix_middleware::Compress::default())
+ .app_data(get_json_err())
+ .wrap(
+ actix_middleware::DefaultHeaders::new()
+ .header("Permissions-Policy", "interest-cohort=()"),
+ )
+ .wrap(actix_middleware::NormalizePath::new(
+ actix_middleware::TrailingSlash::Trim,
+ ))
+ .app_data(data.clone())
+ .configure(services)
+ })
+ .workers(SETTINGS.server.workers.unwrap_or_else(num_cpus::get))
+ .bind(SETTINGS.server.get_ip())
+ .unwrap()
+ .run()
+ .await
+}
+
+#[cfg(not(tarpaulin_include))]
+pub fn get_json_err() -> JsonConfig {
+ JsonConfig::default().error_handler(|err, _| {
+
+ InternalError::new(err, StatusCode::BAD_REQUEST).into()
+ })
+}
+
+pub fn services(cfg: &mut actix_web::web::ServiceConfig) {
+ routes::services(cfg);
+}
+
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
+
+use actix_web::{web, HttpResponse, Responder};
+use serde::{Deserialize, Serialize};
+
+use crate::{GIT_COMMIT_HASH, VERSION};
+
+#[derive(Clone, Debug, Deserialize, Serialize)]
+pub struct BuildDetails {
+ pub version: &'static str,
+ pub git_commit_hash: &'static str,
+}
+
+pub mod routes {
+ pub struct Meta {
+ pub build_details: &'static str,
+ pub health: &'static str,
+ }
+
+ impl Meta {
+ pub const fn new() -> Self {
+ Self {
+ build_details: "/api/v1/meta/build",
+ health: "/api/v1/meta/health",
+ }
+ }
+ }
+}
+
+
+#[my_codegen::get(path = "crate::V1_API_ROUTES.meta.build_details")]
+async fn build_details() -> impl Responder {
+ let build = BuildDetails {
+ version: VERSION,
+ git_commit_hash: GIT_COMMIT_HASH,
+ };
+ HttpResponse::Ok().json(build)
+}
+
+pub fn services(cfg: &mut web::ServiceConfig) {
+ cfg.service(build_details);
+}
+
+#[cfg(test)]
+mod tests {
+ use actix_web::{http::StatusCode, test, App};
+
+ use crate::services;
+ use crate::*;
+
+ #[actix_rt::test]
+ async fn build_details_works() {
+ let app = test::init_service(App::new().configure(services)).await;
+
+ let resp = test::call_service(
+ &app,
+ test::TestRequest::get()
+ .uri(V1_API_ROUTES.meta.build_details)
+ .to_request(),
+ )
+ .await;
+ assert_eq!(resp.status(), StatusCode::OK);
+ }
+}
+
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
+ 85
+ 86
+ 87
+ 88
+ 89
+ 90
+ 91
+ 92
+ 93
+ 94
+ 95
+ 96
+ 97
+ 98
+ 99
+100
+101
+102
+103
+104
+105
+106
+107
+108
+109
+110
+111
+112
+113
+114
+115
+116
+117
+118
+119
+120
+121
+122
+123
+124
+125
+126
+127
+128
+129
+130
+131
+132
+133
+134
+135
+136
+137
+138
+139
+140
+141
+142
+143
+144
+145
+146
+147
+148
+149
+150
+151
+152
+153
+154
+155
+156
+157
+158
+159
+160
+161
+162
+163
+164
+165
+166
+167
+
+use std::ops::{Bound, RangeBounds};
+
+use actix_web::{web, HttpResponse, Responder};
+use graphql_client::{reqwest::post_graphql, GraphQLQuery};
+use sailfish::TemplateOnce;
+
+use crate::AppData;
+
+pub mod routes {
+ pub struct Proxy {
+ pub update: &'static str,
+ pub page: &'static str,
+ }
+
+ impl Proxy {
+ pub const fn new() -> Self {
+ Self {
+ update: "/api/v1/update",
+ page: "/{username}/{post}",
+ }
+ }
+ pub fn get_page(&self, username: &str, post: &str) -> String {
+ self.page
+ .replace("{username}", username)
+ .replace("{post}", post)
+ }
+ }
+}
+
+
+
+trait StringUtils {
+ fn substring(&self, start: usize, len: usize) -> &str;
+ fn slice(&self, range: impl RangeBounds<usize>) -> &str;
+}
+
+impl StringUtils for str {
+ fn substring(&self, start: usize, len: usize) -> &str {
+ let mut char_pos = 0;
+ let mut byte_start = 0;
+ let mut it = self.chars();
+ loop {
+ if char_pos == start {
+ break;
+ }
+ if let Some(c) = it.next() {
+ char_pos += 1;
+ byte_start += c.len_utf8();
+ } else {
+ break;
+ }
+ }
+ char_pos = 0;
+ let mut byte_end = byte_start;
+ loop {
+ if char_pos == len {
+ break;
+ }
+ if let Some(c) = it.next() {
+ char_pos += 1;
+ byte_end += c.len_utf8();
+ } else {
+ break;
+ }
+ }
+ &self[byte_start..byte_end]
+ }
+ fn slice(&self, range: impl RangeBounds<usize>) -> &str {
+ let start = match range.start_bound() {
+ Bound::Included(bound) | Bound::Excluded(bound) => *bound,
+ Bound::Unbounded => 0,
+ };
+ let len = match range.end_bound() {
+ Bound::Included(bound) => *bound + 1,
+ Bound::Excluded(bound) => *bound,
+ Bound::Unbounded => self.len(),
+ } - start;
+ self.substring(start, len)
+ }
+}
+
+#[derive(GraphQLQuery)]
+#[graphql(
+ schema_path = "schemas/schema.graphql",
+ query_path = "schemas/query.graphql",
+ response_derives = "Debug"
+)]
+struct GetPost;
+
+#[derive(TemplateOnce)]
+#[template(path = "index.html")]
+pub struct Post {
+ pub data: get_post::GetPostPost,
+ pub id: String,
+}
+
+#[my_codegen::get(path = "crate::V1_API_ROUTES.proxy.page")]
+async fn page(path: web::Path<(String, String)>, data: AppData) -> impl Responder {
+ let post_id = path.1.split("-").last();
+ if post_id.is_none() {
+ return HttpResponse::BadRequest().finish();
+ }
+ let id = post_id.unwrap().to_string();
+
+ let vars = get_post::Variables { id: id.clone() };
+
+ const URL: &str = "https://medium.com/_/graphql";
+
+ let res = post_graphql::<GetPost, _>(&data.client, URL, vars)
+ .await
+ .unwrap();
+ let response_data: get_post::ResponseData = res.data.expect("missing response data");
+
+ let page = Post {
+ id,
+ data: response_data.post.unwrap(),
+ }
+ .render_once()
+ .unwrap();
+ HttpResponse::Ok()
+ .content_type("text/html; charset=utf-8")
+ .body(page)
+}
+
+pub fn services(cfg: &mut web::ServiceConfig) {
+ cfg.service(page);
+}
+
+#[cfg(test)]
+mod tests {
+ use actix_web::{http::StatusCode, test, App};
+
+ use crate::{services, Data};
+
+ #[actix_rt::test]
+ async fn deploy_update_works() {
+ let data = Data::new();
+ let app = test::init_service(App::new().app_data(data.clone()).configure(services)).await;
+ let urls = vec![
+ "/@ftrain/big-data-small-effort-b62607a43a8c",
+ "/geekculture/rest-api-best-practices-decouple-long-running-tasks-from-http-request-processing-9fab2921ace8",
+ "/illumination/5-bugs-that-turned-into-features-e9a0e972a4e7",
+ ];
+
+ for uri in urls.iter() {
+ let resp =
+ test::call_service(&app, test::TestRequest::get().uri(uri).to_request()).await;
+ assert_eq!(resp.status(), StatusCode::OK);
+ }
+ }
+}
+
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
+
+use actix_web::web;
+
+use crate::meta::routes::Meta;
+use crate::proxy::routes::Proxy;
+
+pub const ROUTES: Routes = Routes::new();
+
+pub struct Routes {
+ pub meta: Meta,
+ pub proxy: Proxy,
+}
+
+impl Routes {
+ pub const fn new() -> Self {
+ Self {
+ meta: Meta::new(),
+ proxy: Proxy::new(),
+ }
+ }
+}
+
+pub fn services(cfg: &mut web::ServiceConfig) {
+ crate::meta::services(cfg);
+ crate::proxy::services(cfg);
+}
+
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
+85
+86
+87
+88
+89
+90
+91
+92
+93
+94
+95
+96
+97
+
+use std::env;
+use std::path::Path;
+
+use config::{Config, ConfigError, Environment, File};
+use log::warn;
+use serde::Deserialize;
+use url::Url;
+
+#[derive(Debug, Clone, Deserialize)]
+pub struct Server {
+ pub port: u32,
+ pub domain: String,
+ pub ip: String,
+ pub proxy_has_tls: bool,
+ pub workers: Option<usize>,
+}
+
+impl Server {
+ #[cfg(not(tarpaulin_include))]
+ pub fn get_ip(&self) -> String {
+ format!("{}:{}", self.ip, self.port)
+ }
+}
+
+#[derive(Debug, Clone, Deserialize)]
+pub struct Settings {
+ pub debug: bool,
+
+ pub server: Server,
+ pub source_code: String,
+}
+
+#[cfg(not(tarpaulin_include))]
+impl Settings {
+ pub fn new() -> Result<Self, ConfigError> {
+ let mut s = Config::new();
+
+
+ #[cfg(test)]
+ s.set_default("database.pool", 2.to_string())
+ .expect("Couldn't get the number of CPUs");
+
+ const CURRENT_DIR: &str = "./config/default.toml";
+ const ETC: &str = "/etc/libmedium/config.toml";
+
+ if let Ok(path) = env::var("LIBMEDIUM") {
+ s.merge(File::with_name(&path))?;
+ } else if Path::new(CURRENT_DIR).exists() {
+
+ s.merge(File::with_name(CURRENT_DIR))?;
+ } else if Path::new(ETC).exists() {
+ s.merge(File::with_name(ETC))?;
+ } else {
+ log::warn!("configuration file not found");
+ }
+
+ s.merge(Environment::with_prefix("PAGES").separator("__"))?;
+
+ check_url(&s);
+
+ match env::var("PORT") {
+ Ok(val) => {
+ s.set("server.port", val).unwrap();
+ }
+ Err(e) => warn!("couldn't interpret PORT: {}", e),
+ }
+
+ let settings: Settings = s.try_into()?;
+
+ Ok(settings)
+ }
+}
+
+#[cfg(not(tarpaulin_include))]
+fn check_url(s: &Config) {
+ let url = s
+ .get::<String>("source_code")
+ .expect("Couldn't access source_code");
+
+ Url::parse(&url).expect("Please enter a URL for source_code in settings");
+}
+