gluon/builder.go at dev · x-mod/gluon · GitHub

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
97
98
99
100
101
102
103
104
105
106
107
package gluon

import (
	"crypto/tls"
	"io"
	"os"
	"time"

	"github.com/ProtonMail/gluon/imap"
	"github.com/ProtonMail/gluon/internal/backend"
	"github.com/ProtonMail/gluon/internal/session"
	"github.com/ProtonMail/gluon/limits"
	"github.com/ProtonMail/gluon/profiling"
	"github.com/ProtonMail/gluon/queue"
	"github.com/ProtonMail/gluon/reporter"
	"github.com/ProtonMail/gluon/store"
	"github.com/ProtonMail/gluon/version"
)

type serverBuilder struct {
	dataDir              string
	databaseDir          string
	delim                string
	loginJailTime        time.Duration
	tlsConfig            *tls.Config
	idleBulkTime         time.Duration
	inLogger             io.Writer
	outLogger            io.Writer
	versionInfo          version.Info
	cmdExecProfBuilder   profiling.CmdProfilerBuilder
	storeBuilder         store.Builder
	reporter             reporter.Reporter
	disableParallelism   bool
	imapLimits           limits.IMAP
	uidValidityGenerator imap.UIDValidityGenerator
}

func newBuilder() (*serverBuilder, error) {
	return &serverBuilder{
		delim:                "/",
		cmdExecProfBuilder:   &profiling.NullCmdExecProfilerBuilder{},
		storeBuilder:         &store.OnDiskStoreBuilder{},
		reporter:             &reporter.NullReporter{},
		idleBulkTime:         500 * time.Millisecond,
		imapLimits:           limits.DefaultLimits(),
		uidValidityGenerator: imap.DefaultEpochUIDValidityGenerator(),
	}, nil
}

func (builder *serverBuilder) build() (*Server, error) {
	if builder.dataDir == "" {
		dir, err := os.MkdirTemp("", "gluon-*")
		if err != nil {
			return nil, err
		}

		builder.dataDir = dir
	}

	if err := os.MkdirAll(builder.dataDir, 0o700); err != nil {
		return nil, err
	}

	if builder.databaseDir == "" {
		dir, err := os.MkdirTemp("", "gluon-*")
		if err != nil {
			return nil, err
		}

		builder.databaseDir = dir
	}

	if err := os.MkdirAll(builder.databaseDir, 0o700); err != nil {
		return nil, err
	}

	backend, err := backend.New(
		builder.dataDir,
		builder.databaseDir,
		builder.storeBuilder,
		builder.delim,
		builder.loginJailTime,
		builder.imapLimits,
	)
	if err != nil {
		return nil, err
	}

	return &Server{
		dataDir:              builder.dataDir,
		databaseDir:          builder.databaseDir,
		backend:              backend,
		sessions:             make(map[int]*session.Session),
		serveErrCh:           queue.NewQueuedChannel[error](1, 1),
		serveDoneCh:          make(chan struct{}),
		inLogger:             builder.inLogger,
		outLogger:            builder.outLogger,
		tlsConfig:            builder.tlsConfig,
		idleBulkTime:         builder.idleBulkTime,
		storeBuilder:         builder.storeBuilder,
		cmdExecProfBuilder:   builder.cmdExecProfBuilder,
		versionInfo:          builder.versionInfo,
		reporter:             builder.reporter,
		disableParallelism:   builder.disableParallelism,
		uidValidityGenerator: builder.uidValidityGenerator,
	}, nil
}