go/libkb/userconfig.go - master - githubmirror/client

go/libkb/userconfig.go
127 строк4 KB
Marcel O'Neil
Migrate to Go Modules (#24023)
22 ноя 2021, 20:07
22 ноя 2021, 20:07a66861b
// Copyright 2015 Keybase, Inc. All rights reserved. Use of
// this source code is governed by the included BSD license.

package libkb

import (
	"encoding/hex"

	"github.com/keybase/client/go/kbun"
	keybase1 "github.com/keybase/client/go/protocol/keybase1"
	jsonw "github.com/keybase/go-jsonw"
)

// ==================================================================

// TODO (CORE-6576): Remove these aliases once everything outside of
// this repo points to kbun.
type NormalizedUsername = kbun.NormalizedUsername

func NewNormalizedUsername(s string) NormalizedUsername {
	return kbun.NewNormalizedUsername(s)
}

// ==================================================================

type UserConfig struct {
	ID              string                    `json:"id"`
	Name            NormalizedUsername        `json:"name"`
	Salt            string                    `json:"salt"`
	Device          *string                   `json:"device"`
	PassphraseState *keybase1.PassphraseState `json:"passphrase_state,omitempty"`

	importedID       keybase1.UID
	importedSalt     []byte
	importedDeviceID keybase1.DeviceID

	isOneshot bool
}

// ==================================================================

func (u UserConfig) GetUID() keybase1.UID                          { return u.importedID }
func (u UserConfig) GetUsername() NormalizedUsername               { return u.Name }
func (u UserConfig) GetDeviceID() keybase1.DeviceID                { return u.importedDeviceID }
func (u UserConfig) GetPassphraseState() *keybase1.PassphraseState { return u.PassphraseState }
func (u UserConfig) IsOneshot() bool                               { return u.isOneshot }

// ==================================================================

func NewUserConfig(id keybase1.UID, name NormalizedUsername, salt []byte, dev keybase1.DeviceID) *UserConfig {
	ret := &UserConfig{
		ID:               id.String(),
		Name:             name,
		Salt:             hex.EncodeToString(salt),
		Device:           nil,
		importedID:       id,
		importedSalt:     salt,
		importedDeviceID: dev,
	}
	if dev.Exists() {
		tmp := dev.String()
		ret.Device = &tmp
	}
	return ret
}

func NewOneshotUserConfig(id keybase1.UID, name NormalizedUsername, salt []byte, dev keybase1.DeviceID) *UserConfig {
	ret := NewUserConfig(id, name, salt, dev)
	ret.isOneshot = true
	return ret
}

// ==================================================================

func (u *UserConfig) Import() (err error) {
	var tmp keybase1.UID
	if tmp, err = UIDFromHex(u.ID); err != nil {
		return
	}
	u.importedID = tmp
	if u.importedSalt, err = hex.DecodeString(u.Salt); err != nil {
		return
	}
	if u.Device != nil {
		if u.importedDeviceID, err = keybase1.DeviceIDFromString(*u.Device); err != nil {
			return
		}
	}
	return
}

// ==================================================================

func ImportUserConfigFromJSONWrapper(jw *jsonw.Wrapper) (ret *UserConfig, err error) {
	var tmp UserConfig
	if jw == nil {
		return
	}
	if err = jw.UnmarshalAgain(&tmp); err != nil {
		return
	}
	if err = tmp.Import(); err != nil {
		return
	}
	ret = &tmp
	return
}

// ==================================================================

func (u *UserConfig) SetDevice(d keybase1.DeviceID) {
	u.importedDeviceID = d
	var s *string
	if d.Exists() {
		tmp := d.String()
		s = &tmp
	}
	u.Device = s
}

// ==================================================================

func (u *UserConfig) SetPassphraseState(passphraseState keybase1.PassphraseState) {
	u.PassphraseState = &passphraseState
}

// ==================================================================