Initial commit

.gitignore

@@ -0,0 +1,3 @@
+/target
+**/*.rs.bk
+.env

Cargo.toml

@@ -0,0 +1,20 @@
+[package]
+name = "gist-server"
+version = "0.1.0"
+authors = ["Dylan Baker"]
+edition = "2018"
+
+[dependencies]
+diesel = { version = "1.0.0", features = ["postgres"] }
+dotenv = "0.9.0"
+r2d2 = "0.8.3"
+r2d2-diesel = "1.0.0"
+rocket = "0.4.0"
+serde = "1.0.82"
+serde_derive = "1.0.82"
+serde_json = "1.0.33"
+
+[dependencies.rocket_contrib]
+version = "0.4.0"
+default-features = false
+features = ["json", "tera_templates"]

diesel.toml

@@ -0,0 +1,5 @@
+# For documentation on how to configure this file,
+# see diesel.rs/guides/configuring-diesel-cli
+
+[print_schema]
+file = "src/schema.rs"

migrations/00000000000000_diesel_initial_setup/down.sql

@@ -0,0 +1,6 @@
+-- This file was automatically created by Diesel to setup helper functions
+-- and other internal bookkeeping. This file is safe to edit, any future
+-- changes will be added to existing projects as new migrations.
+
+DROP FUNCTION IF EXISTS diesel_manage_updated_at(_tbl regclass);
+DROP FUNCTION IF EXISTS diesel_set_updated_at();

migrations/00000000000000_diesel_initial_setup/up.sql

@@ -0,0 +1,36 @@
+-- This file was automatically created by Diesel to setup helper functions
+-- and other internal bookkeeping. This file is safe to edit, any future
+-- changes will be added to existing projects as new migrations.
+
+
+
+
+-- Sets up a trigger for the given table to automatically set a column called
+-- `updated_at` whenever the row is modified (unless `updated_at` was included
+-- in the modified columns)
+--
+-- # Example
+--
+-- ```sql
+-- CREATE TABLE users (id SERIAL PRIMARY KEY, updated_at TIMESTAMP NOT NULL DEFAULT NOW());
+--
+-- SELECT diesel_manage_updated_at('users');
+-- ```
+CREATE OR REPLACE FUNCTION diesel_manage_updated_at(_tbl regclass) RETURNS VOID AS $$
+BEGIN
+    EXECUTE format('CREATE TRIGGER set_updated_at BEFORE UPDATE ON %s
+                    FOR EACH ROW EXECUTE PROCEDURE diesel_set_updated_at()', _tbl);
+END;
+$$ LANGUAGE plpgsql;
+
+CREATE OR REPLACE FUNCTION diesel_set_updated_at() RETURNS trigger AS $$
+BEGIN
+    IF (
+        NEW IS DISTINCT FROM OLD AND
+        NEW.updated_at IS NOT DISTINCT FROM OLD.updated_at
+    ) THEN
+        NEW.updated_at := current_timestamp;
+    END IF;
+    RETURN NEW;
+END;
+$$ LANGUAGE plpgsql;

migrations/2019-04-30-203312_create_gists/down.sql

@@ -0,0 +1 @@
+DROP TABLE gists;

migrations/2019-04-30-203312_create_gists/up.sql

@@ -0,0 +1,5 @@
+CREATE TABLE gists (
+  id SERIAL PRIMARY KEY,
+  title VARCHAR NOT NULL,
+  body TEXT NOT NULL
+)

src/connection.rs

@@ -0,0 +1,41 @@
+use diesel::pg::PgConnection;
+use r2d2;
+use r2d2_diesel::ConnectionManager;
+use rocket::http::Status;
+use rocket::request::{self, FromRequest};
+use rocket::{Outcome, Request, State};
+use std::env;
+use std::ops::Deref;
+
+type Pool = r2d2::Pool<ConnectionManager<PgConnection>>;
+
+pub fn init_pool() -> Pool {
+    let manager = ConnectionManager::<PgConnection>::new(database_url());
+    Pool::new(manager).expect("db pool")
+}
+
+fn database_url() -> String {
+    env::var("DATABASE_URL").expect("DATABASE_URL must be set")
+}
+
+pub struct DbConn(pub r2d2::PooledConnection<ConnectionManager<PgConnection>>);
+
+impl<'a, 'r> FromRequest<'a, 'r> for DbConn {
+    type Error = ();
+
+    fn from_request(request: &'a Request<'r>) -> request::Outcome<DbConn, Self::Error> {
+        let pool = request.guard::<State<Pool>>()?;
+        match pool.get() {
+            Ok(conn) => Outcome::Success(DbConn(conn)),
+            Err(_) => Outcome::Failure((Status::ServiceUnavailable, ())),
+        }
+    }
+}
+
+impl Deref for DbConn {
+    type Target = PgConnection;
+
+    fn deref(&self) -> &Self::Target {
+        &self.0
+    }
+}

src/gists/mod.rs

@@ -0,0 +1,19 @@
+use diesel::pg::PgConnection as PGC;
+use diesel::prelude::*;
+
+use crate::schema::gists;
+
+#[derive(Queryable, AsChangeset, Serialize, Deserialize)]
+pub struct Gist {
+    pub id: i32,
+    pub title: String,
+    pub body: String,
+}
+
+pub fn all(connection: &PGC) -> QueryResult<Vec<Gist>> {
+    gists::table.load::<Gist>(&*connection)
+}
+
+pub fn get(connection: &PGC, id: i32) -> QueryResult<Gist> {
+    gists::table.find(id).get_result::<Gist>(connection)
+}

src/main.rs

@@ -0,0 +1,32 @@
+#![feature(proc_macro_hygiene, decl_macro)]
+
+#[macro_use]
+extern crate diesel;
+extern crate dotenv;
+extern crate r2d2;
+extern crate r2d2_diesel;
+#[macro_use]
+extern crate rocket;
+extern crate rocket_contrib;
+#[macro_use]
+extern crate serde_derive;
+
+use dotenv::dotenv;
+use rocket_contrib::templates::Template;
+
+mod connection;
+mod gists;
+mod routes;
+mod schema;
+
+use crate::routes::static_rocket_route_info_for_index;
+use crate::routes::static_rocket_route_info_for_show_gist;
+
+fn main() {
+    dotenv().ok();
+    rocket::ignite()
+        .attach(Template::fairing())
+        .manage(connection::init_pool())
+        .mount("/", routes![index, show_gist])
+        .launch();
+}

src/routes.rs

@@ -0,0 +1,34 @@
+use diesel::result::Error;
+use rocket::http::Status;
+use rocket_contrib::json::Json;
+use rocket_contrib::templates::Template;
+use std::collections::HashMap;
+
+use crate::connection::DbConn;
+use crate::gists;
+use crate::gists::Gist;
+
+fn error_status(error: Error) -> Status {
+    match error {
+        Error::NotFound => Status::NotFound,
+        _ => Status::InternalServerError,
+    }
+}
+
+#[get("/")]
+pub fn index(connection: DbConn) -> Result<Json<Vec<Gist>>, Status> {
+    gists::all(&connection)
+        .map(|gists| Json(gists))
+        .map_err(|error| error_status(error))
+}
+
+#[get("/gists/<id>")]
+pub fn show_gist(connection: DbConn, id: i32) -> Template {
+    let gist = match gists::get(&connection, id) {
+        Ok(gist) => Some(gist),
+        Err(_) => None,
+    };
+    let mut context = HashMap::new();
+    context.insert("gist", gist);
+    Template::render("gists/show", &context)
+}

src/schema.rs

@@ -0,0 +1,7 @@
+table! {
+    gists (id) {
+        id -> Int4,
+        title -> Varchar,
+        body -> Text,
+    }
+}

templates/gists/show.tera

@@ -0,0 +1,17 @@
+<html>
+  <head>
+    <title>{{ gist.title }}</title>
+  </head>
+  <body>
+    {% if gist %}
+      <h3>{{ gist.title }}</h3>
+      <div>
+        <p>
+          {{ gist.body }}
+        </p>
+      </div>
+    {% else %}
+      <h3>Gist not found</h3>
+    {% endif %}
+  </body>
+</html>