Skip to content

Commit

Permalink
fix: Remove hard requirement for login creds to be able to push (#187)
Browse files Browse the repository at this point in the history
  • Loading branch information
gmpinder authored May 29, 2024
1 parent 02b2fe5 commit 9dd1ec9
Show file tree
Hide file tree
Showing 8 changed files with 108 additions and 99 deletions.
4 changes: 2 additions & 2 deletions Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ version = "0.8.9"
[workspace.dependencies]
anyhow = "1"
chrono = "0.4"
clap = { version = "4", features = ["derive", "cargo", "unicode"] }
clap = "4"
colored = "2"
env_logger = "0.11"
format_serde_error = "0.3"
Expand Down Expand Up @@ -74,7 +74,7 @@ users = "0.11"
# Workspace dependencies
anyhow.workspace = true
chrono.workspace = true
clap.workspace = true
clap = { workspace = true, features = ["derive", "cargo", "unicode", "env"] }
colored.workspace = true
env_logger.workspace = true
indexmap.workspace = true
Expand Down
76 changes: 37 additions & 39 deletions src/commands/build.rs
Original file line number Diff line number Diff line change
Expand Up @@ -7,11 +7,12 @@ use std::{
use anyhow::{bail, Context, Result};
use blue_build_recipe::Recipe;
use blue_build_utils::constants::{
ARCHIVE_SUFFIX, BUILD_ID_LABEL, CI_DEFAULT_BRANCH, CI_PROJECT_NAME, CI_PROJECT_NAMESPACE,
CI_PROJECT_URL, CI_REGISTRY, CI_SERVER_HOST, CI_SERVER_PROTOCOL, CONFIG_PATH, CONTAINER_FILE,
COSIGN_PATH, COSIGN_PRIVATE_KEY, GITHUB_REPOSITORY_OWNER, GITHUB_TOKEN,
GITHUB_TOKEN_ISSUER_URL, GITHUB_WORKFLOW_REF, GITIGNORE_PATH, LABELED_ERROR_MESSAGE,
NO_LABEL_ERROR_MESSAGE, RECIPE_FILE, RECIPE_PATH, SIGSTORE_ID_TOKEN,
ARCHIVE_SUFFIX, BB_PASSWORD, BB_REGISTRY, BB_REGISTRY_NAMESPACE, BB_USERNAME, BUILD_ID_LABEL,
CI_DEFAULT_BRANCH, CI_PROJECT_NAME, CI_PROJECT_NAMESPACE, CI_PROJECT_URL, CI_REGISTRY,
CI_SERVER_HOST, CI_SERVER_PROTOCOL, CONFIG_PATH, CONTAINER_FILE, COSIGN_PATH,
COSIGN_PRIVATE_KEY, GITHUB_REPOSITORY_OWNER, GITHUB_TOKEN, GITHUB_TOKEN_ISSUER_URL,
GITHUB_WORKFLOW_REF, GITIGNORE_PATH, LABELED_ERROR_MESSAGE, NO_LABEL_ERROR_MESSAGE,
RECIPE_FILE, RECIPE_PATH, SIGSTORE_ID_TOKEN,
};
use clap::Args;
use colored::Colorize;
Expand All @@ -20,7 +21,7 @@ use typed_builder::TypedBuilder;

use crate::{
commands::generate::GenerateCommand,
credentials,
credentials::{self, Credentials},
drivers::{
opts::{BuildTagPushOpts, CompressionType, GetMetadataOpts},
Driver,
Expand Down Expand Up @@ -78,26 +79,26 @@ pub struct BuildCommand {
archive: Option<PathBuf>,

/// The registry's domain name.
#[arg(long)]
#[arg(long, env = BB_REGISTRY)]
#[builder(default, setter(into, strip_option))]
registry: Option<String>,

/// The url path to your base
/// project images.
#[arg(long)]
#[arg(long, env = BB_REGISTRY_NAMESPACE)]
#[builder(default, setter(into, strip_option))]
#[arg(visible_alias("registry-path"))]
registry_namespace: Option<String>,

/// The username to login to the
/// container registry.
#[arg(short = 'U', long)]
#[arg(short = 'U', long, env = BB_USERNAME, hide_env_values = true)]
#[builder(default, setter(into, strip_option))]
username: Option<String>,

/// The password to login to the
/// container registry.
#[arg(short = 'P', long)]
#[arg(short = 'P', long, env = BB_PASSWORD, hide_env_values = true)]
#[builder(default, setter(into, strip_option))]
password: Option<String>,

Expand Down Expand Up @@ -129,6 +130,8 @@ impl BlueBuildCommand for BuildCommand {
check_cosign_files()?;
}

Self::login()?;

// Check if the Containerfile exists
// - If doesn't => *Build*
// - If it does:
Expand Down Expand Up @@ -213,10 +216,6 @@ impl BuildCommand {
let tags = recipe.generate_tags(os_version);
let image_name = self.generate_full_image_name(&recipe)?;

if self.push {
Self::login()?;
}

let opts = if let Some(archive_dir) = self.archive.as_ref() {
BuildTagPushOpts::builder()
.archive_path(format!(
Expand Down Expand Up @@ -253,32 +252,31 @@ impl BuildCommand {
trace!("BuildCommand::login()");
info!("Attempting to login to the registry");

let credentials = credentials::get()?;

let (registry, username, password) = (
&credentials.registry,
&credentials.username,
&credentials.password,
);

info!("Logging into the registry, {registry}");
Driver::get_build_driver().login()?;

trace!("cosign login -u {username} -p [MASKED] {registry}");
let login_output = Command::new("cosign")
.arg("login")
.arg("-u")
.arg(username)
.arg("-p")
.arg(password)
.arg(registry)
.output()?;

if !login_output.status.success() {
let err_output = String::from_utf8_lossy(&login_output.stderr);
bail!("Failed to login for cosign: {err_output}");
if let Some(Credentials {
registry,
username,
password,
}) = credentials::get()
{
info!("Logging into the registry, {registry}");
Driver::get_build_driver().login()?;

trace!("cosign login -u {username} -p [MASKED] {registry}");
let login_output = Command::new("cosign")
.arg("login")
.arg("-u")
.arg(username)
.arg("-p")
.arg(password)
.arg(registry)
.output()?;

if !login_output.status.success() {
let err_output = String::from_utf8_lossy(&login_output.stderr);
bail!("Failed to login for cosign: {err_output}");
}
info!("Login success at {registry}");
}
info!("Login success at {registry}");

Ok(())
}
Expand Down
6 changes: 2 additions & 4 deletions src/credentials.rs
Original file line number Diff line number Diff line change
Expand Up @@ -128,9 +128,7 @@ pub fn set_user_creds(
///
/// # Errors
/// Will error if there aren't any credentials available.
pub fn get() -> Result<&'static Credentials> {
pub fn get() -> Option<&'static Credentials> {
trace!("credentials::get()");
ENV_CREDENTIALS
.as_ref()
.ok_or_else(|| anyhow!("No credentials available"))
ENV_CREDENTIALS.as_ref()
}
38 changes: 21 additions & 17 deletions src/drivers/buildah_driver.rs
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ use log::{error, info, trace};
use semver::Version;
use serde::Deserialize;

use crate::credentials;
use crate::credentials::{self, Credentials};

use super::{
opts::{BuildOpts, PushOpts, TagOpts},
Expand Down Expand Up @@ -109,22 +109,26 @@ impl BuildDriver for BuildahDriver {
fn login(&self) -> Result<()> {
trace!("BuildahDriver::login()");

let (registry, username, password) =
credentials::get().map(|c| (&c.registry, &c.username, &c.password))?;

trace!("buildah login -u {username} -p [MASKED] {registry}");
let output = Command::new("buildah")
.arg("login")
.arg("-u")
.arg(username)
.arg("-p")
.arg(password)
.arg(registry)
.output()?;

if !output.status.success() {
let err_out = String::from_utf8_lossy(&output.stderr);
bail!("Failed to login for buildah: {err_out}");
if let Some(Credentials {
registry,
username,
password,
}) = credentials::get()
{
trace!("buildah login -u {username} -p [MASKED] {registry}");
let output = Command::new("buildah")
.arg("login")
.arg("-u")
.arg(username)
.arg("-p")
.arg(password)
.arg(registry)
.output()?;

if !output.status.success() {
let err_out = String::from_utf8_lossy(&output.stderr);
bail!("Failed to login for buildah: {err_out}");
}
}
Ok(())
}
Expand Down
36 changes: 20 additions & 16 deletions src/drivers/docker_driver.rs
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ use once_cell::sync::Lazy;
use semver::Version;
use serde::Deserialize;

use crate::image_metadata::ImageMetadata;
use crate::{credentials::Credentials, image_metadata::ImageMetadata};

use super::{
credentials,
Expand Down Expand Up @@ -169,22 +169,26 @@ impl BuildDriver for DockerDriver {
fn login(&self) -> Result<()> {
trace!("DockerDriver::login()");

let (registry, username, password) =
credentials::get().map(|c| (&c.registry, &c.username, &c.password))?;

trace!("docker login -u {username} -p [MASKED] {registry}");
let output = Command::new("docker")
.arg("login")
.arg("-u")
.arg(username)
.arg("-p")
.arg(password)
.arg(registry)
.output()?;
if let Some(Credentials {
registry,
username,
password,
}) = credentials::get()
{
trace!("docker login -u {username} -p [MASKED] {registry}");
let output = Command::new("docker")
.arg("login")
.arg("-u")
.arg(username)
.arg("-p")
.arg(password)
.arg(registry)
.output()?;

if !output.status.success() {
let err_out = String::from_utf8_lossy(&output.stderr);
bail!("Failed to login for docker: {err_out}");
if !output.status.success() {
let err_out = String::from_utf8_lossy(&output.stderr);
bail!("Failed to login for docker: {err_out}");
}
}
Ok(())
}
Expand Down
38 changes: 21 additions & 17 deletions src/drivers/podman_driver.rs
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ use log::{debug, error, info, trace};
use semver::Version;
use serde::Deserialize;

use crate::image_metadata::ImageMetadata;
use crate::{credentials::Credentials, image_metadata::ImageMetadata};

use super::{
credentials,
Expand Down Expand Up @@ -120,22 +120,26 @@ impl BuildDriver for PodmanDriver {
fn login(&self) -> Result<()> {
trace!("PodmanDriver::login()");

let (registry, username, password) =
credentials::get().map(|c| (&c.registry, &c.username, &c.password))?;

trace!("podman login -u {username} -p [MASKED] {registry}");
let output = Command::new("podman")
.arg("login")
.arg("-u")
.arg(username)
.arg("-p")
.arg(password)
.arg(registry)
.output()?;

if !output.status.success() {
let err_out = String::from_utf8_lossy(&output.stderr);
bail!("Failed to login for podman: {err_out}");
if let Some(Credentials {
registry,
username,
password,
}) = credentials::get()
{
trace!("podman login -u {username} -p [MASKED] {registry}");
let output = Command::new("podman")
.arg("login")
.arg("-u")
.arg(username)
.arg("-p")
.arg(password)
.arg(registry)
.output()?;

if !output.status.success() {
let err_out = String::from_utf8_lossy(&output.stderr);
bail!("Failed to login for podman: {err_out}");
}
}
Ok(())
}
Expand Down
5 changes: 1 addition & 4 deletions utils/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@ which = "6"

anyhow.workspace = true
chrono.workspace = true
clap = { workspace = true, features = ["derive"] }
colored.workspace = true
env_logger.workspace = true
format_serde_error.workspace = true
Expand All @@ -25,10 +26,6 @@ serde.workspace = true
serde_yaml.workspace = true
serde_json.workspace = true

[dependencies.clap]
workspace = true
features = ["derive"]

[build-dependencies]
syntect = "5.2.0"

Expand Down
4 changes: 4 additions & 0 deletions utils/src/constants.rs
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,10 @@ pub const IMAGE_VERSION_LABEL: &str = "org.opencontainers.image.version";

// BlueBuild vars
pub const BB_BUILDKIT_CACHE_GHA: &str = "BB_BUILDKIT_CACHE_GHA";
pub const BB_PASSWORD: &str = "BB_PASSWORD";
pub const BB_REGISTRY: &str = "BB_REGISTRY";
pub const BB_REGISTRY_NAMESPACE: &str = "BB_REGISTRY_NAMESPACE";
pub const BB_USERNAME: &str = "BB_USERNAME";

// Docker vars
pub const DOCKER_HOST: &str = "DOCKER_HOST";
Expand Down

0 comments on commit 9dd1ec9

Please sign in to comment.