diff options
Diffstat (limited to '')
-rw-r--r-- | ui/systemimageprovider.go | 78 |
1 files changed, 78 insertions, 0 deletions
diff --git a/ui/systemimageprovider.go b/ui/systemimageprovider.go new file mode 100644 index 00000000..26764ccc --- /dev/null +++ b/ui/systemimageprovider.go @@ -0,0 +1,78 @@ +/* SPDX-License-Identifier: MIT + * + * Copyright (C) 2019 WireGuard LLC. All Rights Reserved. + */ + +package ui + +import ( + "fmt" + "path" + + "github.com/lxn/walk" + "github.com/lxn/win" + "golang.org/x/sys/windows" +) + +var ( + systemIconAddTunnel, + systemIconAddTunnelFromScratch, + systemIconAddTunnelFromFile, + systemIconDeleteTunnel, + systemIconExportTunnels, + systemIconSaveTunnelsToZip, + _ *walk.Icon +) + +func loadSystemIcon(dll string, index uint) (*walk.Icon, error) { + system32, err := windows.GetSystemDirectory() + if err != nil { + return nil, err + } + hicon := win.ExtractIcon(win.GetModuleHandle(nil), windows.StringToUTF16Ptr(path.Join(system32, dll+".dll")), int32(index)) + if hicon <= 1 { + return nil, fmt.Errorf("Unable to find icon %d of %s", index, dll) + } + return walk.NewIconFromHICON(hicon) +} + +func setIconOnAction(wb *walk.WindowBase, action *walk.Action, icon *walk.Icon) error { + //TODO: this is an unholy hack. Fix walk! + bitmap, err := walk.NewBitmapFromIcon(icon, walk.Size{32, 32}) + if err != nil { + return err + } + wb.AddDisposable(bitmap) + return action.SetImage(bitmap) +} + +func loadSystemIcons() (err error) { + //TODO: this should probably be in an object that is disposable instead of using globals like this + + systemIconAddTunnel, err = loadSystemIcon("shell32", 149) + if err != nil { + return + } + systemIconAddTunnelFromScratch, err = loadSystemIcon("imageres", 2) + if err != nil { + return + } + systemIconAddTunnelFromFile, err = loadSystemIcon("imageres", 3) + if err != nil { + return + } + systemIconDeleteTunnel, err = loadSystemIcon("shell32", 131) + if err != nil { + return + } + systemIconExportTunnels, err = loadSystemIcon("shell32", 45) + if err != nil { + return + } + systemIconSaveTunnelsToZip, err = loadSystemIcon("imageres", 165) + if err != nil { + return + } + + return +} |