From b8a2f20840c44e16e5f5a884435cc7a9a151e07e Mon Sep 17 00:00:00 2001
From: Illia Chub <illia.chub@selfprivacy.org>
Date: Thu, 26 Aug 2021 12:11:21 +0300
Subject: [PATCH] Fixed SSH configuration write issues

---
 main.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/main.py b/main.py
index 84c6155..57299df 100755
--- a/main.py
+++ b/main.py
@@ -159,14 +159,18 @@ echo -n {0} | cryptsetup luksOpen /dev/sdb decryptedVar'''.format(request.header
 
 def enableSSH():
     readOnlyFileDescriptor = open("/etc/nixos/configuration.nix", "rt")
-    readWriteFileDescriptor = open("/etc/nixos/configuration.nix", "wt")
+    
 
-    for line in readOnlyFileDescriptor:
-        readWriteFileDescriptor.write(line.replace("services.openssh.enable = false;", "services.openssh.enable = true;"))
+    fileContent = readOnlyFileDescriptor.read()
 
-    readWriteFileDescriptor.close()
+    fileContent = fileContent.replace("enabled = false;", "enabled = true;")     
     readOnlyFileDescriptor.close()
 
+    readWriteFileDescriptor = open("/etc/nixos/configuration.nix", "wt") 
+
+    readWriteFileDescriptor.write(fileContent)
+    readWriteFileDescriptor.close()
+    
     return jsonify(
         status=0
     )