Thomas Peetz
70 lines
1.2 KiB
Go
70 lines
1.2 KiB
Go
package helpers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log/slog"
|
|
"net/http"
|
|
"net/url"
|
|
)
|
|
|
|
func StripUrl(webLink string, keepParam string) string {
|
|
u, err := url.Parse(webLink)
|
|
if err != nil {
|
|
slog.Info("StripUrl", "error", err)
|
|
return webLink
|
|
}
|
|
values := u.Query()
|
|
for key := range values {
|
|
if key != keepParam {
|
|
values.Del(key)
|
|
}
|
|
}
|
|
u.RawQuery = values.Encode()
|
|
return u.String()
|
|
}
|
|
|
|
func GetParam(webLink string, param string) string {
|
|
u, err := url.Parse(webLink)
|
|
if err != nil {
|
|
slog.Info("StripUrl", "error", err)
|
|
return webLink
|
|
}
|
|
values := u.Query()
|
|
for key := range values {
|
|
if key == param {
|
|
return values.Get(key)
|
|
}
|
|
}
|
|
return ""
|
|
}
|
|
|
|
type OEmbedResponse struct {
|
|
Title string `json:"title"`
|
|
}
|
|
|
|
func GetVideoTitle(videoUrl string) (string, error) {
|
|
videoID := GetParam(videoUrl, "v")
|
|
// Construct the oEmbed URL
|
|
url := fmt.Sprintf("https://www.youtube.com/oembed?url=https://www.youtube.com/watch?v=%s&format=json", videoID)
|
|
|
|
resp, err := http.Get(url)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
var data OEmbedResponse
|
|
if err := json.Unmarshal(body, &data); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return data.Title, nil
|
|
}
|