sriracha/extension.go

78 lines
1.4 KiB
Go
Raw Permalink Normal View History

2023-04-21 03:03:02 +00:00
package sriracha
import (
"io"
"log"
"reflect"
"strings"
)
2024-01-15 20:26:31 +00:00
type ExtensionSettings struct {
// Hidden fields
}
func (s *ExtensionSettings) Get(name string) string {
return ""
}
2024-01-15 20:26:31 +00:00
func (s *ExtensionSettings) Set(name string, value string) {
return
}
type Extension interface {
Init(settings *ExtensionSettings) (description string)
2023-04-21 03:03:02 +00:00
}
var (
allExtensions []Extension
allExtensionNames []string
)
func AddExtension(e Extension) {
for _, ext := range allExtensions {
if ext == e {
return
}
}
allExtensions = append(allExtensions, e)
allExtensionNames = append(allExtensionNames, extensionName(e))
}
func extensionName(e Extension) string {
name := reflect.ValueOf(e).Type().String()
if name[0] == '*' {
name = name[1:]
}
2024-01-15 20:26:31 +00:00
name = strings.TrimPrefix(name, "extension.")
2023-04-21 03:03:02 +00:00
return strings.ToUpper(name[0:1]) + name[1:]
}
func attach(file io.Reader, size int64, mime string) (*Attachment, error) {
for i, ext := range allExtensions {
2023-04-30 03:17:56 +00:00
if extension, ok := ext.(ExtensionAttach); ok {
log.Println("DEBUG ATTACH", allExtensionNames[i])
a, err := extension.Attach(file, size, mime)
if err != nil {
return nil, err
} else if a != nil {
return a, nil
}
2023-04-21 03:03:02 +00:00
}
2023-04-30 03:17:56 +00:00
2023-04-21 03:03:02 +00:00
}
return nil, nil
}
func post(p *Post) error {
for i, ext := range allExtensions {
2023-04-30 03:17:56 +00:00
if extension, ok := ext.(ExtensionPost); ok {
log.Println("DEBUG POST", allExtensionNames[i])
err := extension.Post(p)
if err != nil {
return err
}
2023-04-21 03:03:02 +00:00
}
}
return nil
}