From d3279088e3a816db2c254b957159d5b697dc0f62 Mon Sep 17 00:00:00 2001
From: Cara Salter <cara@devcara.com>
Date: Sun, 10 Jul 2022 11:05:51 -0400
Subject: colonies: Fully support {meta, user}-data

cloud-init should be able to use this as a seed to properly set up the
Planet.
---
 src/handlers/colonies.rs | 55 +++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 52 insertions(+), 3 deletions(-)

(limited to 'src/handlers/colonies.rs')

diff --git a/src/handlers/colonies.rs b/src/handlers/colonies.rs
index 37ec65a..cc9ee0e 100644
--- a/src/handlers/colonies.rs
+++ b/src/handlers/colonies.rs
@@ -1,15 +1,18 @@
 use std::sync::Arc;
 
-use axum::{Extension, Json};
-use sqlx::query_as;
+use axum::{Extension, Json, extract::Path};
+use axum_auth::AuthBearer;
+use sqlx::{query_as, query};
 
 use crate::{
-    errors::{JsonResult, ServiceError, StringResult},
+    errors::{JsonResult, ServiceError, StringResult, NoneResult},
     State,
 };
 
 use solarlib::colony::{User, UserData};
 
+use super::ships::check_bearer;
+
 #[derive(Debug, Clone)]
 pub struct DbUser {
     pub id: i32,
@@ -18,6 +21,12 @@ pub struct DbUser {
     pub ssh_authorized_keys: Vec<String>,
 }
 
+#[derive(Debug)]
+struct DbMeta {
+    pub uuid: String,
+    pub hostname: String,
+}
+
 impl From<DbUser> for User {
     fn from(o: DbUser) -> Self {
         Self {
@@ -44,3 +53,43 @@ pub async fn list_users(state: Extension<Arc<State>>) -> JsonResult<Json<UserDat
 
     Ok(Json(data))
 }
+
+pub async fn create_user(Json(new_user): Json<User>, state: Extension<Arc<State>>, AuthBearer(token): AuthBearer) -> NoneResult {
+    check_bearer(token)?;
+    let mut conn = state.conn.acquire().await?;
+
+    query!("INSERT INTO seed_users (name, groups, ssh_authorized_keys) VALUES ($1, $2, $3)", new_user.name, &new_user.groups, &new_user.ssh_authorized_keys)
+        .execute(&mut conn)
+        .await?;
+
+    Ok(())
+}
+
+pub async fn add_metadata(Path((uuid, hostname)): Path<(String, String)>, state: Extension<Arc<State>>, AuthBearer(token): AuthBearer) -> NoneResult {
+    check_bearer(token)?;
+    let mut conn = state.conn.acquire().await?;
+
+    query!("INSERT INTO planet_metadata (uuid, hostname) VALUES ($1, $2)", uuid, hostname).execute(&mut conn).await?;
+    Ok(())
+}
+
+pub async fn meta_data(Path(uuid): Path<String>, state: Extension<Arc<State>>) -> StringResult {
+    let mut conn = state.conn.acquire().await?;
+
+    let m = query_as!(DbMeta, "SELECT * FROM planet_metadata WHERE uuid=$1", uuid).fetch_one(&mut conn).await?;
+
+    Ok(format!("instance-id: {}\nlocal-hostname: {}", m.uuid, m.hostname))
+}
+
+pub async fn user_data(Path(_uuid): Path<String>, state: Extension<Arc<State>>) -> StringResult {
+    let mut conn = state.conn.acquire().await?;
+
+    let db_users = query_as!(DbUser, "SELECT * FROM seed_users").fetch_all(&mut conn).await?;
+
+    let users = db_users.into_iter().map(|u| u.into()).collect::<Vec<User>>();
+
+    let data = UserData { users };
+
+    Ok(format!("#cloud-config\n{}", serde_yaml::to_string(&data).unwrap()))
+
+}
-- 
cgit v1.2.3