updated hubstats to accept repo for pull count

Signed-off-by: Burton Rheutan <rheutan7@gmail.com>
This commit is contained in:
Burton Rheutan 2018-03-04 20:06:44 -06:00 committed by Alex Ellis
parent d1fb0084b4
commit 26c8671865

View File

@ -10,18 +10,22 @@ import (
"strings" "strings"
) )
type dockerHubStatsType struct { type dockerHubOrgStatsType struct {
Count int `json:"count"` Count int `json:"count"`
} }
type dockerHubRepoStatsType struct {
PullCount int `json:"pull_count"`
}
func sanitizeInput(input string) string { func sanitizeInput(input string) string {
parts := strings.Split(input, "\n") parts := strings.Split(input, "\n")
return strings.Trim(parts[0], " ") return strings.Trim(parts[0], " ")
} }
func requestHubStats(org string) dockerHubStatsType { func requestStats(repo string) []byte {
client := http.Client{} client := http.Client{}
res, err := client.Get("https://hub.docker.com/v2/repositories/" + org) res, err := client.Get("https://hub.docker.com/v2/repositories/" + repo)
if err != nil { if err != nil {
log.Fatalln("Unable to reach Docker Hub server.") log.Fatalln("Unable to reach Docker Hub server.")
} }
@ -31,9 +35,19 @@ func requestHubStats(org string) dockerHubStatsType {
log.Fatalln("Unable to parse response from server.") log.Fatalln("Unable to parse response from server.")
} }
dockerHubStats := dockerHubStatsType{} return body
json.Unmarshal(body, &dockerHubStats) }
return dockerHubStats
func parseOrgStats(response []byte) dockerHubOrgStatsType {
dockerHubOrgStats := dockerHubOrgStatsType{}
json.Unmarshal(response, &dockerHubOrgStats)
return dockerHubOrgStats
}
func parseRepoStats(response []byte) dockerHubRepoStatsType {
dockerHubRepoStats := dockerHubRepoStatsType{}
json.Unmarshal(response, &dockerHubRepoStats)
return dockerHubRepoStats
} }
func main() { func main() {
@ -41,13 +55,19 @@ func main() {
if err != nil { if err != nil {
log.Fatal("Unable to read standard input:", err) log.Fatal("Unable to read standard input:", err)
} }
org := string(input) request := string(input)
if len(input) == 0 { if len(input) == 0 {
log.Fatalln("A username or organisation is required.") log.Fatalln("A username/organisation or repository is required.")
} }
org = sanitizeInput(org) request = sanitizeInput(request)
dockerHubStats := requestHubStats(org) response := requestStats(request)
fmt.Printf("The organisation or user %s has %d repositories on the Docker hub.\n", org, dockerHubStats.Count) if strings.Contains(request, "/") {
dockerHubRepoStats := parseRepoStats(response)
fmt.Printf("Repo: %s has been pulled %d times from the Docker Hub", request, dockerHubRepoStats.PullCount)
} else {
dockerHubOrgStats := parseOrgStats(response)
fmt.Printf("The organisation or user %s has %d repositories on the Docker hub.\n", request, dockerHubOrgStats.Count)
}
} }