upstream-ory-kratos/courier/sms_templates.go

// Copyright © 2023 Ory Corp
// SPDX-License-Identifier: Apache-2.0

package courier

import (
	"context"
	"encoding/json"

	"github.com/pkg/errors"

	"github.com/ory/kratos/courier/template"
	"github.com/ory/kratos/courier/template/sms"
)

type SMSTemplate interface {
	Template
	SMSBody(context.Context) (string, error)
	PhoneNumber() (string, error)
}

func NewSMSTemplateFromMessage(d template.Dependencies, m Message) (SMSTemplate, error) {
	switch m.TemplateType {
	case template.TypeVerificationCodeValid:
		var t sms.VerificationCodeValidModel
		if err := json.Unmarshal(m.TemplateData, &t); err != nil {
			return nil, err
		}
		return sms.NewVerificationCodeValid(d, &t), nil
	case template.TypeRecoveryCodeValid:
		var t sms.RecoveryCodeValidModel
		if err := json.Unmarshal(m.TemplateData, &t); err != nil {
			return nil, err
		}
		return sms.NewRecoveryCodeValid(d, &t), nil
	case template.TypeTestStub:
		var t sms.TestStubModel
		if err := json.Unmarshal(m.TemplateData, &t); err != nil {
			return nil, err
		}
		return sms.NewTestStub(&t), nil
	case template.TypeLoginCodeValid:
		var t sms.LoginCodeValidModel
		if err := json.Unmarshal(m.TemplateData, &t); err != nil {
			return nil, err
		}
		return sms.NewLoginCodeValid(d, &t), nil
	case template.TypeRegistrationCodeValid:
		var t sms.RegistrationCodeValidModel
		if err := json.Unmarshal(m.TemplateData, &t); err != nil {
			return nil, err
		}
		return sms.NewRegistrationCodeValid(d, &t), nil
	case template.TypeVerifiableAddressChanged:
		var t sms.VerifiableAddressChangedModel
		if err := json.Unmarshal(m.TemplateData, &t); err != nil {
			return nil, err
		}
		return sms.NewVerifiableAddressChanged(d, &t), nil
	case template.TypeAuthenticatorKeyAdded:
		var t sms.AuthenticatorKeyAddedModel
		if err := json.Unmarshal(m.TemplateData, &t); err != nil {
			return nil, err
		}
		return sms.NewAuthenticatorKeyAdded(d, &t), nil
	default:
		return nil, errors.Errorf("received unexpected message template type: %s", m.TemplateType)
	}
}