Update entrypoint.sh

entrypoint.sh

@@ -1,73 +1,81 @@
 #!/bin/bash
-
-# エラーが出たら止まるように設定
 set -e
 
-echo "🚀 Starting custom VSCode environment..."
-
-# ---------------------------------------------------------
-# 1. 永続ストレージ (/data) のセットアップ
-# ---------------------------------------------------------
-# /data フォルダが存在するか確認（Persistent Storageが有効なら存在する）
+# --- 永続ストレージ / SQLite の設定 (前と同じなので省略せず書くね) ---
 if [ -d "/data" ]; then
-    echo "💾 Persistent storage detected provided at /data"
-    
-    # ユーザー設定や拡張機能を保存するフォルダを /data 内に作成
-    mkdir -p /data/User
-    mkdir -p /data/extensions
-    mkdir -p /data/workspace
-
-    # 本来の保存場所から /data へリンクを貼る（これでデータが永続化される！）
-    # .local/share/code-server は拡張機能など
-    mkdir -p /home/user/.local/share
-    rm -rf /home/user/.local/share/code-server
-    ln -s /data/extensions /home/user/.local/share/code-server
-
-    # .config/code-server は設定ファイルなど
-    mkdir -p /home/user/.config
-    rm -rf /home/user/.config/code-server
-    ln -s /data/User /home/user/.config/code-server
-    
-    # ワークスペース（作業場）を /data/workspace に設定
+    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."
     WORKING_DIR="/home/user"
 fi
 
-# ---------------------------------------------------------
-# 2. SQLite データベースの準備
-# ---------------------------------------------------------
-# ユーザーごとのデータを保存するためのSQLite DBをセットアップ
+# SQLite初期化
 DB_PATH="$WORKING_DIR/user_data.sqlite"
-
 if [ ! -f "$DB_PATH" ]; then
-    echo "📦 Initializing SQLite database at $DB_PATH..."
-    # 新規作成時にテーブルを作っておく（例としてusersテーブルを作成）
-    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');"
-    echo "✅ SQLite database created."
-else
-    echo "📂 SQLite database found. Ready to use."
+    sqlite3 "$DB_PATH" "CREATE TABLE IF NOT EXISTS system_logs (id INTEGER PRIMARY KEY, message TEXT);"
 fi
 
-# ---------------------------------------------------------
-# 3. code-server (VSCode) の起動
-# ---------------------------------------------------------
-echo "💻 Launching VSCode Server on port 7860..."
+# --- 案内ページ vs VSCode の切り替えロジック ---
 
-# PASSWORD環境変数が設定されていない場合のデフォルト（必ずSpaceのSettingsで設定してね！）
-if [ -z "$PASSWORD" ]; then
-    export PASSWORD="changeme123"
-    echo "⚠️ WARNING: No password set. Using default: 'changeme123'"
-fi
+# Hugging Faceは常にポート7860を監視している
+# code-serverをそのまま起動すると即ログイン画面になるので、
+# フロントに簡単な「案内用」を挟む仕組みを作る。
+
+echo "🔧 Starting Gateway Service..."
+
+# PASSWORD設定の確認
+export PASSWORD=${PASSWORD:-"default_password_123"}
+
+# VSCode本体をバックグラウンドで（ポート8080で）起動しておく
+code-server --bind-addr 127.0.0.1:8080 --auth password --disable-telemetry "$WORKING_DIR" &
+
+# VSCodeが立ち上がるまで少し待つ
+sleep 3
+
+# ここで「リバースプロキシ」という技を使うよ
+# Pythonを使って、アクセスしてきた人に「案内」か「VSCode」を振り分ける
+cat << EOF > proxy.py
+import http.server
+import http.client
+import socketserver
+import urllib.parse
+
+class ProxyHandler(http.server.SimpleHTTPRequestHandler):
+    def do_GET(self):
+        query = urllib.parse.urlparse(self.path).query
+        # URLに app=vscode が入っていれば、後ろで動いてるVSCode(8080)に繋ぐ
+        if 'app=vscode' in query or self.path.startswith('/static') or self.path.startswith('/vscode'):
+            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リクエストもVSCodeに転送
+        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())
+
+print("Proxy running on port 7860...")
+socketserver.TCPServer(("0.0.0.0", 7860), ProxyHandler).serve_forever()
+EOF
 
-# VSCodeを起動
-# --bind-addr: 外部からアクセスできるようにする
-# --auth: パスワード認証を有効にする
-# --user-data-dir: 設定ファイルの場所を指定（リンク済みだが念のため）
-exec code-server \
-    --bind-addr 0.0.0.0:7860 \
-    --auth password \
-    --disable-telemetry \
-    "$WORKING_DIR"
+python3 proxy.py