91 lines
1.8 KiB
Go
91 lines
1.8 KiB
Go
package linkinfo
|
|
|
|
import (
|
|
"net/http"
|
|
"net/url"
|
|
"time"
|
|
)
|
|
|
|
type LinkHandler struct {
|
|
Hosts []string
|
|
Handler func(link string) (*LinkInfo, error)
|
|
}
|
|
|
|
type LinkInfoApi struct {
|
|
Client *http.Client
|
|
|
|
Imgur *ImgurInfoApi
|
|
Youtube *YoutubeInfoApi
|
|
linkHandlers []*LinkHandler
|
|
}
|
|
|
|
type LinkInfo struct {
|
|
Title string `json:"title"`
|
|
Description string `json:"description"`
|
|
ContentType string `json:"type"`
|
|
ContentLength int64 `json:"contentLength"`
|
|
Duration string `json:"duration,omitempty"`
|
|
Redirects []string `json:"redirects,omitempty"`
|
|
}
|
|
|
|
func New(opts ...Option) *LinkInfoApi {
|
|
api := &LinkInfoApi{
|
|
Client: &http.Client{
|
|
Timeout: 60 * time.Second,
|
|
},
|
|
}
|
|
|
|
for _, opt := range opts {
|
|
switch opt.(type) {
|
|
case *ImgurOptions:
|
|
api.Imgur = &ImgurInfoApi{api, opt.(*ImgurOptions)}
|
|
case *YoutubeOptions:
|
|
api.Youtube = &YoutubeInfoApi{api, opt.(*YoutubeOptions)}
|
|
}
|
|
}
|
|
|
|
api.registerDefaultHandlers()
|
|
|
|
return api
|
|
}
|
|
|
|
func (i *LinkInfoApi) registerDefaultHandlers() {
|
|
i.linkHandlers = make([]*LinkHandler, 0)
|
|
|
|
if i.Imgur != nil {
|
|
i.linkHandlers = append(i.linkHandlers, &LinkHandler{
|
|
Hosts: imgurHosts,
|
|
Handler: i.Imgur.Handler,
|
|
})
|
|
}
|
|
|
|
if i.Youtube != nil {
|
|
i.linkHandlers = append(i.linkHandlers, &LinkHandler{
|
|
Hosts: youtubeHosts,
|
|
Handler: i.Youtube.Handler,
|
|
})
|
|
}
|
|
}
|
|
|
|
func (i *LinkInfoApi) AddHandler(handler *LinkHandler) {
|
|
i.linkHandlers = append(i.linkHandlers, handler)
|
|
}
|
|
|
|
func (i *LinkInfoApi) Retrieve(link string) (*LinkInfo, error) {
|
|
u, err := url.Parse(link)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, handler := range i.linkHandlers {
|
|
for _, host := range handler.Hosts {
|
|
if host == u.Hostname() {
|
|
return handler.Handler(link)
|
|
}
|
|
}
|
|
}
|
|
|
|
return i.DefaultLinkHandler(link)
|
|
}
|