🎉 Auto Upgrade Camera Python File

Author: Yuzz1e

api.go

@@ -3,8 +3,10 @@ package main
 import (
 	"encoding/json"
 	"fmt"
+	"io"
 	"log"
 	"net"
+	"net/http"
 	"os"
 	"os/exec"
 	"strconv"
@@ -15,10 +17,12 @@ import (
 var cmd *exec.Cmd
 
 func RunApi(chapi chan bool, MyID uint32) {
-	// python3 main.py を実行する
-	cmd = exec.Command("python3", "main.py")
-	//コマンドを実行
-	cmd.Start()
+	// GitHubから最新のmain.pyを取得してPythonプロセスを開始
+	err := restartPythonProcess()
+	if err != nil {
+		log.Printf("Pythonプロセス開始エラー（プログラムは継続します）: %v", err)
+		// エラーがあってもAPIサーバーは継続
+	}
 	//ポートを開く
 	listener, err := net.Listen("tcp", PORT)
 	if err != nil {
@@ -144,6 +148,20 @@ func HandleRequest(conn net.Conn) {
 
 	}
 
+	if strings.Split(requests[1], "/")[1] == "updatepython" {
+		// OK と表示
+		fmt.Fprintf(conn, "HTTP/1.1 200 OK\r\n")
+		fmt.Fprintf(conn, "Content-Type: text/plain; charset=utf-8\r\n\r\n")
+		fmt.Fprintf(conn, "UPDATE PYTHON OK\r\n")
+
+		// Pythonプロセスを再起動（GitHubからの更新チェック付き）
+		err := restartPythonProcess()
+		if err != nil {
+			log.Printf("Pythonプロセス再起動エラー: %v", err)
+		}
+		return
+	}
+
 	if strings.Split(requests[1], "/")[1] == "changeadjustment" {
 		// /changeadjustment/1,120,100/15,255,255/150/0.2これを受け取る
 		// /120,100,15をとる
@@ -196,12 +214,10 @@ func HandleRequest(conn net.Conn) {
 			fmt.Fprintf(conn, "500 Internal Server Error\r\n")
 		}
 
-		cmd.Process.Kill()
-		cmd = exec.Command("python3", "main.py")
-
-		err = cmd.Start()
+		// Pythonプロセスを再起動（GitHubからの更新チェック付き）
+		err = restartPythonProcess()
 		if err != nil {
-			log.Println(err)
+			log.Printf("Pythonプロセス再起動エラー: %v", err)
 		}
 
 		return
@@ -234,3 +250,83 @@ func HandleRequest(conn net.Conn) {
 	fmt.Fprint(conn, response)
 
 }
+
+// GitHubからmain.pyを取得して更新が必要かチェックし、必要に応じて更新する
+func updateMainPyFromGitHub() error {
+	const githubURL = "https://raw.githubusercontent.com/Rione/ssl-RACOON-Pi2/refs/heads/master/main.py"
+
+	// タイムアウト付きHTTPクライアントを作成
+	client := &http.Client{
+		Timeout: 5 * time.Second,
+	}
+
+	// GitHubからファイルを取得
+	resp, err := client.Get(githubURL)
+	if err != nil {
+		return fmt.Errorf("gitHubからファイルを取得できませんでした: %v", err)
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("httpエラー: %d", resp.StatusCode)
+	}
+
+	// GitHubのファイル内容を読み取り
+	githubContent, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return fmt.Errorf("レスポンス読み取りエラー: %v", err)
+	}
+
+	// 現在のローカルファイルを読み取り
+	localContent, err := os.ReadFile("main.py")
+	if err != nil {
+		// ファイルが存在しない場合は新規作成
+		log.Println("ローカルのmain.pyが見つかりません。新規作成します。")
+	} else {
+		// ファイル内容を比較
+		if string(localContent) == string(githubContent) {
+			log.Println("main.pyは最新です。更新の必要はありません。")
+			return nil
+		}
+	}
+
+	// ファイルを更新
+	err = os.WriteFile("main.py", githubContent, 0644)
+	if err != nil {
+		return fmt.Errorf("ファイル書き込みエラー: %v", err)
+	}
+
+	log.Println("main.pyが正常に更新されました。")
+	return nil
+}
+
+// Pythonプロセスを停止して再起動する
+func restartPythonProcess() error {
+	// 既存のプロセスを停止
+	if cmd != nil && cmd.Process != nil {
+		log.Println("既存のPythonプロセスを停止します。")
+		err := cmd.Process.Kill()
+		if err != nil {
+			log.Printf("プロセス停止エラー: %v", err)
+		}
+		cmd.Wait() // プロセスの終了を待つ
+	}
+
+	// GitHubから最新のmain.pyを取得・更新を試行
+	err := updateMainPyFromGitHub()
+	if err != nil {
+		log.Printf("main.py更新エラー（ローカルファイルで継続）: %v", err)
+		// エラーがあってもローカルのファイルで実行を継続
+	}
+
+	// 新しいプロセスを開始
+	log.Println("Pythonプロセスを開始します。")
+	cmd = exec.Command("python3", "main.py")
+	err = cmd.Start()
+	if err != nil {
+		return fmt.Errorf("pythonプロセス開始エラー: %v", err)
+	}
+
+	log.Println("Pythonプロセスが正常に開始されました。")
+	return nil
+}