You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
2.0 KiB
100 lines
2.0 KiB
package pypihub
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/google/go-github/github"
|
|
)
|
|
|
|
type Client struct {
|
|
config Config
|
|
client *github.Client
|
|
repos []string
|
|
}
|
|
|
|
func NewClient(cfg Config) *Client {
|
|
var t = github.BasicAuthTransport{
|
|
Username: cfg.Username,
|
|
Password: cfg.AccessToken,
|
|
}
|
|
return &Client{
|
|
config: cfg,
|
|
client: github.NewClient(t.Client()),
|
|
repos: cfg.RepoNames,
|
|
}
|
|
}
|
|
|
|
func (c *Client) splitRepoName(r string) (string, string) {
|
|
var p = strings.SplitN(r, "/", 2)
|
|
switch len(p) {
|
|
case 2:
|
|
return p[0], p[1]
|
|
case 1:
|
|
return c.config.Username, p[0]
|
|
default:
|
|
return "", ""
|
|
}
|
|
}
|
|
|
|
func (c *Client) GetRepoAssets(r string) ([]Asset, error) {
|
|
var owner, repo string
|
|
owner, repo = c.splitRepoName(r)
|
|
|
|
var releases []*github.RepositoryRelease
|
|
var err error
|
|
releases, _, err = c.client.Repositories.ListReleases(owner, repo, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var allAssets = make([]Asset, 0)
|
|
for _, rel := range releases {
|
|
var assets []*github.ReleaseAsset
|
|
assets, _, err = c.client.Repositories.ListReleaseAssets(owner, repo, *rel.ID, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, a := range assets {
|
|
allAssets = append(allAssets, Asset{
|
|
ID: *a.ID,
|
|
Name: *a.Name,
|
|
Owner: owner,
|
|
Repo: repo,
|
|
})
|
|
}
|
|
}
|
|
return allAssets, nil
|
|
}
|
|
|
|
func (c *Client) GetAllAssets() ([]Asset, error) {
|
|
var allAssets = make([]Asset, 0)
|
|
for _, r := range c.config.RepoNames {
|
|
var repoAssets = make([]Asset, 0)
|
|
var err error
|
|
repoAssets, err = c.GetRepoAssets(r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
allAssets = append(allAssets, repoAssets...)
|
|
}
|
|
|
|
return allAssets, nil
|
|
}
|
|
|
|
func (c *Client) DownloadAsset(a Asset) (io.ReadCloser, error) {
|
|
var rc io.ReadCloser
|
|
var redirect string
|
|
var err error
|
|
rc, redirect, err = c.client.Repositories.DownloadReleaseAsset(a.Owner, a.Repo, a.ID)
|
|
if rc == nil {
|
|
var resp *http.Response
|
|
resp, err = http.Get(redirect)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
rc = resp.Body
|
|
}
|
|
return rc, err
|
|
}
|