go/updater/keybase/source.go - master - githubmirror/client

go/updater/keybase/source.go
108 строк3 KB
zoom-ua
enable linters: misspell, bodyclose (#28736)
23 дек 2025, 22:37
23 дек 2025, 22:377debd3d
// Copyright 2015 Keybase, Inc. All rights reserved. Use of
// this source code is governed by the included BSD license.

package keybase

import (
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"net/url"
	"time"

	"github.com/keybase/client/go/updater"
	"github.com/keybase/client/go/updater/util"
)

// UpdateSource finds releases/updates on keybase.io
type UpdateSource struct {
	cfg      *config
	log      Log
	endpoint string
}

// NewUpdateSource contructs an update source for keybase.io
func NewUpdateSource(cfg *config, log Log) UpdateSource {
	return newUpdateSource(cfg, defaultEndpoints.update, log)
}

func newUpdateSource(cfg *config, endpoint string, log Log) UpdateSource {
	return UpdateSource{
		cfg:      cfg,
		endpoint: endpoint,
		log:      log,
	}
}

// Description returns description for update source
func (k UpdateSource) Description() string {
	return "Keybase.io"
}

// FindUpdate returns update for updater and options
func (k UpdateSource) FindUpdate(options updater.UpdateOptions) (*updater.Update, error) {
	return k.findUpdate(options, time.Minute)
}

func (k UpdateSource) findUpdate(options updater.UpdateOptions, timeout time.Duration) (*updater.Update, error) {
	if options.URL != "" {
		return nil, fmt.Errorf("Custom URLs not supported for this update source")
	}

	u, err := url.Parse(k.endpoint)
	if err != nil {
		return nil, err
	}

	urlValues := url.Values{}
	urlValues.Add("install_id", k.cfg.GetInstallID())
	urlValues.Add("version", options.Version)
	urlValues.Add("platform", options.Platform)
	urlValues.Add("run_mode", options.Env)
	urlValues.Add("os_version", options.OSVersion)
	urlValues.Add("upd_version", options.UpdaterVersion)
	urlValues.Add("arch", options.Arch)
	urlValues.Add("ignore_snooze", util.URLValueForBool(options.IgnoreSnooze))

	force := util.EnvBool("KEYBASE_UPDATER_FORCE", false)
	if force {
		k.log.Info("KEYBASE_UPDATER_FORCE is true, will force update")
		urlValues.Add("force", util.URLValueForBool(force))
	}

	autoUpdate, _ := k.cfg.GetUpdateAuto()
	urlValues.Add("auto_update", util.URLValueForBool(autoUpdate))

	u.RawQuery = urlValues.Encode()
	urlString := u.String()

	req, err := http.NewRequest("GET", urlString, nil)
	if err != nil {
		return nil, err
	}
	client, err := httpClient(timeout)
	if err != nil {
		return nil, err
	}
	k.log.Infof("Request %#v", urlString)
	resp, err := client.Do(req) //nolint:bodyclose // false positive, body is closed via defer after error check
	if err != nil {
		return nil, err
	}
	defer util.DiscardAndCloseBodyIgnoreError(resp)

	if resp.StatusCode != http.StatusOK {
		return nil, fmt.Errorf("Find update returned bad HTTP status %v", resp.Status)
	}

	var reader io.Reader = resp.Body
	var update updater.Update
	if err = json.NewDecoder(reader).Decode(&update); err != nil {
		return nil, fmt.Errorf("Invalid API response %s", err)
	}

	k.log.Debugf("Received update response: %#v", update)

	return &update, nil
}