Update entrypoint.sh

entrypoint.sh

@@ -8,20 +8,10 @@ echo "🚀 Starting custom VSCode environment..."
 # ---------------------------------------------------------
 if [ -d "/data" ]; then
     echo "💾 Persistent storage detected at /data"
-    
-    mkdir -p /data/User
-    mkdir -p /data/extensions
-    mkdir -p /data/workspace
-
-    # 設定ファイルの保存先を /data にリンク
-    mkdir -p /home/user/.local/share
-    rm -rf /home/user/.local/share/code-server
-    ln -s /data/extensions /home/user/.local/share/code-server
-
-    mkdir -p /home/user/.config
-    rm -rf /home/user/.config/code-server
-    ln -s /data/User /home/user/.config/code-server
-    
+    mkdir -p /data/User /data/extensions /data/workspace
+    mkdir -p /home/user/.local/share /home/user/.config
+    [ ! -L /home/user/.local/share/code-server ] && ln -s /data/extensions /home/user/.local/share/code-server
+    [ ! -L /home/user/.config/code-server ] && ln -s /data/User /home/user/.config/code-server
     WORKING_DIR="/data/workspace"
 else
     echo "⚠️ No persistent storage found. Using ephemeral storage."
@@ -33,74 +23,70 @@ fi
 # ---------------------------------------------------------
 DB_PATH="$WORKING_DIR/user_data.sqlite"
 if [ ! -f "$DB_PATH" ]; then
-    echo "📦 Initializing SQLite database..."
     sqlite3 "$DB_PATH" "CREATE TABLE IF NOT EXISTS system_logs (id INTEGER PRIMARY KEY, message TEXT, created_at DATETIME DEFAULT CURRENT_TIMESTAMP);"
-    sqlite3 "$DB_PATH" "INSERT INTO system_logs (message) VALUES ('System initialized');"
 fi
 
 # ---------------------------------------------------------
-# 3. code-server (VSCode) をバックグラウンドで起動
+# 3. 案内ページ 兼 リバースプロキシ (proxy.py) の作成
 # ---------------------------------------------------------
-# ポートを 8080 にして、Proxy(7860) とぶつからないようにする
-echo "💻 Launching VSCode Server on internal port 8080..."
-
-export PASSWORD=${PASSWORD:-"changeme123"}
-
-code-server \
-    --bind-addr 127.0.0.1:8080 \
-    --auth password \
-    --disable-telemetry \
-    "$WORKING_DIR" &
-
-# VSCodeが立ち上がるまで少し待つ
-sleep 5
-
-# ---------------------------------------------------------
-# 4. 案内ページ 兼 リバースプロキシ (proxy.py) の作成
-# ---------------------------------------------------------
-# 7860番ポートで待ち受け、アクセス内容によって案内かVSCodeかを振り分ける
+# 7860番を先に確保するために、スクリプトを先に書き出すよ
 cat << EOF > proxy.py
 import http.server
 import http.client
 import socketserver
 import urllib.parse
+import sys
 
 class ProxyHandler(http.server.SimpleHTTPRequestHandler):
     def do_GET(self):
         query = urllib.parse.urlparse(self.path).query
-        # URLに app=vscode が含まれているか、VSCodeが必要とするリソースへのアクセスなら転送
+        # パラメータがあるか、VSCodeが必要とするリソースなら8080へ
         if 'app=vscode' in query or any(self.path.startswith(p) for p in ['/static', '/vscode', '/webview', '/healthz']):
             self.proxy_request("127.0.0.1", 8080)
         else:
-            # それ以外は案内ページを表示
             self.path = '/landing.html'
             return http.server.SimpleHTTPRequestHandler.do_GET(self)
 
     def do_POST(self):
-        # ログイン時のパスワード送信などはPOSTなので常に転送
         self.proxy_request("127.0.0.1", 8080)
 
     def proxy_request(self, host, port):
-        content_length = int(self.headers.get('Content-Length', 0))
-        body = self.rfile.read(content_length) if content_length > 0 else None
-        
-        conn = http.client.HTTPConnection(host, port)
-        headers = {k: v for k, v in self.headers.items()}
-        conn.request(self.command, self.path, body, headers)
-        
-        res = conn.getresponse()
-        self.send_response(res.status)
-        for k, v in res.getheaders():
-            self.send_header(k, v)
-        self.end_headers()
-        self.wfile.write(res.read())
+        try:
+            content_length = int(self.headers.get('Content-Length', 0))
+            body = self.rfile.read(content_length) if content_length > 0 else None
+            conn = http.client.HTTPConnection(host, port, timeout=10)
+            headers = {k: v for k, v in self.headers.items()}
+            conn.request(self.command, self.path, body, headers)
+            res = conn.getresponse()
+            self.send_response(res.status)
+            for k, v in res.getheaders():
+                self.send_header(k, v)
+            self.end_headers()
+            self.wfile.write(res.read())
+        except Exception as e:
+            self.send_error(502, f"VSCode is still starting... Please refresh in a moment. ({e})")
 
-print("✅ Gateway Proxy running on port 7860")
-socketserver.TCPServer(("0.0.0.0", 7860), ProxyHandler).serve_forever()
+print("✅ Gateway Proxy starting on port 7860...")
+# 7860番をここでガッチリ掴む！
+server = socketserver.TCPServer(("0.0.0.0", 7860), ProxyHandler)
+server.serve_forever()
 EOF
 
 # ---------------------------------------------------------
-# 5. プロキシサーバーの起動
+# 4. 実行開始
 # ---------------------------------------------------------
-echo "🌐 Opening the gateway..."
+
+# まずは VSCode を 8080 番で、完全に「隠れた状態」で起動する
+echo "💻 Launching VSCode Server in background (Port 8080)..."
+export PASSWORD=${PASSWORD:-"changeme123"}
+
+# 重要な変更: --port 8080 を明示的に指定し、bind-addr も 127.0.0.1 に固定
+code-server \
+    --bind-addr 127.0.0.1:8080 \
+    --auth password \
+    --disable-telemetry \
+    "$WORKING_DIR" &
+
+# VSCode が起動するのを待たずに、すぐにプロキシを起動して 7860 番を確保する！
+echo "🌐 Opening the gateway on 7860..."
 python3 proxy.py