aboutsummaryrefslogtreecommitdiff
path: root/server/src/import.rs
blob: 72fb3998be80d76190d3d1a4419993dc972d9de1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
/*
    This file is part of jellything (https://codeberg.org/metamuffin/jellything)
    which is licensed under the GNU Affero General Public License (version 3); see /COPYING.
    Copyright (C) 2023 metamuffin <metamuffin.org>
*/
use crate::{database::Database, CONF};
use anyhow::{bail, Context, Ok};
use jellycommon::Node;
use log::info;
use std::{ffi::OsStr, fs::File, os::unix::prelude::OsStrExt, path::PathBuf};

pub fn import(db: &Database) -> anyhow::Result<()> {
    info!("clearing node tree");
    db.node.clear()?;
    info!("importing...");
    import_path(CONF.library_path.clone(), db).context("indexing")?;
    Ok(())
}

pub fn import_path(path: PathBuf, db: &Database) -> anyhow::Result<Vec<String>> {
    if path.is_dir() {
        let mpath = path.join("directory.json");
        let children = path.read_dir()?.filter_map(|e| {
            let e = e.unwrap();
            if e.path().extension() == Some(&OsStr::from_bytes(b"jelly"))
                || e.metadata().unwrap().is_dir()
            {
                Some(e.path())
            } else {
                None
            }
        });
        let children = children
            .map(|e| import_path(e, db))
            .collect::<anyhow::Result<Vec<_>>>()?
            .into_iter()
            .flatten()
            .collect();
        if mpath.exists() {
            let mut data: Node =
                serde_json::from_reader(File::open(mpath).context("metadata missing")?)?;

            let identifier = path.file_name().unwrap().to_str().unwrap().to_string();
            
            data.public.children = children;
            info!("insert {identifier}");
            db.node.insert(&identifier, &data)?;

            Ok(vec![identifier])
        } else {
            Ok(children)
        }
    } else if path.is_file() {
        info!("loading item {path:?}");
        let datafile = File::open(path.clone()).context("cant load metadata")?;
        let data: Node = serde_json::from_reader(datafile).context("invalid metadata")?;
        let identifier = path
            .file_name()
            .unwrap()
            .to_str()
            .unwrap()
            .strip_suffix(".jelly")
            .unwrap()
            .to_string();
        info!("insert {identifier}");
        db.node.insert(&identifier, &data)?;
        Ok(vec![identifier])
    } else {
        bail!("did somebody really put a fifo or socket in the library?!")
    }
}