85 lines
1.7 KiB
Go
85 lines
1.7 KiB
Go
|
// Example
|
||
|
|
//
|
||
|
|
// publicdir.MustRegister(app, &publicdir.Options{
|
||
|
|
// FlagsCmd: app.RootCmd,
|
||
|
|
// IndexFallback: false,
|
||
|
|
// })
|
||
|
|
package publicdir
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/pocketbase/pocketbase/apis"
|
||
|
|
"github.com/pocketbase/pocketbase/core"
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Options struct {
|
||
|
|
Dir string
|
||
|
|
IndexFallback bool
|
||
|
|
FlagsCmd *cobra.Command
|
||
|
|
}
|
||
|
|
|
||
|
|
type plugin struct {
|
||
|
|
app core.App
|
||
|
|
options *Options
|
||
|
|
}
|
||
|
|
|
||
|
|
func MustRegister(app core.App, options *Options) {
|
||
|
|
if err := Register(app, options); err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func Register(app core.App, options *Options) error {
|
||
|
|
p := &plugin{app: app}
|
||
|
|
|
||
|
|
if options != nil {
|
||
|
|
p.options = options
|
||
|
|
} else {
|
||
|
|
p.options = &Options{}
|
||
|
|
}
|
||
|
|
|
||
|
|
if options.Dir == "" {
|
||
|
|
options.Dir = defaultPublicDir()
|
||
|
|
}
|
||
|
|
|
||
|
|
if options.FlagsCmd != nil {
|
||
|
|
// add "--publicDir" option flag
|
||
|
|
options.FlagsCmd.PersistentFlags().StringVar(
|
||
|
|
&options.Dir,
|
||
|
|
"publicDir",
|
||
|
|
options.Dir,
|
||
|
|
"the directory to serve static files",
|
||
|
|
)
|
||
|
|
|
||
|
|
// add "--indexFallback" option flag
|
||
|
|
options.FlagsCmd.PersistentFlags().BoolVar(
|
||
|
|
&options.IndexFallback,
|
||
|
|
"indexFallback",
|
||
|
|
options.IndexFallback,
|
||
|
|
"fallback the request to index.html on missing static path (eg. when pretty urls are used with SPA)",
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
p.app.OnBeforeServe().Add(func(e *core.ServeEvent) error {
|
||
|
|
// serves static files from the provided public dir (if exists)
|
||
|
|
e.Router.GET("/*", apis.StaticDirectoryHandler(os.DirFS(options.Dir), options.IndexFallback))
|
||
|
|
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func defaultPublicDir() string {
|
||
|
|
if strings.HasPrefix(os.Args[0], os.TempDir()) {
|
||
|
|
// most likely ran with go run
|
||
|
|
return "./pb_public"
|
||
|
|
}
|
||
|
|
|
||
|
|
return filepath.Join(os.Args[0], "../pb_public")
|
||
|
|
}
|