1 files changed, 115 insertions, 0 deletions
diff --git a/src/git/reference_kind.rs b/src/git/reference_kind.rs
new file mode 100644
index 0000000..f3db0b6
--- /dev/null
+++ b/src/git/reference_kind.rs
@@ -0,0 +1,115 @@
+/// Represents the kind of a reference
+#[derive(Debug, Copy, Clone, PartialEq, Eq)]
+#[allow(clippy::exhaustive_enums)]
+pub(crate) enum ReferenceKind {
+	/// Reference is a branch.
+	Branch,
+	/// Reference is a note.
+	Note,
+	/// Reference is a remote.
+	Remote,
+	/// Reference is a tag.
+	Tag,
+	/// Reference is another kind.
+	Other,
+}
+
+impl ReferenceKind {
+	pub(crate) fn from(reference: &git2::Reference<'_>) -> Self {
+		if reference.is_branch() {
+			Self::Branch
+		}
+		else if reference.is_note() {
+			Self::Note
+		}
+		else if reference.is_remote() {
+			Self::Remote
+		}
+		else if reference.is_tag() {
+			Self::Tag
+		}
+		else {
+			Self::Other
+		}
+	}
+}
+
+#[cfg(test)]
+mod tests {
+	use super::*;
+	use crate::git::testutil::{with_temp_repository, JAN_2021_EPOCH};
+
+	#[test]
+	fn from_git2_reference_branch() {
+		with_temp_repository(|repository| {
+			assert_eq!(
+				ReferenceKind::from(
+					&repository
+						.repository()
+						.lock()
+						.find_reference("refs/heads/main")
+						.unwrap()
+				),
+				ReferenceKind::Branch
+			);
+		});
+	}
+
+	#[test]
+	fn from_git2_reference_note() {
+		with_temp_repository(|repository| {
+			let git2_repository = repository.repository();
+			let git2_lock = git2_repository.lock();
+			let sig = git2::Signature::new("name", "name@example.com", &git2::Time::new(JAN_2021_EPOCH, 0)).unwrap();
+			let head_id = git2_lock.refname_to_id("HEAD").unwrap();
+			_ = git2_lock.note(&sig, &sig, None, head_id, "note", false).unwrap();
+			assert_eq!(
+				ReferenceKind::from(&git2_lock.find_reference("refs/notes/commits").unwrap()),
+				ReferenceKind::Note
+			);
+		});
+	}
+
+	#[test]
+	fn from_git2_reference_remote() {
+		with_temp_repository(|repository| {
+			let git2_repository = repository.repository();
+			let git2_lock = git2_repository.lock();
+			let mut remote = git2_lock.remote("origin", git2_lock.path().to_str().unwrap()).unwrap();
+			remote.fetch(&["main"], None, None).unwrap();
+			assert_eq!(
+				ReferenceKind::from(&git2_lock.find_reference("refs/remotes/origin/main").unwrap()),
+				ReferenceKind::Remote
+			);
+		});
+	}
+
+	#[test]
+	fn from_git2_reference_tag() {
+		with_temp_repository(|repository| {
+			let git2_repository = repository.repository();
+			let git2_lock = git2_repository.lock();
+			let sig = git2::Signature::new("name", "name@example.com", &git2::Time::new(JAN_2021_EPOCH, 0)).unwrap();
+			let head_id = git2_lock.revparse_single("HEAD").unwrap();
+			_ = git2_lock.tag("tag", &head_id, &sig, "note", false).unwrap();
+			assert_eq!(
+				ReferenceKind::from(&git2_lock.find_reference("refs/tags/tag").unwrap()),
+				ReferenceKind::Tag
+			);
+		});
+	}
+
+	#[test]
+	fn from_git2_reference_other() {
+		with_temp_repository(|repository| {
+			let git2_repository = repository.repository();
+			let git2_lock = git2_repository.lock();
+			let blob = git2_lock.blob(b"foo").unwrap();
+			_ = git2_lock.reference("refs/blob", blob, false, "blob").unwrap();
+			assert_eq!(
+				ReferenceKind::from(&git2_lock.find_reference("refs/blob").unwrap()),
+				ReferenceKind::Other
+			);
+		});
+	}
+}