102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package instances
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/FAU-CDI/wisski-distillery/pkg/fsx"
|
|
"github.com/FAU-CDI/wisski-distillery/pkg/slicesx"
|
|
"github.com/tkw1536/goprogram/stream"
|
|
|
|
_ "embed"
|
|
)
|
|
|
|
// NoPrefix checks if this WissKI instance is excluded from generating prefixes.
|
|
// TODO: Move this to the database!
|
|
func (wisski *WissKI) NoPrefix() bool {
|
|
return fsx.IsFile(wisski.instances.Environment, filepath.Join(wisski.FilesystemBase, "prefixes.skip"))
|
|
}
|
|
|
|
//go:embed php/list_uri_prefixes.php
|
|
var listURIPrefixesPHP string
|
|
|
|
// Prefixes returns the prefixes
|
|
func (wisski *WissKI) Prefixes() (prefixes []string, err error) {
|
|
// get all the ugly prefixes
|
|
err = wisski.ExecPHPScript(stream.FromEnv(), &prefixes, listURIPrefixesPHP, "list_prefixes")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// filter out sequential prefixes
|
|
prefixes = slicesx.NonSequential(prefixes, func(prev, now string) bool {
|
|
return strings.HasPrefix(now, prev)
|
|
})
|
|
|
|
// filter out blocked prefixes
|
|
return slicesx.Filter(prefixes, func(uri string) bool { return !IsNonServedURI(uri) }), nil
|
|
}
|
|
|
|
// TODO: Eventually move this into a configuration file.
|
|
// But for now this is fine
|
|
var blockedURIs = []string{
|
|
"http://erlangen-crm.org/",
|
|
"http://www.w3.org/",
|
|
"xsd:",
|
|
}
|
|
|
|
func IsNonServedURI(candidate string) bool {
|
|
return slicesx.Any(
|
|
blockedURIs,
|
|
func(prefix string) bool {
|
|
return strings.HasPrefix(candidate, prefix)
|
|
},
|
|
)
|
|
}
|
|
|
|
var errPrefixExecFailed = errors.New("PrefixConfig: Failed to call list_uri_prefixes")
|
|
|
|
// PrefixConfig returns the prefix config belonging to this instance.
|
|
func (wisski *WissKI) PrefixConfig() (config string, err error) {
|
|
// if the user requested to skip the prefix, then don't do anything with it!
|
|
if wisski.NoPrefix() {
|
|
return "", nil
|
|
}
|
|
|
|
var builder strings.Builder
|
|
|
|
// domain
|
|
builder.WriteString(wisski.URL().String() + ":")
|
|
builder.WriteString("\n")
|
|
|
|
// default prefixes
|
|
prefixes, err := wisski.Prefixes()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
// predefined prefixes
|
|
for _, prefix := range prefixes {
|
|
builder.WriteString(prefix)
|
|
builder.WriteRune('\n')
|
|
}
|
|
|
|
// custom prefixes
|
|
prefixPath := filepath.Join(wisski.FilesystemBase, "prefixes")
|
|
if fsx.IsFile(wisski.instances.Environment, prefixPath) {
|
|
prefix, err := wisski.instances.Core.Environment.Open(prefixPath)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer prefix.Close()
|
|
if _, err := io.Copy(&builder, prefix); err != nil {
|
|
return "", err
|
|
}
|
|
builder.WriteString("\n")
|
|
}
|
|
|
|
// and done!
|
|
return builder.String(), nil
|
|
}
|