Nicotine++

# PowerShell Artifact Collection Script
# Target: Nicotine++
# Run as Administrator

#Requires -RunAsAdministrator

$ErrorActionPreference = "Continue"
$SourceRoot = "C:"
$DestBase   = "D:\Evidence"
$Summary = @{ Copied = 0; Missed = 0; Errors = 0 }

function Collect-Artifact {
    param(
        [Parameter(Mandatory)][string]$SourceDir,
        [Parameter(Mandatory)][string]$FolderName,
        [string]$FileMask = "*"
    )
    # Expand wildcards in any path segment (e.g. 'Program Files*',
    # 'ScreenConnect Client*'). robocopy itself does not glob the source.
    $sources = @(Get-Item -Path $SourceDir -ErrorAction SilentlyContinue |
        Where-Object { $_.PSIsContainer })
    if ($sources.Count -eq 0) {
        $Summary.Missed++
        return
    }
    $FullDest = Join-Path -Path $DestBase -ChildPath $FolderName
    $null = New-Item -ItemType Directory -Force -Path $FullDest -ErrorAction SilentlyContinue
    foreach ($src in $sources) {
        robocopy $src.FullName "$FullDest" "$FileMask" /E /COPY:DAT /R:0 /W:0 /NP /NFL /NDL /NJH /NJS 2>$null | Out-Null
        if ($LASTEXITCODE -le 7) { $Summary.Copied++ } else { $Summary.Errors++ }
    }
}

# Iterate every user profile under the source drive
Get-ChildItem "$SourceRoot\Users" -Directory -ErrorAction SilentlyContinue |
    Where-Object { $_.Name -notin @('All Users', 'Default', 'Default User', 'Public') } |
    ForEach-Object {
        $UserName = $_.Name
        # Nicotine++ Logs
        $UserPath = "$($_.FullName)\AppData\Roaming\nicotine\logs"
        Collect-Artifact -SourceDir $UserPath -FolderName "Nicotine_Logs_$UserName"
        # Nicotine++ Incomplete Downloads
        $UserPath = "$($_.FullName)\AppData\Roaming\nicotine\incomplete"
        Collect-Artifact -SourceDir $UserPath -FolderName "Nicotine_Incomplete_Downloads_$UserName"
        # Nicotine++ Buddyfiles.db
        $UserPath = "$($_.FullName)\AppData\Roaming\nicotine"
        Collect-Artifact -SourceDir $UserPath -FileMask "buddyfiles.db" -FolderName "Nicotine_Buddyfiles_db_$UserName"
        # Nicotine++ Buddystreams.db
        $UserPath = "$($_.FullName)\AppData\Roaming\nicotine"
        Collect-Artifact -SourceDir $UserPath -FileMask "buddystreams.db" -FolderName "Nicotine_Buddystreams_db_$UserName"
        # Nicotine++ Buddymtimes.db
        $UserPath = "$($_.FullName)\AppData\Roaming\nicotine"
        Collect-Artifact -SourceDir $UserPath -FileMask "buddymtimes.db" -FolderName "Nicotine_Buddymtimes_db_$UserName"
        # Nicotine++ Buddyfileindex.db
        $UserPath = "$($_.FullName)\AppData\Roaming\nicotine"
        Collect-Artifact -SourceDir $UserPath -FileMask "buddyfileindex.db" -FolderName "Nicotine_Buddyfileindex_db_$UserName"
        # Nicotine++ Buddywordindex.db
        $UserPath = "$($_.FullName)\AppData\Roaming\nicotine"
        Collect-Artifact -SourceDir $UserPath -FileMask "buddywordindex.db" -FolderName "Nicotine_Buddywordindex_db_$UserName"
        # Nicotine++ Config Files
        $UserPath = "$($_.FullName)\AppData\Roaming\nicotine\config"
        Collect-Artifact -SourceDir $UserPath -FolderName "Nicotine_Config_Files_$UserName"
        # Nicotine++ User Shares
        $UserPath = "$($_.FullName)\AppData\Roaming\nicotine\usershares"
        Collect-Artifact -SourceDir $UserPath -FolderName "Nicotine_User_Shares_$UserName"
        # Nicotine++ Downloads.json
        $UserPath = "$($_.FullName)\AppData\Roaming\nicotine"
        Collect-Artifact -SourceDir $UserPath -FileMask "downloads.json*" -FolderName "Nicotine_Downloads_json_$UserName"
        # Nicotine++ Uploads.json
        $UserPath = "$($_.FullName)\AppData\Roaming\nicotine"
        Collect-Artifact -SourceDir $UserPath -FileMask "uploads.json*" -FolderName "Nicotine_Uploads_json_$UserName"
    }

Write-Host ("Collection complete. Copied: {0}  Missed: {1}  Errors: {2}" -f $Summary.Copied, $Summary.Missed, $Summary.Errors) -ForegroundColor Green