Implement atexit.rmtree

kovidgoyal · Jan 5, 2025 · 25cb735 · 25cb735
1 parent 0d5bcff
commit 25cb735
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/kitty_tests/atexit.py b/kitty_tests/atexit.py
@@ -44,6 +44,14 @@ def read():
                     p.stdin.flush()
                 select.select(readers, [], [], 10)
                 self.ae(read(), str(i+1))
+            sdir = os.path.join(self.tdir, 'd')
+            os.mkdir(sdir)
+            p.stdin.write(f'rmtree {sdir}\n'.encode())
+            p.stdin.flush()
+            open(os.path.join(sdir, 'f'), 'w').close()
+            select.select(readers, [], [], 10)
+            self.ae(read(), str(i+2))
+
             self.assertTrue(os.listdir(self.tdir))
 
             # Ensure child is ignoring signals

diff --git a/tools/cmd/atexit/main.go b/tools/cmd/atexit/main.go
@@ -44,6 +44,11 @@ func main() (rc int, err error) {
 					fmt.Fprintln(os.Stderr, "Failed to remove:", rest, "with error:", err)
 					rc = 1
 				}
+			case "rmtree":
+				if err := os.RemoveAll(rest); err != nil {
+					fmt.Fprintln(os.Stderr, "Failed to remove:", rest, "with error:", err)
+					rc = 1
+				}
 			}
 		}
 	}