mirror of https://github.com/veonik/eokvin
78 lines
1.5 KiB
Go
78 lines
1.5 KiB
Go
package eokvin // import "github.com/veonik/eokvin"
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"encoding/json"
|
|
"errors"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
"time"
|
|
)
|
|
|
|
type ShortURL struct {
|
|
url.URL
|
|
|
|
Original *url.URL
|
|
}
|
|
|
|
type Client struct {
|
|
http *http.Client
|
|
|
|
Endpoint string
|
|
Token string
|
|
}
|
|
|
|
func NewClient(endpoint, token string) *Client {
|
|
return &Client{
|
|
http: &http.Client{},
|
|
Endpoint: endpoint,
|
|
Token: token,
|
|
}
|
|
}
|
|
|
|
func NewInsecureClient(endpoint, token string) *Client {
|
|
return &Client{
|
|
http: &http.Client{Transport: &http.Transport{TLSClientConfig: &tls.Config{InsecureSkipVerify: true}}},
|
|
Endpoint: endpoint,
|
|
Token: token,
|
|
}
|
|
}
|
|
|
|
func (c *Client) NewShortURLString(lu string, ttl time.Duration) (*ShortURL, error) {
|
|
ou, err := url.Parse(lu)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return c.NewShortURL(ou, ttl)
|
|
}
|
|
|
|
func (c *Client) NewShortURL(ou *url.URL, ttl time.Duration) (*ShortURL, error) {
|
|
resp, err := c.http.PostForm(
|
|
c.Endpoint,
|
|
url.Values{"token": {c.Token}, "url": {ou.String()}, "ttl": {ttl.String()}})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer resp.Body.Close()
|
|
b, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
r := struct {
|
|
Error string `json:"error"`
|
|
ShortURL string `json:"short-url"`
|
|
}{}
|
|
if err := json.Unmarshal(b, &r); err != nil {
|
|
return nil, err
|
|
}
|
|
if r.Error != "" {
|
|
return nil, errors.New(r.Error)
|
|
}
|
|
u, err := url.Parse(r.ShortURL)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &ShortURL{URL: *u, Original: ou}, nil
|
|
}
|