1 files changed, 81 insertions, 0 deletions
diff --git a/vendor/go.mau.fi/whatsmeow/update.go b/vendor/go.mau.fi/whatsmeow/update.go
new file mode 100644
index 00000000..ec94aa2f
--- /dev/null
+++ b/vendor/go.mau.fi/whatsmeow/update.go
@@ -0,0 +1,81 @@
+// Copyright (c) 2022 Tulir Asokan
+//
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this
+// file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+package whatsmeow
+
+import (
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"net/url"
+
+	"go.mau.fi/whatsmeow/socket"
+	"go.mau.fi/whatsmeow/store"
+)
+
+// CheckUpdateResponse is the data returned by CheckUpdate.
+type CheckUpdateResponse struct {
+	IsBroken       bool
+	IsBelowSoft    bool
+	IsBelowHard    bool
+	CurrentVersion string
+
+	ParsedVersion store.WAVersionContainer `json:"-"`
+}
+
+// CheckUpdateURL is the base URL to check for WhatsApp web updates.
+const CheckUpdateURL = "https://web.whatsapp.com/check-update"
+
+// CheckUpdate asks the WhatsApp servers if there is an update available
+// (using the HTTP client and proxy settings of this whatsmeow Client instance).
+func (cli *Client) CheckUpdate() (respData CheckUpdateResponse, err error) {
+	return CheckUpdate(http.DefaultClient)
+}
+
+// CheckUpdate asks the WhatsApp servers if there is an update available.
+func CheckUpdate(httpClient *http.Client) (respData CheckUpdateResponse, err error) {
+	var reqURL *url.URL
+	reqURL, err = url.Parse(CheckUpdateURL)
+	if err != nil {
+		err = fmt.Errorf("failed to parse check update URL: %w", err)
+		return
+	}
+	q := reqURL.Query()
+	q.Set("version", store.GetWAVersion().String())
+	q.Set("platform", "web")
+	reqURL.RawQuery = q.Encode()
+	var req *http.Request
+	req, err = http.NewRequest(http.MethodGet, reqURL.String(), nil)
+	if err != nil {
+		err = fmt.Errorf("failed to prepare request: %w", err)
+		return
+	}
+	req.Header.Set("Origin", socket.Origin)
+	req.Header.Set("Referer", socket.Origin+"/")
+	var resp *http.Response
+	resp, err = httpClient.Do(req)
+	if err != nil {
+		err = fmt.Errorf("failed to send request: %w", err)
+		return
+	}
+	defer resp.Body.Close()
+	if resp.StatusCode != 200 {
+		body, _ := io.ReadAll(resp.Body)
+		err = fmt.Errorf("unexpected response with status %d: %s", resp.StatusCode, body)
+		return
+	}
+	err = json.NewDecoder(resp.Body).Decode(&respData)
+	if err != nil {
+		err = fmt.Errorf("failed to decode response body (status %d): %w", resp.StatusCode, err)
+		return
+	}
+	respData.ParsedVersion, err = store.ParseVersion(respData.CurrentVersion)
+	if err != nil {
+		err = fmt.Errorf("failed to parse version string: %w", err)
+	}
+	return
+}