authelia/internal/templates/provider.go

package templates

import (
	"fmt"
	th "html/template"
	"io/fs"
	"path"
	tt "text/template"
)

// New creates a new templates' provider.
func New(config Config) (provider *Provider, err error) {
	provider = &Provider{
		config: config,
	}

	if err = provider.load(); err != nil {
		return nil, err
	}

	return provider, nil
}

// Provider of templates.
type Provider struct {
	config    Config
	templates Templates
}

// LoadTemplatedAssets takes an embed.FS and loads each templated asset document into a Template.
func (p *Provider) LoadTemplatedAssets(fs fs.ReadFileFS) (err error) {
	var (
		data []byte
	)

	if data, err = fs.ReadFile("public_html/index.html"); err != nil {
		return err
	}

	if p.templates.asset.index, err = tt.
		New("assets/public_html/index.html").
		Funcs(FuncMap()).
		Parse(string(data)); err != nil {
		return err
	}

	if data, err = fs.ReadFile("public_html/api/index.html"); err != nil {
		return err
	}

	if p.templates.asset.api.index, err = tt.
		New("assets/public_html/api/index.html").
		Funcs(FuncMap()).
		Parse(string(data)); err != nil {
		return err
	}

	if data, err = fs.ReadFile("public_html/api/openapi.yml"); err != nil {
		return err
	}

	if p.templates.asset.api.spec, err = tt.
		New("api/public_html/openapi.yaml").
		Funcs(FuncMap()).
		Parse(string(data)); err != nil {
		return err
	}

	return nil
}

// GetAssetIndexTemplate returns a Template used to generate the React index document.
func (p *Provider) GetAssetIndexTemplate() (t Template) {
	return p.templates.asset.index
}

// GetAssetOpenAPIIndexTemplate returns a Template used to generate the OpenAPI index document.
func (p *Provider) GetAssetOpenAPIIndexTemplate() (t Template) {
	return p.templates.asset.api.index
}

// GetAssetOpenAPISpecTemplate returns a Template used to generate the OpenAPI specification document.
func (p *Provider) GetAssetOpenAPISpecTemplate() (t Template) {
	return p.templates.asset.api.spec
}

// GetEventEmailTemplate returns an EmailTemplate used for generic event notifications.
func (p *Provider) GetEventEmailTemplate() (t *EmailTemplate) {
	return p.templates.notification.event
}

// GetPasswordResetEmailTemplate returns the EmailTemplate for Identity Verification notifications.
func (p *Provider) GetPasswordResetEmailTemplate() (t *EmailTemplate) {
	return p.templates.notification.reset
}

// GetOneTimePasswordEmailTemplate returns the EmailTemplate for Identity Verification notifications.
func (p *Provider) GetOneTimePasswordEmailTemplate() (t *EmailTemplate) {
	return p.templates.notification.reset
}

// GetOpenIDConnectAuthorizeResponseFormPostTemplate returns a Template used to generate the OpenID Connect 1.0 Form Post Authorize Response.
func (p *Provider) GetOpenIDConnectAuthorizeResponseFormPostTemplate() (t *th.Template) {
	return p.templates.oidc.formpost
}

func (p *Provider) load() (err error) {
	var errs []error

	if p.templates.notification.reset, err = loadEmailTemplate(TemplateNameEmailPasswordReset, p.config.EmailTemplatesPath); err != nil {
		errs = append(errs, err)
	}

	if p.templates.notification.event, err = loadEmailTemplate(TemplateNameEmailEvent, p.config.EmailTemplatesPath); err != nil {
		errs = append(errs, err)
	}

	if p.templates.notification.otp, err = loadEmailTemplate(TemplateNameOneTimePassword, p.config.EmailTemplatesPath); err != nil {
		errs = append(errs, err)
	}

	var data []byte

	if data, err = embedFS.ReadFile(path.Join("src", TemplateCategoryOpenIDConnect, TemplateNameOIDCAuthorizeFormPost)); err != nil {
		errs = append(errs, err)
	} else if p.templates.oidc.formpost, err = th.
		New("oidc/AuthorizeResponseFormPost.html").
		Funcs(FuncMap()).
		Parse(string(data)); err != nil {
		errs = append(errs, err)
	}

	if len(errs) != 0 {
		for i, e := range errs {
			if i == 0 {
				err = e
				continue
			}

			err = fmt.Errorf("%v, %w", err, e)
		}

		return fmt.Errorf("one or more errors occurred loading templates: %w", err)
	}

	return nil
}