flams_router_backend/

server_fns.rs

use crate::FileStates;
use flams_backend_types::{
    archive_json::{ArchiveIndex, Institution},
    archives::{ArchiveData, ArchiveGroupData, DirectoryData, FileData},
};
use ftml_uris::{
    ArchiveId, IsDomainUri, IsNarrativeUri, Language, ModuleUri, Uri, UriWithArchive, UriWithPath,
    components::UriComponents,
};
use leptos::prelude::*;
use std::path::Path;

#[server(prefix = "/api/backend", endpoint = "group_entries")]
pub async fn group_entries(
    r#in: Option<ArchiveId>,
) -> Result<(Vec<ArchiveGroupData>, Vec<ArchiveData>), ServerFnError<String>> {
    server::group_entries(r#in).await
}

#[server(prefix = "/api/backend", endpoint = "archive_entries")]
pub async fn archive_entries(
    archive: ArchiveId,
    path: Option<String>,
) -> Result<(Vec<DirectoryData>, Vec<FileData>), ServerFnError<String>> {
    server::archive_entries(archive, path).await
}

#[server(prefix = "/api/backend", endpoint = "archive_modules")]
pub async fn archive_modules(
    archive: ArchiveId,
    path: Option<String>,
) -> Result<(Vec<DirectoryData>, Vec<ModuleUri>), ServerFnError<String>> {
    server::archive_modules(archive, path).await
}

#[server(prefix = "/api/backend", endpoint = "archive_dependencies")]
pub async fn archive_dependencies(
    archives: Vec<ArchiveId>,
) -> Result<Vec<ArchiveId>, ServerFnError<String>> {
    server::archive_dependencies(archives).await
}

#[server(prefix = "/api/backend", endpoint = "build_status")]
pub async fn build_status(
    archive: Option<ArchiveId>,
    path: Option<String>,
) -> Result<FileStates, ServerFnError<String>> {
    server::build_status(archive, path).await
}

ftml_uris::compfun! {
    #[server(prefix = "/api/backend", endpoint = "source_file",
        input=server_fn::codec::GetUrl,
        output=server_fn::codec::Json)]
    #[allow(clippy::many_single_char_names)]
    #[allow(clippy::too_many_arguments)]
    pub async fn source_file(
        uri: Uri
    ) -> Result<String, ServerFnError<String>> {
        use flams_math_archives::{backend::LocalBackend, LocalArchive,MathArchive};
        use flams_web_utils::not_found;
        fn get_root(
            id: &ArchiveId,
            and_then: impl FnOnce(&LocalArchive, String) -> Result<String, String>,
        ) -> Result<String, String> {
            use flams_git::GitUrlExt;
            flams_math_archives::backend::GlobalBackend.with_local_archive(id, |a| {
                let Some(a) = a else {
                    not_found!("Archive {id} not found")
                };
                let repo = flams_git::repos::GitRepo::open(a.path())
                    .map_err(|_| format!("No git remote for {id} found"))?;
                let url = repo
                    .get_origin_url()
                    .map_err(|_| format!("No git remote for {id} found"))?;
                let https = url.into_https();
                let mut url = https.to_string();
                if https.git_suffix {
                    // remove .git
                    url.pop();
                    url.pop();
                    url.pop();
                    url.pop();
                }
                and_then(a, url)
            })
        }
        fn get_source(id: &ArchiveId, path: Option<&str>) -> Result<String, String> {
            get_root(id, |_, s| Ok(s)).map(|mut s| {
                s.push_str("/-/tree/main/source/");
                if let Some(p) = path {
                    s.push_str(p);
                }
                s
            })
        }
        fn get_source_of_file<'a>(
            id: &ArchiveId,
            path: Option<&str>,
            last: Option<&'a str>,
            mut name: &'a str,
            lang: Option<Language>,
        ) -> Result<String, String> {
            fn find(path: &Path, base: &mut String, name: &str, lang: Option<Language>) -> bool {
                if let Some(lang) = lang {
                    // TODO add other file extensions here!
                    let filename = format!("{name}.{lang}.tex");
                    let p = path.join(&filename);
                    if p.exists() {
                        base.push('/');
                        base.push_str(&filename);
                        return true;
                    }
                } else {
                    // TODO add other file extensions here!
                    let filename = format!("{name}.en.tex");
                    let p = path.join(&filename);
                    if p.exists() {
                        base.push('/');
                        base.push_str(&filename);
                        return true;
                    }
                }
                // TODO add other file extensions here!
                let filename = format!("{name}.tex");
                let p = path.join(&filename);
                p.exists() && {
                    base.push('/');
                    base.push_str(&filename);
                    true
                }
            }
            get_root(id, |a, mut base| {
                base.push_str("/-/blob/main/source");
                let mut source_path = a.source_dir();
                if let Some(path) = path {
                    for s in path.split('/') {
                        source_path = source_path.join(s);
                        base.push('/');
                        base.push_str(s);
                    }
                }
                if let Some(last) = last {
                    let np = source_path.join(last);
                    let mut nb = format!("{base}/{last}");
                    if find(&np, &mut nb, name, lang) {
                        return Ok(base);
                    }
                    name = last;
                }
                if find(&source_path, &mut base, name, lang) {
                    Ok(base)
                } else {
                    not_found!("No source file found")
                }
            })
        }

        tokio::task::spawn_blocking(move || {
            let Result::<UriComponents, _>::Ok(comps) = uri else {
                return Err("invalid uri components".to_string());
            };
            let uri = match comps.parse(flams_router_base::uris::get_uri) {
                Ok(uri) => uri,
                Err(e) => return Err(format!("Invalid uri: {e}")),
            };

            match uri {
                uri @ Uri::Base(_) => Err(format!("BaseUri can not have a source path: {uri}")),
                Uri::Archive(a) => get_root(a.archive_id(), |_, s| Ok(s)),
                Uri::Path(uri) => uri.path().map_or_else(
                    || get_root(uri.archive_id(), |_, s| Ok(s)),
                    |p| get_source(uri.archive_id(), Some(p.as_ref()))
                ),
                Uri::Document(doc) => {
                    let path_str = doc.path().map(ToString::to_string);
                    get_source_of_file(
                        doc.archive_id(),
                        path_str.as_deref(),
                        None,
                        doc.document_name().as_ref(),
                        Some(doc.language()),
                    )
                }
                Uri::DocumentElement(n) => {
                    let doc = n.document_uri();
                    let path_str = doc.path().map(ToString::to_string);
                    get_source_of_file(
                        doc.archive_id(),
                        path_str.as_deref(),
                        None,
                        doc.document_name().as_ref(),
                        Some(doc.language()),
                    )
                }
                Uri::Module(module) => {
                    let (path, last) = module.path().map_or((None,None),|p| {
                        let ps = p.to_string();
                        if let Some((p, l)) = ps.rsplit_once('/') {
                            (Some(p.to_string()), Some(l.to_string()))
                        } else {
                            (None, Some(ps))
                        }
                    });

                    get_source_of_file(
                        module.archive_id(),
                        path.as_deref(),
                        last.as_deref(),
                        module.module_name().first(),
                        None,
                    )
                }
                Uri::Symbol(symbol) => {
                    let (path, last) = symbol.path().map_or((None,None),|p| {
                        let ps = p.to_string();
                        if let Some((p, l)) = ps.rsplit_once('/') {
                            (Some(p.to_string()), Some(l.to_string()))
                        } else {
                            (None, Some(ps))
                        }
                    });

                    get_source_of_file(
                        symbol.archive_id(),
                        path.as_deref(),
                        last.as_deref(),
                        symbol.module_name().first(),
                        None,
                    )
                }
            }
        })
        .await
        .map_err(|e| e.to_string())?
        .map_err(|s: String| s.into())
    }
}

#[server(prefix="/api/backend",endpoint="download",
  input=server_fn::codec::GetUrl,
  output=server_fn::codec::Streaming
)]
pub async fn archive_stream(
    id: ArchiveId,
) -> Result<leptos::server_fn::codec::ByteStream<ServerFnError<String>>, ServerFnError<String>> {
    server::archive_stream(&id)
}

#[server(
  prefix="/api",
  endpoint="index",
  output=server_fn::codec::Json
)]
pub async fn index() -> Result<(Vec<Institution>, Vec<ArchiveIndex>), ServerFnError<String>> {
    Ok(
        flams_math_archives::manager::ArchiveManager::index_async::<flams_system::TokioEngine>(
            || flams_system::settings::Settings::get().external_url(),
        )
        .await,
    )
}

#[cfg(feature = "ssr")]
mod server {
    use flams_backend_types::archives::{ArchiveData, ArchiveGroupData, DirectoryData, FileData};
    use flams_math_archives::{
        Archive, BuildableArchive, LocallyBuilt, MathArchive,
        backend::{GlobalBackend, LocalBackend},
        manager::ArchiveOrGroup as AoG,
        source_files::{SourceEntry, SourceEntryRef},
        utils::path_ext::RelPath,
    };
    use flams_router_base::LoginState;
    use flams_system::LocalArchiveExt;
    use flams_utils::vecmap::VecSet;
    use flams_web_utils::blocking_server_fn;
    use ftml_uris::{ArchiveId, ArchiveUri, FtmlUri, ModuleUri, UriName, UriPath, UriWithArchive};
    use leptos::prelude::*;

    use crate::FileStates;

    pub async fn group_entries(
        id: Option<ArchiveId>,
    ) -> Result<(Vec<ArchiveGroupData>, Vec<ArchiveData>), ServerFnError<String>> {
        let login = LoginState::get_server();
        blocking_server_fn(move || {
            let allowed = matches!(
                login,
                LoginState::Admin
                    | LoginState::NoAccounts
                    | LoginState::User { is_admin: true, .. }
            );
            GlobalBackend.with_tree(|tree| {
                let v = match id {
                    None => &tree.top,
                    Some(id) => match tree.get_group_or_archive(&id) {
                        Some(AoG::Group(g)) => &g.children,
                        _ => return Err(format!("Archive Group {id} not found")),
                    },
                };
                let mut groups = Vec::new();
                let mut archives = Vec::new();
                for a in v {
                    match a {
                        AoG::Archive(id) => {
                            let (summary, git) = if !allowed
                                && flams_system::settings::Settings::get().gitlab_url.is_none()
                            {
                                (None, None)
                            } else {
                                tree.get(id)
                                    .map(|a| {
                                        if let Archive::Local(a) = a {
                                            (
                                                if allowed {
                                                    Some(a.state_summary())
                                                } else {
                                                    None
                                                },
                                                a.is_managed().map(ToString::to_string),
                                            )
                                        } else {
                                            (None, None)
                                        }
                                    })
                                    .unwrap_or_default()
                            };
                            archives.push(ArchiveData {
                                id: id.clone(),
                                summary,
                                git,
                            });
                        }
                        AoG::Group(g) => {
                            let summary = if allowed {
                                Some(g.state.summarize())
                            } else {
                                None
                            };
                            groups.push(ArchiveGroupData {
                                id: g.id.clone(),
                                summary,
                            });
                        }
                    }
                }
                Ok((groups, archives))
            })
        })
        .await
    }

    pub async fn archive_modules(
        archive: ArchiveId,
        path: Option<String>,
    ) -> Result<(Vec<DirectoryData>, Vec<ModuleUri>), ServerFnError<String>> {
        let login = LoginState::get_server();
        blocking_server_fn(move || {
            let allowed = matches!(
                login,
                LoginState::Admin
                    | LoginState::NoAccounts
                    | LoginState::User { is_admin: true, .. }
            );
            GlobalBackend.with_local_archive(&archive, |a| {
                let Some(a) = a else {
                    return Err(format!("Archive {archive} not found"));
                };
                a.with_sources(|d| {
                    let d = match &path {
                        None => d,
                        Some(p) => match d.find(RelPath::new(p)) {
                            Some(SourceEntryRef::Dir(d)) => d,
                            _ => {
                                return Err(format!(
                                    "Directory {p} not found in archive {archive}"
                                ));
                            }
                        },
                    };
                    let mut ds = Vec::new();
                    for d in &d.children {
                        if let SourceEntry::Dir(d) = d {
                            ds.push(DirectoryData {
                                rel_path: d
                                    .relative_path
                                    .as_ref()
                                    .map(UriPath::to_string)
                                    .unwrap_or_default(),
                                summary: if allowed {
                                    Some(d.state.summarize())
                                } else {
                                    None
                                },
                            });
                        }
                    }
                    let mut mods = Vec::new();
                    let out = path.as_ref().map_or_else(
                        || a.out_dir().join(".modules"),
                        |p| a.out_dir().join(p).join(".modules"),
                    );
                    let uripath = path.as_ref().and_then(|p| p.parse::<UriPath>().ok());
                    let path_uri = a.uri().clone() / uripath;
                    if let Ok(dir) = std::fs::read_dir(out) {
                        for f in dir {
                            if let Ok(f) = f
                                && let Ok(name) = f.file_name().into_string()
                                && let Ok(name) = name.replace("__AST__", "*").parse::<UriName>()
                            {
                                mods.push(path_uri.clone() | name);
                            }
                        }
                    }

                    Ok((ds, mods))
                })
            })
        })
        .await
    }

    pub async fn archive_entries(
        archive: ArchiveId,
        path: Option<String>,
    ) -> Result<(Vec<DirectoryData>, Vec<FileData>), ServerFnError<String>> {
        let login = LoginState::get_server();

        blocking_server_fn(move || {
            let allowed = matches!(
                login,
                LoginState::Admin
                    | LoginState::NoAccounts
                    | LoginState::User { is_admin: true, .. }
            );
            GlobalBackend.with_local_archive(&archive, |a| {
                let Some(a) = a else {
                    return Err(format!("Archive {archive} not found"));
                };
                a.with_sources(|d| {
                    let d = match path {
                        None => d,
                        Some(p) => match d.find(RelPath::new(&p)) {
                            Some(SourceEntryRef::Dir(d)) => d,
                            _ => {
                                return Err(format!(
                                    "Directory {p} not found in archive {archive}"
                                ));
                            }
                        },
                    };
                    let mut ds = Vec::new();
                    let mut fs = Vec::new();
                    for d in &d.children {
                        match d {
                            SourceEntry::Dir(d) => ds.push(DirectoryData {
                                rel_path: d
                                    .relative_path
                                    .as_ref()
                                    .map(UriPath::to_string)
                                    .unwrap_or_default(),
                                summary: if allowed {
                                    Some(d.state.summarize())
                                } else {
                                    None
                                },
                            }),
                            SourceEntry::File(f) => fs.push(FileData {
                                rel_path: f.relative_path.to_string(),
                                format: f.format.to_string(),
                            }),
                        }
                    }
                    Ok((ds, fs))
                })
            })
        })
        .await
    }

    pub async fn archive_dependencies(
        archives: Vec<ArchiveId>,
    ) -> Result<Vec<ArchiveId>, ServerFnError<String>> {
        use flams_system::TokioEngine;
        let mut archives: VecSet<_> = archives.into_iter().collect();
        blocking_server_fn(move || {
            let mut ret = VecSet::new();
            let mut dones = VecSet::new();
            while let Some(archive) = archives.0.pop() {
                if dones.0.contains(&archive) {
                    continue;
                }
                dones.insert(archive.clone());
                let Some(iri) = GlobalBackend.with_tree(|tree| {
                    let mut steps = archive.steps();
                    if let Some(mut n) = steps.next() {
                        let mut curr = tree.top.as_slice();
                        while let Some(g) = curr.iter().find_map(|a| match a {
                            AoG::Group(g) if g.id.last() == n => Some(g),
                            _ => None,
                        }) {
                            curr = g.children.as_slice();
                            if let Some(a) = curr.iter().find_map(|a| match a {
                                AoG::Archive(a) if a.is_meta() => Some(a),
                                _ => None,
                            }) && !ret.0.contains(a)
                            {
                                ret.insert(a.clone());
                                archives.insert(a.clone());
                            }
                            if let Some(m) = steps.next() {
                                n = m;
                            } else {
                                break;
                            }
                        }
                    }
                    tree.get(&archive).map(|a| a.uri().to_iri())
                }) else {
                    return Err(format!("Archive {archive} not found"));
                };
                let res = GlobalBackend
                    .triple_store()
                    .query_str::<TokioEngine>(format!(
                        "SELECT DISTINCT ?a WHERE {{
                            <{}> ulo:contains ?d.
                            ?d rdf:type ulo:document .
                            ?d ulo:contains* ?x.
                            ?x (dc:requires|ulo:imports|dc:hasPart) ?m.
                            ?e ulo:contains? ?m.
                            ?e rdf:type ulo:document.
                            ?a ulo:contains ?e.
                        }}",
                        iri.as_str()
                    ))
                    .map_err(|e| e.to_string())?;
                for i in res.into_uris::<ArchiveUri>() {
                    let id = i.archive_id();
                    if !ret.0.contains(id) {
                        archives.insert(id.clone());
                        ret.insert(id.clone());
                    }
                }
            }
            Ok(ret.0)
        })
        .await
    }
    pub async fn build_status(
        archive: Option<ArchiveId>,
        path: Option<String>,
    ) -> Result<FileStates, ServerFnError<String>> {
        use flams_math_archives::Archive;
        use flams_math_archives::backend::LocalBackend;
        use flams_math_archives::manager::ArchiveOrGroup as AoG;
        let login = LoginState::get_server();

        blocking_server_fn(move || {
            let allowed = matches!(
                login,
                LoginState::Admin
                    | LoginState::NoAccounts
                    | LoginState::User { is_admin: true, .. }
            );
            if !allowed {
                return Err("Not logged in".to_string());
            }
            path.map_or_else(
                || {
                    archive.as_ref().map_or_else(
                        || {
                            Ok(GlobalBackend
                                .with_tree(flams_math_archives::manager::ArchiveTree::state)
                                .into())
                        },
                        |archive| {
                            GlobalBackend.with_tree(|tree| {
                                match tree.get_group_or_archive(archive) {
                                    None => Err(format!("Archive {archive} not found")),
                                    Some(AoG::Archive(id)) => {
                                        let Some(Archive::Local(archive)) = tree.get(id) else {
                                            return Err(format!("Archive {archive} not found"));
                                        };
                                        Ok(archive.file_state().into())
                                    }
                                    Some(AoG::Group(g)) => Ok(g.state.clone().into()),
                                }
                            })
                        },
                    )
                },
                |path| {
                    let Some(archive) = archive.as_ref() else {
                        return Err("path without archive".to_string());
                    };
                    GlobalBackend.with_local_archive(archive, |a| {
                        let Some(a) = a else {
                            return Err(format!("Archive {archive} not found"));
                        };
                        a.with_sources(|d| match d.find(RelPath::new(&path)) {
                            Some(SourceEntryRef::Dir(d)) => Ok(d.state.clone().into()),
                            Some(SourceEntryRef::File(f)) => Ok((&*f.target_state).into()),
                            None => Err(format!("Directory {path} not found in archive {archive}")),
                        })
                    })
                },
            )
        })
        .await
    }
    pub fn archive_stream(
        id: &ArchiveId,
    ) -> Result<leptos::server_fn::codec::ByteStream<ServerFnError<String>>, ServerFnError<String>>
    {
        use futures::TryStreamExt;
        let stream = GlobalBackend
            .with_local_archive(id, |a| a.map(flams_system::zip::zip))
            .ok_or_else(|| format!("No archive with id {id} found!"))?;
        Ok(leptos::server_fn::codec::ByteStream::new(
            stream.map_err(|e| e.to_string().into()), //.map_err(|e| ServerFnError::new(e.to_string())),
        ))
    }
}