robin/haze
1
0
Fork 0
mirror of https://codeberg.org/icewind/haze.git synced 2026-06-03 17:14:08 +02:00
Code Issues Projects Releases Wiki Activity

forward OCC_LOG

Browse source
This commit is contained in:
Robin Appelman 2023-06-15 17:54:11 +02:00
commit fe5562a9b5
6 changed files with 90 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

18
src/cloud.rs
View file

@ -436,36 +436,39 @@ impl Cloud {
Ok(())
}
pub async fn exec<S: Into<String>>(
pub async fn exec<S: Into<String>, Env: Into<String>>(
&self,
docker: &Docker,
cmd: Vec<S>,
tty: bool,
env: Vec<Env>,
) -> Result<ExitCode> {
if tty {
exec_tty(docker, &self.id, "haze", cmd, vec![]).await
exec_tty(docker, &self.id, "haze", cmd, env).await
} else {
exec(docker, &self.id, "haze", cmd, vec![], Some(stdout())).await
exec(docker, &self.id, "haze", cmd, env, Some(stdout())).await
}
}
pub async fn occ<'a, S: Into<String> + From<&'a str>>(
pub async fn occ<'a, S: Into<String> + From<&'a str>, Env: Into<String>>(
&self,
docker: &Docker,
mut cmd: Vec<S>,
output: Option<&mut Vec<u8>>,
env: Vec<Env>,
) -> Result<ExitCode> {
cmd.insert(0, "occ".into());
self.exec_with_output(docker, cmd, output).await
self.exec_with_output(docker, cmd, output, env).await
}
pub async fn exec_with_output<S: Into<String>>(
pub async fn exec_with_output<S: Into<String>, Env: Into<String>>(
&self,
docker: &Docker,
cmd: Vec<S>,
output: Option<impl Write>,
env: Vec<Env>,
) -> Result<ExitCode> {
exec(docker, &self.id, "haze", cmd, vec![], output).await
exec(docker, &self.id, "haze", cmd, env, output).await
}
pub async fn list(
@ -607,6 +610,7 @@ impl Cloud {
"--force".to_string(),
],
false,
Vec::<String>::default(),
)
.await?;
Ok(())

26
src/database.rs
View file

@ -252,9 +252,17 @@ impl Database {
_ => format!("{}-db", cloud_id),
};
if tty {
exec_tty(docker, &container, "root", cmd, vec![]).await
exec_tty(docker, &container, "root", cmd, Vec::<String>::default()).await
} else {
exec(docker, &container, "root", cmd, vec![], Some(stdout())).await
exec(
docker,
&container,
"root",
cmd,
Vec::<String>::default(),
Some(stdout()),
)
.await
}
}
@ -266,7 +274,7 @@ impl Database {
cloud_id,
"haze",
vec!["sqlite3", "/var/www/html/data/haze.db"],
vec![],
Vec::<String>::default(),
)
.await
}
@ -282,7 +290,7 @@ impl Database {
"-phaze",
"haze",
],
vec![],
Vec::<String>::default(),
)
.await
}
@ -302,7 +310,7 @@ impl Database {
format!("{}-db", cloud_id),
"root",
vec!["sqlplus", "system/haze"],
vec![],
Vec::<String>::default(),
)
.await
}
@ -355,7 +363,7 @@ impl Database {
format!("{}-db", cloud_id),
"root",
vec!["mysql", "-u", "haze", "-phaze", "-e", "SELECT 1"],
vec![],
Vec::<String>::default(),
Some(&mut output),
)
.await?;
@ -368,7 +376,7 @@ impl Database {
format!("{}-db", cloud_id),
"root",
vec!["pg_isready", "-U", "haze", "-q"],
vec![],
Vec::<String>::default(),
Option::<Stdout>::None,
)
.await?;
@ -378,7 +386,7 @@ impl Database {
format!("{}-db", cloud_id),
"root",
vec!["psql", "-U", "haze", "-qtA", "-c", ""],
vec![],
Vec::<String>::default(),
Option::<Stdout>::None,
)
.await?;
@ -394,7 +402,7 @@ impl Database {
format!("{}-db", cloud_id),
"root",
vec!["sh", "-c", r#"echo "show user" | sqlplus -S system/haze"#],
vec![],
Vec::<String>::default(),
Some(&mut output),
)
.await?;

12
src/exec.rs
View file

@ -11,12 +11,12 @@ use tokio::io::AsyncWriteExt;
use tokio::task::spawn;
use tokio::time::sleep;
pub async fn exec_tty<S1: AsRef<str>, S2: Into<String>>(
pub async fn exec_tty<S1: AsRef<str>, S2: Into<String>, Env: Into<String>>(
docker: &Docker,
container: S1,
user: &str,
cmd: Vec<S2>,
env: Vec<&str>,
env: Vec<Env>,
) -> Result<ExitCode> {
let stdout = stdout();
@ -26,7 +26,7 @@ pub async fn exec_tty<S1: AsRef<str>, S2: Into<String>>(
let tty_size = terminal_size().into_diagnostic()?;
let cmd = cmd.into_iter().map(S2::into).collect();
let env = env.into_iter().map(String::from).collect();
let env = env.into_iter().map(Env::into).collect();
let config = CreateExecOptions {
cmd: Some(cmd),
user: Some(user.to_string()),
@ -97,16 +97,16 @@ pub async fn exec_tty<S1: AsRef<str>, S2: Into<String>>(
.into())
}
pub async fn exec<S1: AsRef<str>, S2: Into<String>>(
pub async fn exec<S1: AsRef<str>, S2: Into<String>, Env: Into<String>>(
docker: &Docker,
container: S1,
user: &str,
cmd: Vec<S2>,
env: Vec<&str>,
env: Vec<Env>,
mut std_out: Option<impl Write>,
) -> Result<ExitCode> {
let cmd = cmd.into_iter().map(S2::into).collect();
let env = env.into_iter().map(String::from).collect();
let env = env.into_iter().map(Env::into).collect();
let config = CreateExecOptions {
cmd: Some(cmd),
user: Some(user.to_string()),

61
src/main.rs
View file

@ -12,6 +12,7 @@ use crate::service::Service;
use crate::service::ServiceTrait;
use bollard::Docker;
use miette::{IntoDiagnostic, Report, Result, WrapErr};
use std::env::vars;
use std::io::stdout;
use std::os::unix::process::CommandExt;
use std::process::Command;
@ -29,6 +30,15 @@ mod php;
mod proxy;
mod service;
static FORWARD_ENV: &[&str] = &["OCC_LOG"];
fn get_forward_env() -> Vec<String> {
vars()
.filter(|(var, _)| FORWARD_ENV.contains(&var.as_str()))
.map(|(var, value)| format!("{var}={value}"))
.collect()
}
#[tokio::main]
async fn main() -> Result<()> {
miette::set_panic_hook();
@ -106,6 +116,7 @@ async fn main() -> Result<()> {
command
},
atty::is(atty::Stream::Stdout),
get_forward_env(),
)
.await?;
}
@ -133,7 +144,12 @@ async fn main() -> Result<()> {
let cloud = Cloud::get_by_filter(&docker, filter, &config).await?;
command.insert(0, "occ".to_string());
cloud
.exec(&docker, command, atty::is(atty::Stream::Stdout))
.exec(
&docker,
command,
atty::is(atty::Stream::Stdout),
get_forward_env(),
)
.await?;
}
HazeArgs::Db { filter, root } => {
@ -161,6 +177,7 @@ async fn main() -> Result<()> {
&config.auto_setup.password,
],
false,
Vec::<String>::default(),
)
.await
{
@ -180,7 +197,7 @@ async fn main() -> Result<()> {
cloud.enable_app(&docker, app).await?;
}
args.insert(0, "tests".to_string());
cloud.exec(&docker, args, false).await?;
cloud.exec(&docker, args, false, get_forward_env()).await?;
cloud.destroy(&docker).await?;
}
HazeArgs::Integration { options, mut args } => {
@ -197,6 +214,7 @@ async fn main() -> Result<()> {
&config.auto_setup.password,
],
false,
Vec::<String>::default(),
)
.await
{
@ -204,7 +222,7 @@ async fn main() -> Result<()> {
return Err(e);
}
args.insert(0, "integration".to_string());
cloud.exec(&docker, args, false).await?;
cloud.exec(&docker, args, false, get_forward_env()).await?;
cloud.destroy(&docker).await?;
}
HazeArgs::Fmt { path } => {
@ -214,7 +232,12 @@ async fn main() -> Result<()> {
cloud.wait_for_start(&docker).await?;
println!("Installing composer");
if let Err(e) = cloud
.exec_with_output(&docker, vec!["composer", "install"], Some(&mut out_buffer))
.exec_with_output(
&docker,
vec!["composer", "install"],
Some(&mut out_buffer),
Vec::<String>::default(),
)
.await
.and_then(|c| c.to_result())
{
@ -229,6 +252,7 @@ async fn main() -> Result<()> {
&docker,
vec!["composer", "run", "cs:fix", path.as_str()],
false,
Vec::<String>::default(),
)
.await
{
@ -241,6 +265,7 @@ async fn main() -> Result<()> {
&docker,
vec!["git", "clean", "-fd", "lib/composer"],
Some(&mut out_buffer),
Vec::<String>::default(),
)
.await
.and_then(|c| c.to_result())
@ -254,6 +279,7 @@ async fn main() -> Result<()> {
&docker,
vec!["git", "checkout", "lib/composer"],
Some(&mut out_buffer),
Vec::<String>::default(),
)
.await
.and_then(|c| c.to_result())
@ -275,6 +301,7 @@ async fn main() -> Result<()> {
command
},
true,
get_forward_env(),
)
.await?;
cloud.destroy(&docker).await?;
@ -355,6 +382,7 @@ async fn setup(docker: &Docker, options: CloudOptions, config: &HazeConfig) -> R
&config.auto_setup.password,
],
false,
Vec::<String>::default(),
)
.await?;
cloud
@ -367,6 +395,7 @@ async fn setup(docker: &Docker, options: CloudOptions, config: &HazeConfig) -> R
&cloud.address,
],
None,
Vec::<String>::default(),
)
.await?;
cloud
@ -374,6 +403,7 @@ async fn setup(docker: &Docker, options: CloudOptions, config: &HazeConfig) -> R
docker,
vec!["config:system:set", "overwritehost", "--value", host],
None,
Vec::<String>::default(),
)
.await?;
if cloud.address.contains("https://") {
@ -382,6 +412,7 @@ async fn setup(docker: &Docker, options: CloudOptions, config: &HazeConfig) -> R
docker,
vec!["config:system:set", "overwriteprotocol", "--value", "https"],
None,
Vec::<String>::default(),
)
.await?;
}
@ -399,6 +430,7 @@ async fn setup(docker: &Docker, options: CloudOptions, config: &HazeConfig) -> R
domain,
],
None,
Vec::<String>::default(),
)
.await?;
}
@ -406,20 +438,35 @@ async fn setup(docker: &Docker, options: CloudOptions, config: &HazeConfig) -> R
for service in &cloud.services {
for app in service.apps() {
cloud
.exec(docker, vec!["occ", "app:enable", *app, "--force"], false)
.exec(
docker,
vec!["occ", "app:enable", *app, "--force"],
false,
Vec::<String>::default(),
)
.await?;
}
}
for service in &cloud.services {
for cmd in service.post_setup(docker, &cloud.id, config).await? {
cloud
.exec(docker, shell_words::split(&cmd).into_diagnostic()?, false)
.exec(
docker,
shell_words::split(&cmd).into_diagnostic()?,
false,
Vec::<String>::default(),
)
.await?;
}
}
for cmd in &config.auto_setup.post_setup {
cloud
.exec(docker, shell_words::split(cmd).into_diagnostic()?, false)
.exec(
docker,
shell_words::split(cmd).into_diagnostic()?,
false,
Vec::<String>::default(),
)
.await?;
}
}

2
src/service/kaspersky.rs
View file

@ -77,7 +77,7 @@ impl ServiceTrait for Kaspersky {
self.container_name(cloud_id),
"root",
vec!["curl", "localhost/licenseinfo"],
vec![],
Vec::<String>::default(),
Option::<Stdout>::None,
)
.await?;

2
src/service/objectstore.rs
View file

@ -126,7 +126,7 @@ impl ServiceTrait for ObjectStore {
format!("{}-object", cloud_id),
"root",
vec!["curl", "localhost:9000/minio/health/ready"],
vec![],
Vec::<String>::default(),
Some(&mut output),
)
.await?;