99 lines
2.6 KiB
Rust
99 lines
2.6 KiB
Rust
use bollard::image::ListImagesOptions;
|
|
use bollard::models::ImageSummary;
|
|
use bollard::Docker;
|
|
use bollard::API_DEFAULT_VERSION;
|
|
use log::info;
|
|
use std::collections::HashMap;
|
|
|
|
use crate::DateArgs;
|
|
|
|
const GHCR_REPO: &str = "ghcr.io/datahearth/clear-docker-images";
|
|
const DOCKER_REPO: &str = "datahearth/clear-docker-images";
|
|
|
|
pub struct DockerActions {
|
|
docker: Docker,
|
|
repository: Option<String>,
|
|
tags: Vec<String>,
|
|
date: DateArgs,
|
|
}
|
|
|
|
impl DockerActions {
|
|
pub fn new(
|
|
socket: String,
|
|
repository: Option<String>,
|
|
tags: Vec<String>,
|
|
date: DateArgs,
|
|
) -> Self {
|
|
Self {
|
|
docker: Docker::connect_with_socket(&socket, 120, API_DEFAULT_VERSION).unwrap(),
|
|
repository,
|
|
tags,
|
|
date,
|
|
}
|
|
}
|
|
|
|
pub async fn get(&self) -> Result<Vec<ImageSummary>, bollard::errors::Error> {
|
|
let mut image_filters = HashMap::new();
|
|
|
|
// why using &self.repository instead of selft.repository ?
|
|
if let Some(r) = &self.repository {
|
|
image_filters.insert("reference", vec![r.as_str()]);
|
|
}
|
|
|
|
self.docker
|
|
.list_images(Some(ListImagesOptions {
|
|
all: true,
|
|
filters: image_filters,
|
|
..Default::default()
|
|
}))
|
|
.await
|
|
}
|
|
|
|
pub async fn delete(
|
|
&self,
|
|
images: Vec<ImageSummary>,
|
|
dry_run: bool,
|
|
) -> Result<i64, bollard::errors::Error> {
|
|
let mut removed_size = 0;
|
|
for image in images {
|
|
info!("deleting: {}", image.id);
|
|
|
|
if !dry_run {
|
|
if let Err(e) = self.docker.delete_service(&image.id).await {
|
|
return Err(e);
|
|
}
|
|
}
|
|
|
|
removed_size += image.size;
|
|
}
|
|
|
|
Ok(removed_size)
|
|
}
|
|
|
|
pub fn filter(&self, images: Vec<ImageSummary>) -> Vec<ImageSummary> {
|
|
let mut to_be_deleted: Vec<ImageSummary> = vec![];
|
|
|
|
for image in images {
|
|
if self
|
|
.date
|
|
.stop
|
|
.map_or(self.date.start > image.created, |stop| {
|
|
self.date.start > image.created && image.created > stop
|
|
})
|
|
&& image.repo_tags.iter().any(|tag| {
|
|
!tag.contains(GHCR_REPO)
|
|
&& !tag.contains(DOCKER_REPO)
|
|
&& !self
|
|
.tags
|
|
.iter()
|
|
.any(|excluded_tag| tag.contains(excluded_tag))
|
|
})
|
|
{
|
|
to_be_deleted.push(image);
|
|
}
|
|
}
|
|
|
|
return to_be_deleted;
|
|
}
|
|
}
|