kellnr-migration 5.14.0

Kellnr is a self-hosted registry for Rust crates with support for rustdocs and crates.io caching.
Documentation 
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85

use kellnr_settings::{Settings, get_settings};
use sea_orm::{ModelTrait, Related};
use sea_orm_migration::prelude::*;
use tracing::debug;

use crate::sea_orm::ActiveValue::Set;
use crate::sea_orm::{ActiveModelTrait, EntityTrait};

#[derive(DeriveMigrationName)]
pub struct Migration;

#[async_trait::async_trait]
impl MigrationTrait for Migration {
    async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
        move_cached_crates(manager.get_connection()).await
    }

    async fn down(&self, _manager: &SchemaManager) -> Result<(), DbErr> {
        Ok(())
    }
}

async fn move_cached_crates(db: &SchemaManagerConnection<'_>) -> Result<(), DbErr> {
    use crate::m20220101_000009_create_table_entities::{cratesio_crate, cratesio_index};

    debug!("Moving cached crates...");
    let settings = get_settings().map_err(|e| DbErr::Custom(e.to_string()))?;

    // Get all cached crate versions
    let cached_indices = cratesio_index::Entity::find()
        .all(db)
        .await?
        .into_iter()
        .collect::<Vec<_>>();

    if cached_indices.is_empty() {
        // There is nothing to do
        debug!("No cached crates to move...");
        return Ok(());
    }

    // Make sure the cratesio bin path exists
    if !settings.crates_io_bin_path().exists() {
        std::fs::create_dir_all(settings.crates_io_bin_path())
            .map_err(|e| DbErr::Custom(e.to_string()))?;
    }

    // Move each crate to the new location
    for cached_index in cached_indices {
        let cached_crate = cached_index
            .find_related(cratesio_crate::Entity)
            .one(db)
            .await?;
        let name = cached_crate
            .ok_or(DbErr::Custom("No crate found".to_string()))?
            .original_name;
        let (old, new) = get_path(&settings, &name, &cached_index.vers);

        if !old.exists() {
            // crate is in the database index but not on disk. Skip the move.
            continue;
        }

        // Move the crate
        debug!("Moving {name} from {old:?} to {new:?}");
        if let Err(e) = std::fs::rename(old, new).map_err(|e| DbErr::Custom(e.to_string())) {
            debug!("Failed to move {name}: {e}");
            continue;
        }
    }
    debug!("Done moving cached crates");
    Ok(())
}

fn get_path(
    settings: &Settings,
    name: &str,
    version: &str,
) -> (std::path::PathBuf, std::path::PathBuf) {
    let crate_name = format!("{name}-{version}.crate");
    (
        settings.bin_path().join(&crate_name),
        settings.crates_io_bin_path().join(&crate_name),
    )
}