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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
|
use bollard::Docker;
use bollard::container::{
Config as ContainerConfig, CreateContainerOptions, RemoveContainerOptions,
StartContainerOptions, StopContainerOptions,
};
use bollard::image::CreateImageOptions;
use bollard::models::HostConfig;
use futures_util::StreamExt;
use std::collections::HashMap;
use crate::config::Config;
use crate::models::Instance;
#[derive(Clone)]
pub struct DockerManager {
client: Docker,
config: Config,
}
impl DockerManager {
pub fn new(config: Config) -> Result<Self, bollard::errors::Error> {
let client = Docker::connect_with_local_defaults()?;
Ok(Self { client, config })
}
pub async fn start_container(&self, instance: &Instance) -> Result<(), String> {
tracing::info!("Pulling itzg/minecraft-server:latest image...");
self.pull_image().await?;
tracing::info!("Creating container for instance '{}'", instance.name);
self.recreate_container(instance).await?;
tracing::info!("Starting container...");
self.start_existing().await
}
async fn pull_image(&self) -> Result<(), String> {
let options = CreateImageOptions {
from_image: "itzg/minecraft-server",
tag: "latest",
..Default::default()
};
let mut stream = self.client.create_image(Some(options), None, None);
while let Some(result) = stream.next().await {
match result {
Ok(_info) => {}
Err(e) => return Err(format!("Image pull failed: {e}")),
}
}
Ok(())
}
async fn recreate_container(&self, instance: &Instance) -> Result<(), String> {
// Remove existing container if present
let remove_options = RemoveContainerOptions {
force: true,
..Default::default()
};
match self
.client
.remove_container("mc", Some(remove_options))
.await
{
Ok(_) => tracing::info!("Removed existing container"),
Err(e) => tracing::debug!("No existing container to remove: {e}"),
}
let server_dir = self
.config
.server_directory
.canonicalize()
.unwrap_or_else(|_| self.config.server_directory.clone())
.display()
.to_string();
let instance_data = format!("{server_dir}/instances/{}/data", instance.name);
let mut env_strings: Vec<String> = vec![
"EULA=TRUE".into(),
format!("VERSION={}", instance.version),
format!("TYPE={}", instance.server_type),
format!("MEMORY={}M", instance.memory_mb),
format!("MAX_PLAYERS={}", instance.max_players),
format!("DIFFICULTY={}", instance.difficulty),
"FORCE_GAMEMODE=true".into(),
"SPAWN_PROTECTION=0".into(),
"ENABLE_WHITELIST=true".into(),
"PAUSE_WHEN_EMPTY_SECONDS=30".into(),
"ONLINE_MODE=false".into(),
"TZ=Europe/Warsaw".into(),
"USE_AIKAR_FLAGS=true".into(),
"ENABLE_RCON=true".into(),
format!("RCON_PASSWORD={}", instance.rcon_password),
"RCON_PORT=25575".into(),
format!("MOTD={}", instance.motd),
format!("GAMEMODE={}", instance.gamemode),
format!("LEVEL_TYPE={}", instance.level_type),
];
if instance.pvp {
env_strings.push("PVP=true".into());
} else {
env_strings.push("PVP=false".into());
}
if let Some(ref seed) = instance.world_seed {
if !seed.is_empty() {
env_strings.push(format!("LEVEL_SEED={seed}"));
}
}
let env_refs: Vec<&str> = env_strings.iter().map(|s| s.as_str()).collect();
let mut exposed_ports: HashMap<&str, HashMap<(), ()>> = HashMap::new();
exposed_ports.insert("25565/tcp", HashMap::new());
exposed_ports.insert("25575/tcp", HashMap::new());
let mut port_bindings = HashMap::new();
port_bindings.insert(
"25565/tcp".to_string(),
Some(vec![bollard::models::PortBinding {
host_ip: None,
host_port: Some("25565".to_string()),
}]),
);
port_bindings.insert(
"25575/tcp".to_string(),
Some(vec![bollard::models::PortBinding {
host_ip: None,
host_port: Some("25575".to_string()),
}]),
);
let host_config = HostConfig {
binds: Some(vec![format!("{instance_data}:/data")]),
port_bindings: Some(port_bindings),
..Default::default()
};
let container_config = ContainerConfig {
image: Some("itzg/minecraft-server:latest"),
tty: Some(true),
env: Some(env_refs),
exposed_ports: Some(exposed_ports),
host_config: Some(host_config),
..Default::default()
};
let options = CreateContainerOptions {
name: "mc",
..Default::default()
};
self.client
.create_container(Some(options), container_config)
.await
.map_err(|e| format!("Failed to create container: {e}"))?;
tracing::info!("Container 'mc' created successfully");
Ok(())
}
async fn start_existing(&self) -> Result<(), String> {
self.client
.start_container("mc", None::<StartContainerOptions<&str>>)
.await
.map_err(|e| format!("Failed to start container: {e}"))?;
Ok(())
}
pub async fn stop_container(&self) -> Result<(), String> {
let options = StopContainerOptions { t: 30 };
self.client
.stop_container("mc", Some(options))
.await
.map_err(|e| format!("Failed to stop container: {e}"))?;
Ok(())
}
}
|
