github.com/keybase/client/go@v0.0.0-20240309051027-028f7c731f8b/service/phone_numbers.go

package service

import (
	"encoding/json"
	"fmt"
	"strings"

	"github.com/keybase/client/go/gregor"
	"github.com/keybase/client/go/libkb"
	"github.com/keybase/client/go/phonenumbers"
	gregor1 "github.com/keybase/client/go/protocol/gregor1"
	keybase1 "github.com/keybase/client/go/protocol/keybase1"
	"github.com/keybase/go-framed-msgpack-rpc/rpc"

	"golang.org/x/net/context"
)

type PhoneNumbersHandler struct {
	libkb.Contextified
	*BaseHandler
}

func NewPhoneNumbersHandler(xp rpc.Transporter, g *libkb.GlobalContext) *PhoneNumbersHandler {
	handler := &PhoneNumbersHandler{
		Contextified: libkb.NewContextified(g),
		BaseHandler:  NewBaseHandler(g, xp),
	}
	return handler
}

var _ keybase1.PhoneNumbersInterface = (*PhoneNumbersHandler)(nil)

func (h *PhoneNumbersHandler) AddPhoneNumber(ctx context.Context, arg keybase1.AddPhoneNumberArg) (err error) {
	mctx := libkb.NewMetaContext(ctx, h.G())
	defer mctx.Trace("PhoneNumbersHandler#AddPhoneNumber", &err)()
	if err = libkb.IsPossiblePhoneNumber(arg.PhoneNumber); err != nil {
		return err
	}
	return phonenumbers.AddPhoneNumber(mctx, arg.PhoneNumber, arg.Visibility)
}

func (h *PhoneNumbersHandler) EditPhoneNumber(ctx context.Context, arg keybase1.EditPhoneNumberArg) (err error) {
	mctx := libkb.NewMetaContext(ctx, h.G())
	defer mctx.Trace("PhoneNumbersHandler#AddPhoneNumber", &err)()
	if err = libkb.IsPossiblePhoneNumber(arg.OldPhoneNumber); err != nil {
		return err
	}
	if err = libkb.IsPossiblePhoneNumber(arg.PhoneNumber); err != nil {
		return err
	}

	if err = phonenumbers.DeletePhoneNumber(mctx, arg.OldPhoneNumber); err != nil {
		return err
	}
	return phonenumbers.AddPhoneNumber(mctx, arg.PhoneNumber, arg.Visibility)
}

func (h *PhoneNumbersHandler) VerifyPhoneNumber(ctx context.Context, arg keybase1.VerifyPhoneNumberArg) (err error) {
	mctx := libkb.NewMetaContext(ctx, h.G())
	defer mctx.Trace("PhoneNumbersHandler#VerifyPhoneNumber", &err)()
	if err = libkb.IsPossiblePhoneNumber(arg.PhoneNumber); err != nil {
		return err
	}

	return phonenumbers.VerifyPhoneNumber(mctx, arg.PhoneNumber, arg.Code)
}

func (h *PhoneNumbersHandler) ResendVerificationForPhoneNumber(ctx context.Context, arg keybase1.ResendVerificationForPhoneNumberArg) (err error) {
	mctx := libkb.NewMetaContext(ctx, h.G())
	defer mctx.Trace("PhoneNumbersHandler#ResendVerificationForPhoneNumber", &err)()
	if err = libkb.IsPossiblePhoneNumber(arg.PhoneNumber); err != nil {
		return err
	}

	numbers, err := phonenumbers.GetPhoneNumbers(mctx)
	if err != nil {
		return err
	}

	// We are going to "re-add" phone number with the same visibility setting.
	var readdVisibility keybase1.IdentityVisibility
	var found bool
	for _, v := range numbers {
		if v.Superseded {
			continue
		}

		if v.PhoneNumber == arg.PhoneNumber {
			if v.Verified {
				return fmt.Errorf("Number %s is already verified", v.PhoneNumber)
			}
			readdVisibility = v.Visibility
			found = true
			break
		}
	}

	if !found {
		return fmt.Errorf("Number %s was not found, it has to be added first before resending verification code",
			arg.PhoneNumber)
	}

	return phonenumbers.AddPhoneNumber(mctx, arg.PhoneNumber, readdVisibility)
}

func (h *PhoneNumbersHandler) GetPhoneNumbers(ctx context.Context, sessionID int) (ret []keybase1.UserPhoneNumber, err error) {
	mctx := libkb.NewMetaContext(ctx, h.G())
	defer mctx.Trace("PhoneNumbersHandler#GetPhoneNumbers", &err)()
	return phonenumbers.GetPhoneNumbers(mctx)
}

func (h *PhoneNumbersHandler) DeletePhoneNumber(ctx context.Context, arg keybase1.DeletePhoneNumberArg) (err error) {
	mctx := libkb.NewMetaContext(ctx, h.G())
	defer mctx.Trace("PhoneNumbersHandler#DeletePhoneNumber", &err)()
	if err = libkb.IsPossiblePhoneNumber(arg.PhoneNumber); err != nil {
		return err
	}

	return phonenumbers.DeletePhoneNumber(mctx, arg.PhoneNumber)
}

func (h *PhoneNumbersHandler) SetVisibilityPhoneNumber(ctx context.Context, arg keybase1.SetVisibilityPhoneNumberArg) (err error) {
	mctx := libkb.NewMetaContext(ctx, h.G())
	defer mctx.Trace("PhoneNumbersHandler#SetVisibilityPhoneNumber", &err)()
	if err = libkb.IsPossiblePhoneNumber(arg.PhoneNumber); err != nil {
		return err
	}

	return phonenumbers.SetVisibilityPhoneNumber(mctx, arg.PhoneNumber, arg.Visibility)
}

func (h *PhoneNumbersHandler) SetVisibilityAllPhoneNumber(ctx context.Context, arg keybase1.SetVisibilityAllPhoneNumberArg) (err error) {
	mctx := libkb.NewMetaContext(ctx, h.G())
	defer mctx.Trace("PhoneNumbersHandler#SetVisibilityAllPhoneNumber", &err)()
	return phonenumbers.SetVisibilityAllPhoneNumber(mctx, arg.Visibility)
}

const phoneNumbersGregorHandlerName = "phoneHandler"

type phoneNumbersGregorHandler struct {
	libkb.Contextified
}

var _ libkb.GregorInBandMessageHandler = (*phoneNumbersGregorHandler)(nil)

func newPhoneNumbersGregorHandler(g *libkb.GlobalContext) *phoneNumbersGregorHandler {
	return &phoneNumbersGregorHandler{
		Contextified: libkb.NewContextified(g),
	}
}

func (r *phoneNumbersGregorHandler) Create(ctx context.Context, cli gregor1.IncomingInterface, category string, item gregor.Item) (bool, error) {
	switch category {
	case "phone.added", "phone.verified", "phone.superseded",
		"phone.visibility_changed", "phone.deleted":
		return true, r.handlePhoneMsg(ctx, cli, category, item)
	case "phone.unverified_count":
		// handled in badgestate
		return true, nil
	default:
		if strings.HasPrefix(category, "phone.") {
			return false, fmt.Errorf("unknown phoneNumbersGregorHandler category: %q", category)
		}
		return false, nil
	}
}

func (r *phoneNumbersGregorHandler) Dismiss(ctx context.Context, cli gregor1.IncomingInterface, category string, item gregor.Item) (bool, error) {
	return false, nil
}

func (r *phoneNumbersGregorHandler) IsAlive() bool {
	return true
}

func (r *phoneNumbersGregorHandler) Name() string {
	return phoneNumbersGregorHandlerName
}

func (r *phoneNumbersGregorHandler) handlePhoneMsg(ctx context.Context, cli gregor1.IncomingInterface, category string, item gregor.Item) error {
	mctx := libkb.NewMetaContext(ctx, r.G())
	mctx.Debug("phoneNumbersGregorHandler: %s received", category)
	var msg keybase1.PhoneNumberChangedMsg
	if err := json.Unmarshal(item.Body().Bytes(), &msg); err != nil {
		mctx.Debug("error unmarshaling %s item: %s", category, err)
		return err
	}
	mctx.Debug("%s unmarshaled: %+v", category, msg)

	phoneNumbers, err := phonenumbers.GetPhoneNumbers(mctx)
	if err != nil {
		mctx.Error("Could not get current phone number list during handlePhoneMsg: %s", err)
	} else {
		r.G().NotifyRouter.HandlePhoneNumbersChanged(ctx, phoneNumbers, category, msg.PhoneNumber)
	}
	// Do not dismiss these notifications so other devices can see it and
	// update their state as well.
	return nil
}