From 81e397358ffdec7229ef9648eacbb678294111ec Mon Sep 17 00:00:00 2001
From: WooKyoung Noh <wookay.noh@gmail.com>
Date: Sat, 3 Jan 2026 13:32:11 +0900
Subject: [PATCH] set PROGRAM_FILE

---
 src/DaemonMode.jl    | 2 ++
 test/program_file.jl | 1 +
 test/runtests.jl     | 9 +++++++++
 3 files changed, 12 insertions(+)
 create mode 100644 test/program_file.jl

diff --git a/src/DaemonMode.jl b/src/DaemonMode.jl
index 061b5a3..680ea72 100644
--- a/src/DaemonMode.jl
+++ b/src/DaemonMode.jl
@@ -440,6 +440,8 @@ function serverRunFile(sock, shared, print_stack, reviser)
             push!(ARGS, arg)
         end
 
+        Base.PROGRAM_FILE = fname
+
         first_time[] = true
 
         cd(dir) do
diff --git a/test/program_file.jl b/test/program_file.jl
new file mode 100644
index 0000000..2106e9f
--- /dev/null
+++ b/test/program_file.jl
@@ -0,0 +1 @@
+println("PROGRAM_FILE: ", PROGRAM_FILE)
diff --git a/test/runtests.jl b/test/runtests.jl
index 8d07bc5..7378563 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -184,3 +184,12 @@ end
     @test endswith(l[1], joinpath("test", "fileandline.jl"))
     @test l[2] == "7"
 end
+
+@testset "test PROGRAM_FILE" begin
+    port = 3012
+    task = init_server(port)
+    eval_expr = ["-e", "using DaemonMode; runargs($port)"]
+    cmd = Cmd([Base.julia_cmd()..., eval_expr..., "program_file.jl"])
+    @test "PROGRAM_FILE: program_file.jl\n" == read(cmd, String)
+    end_server(task, port)
+end