mirror of
https://github.com/ProtonMail/proton-bridge.git
synced 2025-12-11 13:16:53 +00:00
138 lines
3.9 KiB
Go
138 lines
3.9 KiB
Go
// Copyright (c) 2020 Proton Technologies AG
|
|
//
|
|
// This file is part of ProtonMail Bridge.
|
|
//
|
|
// ProtonMail Bridge is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// ProtonMail Bridge is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with ProtonMail Bridge. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
// Package bridge implements the bridge CLI application.
|
|
package bridge
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/ProtonMail/proton-bridge/internal/api"
|
|
"github.com/ProtonMail/proton-bridge/internal/app/base"
|
|
"github.com/ProtonMail/proton-bridge/internal/bridge"
|
|
"github.com/ProtonMail/proton-bridge/internal/config/settings"
|
|
"github.com/ProtonMail/proton-bridge/internal/constants"
|
|
"github.com/ProtonMail/proton-bridge/internal/frontend"
|
|
"github.com/ProtonMail/proton-bridge/internal/imap"
|
|
"github.com/ProtonMail/proton-bridge/internal/smtp"
|
|
"github.com/ProtonMail/proton-bridge/internal/updater"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
func New(base *base.Base) *cli.App {
|
|
app := base.NewApp(run)
|
|
|
|
app.Flags = append(app.Flags, []cli.Flag{
|
|
&cli.StringFlag{
|
|
Name: "log-imap",
|
|
Usage: "Enable logging of IMAP communications (all|client|server) (may contain decrypted data!)"},
|
|
&cli.BoolFlag{
|
|
Name: "log-smtp",
|
|
Usage: "Enable logging of SMTP communications (may contain decrypted data!)"},
|
|
&cli.BoolFlag{
|
|
Name: "no-window",
|
|
Usage: "Don't show window after start"},
|
|
&cli.BoolFlag{
|
|
Name: "noninteractive",
|
|
Usage: "Start Bridge entirely noninteractively"},
|
|
}...)
|
|
|
|
return app
|
|
}
|
|
|
|
func run(b *base.Base, c *cli.Context) error { // nolint[funlen]
|
|
tls, err := b.TLS.GetConfig()
|
|
if err != nil {
|
|
logrus.WithError(err).Fatal("Failed to create TLS config")
|
|
}
|
|
|
|
bridge := bridge.New(b.Locations, b.Cache, b.Settings, b.CrashHandler, b.Listener, b.CM, b.Creds)
|
|
imapBackend := imap.NewIMAPBackend(b.CrashHandler, b.Listener, b.Cache, bridge)
|
|
smtpBackend := smtp.NewSMTPBackend(b.CrashHandler, b.Listener, b.Settings, bridge)
|
|
|
|
go func() {
|
|
defer b.CrashHandler.HandlePanic()
|
|
api.NewAPIServer(b.Settings, b.Listener).ListenAndServe()
|
|
}()
|
|
|
|
go func() {
|
|
defer b.CrashHandler.HandlePanic()
|
|
imapPort := b.Settings.GetInt(settings.IMAPPortKey)
|
|
imap.NewIMAPServer(
|
|
c.String("log-imap") == "client" || c.String("log-imap") == "all",
|
|
c.String("log-imap") == "server" || c.String("log-imap") == "all",
|
|
imapPort, tls, imapBackend, b.Listener).ListenAndServe()
|
|
}()
|
|
|
|
go func() {
|
|
defer b.CrashHandler.HandlePanic()
|
|
smtpPort := b.Settings.GetInt(settings.SMTPPortKey)
|
|
useSSL := b.Settings.GetBool(settings.SMTPSSLKey)
|
|
smtp.NewSMTPServer(
|
|
c.Bool("log-smtp"),
|
|
smtpPort, useSSL, tls, smtpBackend, b.Listener).ListenAndServe()
|
|
}()
|
|
|
|
var frontendMode string
|
|
|
|
switch {
|
|
case c.Bool("cli"):
|
|
frontendMode = "cli"
|
|
case c.Bool("noninteractive"):
|
|
frontendMode = "noninteractive"
|
|
default:
|
|
frontendMode = "qt"
|
|
}
|
|
|
|
if frontendMode == "noninteractive" {
|
|
<-(make(chan struct{}))
|
|
return nil
|
|
}
|
|
|
|
f := frontend.New(
|
|
constants.Version,
|
|
constants.BuildVersion,
|
|
frontendMode,
|
|
!c.Bool("no-window"),
|
|
b.CrashHandler,
|
|
b.Locations,
|
|
b.Settings,
|
|
b.Listener,
|
|
b.Updater,
|
|
bridge,
|
|
smtpBackend,
|
|
b,
|
|
)
|
|
|
|
b.Updater.Watch(
|
|
time.Hour,
|
|
func(update updater.VersionInfo) error {
|
|
if !b.Settings.GetBool(settings.AutoUpdateKey) {
|
|
return f.NotifyManualUpdate(update)
|
|
}
|
|
|
|
return b.Updater.InstallUpdate(update)
|
|
},
|
|
func(err error) {
|
|
logrus.WithError(err).Error("An error occurred while watching for updates")
|
|
},
|
|
)
|
|
|
|
return f.Loop()
|
|
}
|