mirror of
https://github.com/pocketbase/pocketbase.git
synced 2025-03-21 22:49:42 +02:00
always register the installer hooks in case the superuser is created by a console command
This commit is contained in:
parent
f38700982c
commit
e4cd6810ab
@ -13,11 +13,9 @@ import (
|
|||||||
"github.com/go-ozzo/ozzo-validation/v4/is"
|
"github.com/go-ozzo/ozzo-validation/v4/is"
|
||||||
"github.com/pocketbase/dbx"
|
"github.com/pocketbase/dbx"
|
||||||
"github.com/pocketbase/pocketbase/core"
|
"github.com/pocketbase/pocketbase/core"
|
||||||
"github.com/pocketbase/pocketbase/tools/hook"
|
|
||||||
"github.com/pocketbase/pocketbase/tools/security"
|
"github.com/pocketbase/pocketbase/tools/security"
|
||||||
)
|
)
|
||||||
|
|
||||||
const installerEmail = "__pbinstaller@example.com"
|
|
||||||
const installerHookId = "__pbinstallerHook"
|
const installerHookId = "__pbinstallerHook"
|
||||||
|
|
||||||
func loadInstaller(app core.App, hostURL string) error {
|
func loadInstaller(app core.App, hostURL string) error {
|
||||||
@ -35,39 +33,6 @@ func loadInstaller(app core.App, hostURL string) error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
// prevent sending password reset emails to the installer address
|
|
||||||
app.OnMailerRecordPasswordResetSend(core.CollectionNameSuperusers).Bind(&hook.Handler[*core.MailerRecordEvent]{
|
|
||||||
Id: installerHookId,
|
|
||||||
Func: func(e *core.MailerRecordEvent) error {
|
|
||||||
if e.Record.Email() == installerEmail {
|
|
||||||
return errors.New("cannot reset the password for the installer account")
|
|
||||||
}
|
|
||||||
|
|
||||||
return e.Next()
|
|
||||||
},
|
|
||||||
})
|
|
||||||
|
|
||||||
// cleanup the installer account after the first superuser creation
|
|
||||||
app.OnRecordCreate(core.CollectionNameSuperusers).Bind(&hook.Handler[*core.RecordEvent]{
|
|
||||||
Id: installerHookId,
|
|
||||||
Func: func(e *core.RecordEvent) error {
|
|
||||||
if err := e.Next(); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
color.Green("Successfully created superuser %s! This message will no longer show on the next startup.\n\n", e.Record.Email())
|
|
||||||
|
|
||||||
if err = e.App.Delete(installerRecord); err != nil {
|
|
||||||
e.App.Logger().Error("Failed to remove installer superuser", "error", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
app.OnRecordCreate().Unbind(installerHookId)
|
|
||||||
app.OnMailerRecordPasswordResetSend().Unbind(installerHookId)
|
|
||||||
|
|
||||||
return nil
|
|
||||||
},
|
|
||||||
})
|
|
||||||
|
|
||||||
// launch url (ignore errors and always print a help text as fallback)
|
// launch url (ignore errors and always print a help text as fallback)
|
||||||
url := fmt.Sprintf("%s/_/#/pbinstal/%s", hostURL, token)
|
url := fmt.Sprintf("%s/_/#/pbinstal/%s", hostURL, token)
|
||||||
_ = launchURL(url)
|
_ = launchURL(url)
|
||||||
@ -80,7 +45,7 @@ func loadInstaller(app core.App, hostURL string) error {
|
|||||||
|
|
||||||
func needInstallerSuperuser(app core.App) bool {
|
func needInstallerSuperuser(app core.App) bool {
|
||||||
total, err := app.CountRecords(core.CollectionNameSuperusers, dbx.Not(dbx.HashExp{
|
total, err := app.CountRecords(core.CollectionNameSuperusers, dbx.Not(dbx.HashExp{
|
||||||
"email": installerEmail,
|
"email": core.DefaultInstallerEmail,
|
||||||
}))
|
}))
|
||||||
return err == nil && total == 0
|
return err == nil && total == 0
|
||||||
}
|
}
|
||||||
@ -91,14 +56,14 @@ func findOrCreateInstallerSuperuser(app core.App) (*core.Record, error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
record, err := app.FindAuthRecordByEmail(col, installerEmail)
|
record, err := app.FindAuthRecordByEmail(col, core.DefaultInstallerEmail)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if !errors.Is(err, sql.ErrNoRows) {
|
if !errors.Is(err, sql.ErrNoRows) {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
record = core.NewRecord(col)
|
record = core.NewRecord(col)
|
||||||
record.SetEmail(installerEmail)
|
record.SetEmail(core.DefaultInstallerEmail)
|
||||||
record.SetPassword(security.RandomString(30))
|
record.SetPassword(security.RandomString(30))
|
||||||
|
|
||||||
err = app.Save(record)
|
err = app.Save(record)
|
||||||
|
@ -270,7 +270,12 @@ func Serve(app core.App, config ServeConfig) error {
|
|||||||
regular.Printf("└─ Dashboard: %s\n", color.CyanString("%s/_/", fullAddr))
|
regular.Printf("└─ Dashboard: %s\n", color.CyanString("%s/_/", fullAddr))
|
||||||
}
|
}
|
||||||
|
|
||||||
go loadInstaller(app, fullAddr)
|
go func() {
|
||||||
|
installerErr := loadInstaller(app, fullAddr)
|
||||||
|
if installerErr != nil {
|
||||||
|
app.Logger().Warn("Failed to initialize installer", "error", installerErr)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
var serveErr error
|
var serveErr error
|
||||||
if config.HttpsAddr != "" {
|
if config.HttpsAddr != "" {
|
||||||
|
@ -1,6 +1,8 @@
|
|||||||
package core
|
package core
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"database/sql"
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
|
||||||
"github.com/pocketbase/pocketbase/tools/hook"
|
"github.com/pocketbase/pocketbase/tools/hook"
|
||||||
@ -9,6 +11,10 @@ import (
|
|||||||
|
|
||||||
const CollectionNameSuperusers = "_superusers"
|
const CollectionNameSuperusers = "_superusers"
|
||||||
|
|
||||||
|
// DefaultInstallerEmail is the default superuser email address
|
||||||
|
// for the initial autogenerated superuser account.
|
||||||
|
const DefaultInstallerEmail = "__pbinstaller@example.com"
|
||||||
|
|
||||||
func (app *BaseApp) registerSuperuserHooks() {
|
func (app *BaseApp) registerSuperuserHooks() {
|
||||||
app.OnRecordDelete(CollectionNameSuperusers).Bind(&hook.Handler[*RecordEvent]{
|
app.OnRecordDelete(CollectionNameSuperusers).Bind(&hook.Handler[*RecordEvent]{
|
||||||
Id: "pbSuperusersRecordDelete",
|
Id: "pbSuperusersRecordDelete",
|
||||||
@ -39,13 +45,45 @@ func (app *BaseApp) registerSuperuserHooks() {
|
|||||||
Id: "pbSuperusersRecordSaveExec",
|
Id: "pbSuperusersRecordSaveExec",
|
||||||
Func: func(e *RecordEvent) error {
|
Func: func(e *RecordEvent) error {
|
||||||
e.Record.SetVerified(true) // always mark superusers as verified
|
e.Record.SetVerified(true) // always mark superusers as verified
|
||||||
return e.Next()
|
|
||||||
|
if err := e.Next(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// ensure that the installer superuser is deleted
|
||||||
|
if e.Type == ModelEventTypeCreate && e.Record.Email() != DefaultInstallerEmail {
|
||||||
|
record, err := app.FindAuthRecordByEmail(CollectionNameSuperusers, DefaultInstallerEmail)
|
||||||
|
if errors.Is(err, sql.ErrNoRows) {
|
||||||
|
// already deleted
|
||||||
|
} else if err != nil {
|
||||||
|
e.App.Logger().Warn("Failed to fetch installer superuser", "error", err)
|
||||||
|
} else {
|
||||||
|
err = e.App.Delete(record)
|
||||||
|
if err != nil {
|
||||||
|
e.App.Logger().Warn("Failed to delete installer superuser", "error", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
},
|
},
|
||||||
Priority: -99,
|
Priority: -99,
|
||||||
}
|
}
|
||||||
app.OnRecordCreateExecute(CollectionNameSuperusers).Bind(recordSaveHandler)
|
app.OnRecordCreateExecute(CollectionNameSuperusers).Bind(recordSaveHandler)
|
||||||
app.OnRecordUpdateExecute(CollectionNameSuperusers).Bind(recordSaveHandler)
|
app.OnRecordUpdateExecute(CollectionNameSuperusers).Bind(recordSaveHandler)
|
||||||
|
|
||||||
|
// prevent sending password reset emails to the installer address
|
||||||
|
app.OnMailerRecordPasswordResetSend(CollectionNameSuperusers).Bind(&hook.Handler[*MailerRecordEvent]{
|
||||||
|
Id: "pbSuperusersInstallerPasswordReset",
|
||||||
|
Func: func(e *MailerRecordEvent) error {
|
||||||
|
if e.Record.Email() == DefaultInstallerEmail {
|
||||||
|
return errors.New("cannot reset the password for the installer superuser")
|
||||||
|
}
|
||||||
|
|
||||||
|
return e.Next()
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
collectionSaveHandler := &hook.Handler[*CollectionEvent]{
|
collectionSaveHandler := &hook.Handler[*CollectionEvent]{
|
||||||
Id: "pbSuperusersCollectionSaveExec",
|
Id: "pbSuperusersCollectionSaveExec",
|
||||||
Func: func(e *CollectionEvent) error {
|
Func: func(e *CollectionEvent) error {
|
||||||
|
Loading…
x
Reference in New Issue
Block a user