From 007c4a7f3a4de8ed1f9f1d959ef58b2a0ff1da49 Mon Sep 17 00:00:00 2001
From: regnat <rg@regnat.ovh>
Date: Fri, 5 Feb 2021 13:35:31 +0100
Subject: [PATCH] Make NIX_SHOW_STATS work with new-style commands

---
 src/libcmd/command.hh      | 2 ++
 src/libcmd/installables.cc | 6 ++++++
 2 files changed, 8 insertions(+)

diff --git a/src/libcmd/command.hh b/src/libcmd/command.hh
index 8c0b3a94a..25e15f501 100644
--- a/src/libcmd/command.hh
+++ b/src/libcmd/command.hh
@@ -48,6 +48,8 @@ struct EvalCommand : virtual StoreCommand, MixEvalArgs
     ref<EvalState> getEvalState();
 
     std::shared_ptr<EvalState> evalState;
+
+    ~EvalCommand();
 };
 
 struct MixFlakeOptions : virtual Args, EvalCommand
diff --git a/src/libcmd/installables.cc b/src/libcmd/installables.cc
index 74eafddc3..a4cd03878 100644
--- a/src/libcmd/installables.cc
+++ b/src/libcmd/installables.cc
@@ -283,6 +283,12 @@ ref<EvalState> EvalCommand::getEvalState()
     return ref<EvalState>(evalState);
 }
 
+EvalCommand::~EvalCommand()
+{
+    if (evalState)
+        evalState->printStats();
+}
+
 void completeFlakeRef(ref<Store> store, std::string_view prefix)
 {
     if (prefix == "")