62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
type AMAPIClient struct {
|
|
BaseURL string
|
|
HTTP *http.Client
|
|
}
|
|
|
|
func (c *AMAPIClient) Search(keyword, source string, page, limit int) (*http.Response, error) {
|
|
u, err := url.Parse(strings.TrimRight(c.BaseURL, "/") + "/search")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
q := u.Query()
|
|
q.Set("keyword", keyword)
|
|
q.Set("source", source)
|
|
q.Set("page", strconv.Itoa(page))
|
|
q.Set("limit", strconv.Itoa(limit))
|
|
u.RawQuery = q.Encode()
|
|
return c.HTTP.Get(u.String())
|
|
}
|
|
|
|
func (c *AMAPIClient) Resolve(source, rid, quality string) (playInfo, error) {
|
|
u, err := url.Parse(strings.TrimRight(c.BaseURL, "/") + "/song/play-url")
|
|
if err != nil {
|
|
return playInfo{}, err
|
|
}
|
|
q := u.Query()
|
|
q.Set("source", source)
|
|
q.Set("rid", rid)
|
|
q.Set("level", quality)
|
|
u.RawQuery = q.Encode()
|
|
resp, err := c.HTTP.Get(u.String())
|
|
if err != nil {
|
|
return playInfo{}, err
|
|
}
|
|
defer resp.Body.Close()
|
|
var x struct {
|
|
Code int `json:"code"`
|
|
Data playInfo `json:"data"`
|
|
Message string `json:"message"`
|
|
}
|
|
if err := json.NewDecoder(resp.Body).Decode(&x); err != nil {
|
|
return playInfo{}, err
|
|
}
|
|
if x.Code != 0 || x.Data.URL == "" {
|
|
if x.Message == "" {
|
|
x.Message = "music API returned no playable URL"
|
|
}
|
|
return playInfo{}, errors.New(x.Message)
|
|
}
|
|
return x.Data, nil
|
|
}
|