$siteUrl = "https://linyus.sharepoint.com/sites/Develop"  # 替换为你的SPO站点地址

if (-not (Get-Module -ListAvailable -Name PnP.PowerShell)) {
    Write-Host "正在安装最新版PnP.PowerShell..." -ForegroundColor Cyan
    Set-PSRepository -Name PSGallery -InstallationPolicy Trusted -ErrorAction SilentlyContinue
    Install-Module -Name PnP.PowerShell -Force -AllowClobber -Scope CurrentUser -ErrorAction Stop
}

Import-Module PnP.PowerShell -DisableNameChecking -Force -ErrorAction Stop
Write-Host "✅ PnP.PowerShell模块加载成功`n" -ForegroundColor Green

$isConnected = $false

try {
    Write-Host "🔗 正在连接站点: $siteUrl" -ForegroundColor Cyan
    Connect-PnPOnline -Url $siteUrl -UseWebLogin -ErrorAction Stop
    $isConnected = $true
    Write-Host "✅ 站点连接成功`n" -ForegroundColor Green

    Write-Host "📊 开始获取站点信息...`n" -ForegroundColor Cyan

    $web = Get-PnPWeb -ErrorAction Stop
    $site = Get-PnPSite -ErrorAction Stop

    $allLists = Get-PnPList -ErrorAction Stop
    $documentLibraries = $allLists | Where-Object { $_.BaseTemplate -eq 101 }  # 101=文档库
    $customLists = $allLists | Where-Object { $_.BaseTemplate -eq 100 }      # 100=自定义列表

    Write-Host "========== SharePoint Online 站点核心信息 ==========" -ForegroundColor Green
    [PSCustomObject]@{
        "站点标题"         = $web.Title
        "站点URL"          = $web.Url
        "列表/库总数"      = $allLists.Count
        "文档库数量"       = $documentLibraries.Count
        "自定义列表数量"   = $customLists.Count
    } | Format-List

    Write-Host "`n========== 站点内前5个列表/库（名称+类型） ==========" -ForegroundColor Green
    $allLists | Select-Object Title, BaseTemplate -First 5 | Format-Table -AutoSize

}
catch {
    Write-Host "`n❌ 执行出错: $($_.Exception.Message)" -ForegroundColor Red
    exit 1
}
finally {
    # 修复连接检测逻辑：仅在确认连接成功时断开
    if ($isConnected) {
        try {
            Disconnect-PnPOnline -ErrorAction SilentlyContinue
            Write-Host "`n🔌 已断开SPO站点连接" -ForegroundColor Cyan
        }
        catch {
            Write-Host "`n⚠️ 断开连接时出现小问题，可忽略: $($_.Exception.Message)" -ForegroundColor Yellow
        }
    }
}