101 lines
3.6 KiB
Go
101 lines
3.6 KiB
Go
/*
|
|
Copyright 2022 Zheng Dayu
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package windowsupdate
|
|
|
|
import (
|
|
"github.com/go-ole/go-ole"
|
|
"github.com/go-ole/go-ole/oleutil"
|
|
)
|
|
|
|
// IUpdateSession represents a session in which the caller can perform operations that involve updates.
|
|
// For example, this interface represents sessions in which the caller performs a search, download, installation, or uninstallation operation.
|
|
// https://docs.microsoft.com/en-us/windows/win32/api/wuapi/nn-wuapi-iupdatesession
|
|
type IUpdateSession struct {
|
|
disp *ole.IDispatch
|
|
ClientApplicationID string
|
|
ReadOnly bool
|
|
WebProxy *IWebProxy
|
|
}
|
|
|
|
func toIUpdateSession(updateSessionDisp *ole.IDispatch) (*IUpdateSession, error) {
|
|
var err error
|
|
iUpdateSession := &IUpdateSession{
|
|
disp: updateSessionDisp,
|
|
}
|
|
|
|
if iUpdateSession.ClientApplicationID, err = toStringErr(oleutil.GetProperty(updateSessionDisp, "ClientApplicationID")); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if iUpdateSession.ReadOnly, err = toBoolErr(oleutil.GetProperty(updateSessionDisp, "ReadOnly")); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
webProxyDisp, err := toIDispatchErr(oleutil.GetProperty(updateSessionDisp, "WebProxy"))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if webProxyDisp != nil {
|
|
if iUpdateSession.WebProxy, err = toIWebProxy(webProxyDisp); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return iUpdateSession, nil
|
|
}
|
|
|
|
// NewUpdateSession creates a new IUpdateSession interface.
|
|
func NewUpdateSession() (*IUpdateSession, error) {
|
|
unknown, err := oleutil.CreateObject("Microsoft.Update.Session")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
disp, err := unknown.QueryInterface(ole.IID_IDispatch)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return toIUpdateSession(disp)
|
|
}
|
|
|
|
// CreateUpdateDownloader returns an IUpdateDownloader interface for this session.
|
|
// https://docs.microsoft.com/en-us/windows/win32/api/wuapi/nf-wuapi-iupdatesession-createupdatedownloader
|
|
func (iUpdateSession *IUpdateSession) CreateUpdateDownloader() (*IUpdateDownloader, error) {
|
|
updateDownloaderDisp, err := toIDispatchErr(oleutil.CallMethod(iUpdateSession.disp, "CreateUpdateDownloader"))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return toIUpdateDownloader(updateDownloaderDisp)
|
|
}
|
|
|
|
// CreateUpdateInstaller returns an IUpdateInstaller interface for this session.
|
|
// https://docs.microsoft.com/en-us/windows/win32/api/wuapi/nf-wuapi-iupdatesession-createupdateinstaller
|
|
func (iUpdateSession *IUpdateSession) CreateUpdateInstaller() (*IUpdateInstaller, error) {
|
|
updateInstallerDisp, err := toIDispatchErr(oleutil.CallMethod(iUpdateSession.disp, "CreateUpdateInstaller"))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return toIUpdateInstaller(updateInstallerDisp)
|
|
}
|
|
|
|
// CreateUpdateSearcher returns an IUpdateSearcher interface for this session.
|
|
// https://docs.microsoft.com/en-us/windows/win32/api/wuapi/nf-wuapi-iupdatesession-createupdatesearcher
|
|
func (iUpdateSession *IUpdateSession) CreateUpdateSearcher() (*IUpdateSearcher, error) {
|
|
updateSearcherDisp, err := toIDispatchErr(oleutil.CallMethod(iUpdateSession.disp, "CreateUpdateSearcher"))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return toIUpdateSearcher(updateSearcherDisp)
|
|
}
|