proxy github gists and render them on posts

Cargo.lock

@@ -1049,6 +1049,7 @@ dependencies = [
  "serde_json",
  "sled",
  "url",
+ "urlencoding",
 ]
 
 [[package]]
@@ -2121,6 +2122,12 @@ dependencies = [
  "percent-encoding",
 ]
 
+[[package]]
+name = "urlencoding"
+version = "2.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "68b90931029ab9b034b300b797048cf23723400aa757e8a2bfb9d748102f9821"
+
 [[package]]
 name = "vcpkg"
 version = "0.2.15"

Cargo.toml

@@ -41,3 +41,4 @@ chrono = "0.4.19"
 
 sled = "0.34.7"
 bincode = "1.3.3"
+urlencoding = "2.1.0"

src/proxy.rs

@@ -17,8 +17,9 @@
 use std::ops::{Bound, RangeBounds};
 
 use actix_web::{http::header, web, HttpResponse, Responder};
-use reqwest::header::CONTENT_TYPE;
+use reqwest::header::{CONTENT_TYPE, USER_AGENT};
 use sailfish::TemplateOnce;
+use serde::{Deserialize, Serialize};
 
 use crate::data::PostResp;
 use crate::AppData;
@@ -30,6 +31,7 @@ pub mod routes {
         pub index: &'static str,
         pub page: &'static str,
         pub asset: &'static str,
+        pub gist: &'static str,
     }
 
     impl Proxy {
@@ -38,6 +40,7 @@ pub mod routes {
                 index: "/",
                 page: "/{username}/{post}",
                 asset: "/asset/medium/{name}",
+                gist: "/asset/github-gist",
             }
         }
         pub fn get_page(&self, username: &str, post: &str) -> String {
@@ -49,6 +52,14 @@ pub mod routes {
         pub fn get_medium_asset(&self, asset_name: &str) -> String {
             self.asset.replace("{name}", asset_name)
         }
+
+        pub fn get_gist(&self, url: &str) -> String {
+            if let Some(gist_id) = url.split('/').last() {
+                format!("{}?gist={}", self.gist, urlencoding::encode(gist_id))
+            } else {
+                url.to_owned()
+            }
+        }
     }
 }
 
@@ -141,6 +152,104 @@ async fn assets(path: web::Path<String>, data: AppData) -> impl Responder {
         .body(res.bytes().await.unwrap())
 }
 
+#[derive(Deserialize, Serialize)]
+struct GistQuery {
+    gist: String,
+}
+
+#[derive(Deserialize, Serialize, TemplateOnce)]
+#[template(path = "gist.html")]
+#[template(rm_whitespace = true)]
+pub struct GistContent {
+    pub files: Vec<GistFile>,
+    pub html_url: String,
+}
+
+#[derive(TemplateOnce)]
+#[template(path = "gist_error.html")]
+#[template(rm_whitespace = true)]
+pub struct GistContentError;
+
+#[derive(Deserialize, Serialize)]
+pub struct GistFile {
+    pub file_name: String,
+    pub content: String,
+    pub language: String,
+    pub raw_url: String,
+}
+
+impl GistFile {
+    pub fn get_html_content(&self) -> String {
+        let mut content = self.content.as_str();
+        if self.content.starts_with('"') {
+            content = self.content.slice(1..);
+        }
+
+        if content.ends_with('"') {
+            content = content.slice(..content.len() - 1);
+        }
+        content.replace("\\t", "  ")
+    }
+}
+
+#[my_codegen::get(path = "crate::V1_API_ROUTES.proxy.gist")]
+async fn get_gist(query: web::Query<GistQuery>, data: AppData) -> impl Responder {
+    const URL: &str = "https://api.github.com/gists/";
+    let url = format!("{}{}", URL, query.gist);
+
+    let resp = data
+        .client
+        .get(&url)
+        .header(USER_AGENT, "libmedium")
+        .send()
+        .await
+        .unwrap()
+        .json::<serde_json::Value>()
+        .await
+        .unwrap();
+    if let Some(files) = resp.get("files") {
+        if let serde_json::Value::Object(v) = files {
+            let mut files = Vec::with_capacity(v.len());
+            v.iter().for_each(|(name, file_obj)| {
+                let file = GistFile {
+                    file_name: name.to_string(),
+                    content: file_obj
+                        .get("content")
+                        .unwrap()
+                        .as_str()
+                        .unwrap()
+                        .to_owned(),
+                    language: file_obj
+                        .get("language")
+                        .unwrap()
+                        .as_str()
+                        .unwrap()
+                        .to_owned(),
+                    raw_url: file_obj
+                        .get("raw_url")
+                        .unwrap()
+                        .as_str()
+                        .unwrap()
+                        .to_owned(),
+                };
+                files.push(file);
+            });
+            let gist = GistContent {
+                files,
+                html_url: resp.get("html_url").unwrap().to_string(),
+            };
+
+            return HttpResponse::Ok()
+                .content_type("text/html")
+                .body(gist.render_once().unwrap());
+        }
+    };
+    let err = GistContentError {};
+    HttpResponse::Ok()
+        .content_type("text/html")
+        .body(err.render_once().unwrap())
+}
+
 #[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();
@@ -162,6 +271,7 @@ async fn page(path: web::Path<(String, String)>, data: AppData) -> impl Responde
 
 pub fn services(cfg: &mut web::ServiceConfig) {
     cfg.service(assets);
+    cfg.service(get_gist);
     cfg.service(page);
     cfg.service(index);
 }

templates/gist.html

@@ -0,0 +1,18 @@
+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>Gist</title>
+  </head>
+  <body>
+    <. for file in files {.>
+    <code> <.= file.get_html_content() .> </code>
+
+    <.}.>
+    <a href="<.= &html_url[1..html_url.len() -1] .>" target="_blank">See on GitHub</a>
+  </body>
+  <style>
+    <. include!("./main.css"); .>
+  </style>
+</html>

templates/gist_error.html

@@ -0,0 +1,14 @@
+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>Gist</title>
+  </head>
+  <body>
+    Error
+  </body>
+  <style>
+    <. include!("./main.css"); .>
+  </style>
+</html>

templates/main.css

@@ -62,6 +62,17 @@ figcaption {
 code {
   font-family: monospace;
   font-size: 15px;
+  white-space: pre-wrap;
+  background-color: #888;
+  color: #fff;
+  font-weight: 600;
+}
+
+iframe {
+  width: 100%;
+  display: block;
+  margin: auto;
+  min-height: 100px;
 }
 
 @media screen and (max-width: 1200px) {

templates/post.html

@@ -44,7 +44,12 @@
             <h6><.= p.text .></h6>
           <.} else if p.type_ == "IFRAME" {.>
             <. let src = &p.iframe.as_ref().unwrap().media_resource.as_ref().unwrap().href; .>
+            <. if src.contains("gist.github.com"){.>
+              <iframe src="<.= crate::V1_API_ROUTES.proxy.get_gist(&src) .>" frameborder="0"></iframe>
+              <a href="<.= src .>">Click here to open gist on GitHub</a>
+            <.} else {.>
               <iframe src="<.= src .>" frameborder="0"></iframe>
+            <.}.>
           <.}.>
         <.}.>
       </article>