bfile/filed/src/web/forms.rs

/*
 forms.rs - All the forms
*/

use std::collections::HashMap;

use askama::Template;
use warp::{Filter, reply::{Reply, json, with_status, html}, reject::Rejection, filters::multipart::FormData, http::StatusCode};
use futures_util::TryStreamExt;
use bytes::BufMut;
use serde::Serialize;

use crate::files::{File, lookup::LookupKind, DeleteMode};

use super::{state::SharedState, pages::{BadActionReq, UploadSuccessPage, self, ErrorPage}, rejection::HttpReject};

#[derive(Debug, Serialize, Clone)]
struct FormElement {
    data: Vec<u8>,
    mime: String
}
impl FormElement {
    pub fn as_str_or_reject(self: &Self) -> Result<String, Rejection> {
        Ok(String::from_utf8(self.data.clone()).map_err(|err| warp::reject::custom(HttpReject::FromUtf8Error(err)))?)
    }

    pub fn as_atr_or_none(self: &Self) -> Option<String> {
        if let Ok(res) = String::from_utf8(self.data.clone()) {
            Some(res)
        } else {
            None
        }
    }

    pub fn is_checked(self: &Self) -> bool {
        if self.data.len() != 2 {
            return false
        }
        let data = self.data.clone();
        let on = "on".bytes().collect::<Vec<u8>>();
        data == on
    }

    pub async fn from_formdata(form: FormData) -> Result<HashMap<String, FormElement>, warp::Error> {
        form.and_then(|mut field| async move {
            let mut bytes: Vec<u8> = vec![];
            while let Some(byte) = field.data().await {
                bytes.put(byte.unwrap())
            }
            
            Ok((field.name().into(), FormElement { data: bytes, mime: field.content_type().unwrap_or("text/plain").to_string() }))
        }).try_collect()
          .await
    }
}

struct UploadFormData {
    filename: Option<String>,
    password: Option<String>,
    lookup_kind: LookupKind,
    delmode: DeleteMode,
    file: Vec<u8>,
    mime: String,
    tos_consent: bool
}

impl Default for UploadFormData {
    fn default() -> Self {
        UploadFormData {
            filename: None,
            password: None,
            lookup_kind: LookupKind::ByHash,
            delmode: DeleteMode::Time,
            file: vec![],
            mime: "application/x-octet-stream".into(),
            tos_consent: false
        }
    }
}

impl UploadFormData {

    pub fn from_formdata(data: HashMap<String, FormElement>) -> Option<UploadFormData> {
        let mut out = Self::default();
        
        // Add a name
        match data.get("named") {
            Some(val) => {
                if val.is_checked() {
                    let name = data.get("filename")?;
                    out.filename = Some(name.as_atr_or_none()?);
                    out.lookup_kind = LookupKind::ByHash
                }
            },
            None => ()
        }

        // Add a password
        match data.get("passworded") {
            Some(val) => {
                if val.is_checked() {
                    let pass = data.get("password")?;
                    out.password = Some(pass.as_atr_or_none()?);
                    out.lookup_kind = LookupKind::ByName
                }
            },
            None => ()
        }

        // Delete mode
        match data.get("delmode") {
            Some(val) => {
                let val = val.data.clone();
                let is_30 = val == "30".bytes().collect::<Vec<u8>>();
                if is_30 {
                    out.delmode = DeleteMode::Time
                } else {
                    out.delmode = DeleteMode::TimeOrDownload
                }
            },
            None => {
                return None
            }
        }

        let file = data.get("file")?;
        out.file = file.data.clone();
        out.mime = file.mime.clone();
        out.tos_consent = match data.get("tos_consent") {
            Some(v) => v.is_checked(),
            None => false
        };

        Some(out)
    }
}

fn bad_req_html(data: String) -> Box<dyn Reply> {
    Box::new(
        with_status(
            html(data),
            StatusCode::BAD_REQUEST
        )
    )
}

pub async fn upload(form: FormData, state: SharedState) -> Result<Box<dyn Reply>, Rejection> {

    if ! state.config.files.allow_uploads {
        return Ok(
            Box::new(warp::redirect(warp::http::Uri::from_static("/")))
        )
    }

    let params: HashMap<String, FormElement> = FormElement::from_formdata(form).await.map_err(|x| HttpReject::WarpError(x))?;
    let formdata = UploadFormData::from_formdata(params.clone());

    if let Some(formdata) = formdata {

        let mut breaks_conf = false;
        if (!state.config.files.allow_custom_names) && formdata.filename.is_some() {
            breaks_conf = true;
        }
        if (!state.config.files.allow_pass_protection) && formdata.password.is_some() {
            breaks_conf = true;
        }
        
        if breaks_conf {
            let error = ErrorPage {
                env: state.env,
                conf: state.config,
                error_text: "Attempt to set name or password when they are disabled".into(),
                link: None,
                link_text: None
            };
            return Ok(
                bad_req_html(
                    error.render()
                        .map_err(|x| HttpReject::AskamaError(x))?
                )
            );
        }
        if ! formdata.tos_consent {
            let error = ErrorPage {
                env: state.env,
                conf: state.config,
                error_text: "You must agree to the ToS".into(),
                link: None,
                link_text: None
            };

            return Ok(
                bad_req_html(
                    error.render()
                        .map_err(|x| HttpReject::AskamaError(x))?
                )
            )
        }

        let file = File::create(
            formdata.file,
            formdata.mime,
            formdata.filename.clone(),
            state.env.clone(),
            formdata.delmode,
            formdata.password
        ).await
         .map_err(|x| HttpReject::StringError(x.to_string()))
         ?;
        
        state.file_mgr.save(&file, formdata.lookup_kind)
                      .map_err(|x| HttpReject::StringError(x.to_string()))?;
        
        let page = UploadSuccessPage {
            env: state.env,
            conf: state.config,
            link: match formdata.filename {
                Some(v) => v,
                None => file.hash()
            }
        };

        return Ok(
            Box::new(
                html(
                    page.render()
                        .map_err(|x| HttpReject::AskamaError(x))?
                )
            )
        )
    }

    let error = ErrorPage {
        env: state.env,
        conf: state.config,
        error_text: "Form is malformed".into(),
        link: None,
        link_text: None
    };

    return Ok(
        bad_req_html(
            error.render()
                .map_err(|x| HttpReject::AskamaError(x))?
        )
    )

}

pub fn get_routes(state: SharedState) -> impl Filter<Extract = impl Reply, Error = Rejection> + Clone {
    warp::post()
        .and(warp::multipart::form())
        .and(
            warp::any().map(move || state.clone())
        )
        .and_then(upload)
}