From 47d7a1cc045b3869af5313b9aa788ed86af7ac67 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=AD=9F=E5=AE=81?= <mengning997@163.com>
Date: Sat, 21 Dec 2024 11:32:45 +0000
Subject: [PATCH] =?UTF-8?q?[Chore]=20public/assets/install.sh:=20=E6=8C=87?=
 =?UTF-8?q?=E5=AE=9A=E7=89=88=E6=9C=AC=E9=95=9C=E5=83=8F=E6=8B=89=E5=8F=96?=
 =?UTF-8?q?,=E7=94=A8=E6=88=B7=E6=95=B0=E6=8D=AE=E7=9B=AE=E5=BD=95?=
 =?UTF-8?q?=E6=9D=83=E9=99=90=E9=97=AE=E9=A2=98=20*=20=E6=94=AF=E6=8C=81do?=
 =?UTF-8?q?cker.io=E9=95=9C=E5=83=8F=E6=8B=89=E5=8F=96=E5=92=8C=E6=8C=87?=
 =?UTF-8?q?=E5=AE=9A=E9=95=9C=E5=83=8F=E7=89=88=E6=9C=AC;fixed=20bug:=20?=
 =?UTF-8?q?=E7=94=A8=E6=88=B7=E6=95=B0=E6=8D=AE=E7=9B=AE=E5=BD=95=E6=9D=83?=
 =?UTF-8?q?=E9=99=90=E9=97=AE=E9=A2=98=20*=20fixed=20bug:=20=E7=94=A8?=
 =?UTF-8?q?=E6=88=B7=E6=95=B0=E6=8D=AE=E7=9B=AE=E5=BD=95=E6=9D=83=E9=99=90?=
 =?UTF-8?q?=E9=97=AE=E9=A2=98=20*=20=E6=94=AF=E6=8C=81docker.io=E9=95=9C?=
 =?UTF-8?q?=E5=83=8F=E6=8B=89=E5=8F=96=E5=92=8C=E6=8C=87=E5=AE=9A=E9=95=9C?=
 =?UTF-8?q?=E5=83=8F=E7=89=88=E6=9C=AC=20*=20add=20install.sh?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 public/assets/install.sh | 26 +++++++++++++++++++-------
 1 file changed, 19 insertions(+), 7 deletions(-)

diff --git a/public/assets/install.sh b/public/assets/install.sh
index 213d87f103..266f8634e9 100755
--- a/public/assets/install.sh
+++ b/public/assets/install.sh
@@ -3,7 +3,8 @@
 
 # 默认值
 NAME=DevStar-Studio
-IMAGE_NAME=devstar.cn/devstar/devstar-studio
+IMAGE_REGISTRY_USER=mengning997
+IMAGE_NAME=devstar-studio
 VERSION=latest    # DevStar Studio的默认版本为最新版本
 PORT=8080         # 设置端口默认值为 8080
 DATA_DIR=~/devstar_data
@@ -77,7 +78,12 @@ function install_dependencies {
 # Function to install
 function install {
   install_dependencies
-  sudo docker pull $IMAGE_NAME:$VERSION
+  if sudo docker pull mengning997/$IMAGE_NAME:$VERSION; then
+      success "Successfully pulled mengning997/$IMAGE_NAME:$VERSION"  
+  else 
+    sudo docker pull devstar.cn/devstar/$IMAGE_NAME:$VERSION
+    IMAGE_REGISTRY_USER=devstar.cn/devstar
+  fi
 }
 
 # Function to start
@@ -85,9 +91,10 @@ function start {
   install
   # 创建devstar_data目录用于持久化存储DevStar相关的配置和用户数据
   mkdir -p $DATA_DIR
+  sudo chown 1000:1000 $DATA_DIR
   # 启动devstar-studio容器
   stop
-  sudo docker run --restart=always --name $NAME -d  -p $PORT:3000 -v /var/run/docker.sock:/var/run/docker.sock -v ~/devstar_data:/var/lib/gitea -v ~/devstar_data:/etc/gitea $IMAGE_NAME:$VERSION
+  sudo docker run --restart=always --name $NAME -d  -p $PORT:3000 -v /var/run/docker.sock:/var/run/docker.sock -v ~/devstar_data:/var/lib/gitea -v ~/devstar_data:/etc/gitea $IMAGE_REGISTRY_USER/$IMAGE_NAME:$VERSION
   # 打开 `http://localhost:8080` 完成安装。
   success "-------------------------------------------------------"
   success "DevStar started in http://localhost:$PORT successfully!"
@@ -121,7 +128,8 @@ function usage {
   success "DevStar usage help:"
   success "  help, -h, --help,     Help information"  
   success "  start                 Start DevStar Studio"  
-  success "      --port <arg>      Specify the port number (default port is 8080)"  
+  success "      --port=<arg>      Specify the port number (default port is 8080)" 
+  success "      --version=<arg>   Specify the DevStar Studio Image Version (default verson is latest)" 
   success "  stop                  Stop the running DevStar Studio"  
   success "  logs                  View the logs of the devstar-studio container"  
   failure "  clean                 Clean up the running DevStar Studio, including deleting user data. Please use with caution."  
@@ -135,7 +143,7 @@ case "$1" in
     usage
     ;;
   start)
-    ARGS=$(getopt --long port:: -- "$@")  
+    ARGS=$(getopt --long port::,version:: -- "$@")  
     if [ $? -ne 0 ]; then  
         failure "ARGS ERROR!"  
         exit 1  
@@ -147,6 +155,10 @@ case "$1" in
             --port)  
                 PORT="$2"  
                 echo "The Port is: $PORT"  
+                shift 2 ;; 
+            --version)  
+                VERSION="$2"  
+                echo "The DevStar Studio Image Version is: $VERSION"  
                 shift 2 ;;  
             --)  
                 shift  
@@ -174,9 +186,9 @@ case "$1" in
     if [ "$script_name" != "devstar" ]; then  
         sudo mv ./install.sh /usr/bin/devstar
         rm -rf install.sh
-        success "---------------------------------"
+        success "--------------------------------------"
         success "DevStar Studio installed successfully!"
-        success "---------------------------------"
+        success "--------------------------------------"
         success "Copyright 2024 Mengning Software All rights reserved." 
         devstar help
     fi