diff options
Diffstat (limited to 'rust')
| -rw-r--r-- | rust/src/bin/fatcat-auth.rs | 8 | ||||
| -rw-r--r-- | rust/src/bin/fatcat-export.rs | 10 | ||||
| -rw-r--r-- | rust/src/bin/fatcatd.rs | 6 | ||||
| -rw-r--r-- | rust/src/editing.rs | 6 | ||||
| -rw-r--r-- | rust/src/endpoints.rs | 4 | ||||
| -rw-r--r-- | rust/src/entity_crud.rs | 4 | ||||
| -rw-r--r-- | rust/src/lib.rs | 1 | ||||
| -rw-r--r-- | rust/src/server.rs | 6 | 
8 files changed, 21 insertions, 24 deletions
| diff --git a/rust/src/bin/fatcat-auth.rs b/rust/src/bin/fatcat-auth.rs index 9efd47ed..85132023 100644 --- a/rust/src/bin/fatcat-auth.rs +++ b/rust/src/bin/fatcat-auth.rs @@ -2,11 +2,9 @@  use clap::{App, SubCommand}; -use fatcat::auth; -use fatcat::editing; -use fatcat::errors::*; +use fatcat::{auth, editing, server}; +use fatcat::errors::Result;  use fatcat::identifiers::FatcatId; -use fatcat::server::*;  use std::process;  use std::str::FromStr; @@ -68,7 +66,7 @@ fn main() -> Result<()> {      }      // Then the ones that do -    let db_conn = database_worker_pool()?.get().expect("database pool"); +    let db_conn = server::database_worker_pool()?.get().expect("database pool");      let confectionary = auth::env_confectionary()?;      match m.subcommand() {          ("list-editors", Some(_subm)) => { diff --git a/rust/src/bin/fatcat-export.rs b/rust/src/bin/fatcat-export.rs index 82f40048..d6764dc6 100644 --- a/rust/src/bin/fatcat-export.rs +++ b/rust/src/bin/fatcat-export.rs @@ -13,17 +13,15 @@ extern crate failure;  use clap::{App, Arg};  use fatcat::entity_crud::*; -use fatcat::errors::*; -use fatcat::identifiers::*; -use fatcat::server::*; +use fatcat::errors::Result; +use fatcat::identifiers::FatcatId; +use fatcat::server::{DbConn, self};  use fatcat_api_spec::models::*;  use std::str::FromStr;  use uuid::Uuid;  use std::thread; -//use std::io::{Stdout,StdoutLock};  use crossbeam_channel as channel; -//use num_cpus; TODO:  use std::io::prelude::*;  use std::io::{BufReader, BufWriter}; @@ -158,7 +156,7 @@ pub fn do_export(      entity_type: ExportEntityType,      redirects: bool,  ) -> Result<()> { -    let db_pool = database_worker_pool()?; +    let db_pool = server::database_worker_pool()?;      let buf_input = BufReader::new(std::io::stdin());      let (row_sender, row_receiver) = channel::bounded(CHANNEL_BUFFER_LEN);      let (output_sender, output_receiver) = channel::bounded(CHANNEL_BUFFER_LEN); diff --git a/rust/src/bin/fatcatd.rs b/rust/src/bin/fatcatd.rs index df5c390e..816ea1f0 100644 --- a/rust/src/bin/fatcatd.rs +++ b/rust/src/bin/fatcatd.rs @@ -6,8 +6,8 @@ extern crate slog;  extern crate hyper;  use clap::{App, Arg}; -use fatcat::errors::*; -use fatcat::server::*; +use fatcat::errors::Result; +use fatcat::server;  use iron::middleware::AfterMiddleware;  use iron::modifiers::RedirectRaw;  use iron::{status, Chain, Iron, IronResult, Request, Response}; @@ -64,7 +64,7 @@ fn main() -> Result<()> {          None      }; -    let mut server = create_server()?; +    let mut server = server::create_server()?;      // metrics reporting      match env::var("FATCAT_STATSD_HOST") { diff --git a/rust/src/editing.rs b/rust/src/editing.rs index 2dfabc75..9dac30ac 100644 --- a/rust/src/editing.rs +++ b/rust/src/editing.rs @@ -1,9 +1,9 @@  use crate::database_models::*;  use crate::database_schema::*;  use crate::entity_crud::EntityCrud; -use crate::errors::*; -use crate::identifiers::*; -use crate::server::*; +use crate::errors::{Result, FatcatError}; +use crate::identifiers::{FatcatId, check_username}; +use crate::server::DbConn;  use diesel;  use diesel::prelude::*;  use fatcat_api_spec::models::*; diff --git a/rust/src/endpoints.rs b/rust/src/endpoints.rs index 027e25bc..8c7700f1 100644 --- a/rust/src/endpoints.rs +++ b/rust/src/endpoints.rs @@ -7,12 +7,12 @@  // using closures as a Result/? hack  #![allow(clippy::redundant_closure_call)] -use crate::auth::*; +use crate::auth::FatcatRole;  use crate::database_models::EntityEditRow;  use crate::editing::*;  use crate::entity_crud::{EntityCrud, ExpandFlags, HideFlags};  use crate::errors::*; -use crate::identifiers::*; +use crate::identifiers::FatcatId;  use crate::server::*;  use diesel::Connection;  use fatcat_api_spec::models; diff --git a/rust/src/entity_crud.rs b/rust/src/entity_crud.rs index 0c1e29b0..a0101de8 100644 --- a/rust/src/entity_crud.rs +++ b/rust/src/entity_crud.rs @@ -1,10 +1,10 @@  use crate::database_models::*;  use crate::database_schema::*; -use crate::editing::*; +use crate::editing::EditContext;  use crate::endpoint_handlers::get_release_files;  use crate::errors::*;  use crate::identifiers::*; -use crate::server::*; +use crate::server::DbConn;  use chrono;  use diesel::prelude::*;  use diesel::{self, insert_into}; diff --git a/rust/src/lib.rs b/rust/src/lib.rs index eeb7e83f..b7661334 100644 --- a/rust/src/lib.rs +++ b/rust/src/lib.rs @@ -14,6 +14,7 @@ pub mod auth;  pub mod database_models;  pub mod database_schema; // only public for tests  pub mod editing; +pub mod editing_crud;  mod endpoint_handlers;  mod endpoints;  pub mod entity_crud; diff --git a/rust/src/server.rs b/rust/src/server.rs index 66b215fc..c7cfa363 100644 --- a/rust/src/server.rs +++ b/rust/src/server.rs @@ -1,7 +1,7 @@  //! API endpoint handlers -use crate::auth::*; -use crate::errors::*; +use crate::auth::{AuthConfectionary, self}; +use crate::errors::Result;  use chrono;  use diesel;  use diesel::pg::PgConnection; @@ -44,7 +44,7 @@ pub fn create_server() -> Result<Server> {      let db_pool = diesel::r2d2::Pool::builder()          .build(manager)          .expect("Failed to create database pool."); -    let auth_confectionary = env_confectionary()?; +    let auth_confectionary = auth::env_confectionary()?;      let metrics = StatsdClient::from_sink("blackhole", NopMetricSink);      Ok(Server {          db_pool, | 
