mirror of
https://github.com/safing/portbase
synced 2025-09-01 18:19:57 +00:00
78 lines
2.5 KiB
Go
78 lines
2.5 KiB
Go
package osdetail
|
|
|
|
import (
|
|
"fmt"
|
|
)
|
|
|
|
const powershellGetFileDescription = `Get-ItemProperty %q | Select -ExpandProperty VersionInfo | Select -ExpandProperty FileDescription`
|
|
|
|
// GetBinaryNameFromSystem queries the operating system for a human readable
|
|
// name for the given binary path.
|
|
func GetBinaryNameFromSystem(path string) (string, error) {
|
|
// Get FileProperties via Powershell call.
|
|
output, err := RunPowershellCmd(fmt.Sprintf(powershellGetFileDescription, path))
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to get file properties of %s: %s", path, err)
|
|
}
|
|
|
|
// Clean name.
|
|
binName := cleanFileDescription(string(output))
|
|
if binName != "" {
|
|
return binName, nil
|
|
}
|
|
|
|
// Generate a default name as default.
|
|
return "", ErrNotFound
|
|
}
|
|
|
|
const powershellGetIcon = `Add-Type -AssemblyName System.Drawing
|
|
$Icon = [System.Drawing.Icon]::ExtractAssociatedIcon(%q)
|
|
$MemoryStream = New-Object System.IO.MemoryStream
|
|
$Icon.save($MemoryStream)
|
|
$Bytes = $MemoryStream.ToArray()
|
|
$MemoryStream.Flush()
|
|
$MemoryStream.Dispose()
|
|
[convert]::ToBase64String($Bytes)`
|
|
|
|
// TODO: This returns a small and crappy icon.
|
|
|
|
// Saving a better icon to file works:
|
|
/*
|
|
Add-Type -AssemblyName System.Drawing
|
|
$ImgList = New-Object System.Windows.Forms.ImageList
|
|
$ImgList.ImageSize = New-Object System.Drawing.Size(256,256)
|
|
$ImgList.ColorDepth = 32
|
|
$Icon = [System.Drawing.Icon]::ExtractAssociatedIcon("C:\Program Files (x86)\Mozilla Firefox\firefox.exe")
|
|
$ImgList.Images.Add($Icon);
|
|
$BigIcon = $ImgList.Images.Item(0)
|
|
$BigIcon.Save("test.png")
|
|
*/
|
|
|
|
// But not saving to a memory stream:
|
|
/*
|
|
Add-Type -AssemblyName System.Drawing
|
|
$ImgList = New-Object System.Windows.Forms.ImageList
|
|
$ImgList.ImageSize = New-Object System.Drawing.Size(256,256)
|
|
$ImgList.ColorDepth = 32
|
|
$Icon = [System.Drawing.Icon]::ExtractAssociatedIcon("C:\Program Files (x86)\Mozilla Firefox\firefox.exe")
|
|
$ImgList.Images.Add($Icon);
|
|
$MemoryStream = New-Object System.IO.MemoryStream
|
|
$BigIcon = $ImgList.Images.Item(0)
|
|
$BigIcon.Save($MemoryStream)
|
|
$Bytes = $MemoryStream.ToArray()
|
|
$MemoryStream.Flush()
|
|
$MemoryStream.Dispose()
|
|
[convert]::ToBase64String($Bytes)
|
|
*/
|
|
|
|
// GetBinaryIconFromSystem queries the operating system for the associated icon
|
|
// for a given binary path and returns it as a data-URL.
|
|
func GetBinaryIconFromSystem(path string) (string, error) {
|
|
// Get Associated File Icon via Powershell call.
|
|
output, err := RunPowershellCmd(fmt.Sprintf(powershellGetIcon, path))
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to get file properties of %s: %s", path, err)
|
|
}
|
|
|
|
return "data:image/png;base64," + string(output), nil
|
|
}
|