File: auto.go

package info (click to toggle)
golang-agwa-go-listener 0.6.1-1.1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 168 kB
  • sloc: makefile: 2
file content (96 lines) | stat: -rw-r--r-- 3,355 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
// Copyright (C) 2022 Andrew Ayer
//
// Permission is hereby granted, free of charge, to any person obtaining a
// copy of this software and associated documentation files (the "Software"),
// to deal in the Software without restriction, including without limitation
// the rights to use, copy, modify, merge, publish, distribute, sublicense,
// and/or sell copies of the Software, and to permit persons to whom the
// Software is furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included
// in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
// THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
// OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
// ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
// OTHER DEALINGS IN THE SOFTWARE.
//
// Except as contained in this notice, the name(s) of the above copyright
// holders shall not be used in advertising or otherwise to promote the
// sale, use or other dealings in this Software without prior written
// authorization.

package cert

import (
	"encoding/base64"
	"golang.org/x/crypto/acme"
	"golang.org/x/crypto/acme/autocert"
	"os"
	"path/filepath"
)

func getAutocertCache() autocert.Cache {
	if cacheDir := os.Getenv("AUTOCERT_CACHE_DIR"); cacheDir != "" {
		return autocert.DirCache(cacheDir)
	} else if os.Getuid() == 0 {
		return autocert.DirCache("/var/lib/autocert-cache")
	} else if dataDir := os.Getenv("XDG_DATA_HOME"); dataDir != "" {
		return autocert.DirCache(filepath.Join(dataDir, "autocert-cache"))
	} else if homeDir, err := os.UserHomeDir(); err == nil {
		return autocert.DirCache(filepath.Join(homeDir, ".local/share/autocert-cache"))
	} else {
		return nil
	}
}

func getAutocertEAB() *acme.ExternalAccountBinding {
	kidString := os.Getenv("AUTOCERT_EAB_KID")
	keyString := os.Getenv("AUTOCERT_EAB_KEY")

	if kidString == "" || keyString == "" {
		return nil
	}

	key, err := base64.RawURLEncoding.DecodeString(keyString)
	if err != nil {
		return nil
	}

	return &acme.ExternalAccountBinding{
		KID: kidString,
		Key: key,
	}
}

func getCertificateAutomatically(hostPolicy autocert.HostPolicy) GetCertificateFunc {
	manager := &autocert.Manager{
		Client: &acme.Client{
			DirectoryURL: os.Getenv("AUTOCERT_ACME_SERVER"),
		},

		Prompt:     autocert.AcceptTOS,
		Cache:      getAutocertCache(),
		HostPolicy: hostPolicy,
		Email:      os.Getenv("AUTOCERT_EMAIL"),

		ExternalAccountBinding: getAutocertEAB(),
	}
	return manager.GetCertificate
}

// Returns a GetCertificateFunc that automatically obtains certificates using ACME
// for the given hostnames.  Various environment variables can be used to customize the ACME
// client.  See the [go-listener README] for details.
//
// [go-listener README]: https://pkg.go.dev/src.agwa.name/go-listener#readme-acme-configuration
func GetCertificateAutomatically(hostnames []string) GetCertificateFunc {
	if hostnames == nil {
		return getCertificateAutomatically(nil)
	} else {
		return getCertificateAutomatically(autocert.HostWhitelist(hostnames...))
	}
}