mii/rs-docker-bot
1
0
Fork 0
mirror of https://github.com/mii443/rs-docker-bot.git synced 2025-08-22 16:15:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

upload code without writing tar.gz file

Browse Source
This commit is contained in:
mii443
2024-10-09 05:24:06 +00:00
parent 0d69168c26
commit 52bab5bb0b
2 changed files with 24 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Cargo.toml
View File

@ -6,7 +6,7 @@ edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
bollard = "0.13"
bollard = "0.17.1"
futures-util = "0.3"
serde_json = "1.0"
serde_yaml = "0.9"

43
src/docker.rs
View File

@ -16,6 +16,7 @@ use bollard::{
};
use flate2::{write::GzEncoder, Compression};
use futures_util::StreamExt;
use tar::Header;
use tokio::task::JoinHandle;
use crate::language::Language;
@ -46,7 +47,13 @@ impl Container {
let name = format!("dockerbot-{}", uuid::Uuid::new_v4().to_string());
let id = docker
.create_container(Some(CreateContainerOptions { name: name.clone() }), config)
.create_container(
Some(CreateContainerOptions {
name: name.clone(),
platform: None,
}),
config,
)
.await
.unwrap()
.id;
@ -171,28 +178,24 @@ impl Container {
let docker = Docker::connect_with_local_defaults().unwrap();
let path = self.language.clone().unwrap().get_path(file_name.clone());
{
let mut f = File::create(&format!("code/{}", path)).unwrap();
f.write_all(content.as_bytes()).unwrap();
f.flush().unwrap();
f.sync_all().unwrap();
}
{
let mut f = File::open(&format!("code/{}", path)).unwrap();
let tar_gz = File::create(&format!("code/{}.tar.gz", file_name)).unwrap();
let encoder = GzEncoder::new(tar_gz, Compression::default());
let code = {
let encoder = GzEncoder::new(vec![], Compression::default());
let mut tar = tar::Builder::new(encoder);
tar.append_file(path.clone(), &mut f).unwrap();
let mut header = Header::new_gnu();
header.set_path(&path).unwrap();
header.set_size(content.as_bytes().len() as u64);
header.set_cksum();
tar.append(&header, content.as_bytes()).unwrap();
tar.finish().unwrap();
}
let encoder = tar.into_inner().unwrap();
encoder.finish().unwrap()
};
{
let mut f = File::open(&format!("code/{}.tar.gz", file_name)).unwrap();
let mut contents = Vec::new();
f.read_to_end(&mut contents).unwrap();
docker
.start_container::<String>(&self.id, None)
.await
@ -205,7 +208,7 @@ impl Container {
path: "/",
..Default::default()
}),
contents.into(),
code.into(),
)
.await
.unwrap();