cargo-v 0.1.5

pub enum VersionLabel {
  Major,
  Minor,
  Patch,
  NumericVersion(String),
}

pub fn update_version(
  cargo_toml_content: &str,
  label: &VersionLabel,
) -> String {
  let current_version_tuple = get_version(&cargo_toml_content);
  let (major, minor, patch) = current_version_tuple;
  let new_version = match label {
    VersionLabel::Major => format!("{}.0.0", major + 1),
    VersionLabel::Minor => format!("{}.{}.0", major, minor + 1),
    VersionLabel::Patch => format!("{}.{}.{}", major, minor, patch + 1),
    VersionLabel::NumericVersion(v) => {
      parse_numeric_version(&current_version_tuple, v)
    }
  };

  cargo_toml_content.replace(
    &tuple_version_to_string(&current_version_tuple),
    &new_version,
  )
}

fn parse_numeric_version(
  current_version_tuple: &(u32, u32, u32),
  numeric_version: &str,
) -> String {
  let new_version = numeric_version.replace("v", "");
  let current_version_string = tuple_version_to_string(current_version_tuple);
  let current_version_number =
    string_version_to_number(&current_version_string);
  let new_version_number = string_version_to_number(&new_version);

  if new_version_number < current_version_number {
    panic!("You can not set a version lower than the current version");
  }

  new_version
}

fn string_version_to_number(version: &str) -> u32 {
  version.replace(".", "").parse().unwrap()
}

pub fn tuple_version_to_string(tuple_version: &(u32, u32, u32)) -> String {
  format!(
    "{}.{}.{}",
    tuple_version.0, tuple_version.1, tuple_version.2,
  )
}

pub fn get_version(cargo_toml_content: &str) -> (u32, u32, u32) {
  let version = cargo_toml_content
    .lines()
    .filter(|line| line.contains("version"))
    .map(|line| {
      line
        .replace("version", "")
        .replace('=', "")
        .replace('"', "")
        .trim()
        .to_string()
    })
    .next();

  let version =
    version.expect("Your Cargo.toml file does not have a \"version\" entry");

  let mut version_split = version.split('.');
  let major = version_split.next().unwrap().parse().unwrap();
  let minor = version_split.next().unwrap().parse().unwrap();
  let patch = version_split.next().unwrap().parse().unwrap();
  (major, minor, patch)
}

#[cfg(test)]
mod tests {
  use super::*;

  #[test]
  fn should_get_version_from_cargo_toml() {
    let cargo_toml =
      String::from("[package]\n name = \"cargo-v\"\n version = \"2.8.1\"\n");
    let expected = (2, 8, 1);

    assert_eq!(get_version(&cargo_toml), expected);
  }

  #[test]
  #[should_panic(
    expected = "Your Cargo.toml file does not have a \"version\" entry"
  )]
  fn should_fail_if_cargo_toml_does_not_have_version() {
    let cargo_toml = String::from("[package]\n name = \"cargo-v\"\n");
    get_version(&cargo_toml);
  }

  #[test]
  fn should_update_version_by_patch_label() {
    let cargo_toml =
      String::from("[package]\n name = \"cargo-v\"\n version = \"0.0.1\"\n");
    let new_version = VersionLabel::Patch;
    let expected = "[package]\n name = \"cargo-v\"\n version = \"0.0.2\"\n";

    assert_eq!(update_version(&cargo_toml, &new_version), expected);
  }

  #[test]
  fn should_update_version_by_minor_label() {
    let cargo_toml =
      String::from("[package]\n name = \"cargo-v\"\n version = \"0.0.2\"\n");
    let new_version = VersionLabel::Minor;
    let expected = "[package]\n name = \"cargo-v\"\n version = \"0.1.0\"\n";

    assert_eq!(update_version(&cargo_toml, &new_version), expected);
  }

  #[test]
  fn should_update_version_by_major_label() {
    let cargo_toml =
      String::from("[package]\n name = \"cargo-v\"\n version = \"0.1.2\"\n");
    let new_version = VersionLabel::Major;
    let expected = "[package]\n name = \"cargo-v\"\n version = \"1.0.0\"\n";

    assert_eq!(update_version(&cargo_toml, &new_version), expected);
  }

  #[test]
  fn should_update_patch_version_by_hand() {
    let cargo_toml =
      String::from("[package]\n name = \"cargo-v\"\n version = \"0.0.1\"\n");
    let new_version = VersionLabel::NumericVersion(String::from("0.0.2"));
    let expected = "[package]\n name = \"cargo-v\"\n version = \"0.0.2\"\n";

    assert_eq!(update_version(&cargo_toml, &new_version), expected);
  }

  #[test]
  fn should_update_minor_version_by_hand() {
    let cargo_toml =
      String::from("[package]\n name = \"cargo-v\"\n version = \"0.0.1\"\n");
    let new_version = VersionLabel::NumericVersion(String::from("0.1.0"));
    let expected = "[package]\n name = \"cargo-v\"\n version = \"0.1.0\"\n";

    assert_eq!(update_version(&cargo_toml, &new_version), expected);
  }

  #[test]
  fn should_update_major_version_by_hand() {
    let cargo_toml =
      String::from("[package]\n name = \"cargo-v\"\n version = \"2.1.8\"\n");
    let new_version = VersionLabel::NumericVersion(String::from("3.0.0"));
    let expected =
      String::from("[package]\n name = \"cargo-v\"\n version = \"3.0.0\"\n");

    assert_eq!(update_version(&cargo_toml, &new_version), expected);
  }

  #[test]
  fn should_accept_v_char_in_front_of_version() {
    let cargo_toml =
      String::from("[package]\n name = \"cargo-v\"\n version = \"2.1.8\"\n");
    let new_version = VersionLabel::NumericVersion(String::from("v3.0.0"));
    let expected =
      String::from("[package]\n name = \"cargo-v\"\n version = \"3.0.0\"\n");

    assert_eq!(update_version(&cargo_toml, &new_version), expected);
  }

  #[test]
  #[should_panic(
    expected = "You can not set a version lower than the current version"
  )]
  fn should_not_set_a_new_version_lower_than_current_version() {
    let cargo_toml =
      String::from("[package]\n name = \"cargo-v\"\n version = \"2.2.0\"\n");
    let new_version = VersionLabel::NumericVersion(String::from("2.1.1"));
    update_version(&cargo_toml, &new_version);
  }

  #[test]
  // TODO: Give a more friendly error message
  #[should_panic(expected = "InvalidDigit")]
  fn should_not_allow_set_a_negative_version() {
    let cargo_toml =
      String::from("[package]\n name = \"cargo-v\"\n version = \"2.2.0\"\n");
    let new_version = VersionLabel::NumericVersion(String::from("-2.2.1"));
    update_version(&cargo_toml, &new_version);
  }
}