use crate::command::{CommandExecutor, ComposeCommand, ComposeConfig, DockerCommand};
use crate::error::Result;
use async_trait::async_trait;
#[derive(Debug, Clone)]
pub struct ComposePushCommand {
pub executor: CommandExecutor,
pub config: ComposeConfig,
pub services: Vec<String>,
pub ignore_build_failures: bool,
pub include_deps: bool,
pub quiet: bool,
}
#[derive(Debug, Clone)]
pub struct ComposePushResult {
pub stdout: String,
pub stderr: String,
pub success: bool,
pub services: Vec<String>,
}
impl ComposePushCommand {
#[must_use]
pub fn new() -> Self {
Self {
executor: CommandExecutor::new(),
config: ComposeConfig::new(),
services: Vec::new(),
ignore_build_failures: false,
include_deps: false,
quiet: false,
}
}
#[must_use]
pub fn service(mut self, service: impl Into<String>) -> Self {
self.services.push(service.into());
self
}
#[must_use]
pub fn services<I, S>(mut self, services: I) -> Self
where
I: IntoIterator<Item = S>,
S: Into<String>,
{
self.services.extend(services.into_iter().map(Into::into));
self
}
#[must_use]
pub fn ignore_build_failures(mut self) -> Self {
self.ignore_build_failures = true;
self
}
#[must_use]
pub fn include_deps(mut self) -> Self {
self.include_deps = true;
self
}
#[must_use]
pub fn quiet(mut self) -> Self {
self.quiet = true;
self
}
}
impl Default for ComposePushCommand {
fn default() -> Self {
Self::new()
}
}
#[async_trait]
impl DockerCommand for ComposePushCommand {
type Output = ComposePushResult;
fn get_executor(&self) -> &CommandExecutor {
&self.executor
}
fn get_executor_mut(&mut self) -> &mut CommandExecutor {
&mut self.executor
}
fn build_command_args(&self) -> Vec<String> {
<Self as ComposeCommand>::build_command_args(self)
}
async fn execute(&self) -> Result<Self::Output> {
let args = <Self as ComposeCommand>::build_command_args(self);
let output = self.execute_command(args).await?;
Ok(ComposePushResult {
stdout: output.stdout,
stderr: output.stderr,
success: output.success,
services: self.services.clone(),
})
}
}
impl ComposeCommand for ComposePushCommand {
fn get_config(&self) -> &ComposeConfig {
&self.config
}
fn get_config_mut(&mut self) -> &mut ComposeConfig {
&mut self.config
}
fn subcommand(&self) -> &'static str {
"push"
}
fn build_subcommand_args(&self) -> Vec<String> {
let mut args = Vec::new();
if self.ignore_build_failures {
args.push("--ignore-build-failures".to_string());
}
if self.include_deps {
args.push("--include-deps".to_string());
}
if self.quiet {
args.push("--quiet".to_string());
}
args.extend(self.services.clone());
args
}
}
impl ComposePushResult {
#[must_use]
pub fn success(&self) -> bool {
self.success
}
#[must_use]
pub fn services(&self) -> &[String] {
&self.services
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_compose_push_basic() {
let cmd = ComposePushCommand::new();
let args = cmd.build_subcommand_args();
assert!(args.is_empty());
let full_args = ComposeCommand::build_command_args(&cmd);
assert_eq!(full_args[0], "compose");
assert!(full_args.contains(&"push".to_string()));
}
#[test]
fn test_compose_push_with_options() {
let cmd = ComposePushCommand::new()
.ignore_build_failures()
.include_deps()
.quiet()
.service("web");
let args = cmd.build_subcommand_args();
assert!(args.contains(&"--ignore-build-failures".to_string()));
assert!(args.contains(&"--include-deps".to_string()));
assert!(args.contains(&"--quiet".to_string()));
assert!(args.contains(&"web".to_string()));
}
}