This repository has been archived by the owner on Aug 20, 2023. It is now read-only.
/
util.go
82 lines (69 loc) · 1.82 KB
/
util.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
package main
import (
"fmt"
"log"
"net/http"
"net/url"
"regexp"
"strings"
"github.com/gorilla/mux"
"github.com/pkg/errors"
)
type stackTracer interface {
StackTrace() errors.StackTrace
}
func logStacktrace(err error) {
if err, ok := err.(stackTracer); ok {
for _, f := range err.StackTrace() {
log.Printf("%+s:%d\n", f, f)
}
}
}
func extractURL(r *http.Request) (*url.URL, error) {
vars := mux.Vars(r)
if vars["v"] != "" {
return url.Parse(fmt.Sprintf("https://www.youtube.com/watch?v=%s", vars["v"]))
}
if vars["_u"] != "" {
url, err := url.Parse(strings.Split(r.URL.Path, "/_/")[1])
url.RawQuery = r.URL.RawQuery
if err != nil {
return nil, err
}
return url, nil
}
return nil, errors.New("not found")
}
func errorHandler(w http.ResponseWriter, r *http.Request, err error, code int) {
if code > 499 {
logStacktrace(err)
}
http.Error(w, err.Error(), code)
}
func err404(w http.ResponseWriter, r *http.Request) {
errorHandler(w, r, errors.New("not found"), http.StatusNotFound)
}
func err403(w http.ResponseWriter, r *http.Request) {
errorHandler(w, r, errors.New("unauthorized"), http.StatusUnauthorized)
}
func err500(w http.ResponseWriter, r *http.Request, err error) {
errorHandler(w, r, err, http.StatusInternalServerError)
}
// wildCardToRegexp converts a wildcard pattern to a regular expression pattern.
func wildCardToRegexp(pattern string) string {
var result strings.Builder
for i, literal := range strings.Split(pattern, "*") {
// Replace * with .*
if i > 0 {
result.WriteString(".*")
}
// Quote any regular expression meta characters in the
// literal text.
result.WriteString(regexp.QuoteMeta(literal))
}
return result.String()
}
func matchWildcard(pattern string, value string) bool {
result, _ := regexp.MatchString(wildCardToRegexp(pattern), value)
return result
}