Issue #15 is finished

1 files changed, 15 insertions, 0 deletions

diff --git a/src/libexpr/primops/flakeref.cc b/src/libexpr/primops/flakeref.cc
index a2700f102..8e7c1f8df 100644
--- a/src/libexpr/primops/flakeref.cc
+++ b/src/libexpr/primops/flakeref.cc
@@ -152,4 +152,19 @@ bool FlakeRef::isImmutable() const
     else abort();
 }
 
+FlakeRef FlakeRef::baseRef() const // Removes the ref and rev from a FlakeRef.
+{
+    FlakeRef result(*this);
+    if (auto refData = std::get_if<FlakeRef::IsGitHub>(&result.data)) {
+        refData->ref = std::nullopt;
+        refData->rev = std::nullopt;
+    } else if (auto refData = std::get_if<FlakeRef::IsGit>(&result.data)) {
+        refData->ref = std::nullopt;
+        refData->rev = std::nullopt;
+    } else if (auto refData = std::get_if<FlakeRef::IsGit>(&result.data)) {
+        refData->ref = std::nullopt;
+        refData->rev = std::nullopt;
+    }
+    return result;
+}
 }