122 lines
2.7 KiB
Go
122 lines
2.7 KiB
Go
// Package interwiki provides interwiki capabilities. Most of them, at least.
|
||
package interwiki
|
||
|
||
import (
|
||
"encoding/json"
|
||
"errors"
|
||
"github.com/bouncepaw/mycomarkup/v5/options"
|
||
"github.com/bouncepaw/mycorrhiza/files"
|
||
"github.com/bouncepaw/mycorrhiza/util"
|
||
"log"
|
||
"os"
|
||
"sync"
|
||
)
|
||
|
||
func Init() {
|
||
var (
|
||
record, err = readInterwiki()
|
||
)
|
||
if err != nil {
|
||
log.Fatalln(err)
|
||
}
|
||
for _, wiki := range record {
|
||
wiki := wiki // This line is required
|
||
wiki.canonize()
|
||
if err := addEntry(&wiki); err != nil {
|
||
log.Fatalln(err.Error())
|
||
}
|
||
}
|
||
log.Printf("Loaded %d interwiki entries\n", len(listOfEntries))
|
||
}
|
||
|
||
func areNamesFree(names []string) (bool, string) {
|
||
for _, name := range names {
|
||
if _, found := entriesByName[name]; found {
|
||
return false, name
|
||
}
|
||
}
|
||
return true, ""
|
||
}
|
||
|
||
var mutex sync.Mutex
|
||
|
||
func addEntry(wiki *Wiki) error {
|
||
mutex.Lock()
|
||
defer mutex.Unlock()
|
||
|
||
var (
|
||
// non-empty names only
|
||
names = func(names []string) []string {
|
||
var result []string
|
||
for _, name := range names {
|
||
if name != "" {
|
||
result = append(result, name)
|
||
}
|
||
}
|
||
return result
|
||
}(append(wiki.Aliases, wiki.Name))
|
||
ok, name = areNamesFree(names)
|
||
)
|
||
if !ok {
|
||
log.Printf("There are multiple uses of the same name ‘%s’\n", name)
|
||
return errors.New(name)
|
||
}
|
||
if len(names) == 0 {
|
||
log.Println("No names passed for a new interwiki entry")
|
||
// There is something clearly wrong with error-returning in this function.
|
||
return errors.New("")
|
||
}
|
||
|
||
listOfEntries = append(listOfEntries, wiki)
|
||
for _, name := range names {
|
||
entriesByName[name] = wiki
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func HrefLinkFormatFor(prefix string) (string, options.InterwikiError) {
|
||
prefix = util.CanonicalName(prefix)
|
||
if wiki, ok := entriesByName[prefix]; ok {
|
||
return wiki.LinkHrefFormat, options.Ok
|
||
}
|
||
return "", options.UnknownPrefix
|
||
}
|
||
|
||
func ImgSrcFormatFor(prefix string) (string, options.InterwikiError) {
|
||
prefix = util.CanonicalName(prefix)
|
||
if wiki, ok := entriesByName[prefix]; ok {
|
||
return wiki.ImgSrcFormat, options.Ok
|
||
}
|
||
return "", options.UnknownPrefix
|
||
}
|
||
|
||
func readInterwiki() ([]Wiki, error) {
|
||
var (
|
||
record []Wiki
|
||
fileContents, err = os.ReadFile(files.InterwikiJSON())
|
||
)
|
||
if os.IsNotExist(err) {
|
||
return record, nil
|
||
}
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
err = json.Unmarshal(fileContents, &record)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return record, nil
|
||
}
|
||
|
||
func saveInterwikiJson() {
|
||
// Trust me, wiki crashing when an admin takes an administrative action totally makes sense.
|
||
if data, err := json.MarshalIndent(listOfEntries, "", "\t"); err != nil {
|
||
log.Fatalln(err)
|
||
} else if err = os.WriteFile(files.InterwikiJSON(), data, 0666); err != nil {
|
||
log.Fatalln(err)
|
||
} else {
|
||
log.Println("Saved interwiki.json")
|
||
}
|
||
}
|