linkinfo/linkinfo.go

86 lines
1.6 KiB
Go
Raw Normal View History

package linkinfo
import (
"net/http"
"net/url"
"time"
)
type LinkHandler struct {
Hosts []string
Handler func(link string) (*LinkInfo, error)
}
type LinkInfoApi struct {
2019-10-04 03:31:24 +00:00
Client *http.Client
Imgur *ImgurInfoApi
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"`
}
2019-10-04 03:31:24 +00:00
func New(opts ...Option) *LinkInfoApi {
api := &LinkInfoApi{
Client: &http.Client{
Timeout: 60 * time.Second,
},
}
2019-10-04 03:31:24 +00:00
for _, opt := range opts {
switch opt.(type) {
case *ImgurOptions:
api.Imgur = &ImgurInfoApi{api, opt.(*ImgurOptions)}
}
}
api.registerDefaultHandlers()
return api
}
func (i *LinkInfoApi) registerDefaultHandlers() {
2019-10-04 03:31:24 +00:00
i.linkHandlers = make([]*LinkHandler, 0)
if i.Imgur != nil {
i.linkHandlers = append(i.linkHandlers, &LinkHandler{
Hosts: imgurHosts,
Handler: i.Imgur.Handler,
})
}
i.linkHandlers = []*LinkHandler{
{Hosts: youtubeHosts, Handler: i.YoutubeLinkHandler},
{Hosts: twitterHosts, Handler: i.TwitterLinkHandler},
}
}
2019-10-04 03:31:24 +00:00
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)
}