From 955b031b8a47a6b82f17656843210ebe01fa5511 Mon Sep 17 00:00:00 2001
From: Wuzzy <Wuzzy2@mail.ru>
Date: Thu, 20 Jan 2022 14:30:34 +0100
Subject: [PATCH] Fix /restart not working in editor

---
 mods/lzr_levels/init.lua | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/mods/lzr_levels/init.lua b/mods/lzr_levels/init.lua
index cbf4230..34cc717 100644
--- a/mods/lzr_levels/init.lua
+++ b/mods/lzr_levels/init.lua
@@ -221,9 +221,12 @@ minetest.register_chatcommand("restart", {
 	params = "",
 	description = S("Restart current level"),
 	func = function(name, param)
-		if lzr_gamestate.get_state() == lzr_gamestate.LEVEL then
+		local state = lzr_gamestate.get_state()
+		if state == lzr_gamestate.LEVEL or state == lzr_gamestate.EDITOR then
 			lzr_levels.start_level(current_level)
 			return true
+		elseif state == lzr_gamestate.LEVEL_COMPLETE then
+			return false, S("Can’t restart level right now.")
 		else
 			return false, S("Not playing in a level!")
 		end