use crate::analyzer::hadolint::parser::instruction::Instruction;
use crate::analyzer::hadolint::rules::{SimpleRule, simple_rule};
use crate::analyzer::hadolint::shell::ParsedShell;
use crate::analyzer::hadolint::types::Severity;
pub fn rule() -> SimpleRule<impl Fn(&Instruction, Option<&ParsedShell>) -> bool + Send + Sync> {
simple_rule(
"DL3028",
Severity::Warning,
"Pin versions in gem install. Instead of `gem install <gem>` use `gem install <gem>:<version>`.",
|instr, shell| {
match instr {
Instruction::Run(_) => {
if let Some(shell) = shell {
!shell.any_command(|cmd| {
if cmd.name == "gem" && cmd.has_any_arg(&["install"]) {
let gems = get_gem_packages(cmd);
gems.iter().any(|gem| !is_pinned_gem(gem))
} else {
false
}
})
} else {
true
}
}
_ => true,
}
},
)
}
fn get_gem_packages(cmd: &crate::analyzer::hadolint::shell::Command) -> Vec<&str> {
let mut gems = Vec::new();
let mut found_install = false;
for arg in &cmd.arguments {
if arg == "install" {
found_install = true;
continue;
}
if found_install && !arg.starts_with('-') {
gems.push(arg.as_str());
}
}
gems
}
fn is_pinned_gem(gem: &str) -> bool {
if gem.starts_with('-') {
return true;
}
gem.contains(':')
}
#[cfg(test)]
mod tests {
use super::*;
use crate::analyzer::hadolint::config::HadolintConfig;
use crate::analyzer::hadolint::lint::{LintResult, lint};
fn lint_dockerfile(content: &str) -> LintResult {
lint(content, &HadolintConfig::default())
}
#[test]
fn test_gem_install_unpinned() {
let result = lint_dockerfile("FROM ruby:3.2\nRUN gem install rails");
assert!(result.failures.iter().any(|f| f.code.as_str() == "DL3028"));
}
#[test]
fn test_gem_install_pinned() {
let result = lint_dockerfile("FROM ruby:3.2\nRUN gem install rails:7.0.0");
assert!(!result.failures.iter().any(|f| f.code.as_str() == "DL3028"));
}
#[test]
fn test_gem_install_multiple_unpinned() {
let result = lint_dockerfile("FROM ruby:3.2\nRUN gem install bundler rake");
assert!(result.failures.iter().any(|f| f.code.as_str() == "DL3028"));
}
#[test]
fn test_bundle_install() {
let result = lint_dockerfile("FROM ruby:3.2\nRUN bundle install");
assert!(!result.failures.iter().any(|f| f.code.as_str() == "DL3028"));
}
}