Skip to content

Commit

Permalink
#502 WIP urls
Browse files Browse the repository at this point in the history
  • Loading branch information
joepio committed Oct 12, 2022
1 parent bad0dad commit 6f14f7e
Show file tree
Hide file tree
Showing 12 changed files with 71 additions and 51 deletions.
2 changes: 1 addition & 1 deletion cli/src/main.rs
Original file line number Diff line number Diff line change
Expand Up @@ -301,7 +301,7 @@ fn tpf(context: &Context) -> AtomicResult<()> {
let endpoint = context
.store
.get_server_url()
.get_route(Routes::Tpf)
.set_route(Routes::Tpf)
.to_string();
let resources =
atomic_lib::client::fetch_tpf(&endpoint, subject, property, value, &context.store)?;
Expand Down
35 changes: 22 additions & 13 deletions lib/src/atomic_url.rs
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ use url::Url;
use crate::{errors::AtomicResult, utils::random_string};

pub enum Routes {
Agents,
AllVersions,
Collections,
Commits,
Expand All @@ -12,6 +13,7 @@ pub enum Routes {
Import,
Tpf,
Version,
Setup,
}

#[derive(Debug, Clone, PartialEq, Eq)]
Expand All @@ -31,16 +33,18 @@ impl AtomicUrl {
}

/// Returns the route to some common Endpoint
pub fn get_route(&self, route: Routes) -> Self {
pub fn set_route(&self, route: Routes) -> Self {
let path = match route {
Routes::AllVersions => "/all-versions".to_string(),
Routes::Agents => "/agents".to_string(),
Routes::Collections => "/collections".to_string(),
Routes::Commits => "/commits".to_string(),
Routes::CommitsUnsigned => "/commits-unsigned".to_string(),
Routes::Endpoints => "/endpoints".to_string(),
Routes::Import => "/import".to_string(),
Routes::Tpf => "/tpf".to_string(),
Routes::Version => "/version".to_string(),
Routes::Setup => "/setup".to_string(),
};
let mut new = self.url.clone();
new.set_path(&path);
Expand Down Expand Up @@ -68,25 +72,30 @@ impl AtomicUrl {
/// let start = "http://localhost";
/// let mut url = AtomicUrl::try_from(start).unwrap();
/// assert_eq!(url.to_string(), "http://localhost/");
/// url.append("/");
/// assert_eq!(url.to_string(), "http://localhost/");
/// url.append("someUrl/123");
/// assert_eq!(url.to_string(), "http://localhost/someUrl/123");
/// url.append("345");
/// assert_eq!(url.to_string(), "http://localhost/someUrl/123/345");
/// url.append("/");
/// url.append("/345");
/// assert_eq!(url.to_string(), "http://localhost/someUrl/123/345");
/// ```
pub fn append(&mut self, path: &str) -> &Self {
// Remove first slash if it exists
let path = if path.starts_with('/') {
if path.len() == 1 {
return self;
let mut new_path = self.url.path().to_string();
match (new_path.ends_with('/'), path.starts_with('/')) {
(true, true) => {
new_path.pop();
}
path.to_string()
} else {
format!("/{path}")
(false, false) => new_path.push('/'),
_other => {}
};
let mut new_path = self.url.path().to_string();
new_path.push_str(&path);

// Remove first slash if it exists
if new_path.starts_with('/') {
new_path.remove(0);
}

new_path.push_str(path);

self.url.set_path(&new_path);
self
}
Expand Down
4 changes: 2 additions & 2 deletions lib/src/collections.rs
Original file line number Diff line number Diff line change
Expand Up @@ -430,7 +430,7 @@ pub fn create_collection_resource_for_class(
drive.to_string()
} else {
drive
.get_route(crate::atomic_url::Routes::Collections)
.set_route(crate::atomic_url::Routes::Collections)
.to_string()
};

Expand Down Expand Up @@ -535,7 +535,7 @@ mod test {
println!("{:?}", subjects);
let collections_collection = store
.get_resource_extended(
&format!("{}/collections", store.get_server_url()),
&format!("{}collections", store.get_server_url()),
false,
None,
)
Expand Down
4 changes: 2 additions & 2 deletions lib/src/commit.rs
Original file line number Diff line number Diff line change
Expand Up @@ -427,14 +427,14 @@ impl Commit {
let commit_subject = match self.signature.as_ref() {
Some(sig) => store
.get_server_url()
.get_route(Routes::Commits)
.set_route(Routes::Commits)
.append(sig)
.to_string(),
None => {
let now = crate::utils::now();
store
.get_server_url()
.get_route(Routes::CommitsUnsigned)
.set_route(Routes::CommitsUnsigned)
.append(&now.to_string())
.to_string()
}
Expand Down
15 changes: 9 additions & 6 deletions lib/src/db/test.rs
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
use crate::urls;
use crate::{atomic_url::Routes, urls};

use super::*;
use ntest::timeout;
Expand Down Expand Up @@ -64,7 +64,7 @@ fn populate_collections() {
.map(|r| r.get_subject().into())
.collect();
println!("{:?}", subjects);
let collections_collection_url = format!("{}/collections", store.get_server_url());
let collections_collection_url = format!("{}collections", store.get_server_url());
let collections_resource = store
.get_resource_extended(&collections_collection_url, false, None)
.unwrap();
Expand Down Expand Up @@ -117,7 +117,7 @@ fn add_atom_to_index() {
/// Also counts commits.
fn destroy_resource_and_check_collection_and_commits() {
let store = Db::init_temp("counter").unwrap();
let agents_url = format!("{}/agents", store.get_server_url());
let agents_url = store.get_server_url().set_route(Routes::Agents).to_string();
let agents_collection_1 = store
.get_resource_extended(&agents_url, false, None)
.unwrap();
Expand All @@ -132,7 +132,10 @@ fn destroy_resource_and_check_collection_and_commits() {
);

// We will count the commits, and check if they've incremented later on.
let commits_url = format!("{}/commits", store.get_server_url());
let commits_url = store
.get_server_url()
.set_route(Routes::Commits)
.to_string();
let commits_collection_1 = store
.get_resource_extended(&commits_url, false, None)
.unwrap();
Expand Down Expand Up @@ -179,7 +182,7 @@ fn destroy_resource_and_check_collection_and_commits() {

_res.resource_new.unwrap().destroy(&store).unwrap();
let agents_collection_3 = store
.get_resource_extended(&agents_url, false, None)
.get_resource_extended(&agents_url.to_string(), false, None)
.unwrap();
let agents_collection_count_3 = agents_collection_3
.get(crate::urls::COLLECTION_MEMBER_COUNT)
Expand Down Expand Up @@ -210,7 +213,7 @@ fn destroy_resource_and_check_collection_and_commits() {
#[test]
fn get_extended_resource_pagination() {
let store = Db::init_temp("get_extended_resource_pagination").unwrap();
let subject = format!("{}/commits?current_page=2", store.get_server_url());
let subject = format!("{}commits?current_page=2", store.get_server_url());
// Should throw, because page 2 is out of bounds for default page size
let _wrong_resource = store
.get_resource_extended(&subject, false, None)
Expand Down
16 changes: 6 additions & 10 deletions lib/src/populate.rs
Original file line number Diff line number Diff line change
Expand Up @@ -185,11 +185,7 @@ pub fn create_drive(
store.get_resource_new(&drive_subject)
};
drive.set_class(urls::DRIVE);
drive.set_propval_string(
urls::NAME.into(),
drive_name.unwrap_or_else(|| "Main drive"),
store,
)?;
drive.set_propval_string(urls::NAME.into(), drive_name.unwrap_or("Main drive"), store)?;

// Set rights
drive.push_propval(urls::WRITE, for_agent.into(), true)?;
Expand Down Expand Up @@ -257,7 +253,7 @@ pub fn populate_endpoints(store: &crate::Db) -> AtomicResult<()> {
use crate::atomic_url::Routes;

let endpoints = crate::endpoints::default_endpoints();
let endpoints_collection = store.get_server_url().get_route(Routes::Endpoints);
let endpoints_collection = store.get_server_url().set_route(Routes::Endpoints);
for endpoint in endpoints {
let mut resource = endpoint.to_resource(store)?;
resource.set_propval(
Expand All @@ -279,7 +275,7 @@ pub fn populate_importer(store: &crate::Db) -> AtomicResult<()> {
let base = store
.get_self_url()
.ok_or("No self URL in this Store - required for populating importer")?;
let mut importer = Resource::new(base.get_route(Routes::Import).to_string());
let mut importer = Resource::new(base.set_route(Routes::Import).to_string());
importer.set_class(urls::IMPORTER);
importer.set_propval(
urls::PARENT.into(),
Expand All @@ -298,9 +294,9 @@ pub fn populate_sidebar_items(store: &crate::Db) -> AtomicResult<()> {
let base = store.get_self_url().ok_or("No self_url")?;
let mut drive = store.get_resource(base.as_str())?;
let arr = vec![
format!("{}setup", base),
format!("{}import", base),
format!("{}collections", base),
base.set_route(crate::atomic_url::Routes::Setup),
base.set_route(crate::atomic_url::Routes::Import),
base.set_route(crate::atomic_url::Routes::Collections),
];
drive.set_propval(urls::SUBRESOURCES.into(), arr.into(), store)?;
drive.save_locally(store)?;
Expand Down
12 changes: 11 additions & 1 deletion lib/src/values.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
use crate::{
datatype::match_datatype, datatype::DataType, errors::AtomicResult, resources::PropVals,
utils::check_valid_url, Resource,
utils::check_valid_url, AtomicUrl, Resource,
};
use regex::Regex;
use serde::{Deserialize, Serialize};
Expand Down Expand Up @@ -330,6 +330,16 @@ impl From<Vec<Resource>> for Value {
}
}

impl From<Vec<AtomicUrl>> for Value {
fn from(val: Vec<AtomicUrl>) -> Self {
let mut vec = Vec::new();
for i in val {
vec.push(SubResource::Subject(i.to_string()));
}
Value::ResourceArray(vec)
}
}

use std::fmt;
impl fmt::Display for Value {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
Expand Down
3 changes: 2 additions & 1 deletion server/src/appstate.rs
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ use crate::{
};
use atomic_lib::{
agents::{generate_public_key, Agent},
atomic_url::Routes,
commit::CommitResponse,
Storelike,
};
Expand Down Expand Up @@ -155,7 +156,7 @@ fn set_default_agent(config: &Config, store: &impl Storelike) -> AtomicServerRes

/// Creates the first Invitation that is opened by the user on the Home page.
fn set_up_initial_invite(store: &impl Storelike) -> AtomicServerResult<()> {
let subject = format!("{}/setup", store.get_server_url());
let subject = store.get_server_url().set_route(Routes::Setup).to_string();
tracing::info!("Creating initial Invite at {}", subject);
let mut invite = store.get_resource_new(&subject);
invite.set_class(atomic_lib::urls::INVITE);
Expand Down
2 changes: 1 addition & 1 deletion server/src/bin.rs
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,7 @@ async fn main_wrapped() -> errors::AtomicServerResult<()> {
.store
.get_self_url()
.expect("No self URL")
.get_route(Routes::Import)
.set_route(Routes::Import)
.to_string()
};
let parse_opts = atomic_lib::parse::ParseOpts {
Expand Down
8 changes: 4 additions & 4 deletions server/src/errors.rs
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ pub enum AppErrorType {
Other,
}

// More strict error type, supports HTTP responses
/// Error type that includes a Resource representation of the Error, which can be sent to the client.
pub struct AtomicServerError {
pub message: String,
pub error_type: AppErrorType,
Expand Down Expand Up @@ -47,8 +47,8 @@ impl ResponseError for AtomicServerError {
}
fn error_response(&self) -> HttpResponse {
// Creates a JSON-AD resource representing the Error.
let r = match &self.error_resource {
Some(r) => r.to_owned(),
let r: Resource = match &self.error_resource {
Some(r) => *r.clone(),
None => {
let mut r = Resource::new("subject".into());
r.set_class(urls::ERROR);
Expand All @@ -61,7 +61,7 @@ impl ResponseError for AtomicServerError {
};

let body = r.to_json_ad().unwrap();
tracing::info!("Error response: {}", self.message);
// tracing::info!("Error response: {}", self.message);
HttpResponse::build(self.status_code())
.content_type(JSON_AD_MIME)
.body(body)
Expand Down
4 changes: 1 addition & 3 deletions server/src/handlers/resource.rs
Original file line number Diff line number Diff line change
Expand Up @@ -34,12 +34,10 @@ pub async fn handle_get_resource(
// You'd think there would be a simpler way of getting the requested URL...
// See https://github.com/actix/actix-web/issues/2895
let mut subject = appstate.store.get_server_url().clone();
println!("server_url: {}", &subject);

// Doe this include the query params?
subject.append(&req.uri().to_string());
subject.set_path(&req.uri().to_string());

println!("server uri {} ", &req.uri());
if let Some(sd) = subdomain {
subject.set_subdomain(Some(&sd))?;
}
Expand Down
17 changes: 10 additions & 7 deletions server/src/tests.rs
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,9 @@ async fn server_tests() {
assert!(resp.status().is_client_error());

// Edit the properties collection, make it hidden to the public agent
let mut drive = store.get_resource(&appstate.config.server_url).unwrap();
let mut drive = store
.get_resource(appstate.store.get_server_url().as_str())
.unwrap();
drive
.set_propval(
urls::READ.into(),
Expand All @@ -100,7 +102,7 @@ async fn server_tests() {

// Should 401 (Unauthorized)
let req =
test::TestRequest::with_uri("/properties").insert_header(("Accept", "application/ad+json"));
test::TestRequest::with_uri("properties").insert_header(("Accept", "application/ad+json"));
let resp = test::call_service(&app, req.to_request()).await;
assert_eq!(
resp.status().as_u16(),
Expand All @@ -109,17 +111,18 @@ async fn server_tests() {
);

// Get JSON-AD
let req = build_request_authenticated("/properties", &appstate);
let req = build_request_authenticated("properties", &appstate);
let resp = test::call_service(&app, req.to_request()).await;
assert!(resp.status().is_success(), "setup not returning JSON-AD");
let body = get_body(resp);
println!("DEBUG: {:?}", body);
// assert!(resp.status().is_success(), "setup not returning JSON-AD");
assert!(
body.as_str().contains("{\n \"@id\""),
"response should be json-ad"
);

// Get JSON-LD
let req = build_request_authenticated("/properties", &appstate)
let req = build_request_authenticated("properties", &appstate)
.insert_header(("Accept", "application/ld+json"));
let resp = test::call_service(&app, req.to_request()).await;
assert!(resp.status().is_success(), "setup not returning JSON-LD");
Expand All @@ -130,7 +133,7 @@ async fn server_tests() {
);

// Get turtle
let req = build_request_authenticated("/properties", &appstate)
let req = build_request_authenticated("properties", &appstate)
.insert_header(("Accept", "text/turtle"));
let resp = test::call_service(&app, req.to_request()).await;
assert!(resp.status().is_success());
Expand All @@ -142,7 +145,7 @@ async fn server_tests() {

// Get Search
// Does not test the contents of the results - the index isn't built at this point
let req = build_request_authenticated("/search?q=setup", &appstate);
let req = build_request_authenticated("search?q=setup", &appstate);
let resp = test::call_service(&app, req.to_request()).await;
assert!(resp.status().is_success());
let body = get_body(resp);
Expand Down

0 comments on commit 6f14f7e

Please sign in to comment.