SEMS-system-guardian/main.ps1

#$UPDATE_SOURCE = "C:\SEMS-development"   #如果为http开头则是网络更新，否则则是本地路径更新
$UPDATE_SOURCE = "https://1.u1227383.nyat.app:23587/api/v1/repos/zhanpeng"
$TOCKEN = "e9aaae4b5f7756d0a2977764dbcbdfeb68bb2e13"
$PRODUCTION_FOLDER = "C:\SEMS-production"
$RAWDATA_FOLDER = "C:\tmp"


if (Test-Path -Path $PRODUCTION_FOLDER -PathType Container) {
    Write-Output "[初始化]生产环境文件夹存在：$PRODUCTION_FOLDER"
}
else {
    New-Item -Path  $PRODUCTION_FOLDER -ItemType Directory -ErrorAction SilentlyContinue | Out-Null
    Write-Output "[初始化]生产环境文件夹不存在，已新建：$PRODUCTION_FOLDER"
}
Write-Output "[初始化]程序更新源路径为：$UPDATE_SOURCE" 



function UpdateProgram {
    param (
        $name        #程序名称
    )

    #在生产部署文件夹中找以$name命名的文件夹，其中包含了版本信息
    $nowVersion = Get-ChildItem -Path $PRODUCTION_FOLDER -Directory | Where-Object { $_.Name -like "*$name*" }  #当前部署的版本，是一个文件夹对象，可能为空

    #判断是否为网络更新，网络更新在生产环境中用，本地更新便于开发时测试
    if ($UPDATE_SOURCE.Contains("http")) {

        #尝试获取最新版本信息，若获取不到则停止更新
        try {
            $response = Invoke-RestMethod -Uri "$UPDATE_SOURCE/$name/releases/latest?token=$TOCKEN"

                

            #判断最新版本号是否大于当前部署版本号
            if ($response.name -gt $nowVersion.name) {
                #新建缓存文件夹
                New-Item -Path  "$PRODUCTION_FOLDER/caches" -ItemType Directory -ErrorAction SilentlyContinue | Out-Null

                foreach ($asset in $response.assets) {
                    if ($asset.name.Contains("zip")) {
                        Write-Output "[更新]正在下载安装包$($asset.name.Replace(".zip",'')): $($asset.browser_download_url)"
                        #要把安装包保存的路径
                        $latestFile = "$PRODUCTION_FOLDER\caches\$($asset.name)"
                        Invoke-WebRequest -Uri $asset.browser_download_url -OutFile $latestFile
                        #把路径字符串转换为文件对象，便于后续处理
                        $latestFile = Get-Item -Path $latestFile
                        Write-Output  "[更新]下载完成，保存在$latestFile"
                    }
                    else {
                        Write-Output "[更新]正在下载配置文件$($asset.name): $($asset.browser_download_url)"

                        #要把配置文件保存的路径
                        $latestConfigFile = "$PRODUCTION_FOLDER\$($asset.name)"
                        Invoke-WebRequest -Uri $asset.browser_download_url -OutFile $latestConfigFile
                        Write-Output  "[更新]下载完成，保存在$latestConfigFile"
                        #把路径字符串转换为文件对象，便于后续处理
                        $latestConfigFile = Get-Item -Path $latestConfigFile
                    }
                }
                # Write-Output "[更新]正在下载$($response.assets.name.Replace(".zip",'')): $($response.assets.browser_download_url)"
                # #要把安装包保存的路径
                # $latestFile = "$PRODUCTION_FOLDER\$($response.assets.name)"
                # Invoke-WebRequest -Uri $response.assets.browser_download_url -OutFile $latestFile
                # Write-Output  "[更新]下载完成，保存在$latestFile"
            }
            else {
                Write-Output  "[更新] $name 不需要更新，部署版本为$($nowVersion.name), 最新版本为$($response.name)"
                Return
            }
                
            

        }
        catch {
            Write-Output "[更新]最新版本请求失败，跳过处理。错误信息：$_"
            Return
        }

    }
    else {

        $latestFile = Get-ChildItem -Path $UPDATE_SOURCE -File -Depth 3 | Where-Object { $_.Name -like "*$name*.zip" } | Sort-Object -Property Name | Select-Object -Last 1 

        #判断最新版本号是否大于当前部署版本号
        if ($latestFile.BaseName -gt $nowVersion.Name) {
            Write-Output "[更新]正在复制$($latestFile.FullName)到 $PRODUCTION_FOLDER\$($latestFile.Name)"
            Copy-Item -Path $latestFile.FullName -Destination $PRODUCTION_FOLDER\$($latestFile.Name) -Force
            $latestFile = Get-Item -Path $PRODUCTION_FOLDER\$($latestFile.Name)
        }
        else {
            Write-Output  "[更新] $name 不需要更新，部署版本为$($nowVersion.name), 最新版本为$($latestFile.BaseName)"
            Return
        }

    }


    #停止运行程序

    if ($name -eq "SEMS-data-collection") {
        $process = Get-Process -Name "SEMS-data-collection" -ErrorAction SilentlyContinue
        if ($process) {
            # 检查进程是否存在，若存在则杀掉
            Stop-Process -Id $process.Id -Force
            Write-Output  "[更新] $name 进程存在，已执行终止命令"
        }
        else {
            Write-Output  "[更新] $name 进程不存在"
        }
    }
    if ($name.ToLower() -eq "sems-on-device-server" -or $name.ToLower() -eq "sems-model-inference" ) {
        $container = docker inspect -f "{{.State.Running}}" $name.ToLower()

        if ($container -eq "true") {
            Write-Output "[更新] 容器 $($name.ToLower()) 正在运行, 正在执行终止命令"
            docker stop $name.ToLower()
        }
        else {
            Write-Output "[更新] 容器 $($name.ToLower()) 不在运行"
        }


    }

        
    Write-Output "[更新]${name}程序有更新：$($latestFile.BaseName.Replace("$name-", ''))" 
    #删除当前部署的旧版本
    if (-not [string]::IsNullOrEmpty($nowVersion.FullName)) {
        Remove-Item -Path $nowVersion.FullName -Recurse -Force
        Write-Output "[更新]已删除旧版本${name}程序：$($nowVersion.Name.Replace("$name-", '')) "  
    }

    #解压安装包到生产部署文件夹
    if ($name -eq "SEMS-data-collection") {
        $destinationPath = $PRODUCTION_FOLDER
    }
    else {
        $destinationPath = "$PRODUCTION_FOLDER\$($latestFile.BaseName)"
    }
    Expand-Archive -Path $latestFile.FullName -DestinationPath $destinationPath -Force

    #删除安装包
    Remove-Item -Path $latestFile.FullName -Force

    Write-Output "[更新]已更新${name}程序到：$PRODUCTION_FOLDER\$($latestFile.BaseName), 并删除安装包 $($latestFile.FullName)"  

    #判断配置是否有更新，若有更新
    if (![string]::IsNullOrEmpty($latestConfigFile)) {
        Write-Output "[更新]正在更新${name}的配置文件：$($latestConfigFile.BaseName)"

        #？？？？？？？此功能需要进一步撰写，如何实现对比两个yaml文件，分别为新文件与旧文件，若新文件中有旧文件中没有，则添加；若新旧文件中都有，则不做任何改动，若新文件中没有，旧文件中有则删除。若旧文件为空或不存在则直接用新文件覆盖。请将上述功能封装为一个函数，输入为新旧两个文件的路径

    }

}








#######################
#  更新生产环境程序    #
######################

UpdateProgram -name "SEMS-data-collection" 
UpdateProgram -name "SEMS-model-inference"  
UpdateProgram -name "SEMS-on-device-server"  


#################################################
#  检查程序是否在运行，若不在运行则重启              #
##################################################

Write-Output "[状态]准备开始检查程序运行状态"



# 检查 Docker Desktop 服务状态
$process = Get-Process -Name "Docker Desktop" -ErrorAction SilentlyContinue

if ($process) {
    Write-Output "[状态]Docker Desktop 正在运行"
}
else {

    Write-Output "[状态]Docker Desktop运行正常不在运行，正在启动，请等待10s"
    Start-Process -FilePath "C:\Program Files\Docker\Docker\Docker Desktop.exe"
    Start-Sleep -Seconds 10 # 等待 Docker Desktop 启动
    $process = Get-Process -Name "Docker Desktop" -ErrorAction SilentlyContinue
    if ($process) {
        Write-Output "[状态]Docker Desktop已成功运行"
    }
    else {
        Write-Output "[状态]Docker Desktop启动失败"
    }
}





$folders = Get-ChildItem -Path $PRODUCTION_FOLDER -Directory

$folderDataCollection = $folders | Where-Object { $_.Name -like "*SEMS-data-collection*" }

Write-Output $folderDataCollection.Name 
$folderModelInference = $folders | Where-Object { $_.Name -like "*SEMS-model-inference*" }
Write-Output $folderModelInference.Name 
$folderOnDeviceServer = $folders | Where-Object { $_.Name -like "*SEMS-on-device-server*" }
Write-Output $folderOnDeviceServer.Name 

$process = Get-Process -Name "SEMS-data-collection" -ErrorAction SilentlyContinue

$restart_flag = "false"
if ($process) {
    # 检查进程是否正在运行
    if ($process.Responding) {
        Write-Output "[状态]进程 'SEMS-data-collection' 正在运行"
    }
    else {
        $restart_flag = "true"
        Write-Output "[状态]进程  'SEMS-data-collection' 存在但未响应"
    }
}
else {
    $restart_flag = "true"
    Write-Output "[状态]进程 'SEMS-data-collection' 不存在"
}
if ($restart_flag -eq "true") {
    $exeFile = Get-ChildItem -Path "$folderDataCollection" -Recurse -File -Depth 2 -Filter "SEMS-data-collection.exe" 
    Write-Output "[状态]启动进程 $($exeFile.FullName)"
    Start-Process -FilePath $exeFile.FullName

}

$container = docker inspect -f "{{.State.Running}}" sems-on-device-server


if ($container -eq "true") {
    Write-Output "[状态]容器 'sems-model-inference' 正在运行"
}
else {
    Write-Output "[状态]容器 'sems-model-inference' 不在运行，正在启动"
    docker run --rm -d -p  "22111:22111" -v "$($folderModelInference.FullName):/app" --name sems-model-inference  sems-model-inference:latest
}


if ($container -eq "true") {
    Write-Output "[状态]容器 'sems-on-device-server' 正在运行"
}
else {
    Write-Output "[状态]容器 'sems-on-device-server' 不在运行，正在运行"
    docker run --rm -d -p   "22110:22110" -v "$($folderOnDeviceServer.FullName):/app" -v "${RAWDATA_FOLDER}:/data" --env-file "C:\SEMS-production\configs\.env" --name sems-on-device-server  --link  sems-model-inference sems-on-device-server:latest 
}