handlers.go

package main

import (
	"context"
	"fmt"
	"net/http"
	"time"

	"github.com/google/go-github/v48/github"
	"go.mongodb.org/mongo-driver/mongo"
)

// Badge with the number of your GitHub Organizations
func organizationsHandler(client github.Client) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		if r.URL.Path != "/organizations" {
			http.Error(w, "404 not found", http.StatusNotFound)
		}
		if r.Method != "GET" {
			http.Error(w, "Method is not supported", http.StatusNotFound)
		}

		values := r.URL.Query()
		// Get GitHub username, i.e. /organizations?username=dtemir
		username := values.Get("username")

		// Get shields.io color, default to brightgreen
		color := values.Get("color")
		if len(color) == 0 {
			color = "brigthgreen"
		}

		// Get shields.io style, default to empty string (shield.io's flat style)
		style := values.Get("style")

		// Get logo, default to none
		logo := values.Get("logo")

		// Fetch the number of Organizations the user has
		// docs.github.com/en/rest/orgs/orgs?apiVersion=2022-11-28#list-organizations-for-a-user
		orgs, _, err := client.Organizations.List(context.Background(), username, nil)
		if err != nil {
			WarningLogger.Println("GitHub: Error fetching organization list for user", username, err)
			http.Error(w, "Incorrect GitHub username. Please make sure it's correct and try again", http.StatusBadRequest)
		}

		// Create a shields.io badge
		url := fmt.Sprintf("https://img.shields.io/badge/Organizations-%d-%s?style=%s&logo=%s", len(orgs), color, style, logo)

		svg := getSVG(url)
		InfoLogger.Println("Successfully created a badge for", username, url)

		// Display SVG badge
		w.Header().Set("content-type", "image/svg+xml")
		w.Header().Set("cache-control", "no-cache")
		w.WriteHeader(200)
		fmt.Fprintf(w, "%s", svg)
	}
}

// Badge with the number of years you have been a GitHub member
func yearsHandler(client github.Client) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		if r.URL.Path != "/years" {
			http.Error(w, "404 not found", http.StatusNotFound)
		}
		if r.Method != "GET" {
			http.Error(w, "Method is not supported", http.StatusNotFound)
		}

		values := r.URL.Query()
		username := values.Get("username")

		color := values.Get("color")
		if len(color) == 0 {
			color = "brigthgreen"
		}

		style := values.Get("style")

		logo := values.Get("logo")

		// Fetch the user data
		// https://docs.github.com/en/rest/users/users?apiVersion=2022-11-28#get-a-user
		user, _, err := client.Users.Get(context.Background(), username)
		if err != nil {
			WarningLogger.Println("GitHub: Error fetching years for user", username, err)
			http.Error(w, "Incorrect GitHub username. Please make sure it's correct and try again", http.StatusBadRequest)
		}

		// Calculate the number of years passed since user creation
		created := user.GetCreatedAt().Time
		now := time.Now()
		years := int64(now.Sub(created).Hours() / 24 / 365)

		url := fmt.Sprintf("https://img.shields.io/badge/Years-%d-%s?style=%s&logo=%s", years, color, style, logo)

		svg := getSVG(url)
		InfoLogger.Println("Successfully created a badge for", username, url)

		w.Header().Set("content-type", "image/svg+xml")
		w.Header().Set("cache-control", "no-cache")
		w.WriteHeader(200)
		fmt.Fprintf(w, "%s", svg)
	}
}

// Badge with the number of public repos you have
func reposHandler(client github.Client) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		if r.URL.Path != "/repos" {
			http.Error(w, "404 not found", http.StatusNotFound)
		}
		if r.Method != "GET" {
			http.Error(w, "Method is not supported", http.StatusNotFound)
		}

		values := r.URL.Query()
		username := values.Get("username")

		color := values.Get("color")
		if len(color) == 0 {
			color = "brigthgreen"
		}

		style := values.Get("style")

		logo := values.Get("logo")

		// Fetch the user data
		// https://docs.github.com/en/rest/users/users?apiVersion=2022-11-28#get-a-user
		user, _, err := client.Users.Get(context.Background(), username)
		if err != nil {
			WarningLogger.Println("GitHub: Error fetching repos for user", username, err)
			http.Error(w, "Incorrect GitHub username. Please make sure it's correct and try again", http.StatusBadRequest)
		}

		// Get the number of public repos the user has
		repos := user.GetPublicRepos()

		url := fmt.Sprintf("https://img.shields.io/badge/Repos-%d-%s?style=%s&logo=%s", repos, color, style, logo)

		svg := getSVG(url)
		InfoLogger.Println("Successfully created a badge for", username, url)

		w.Header().Set("content-type", "image/svg+xml")
		w.Header().Set("cache-control", "no-cache")
		w.WriteHeader(200)
		fmt.Fprintf(w, "%s", svg)

	}
}

// Badge with the number of visits your repo had so far
func visitsHandler(client github.Client, collection mongo.Collection) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		if r.URL.Path != "/visits" {
			http.Error(w, "404 not found", http.StatusNotFound)
		}
		if r.Method != "GET" {
			http.Error(w, "Method is not supported", http.StatusNotFound)
		}

		values := r.URL.Query()
		username := values.Get("username")

		color := values.Get("color")
		if len(color) == 0 {
			color = "brigthgreen"
		}

		style := values.Get("style")

		logo := values.Get("logo")

		// Get a repo name, i.e. dtemir/github-badges-golang
		repo := username + "/" + values.Get("repo")

		visits := getVisitsCount(repo, collection)

		url := fmt.Sprintf("https://img.shields.io/badge/Visits-%d-%s?style=%s&logo=%s", visits, color, style, logo)

		svg := getSVG(url)
		InfoLogger.Println("Successfully created a badge for", username, url)

		w.Header().Set("content-type", "image/svg+xml")
		w.Header().Set("cache-control", "no-cache")
		w.WriteHeader(200)
		fmt.Fprintf(w, "%s", svg)
	}
}

// Show a message if people land at "/" instead of any of the other endpoints
func indexHandler(w http.ResponseWriter, r *http.Request) {
	http.Redirect(w, r, "https://github.com/dtemir/git-badges-golang", http.StatusMovedPermanently)
}