updated

Author: wpcodevo

config/default.go

@@ -25,6 +25,10 @@ type Config struct {
 	GoogleClientID         string `mapstructure:"GOOGLE_OAUTH_CLIENT_ID"`
 	GoogleClientSecret     string `mapstructure:"GOOGLE_OAUTH_CLIENT_SECRET"`
 	GoogleOAuthRedirectUrl string `mapstructure:"GOOGLE_OAUTH_REDIRECT_URL"`
+
+	GitHubClientID         string `mapstructure:"GITHUB_OAUTH_CLIENT_ID"`
+	GitHubClientSecret     string `mapstructure:"GITHUB_OAUTH_CLIENT_SECRET"`
+	GitHubOAuthRedirectUrl string `mapstructure:"GITHUB_OAUTH_REDIRECT_URL"`
 }
 
 func LoadConfig(path string) (config Config, err error) {

controllers/auth.controller.go

@@ -193,3 +193,68 @@ func (ac *AuthController) GoogleOAuth(ctx *gin.Context) {
 
 	ctx.Redirect(http.StatusTemporaryRedirect, fmt.Sprint(config.ClientOrigin, pathUrl))
 }
+
+func (ac *AuthController) GitHubOAuth(ctx *gin.Context) {
+	code := ctx.Query("code")
+	var pathUrl string = "/"
+
+	if ctx.Query("state") != "" {
+		pathUrl = ctx.Query("state")
+	}
+
+	if code == "" {
+		ctx.JSON(http.StatusUnauthorized, gin.H{"status": "fail", "message": "Authorization code not provided!"})
+		return
+	}
+
+	// Use the code to get the id and access tokens
+	tokenRes, err := utils.GetGitHubOauthToken(code)
+
+	if err != nil {
+		ctx.JSON(http.StatusBadGateway, gin.H{"status": "fail", "message": err.Error()})
+	}
+
+	user, err := utils.GetGitHubUser(tokenRes.Access_token)
+
+	if err != nil {
+		ctx.JSON(http.StatusBadGateway, gin.H{"status": "fail", "message": err.Error()})
+	}
+
+	createdAt := time.Now()
+	resBody := &models.UpdateDBUser{
+		Email:     user.Email,
+		Name:      user.Name,
+		Photo:     user.Photo,
+		Provider:  "github",
+		Role:      "user",
+		Verified:  true,
+		CreatedAt: createdAt,
+		UpdatedAt: createdAt,
+	}
+
+	updatedUser, err := ac.userService.UpsertUser(user.Email, resBody)
+	if err != nil {
+		ctx.JSON(http.StatusBadGateway, gin.H{"status": "fail", "message": err.Error()})
+	}
+
+	config, _ := config.LoadConfig(".")
+
+	// Generate Tokens
+	access_token, err := utils.CreateToken(config.AccessTokenExpiresIn, updatedUser.ID.Hex(), config.AccessTokenPrivateKey)
+	if err != nil {
+		ctx.JSON(http.StatusBadRequest, gin.H{"status": "fail", "message": err.Error()})
+		return
+	}
+
+	refresh_token, err := utils.CreateToken(config.RefreshTokenExpiresIn, updatedUser.ID.Hex(), config.RefreshTokenPrivateKey)
+	if err != nil {
+		ctx.JSON(http.StatusBadRequest, gin.H{"status": "fail", "message": err.Error()})
+		return
+	}
+
+	ctx.SetCookie("access_token", access_token, config.AccessTokenMaxAge*60, "/", "localhost", false, true)
+	ctx.SetCookie("refresh_token", refresh_token, config.RefreshTokenMaxAge*60, "/", "localhost", false, true)
+	ctx.SetCookie("logged_in", "true", config.AccessTokenMaxAge*60, "/", "localhost", false, false)
+
+	ctx.Redirect(http.StatusTemporaryRedirect, fmt.Sprint(config.ClientOrigin, pathUrl))
+}

main.go

@@ -118,5 +118,10 @@ func main() {
 	AuthRouteController.AuthRoute(router)
 	UserRouteController.UserRoute(router, userService)
 	SessionRouteController.SessionRoute(router)
+
+	server.NoRoute(func(ctx *gin.Context) {
+		ctx.JSON(http.StatusNotFound, gin.H{"status": "error", "message": "Route Not Found"})
+	})
+
 	log.Fatal(server.Run(":" + config.Port))
 }

routes/session.routes.go

@@ -17,4 +17,5 @@ func (rc *SessionRouteController) SessionRoute(rg *gin.RouterGroup) {
 	router := rg.Group("/sessions/oauth")
 
 	router.GET("/google", rc.authController.GoogleOAuth)
+	router.GET("/github", rc.authController.GitHubOAuth)
 }

utils/githubOAuth.go

@@ -0,0 +1,115 @@
+package utils
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"time"
+
+	"github.com/wpcodevo/golang-mongodb/config"
+)
+
+type GitHubOauthToken struct {
+	Access_token string
+}
+
+type GitHubUserResult struct {
+	Name  string
+	Photo string
+	Email string
+}
+
+func GetGitHubOauthToken(code string) (*GitHubOauthToken, error) {
+	const rootURl = "https://github.com/login/oauth/access_token"
+
+	config, _ := config.LoadConfig(".")
+	values := url.Values{}
+	values.Add("code", code)
+	values.Add("client_id", config.GitHubClientID)
+	values.Add("client_secret", config.GitHubClientSecret)
+
+	query := values.Encode()
+
+	queryString := fmt.Sprintf("%s?%s", rootURl, bytes.NewBufferString(query))
+	req, err := http.NewRequest("POST", queryString, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
+	client := http.Client{
+		Timeout: time.Second * 30,
+	}
+
+	res, err := client.Do(req)
+	if err != nil {
+		return nil, err
+	}
+
+	if res.StatusCode != http.StatusOK {
+		return nil, errors.New("could not retrieve token")
+	}
+
+	resBody, err := ioutil.ReadAll(res.Body)
+	if err != nil {
+		return nil, err
+	}
+
+	parsedQuery, err := url.ParseQuery(string(resBody))
+	if err != nil {
+		return nil, err
+	}
+
+	tokenBody := &GitHubOauthToken{
+		Access_token: parsedQuery["access_token"][0],
+	}
+
+	return tokenBody, nil
+}
+
+func GetGitHubUser(access_token string) (*GitHubUserResult, error) {
+	rootUrl := "https://api.github.com/user"
+
+	req, err := http.NewRequest("GET", rootUrl, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", access_token))
+
+	client := http.Client{
+		Timeout: time.Second * 30,
+	}
+
+	res, err := client.Do(req)
+	if err != nil {
+		return nil, err
+	}
+
+	if res.StatusCode != http.StatusOK {
+		return nil, errors.New("could not retrieve user")
+	}
+
+	resBody, err := ioutil.ReadAll(res.Body)
+	if err != nil {
+		return nil, err
+	}
+
+	var GitHubUserRes map[string]interface{}
+
+	if err := json.Unmarshal(resBody, &GitHubUserRes); err != nil {
+		return nil, err
+	}
+
+	userBody := &GitHubUserResult{
+		Email: GitHubUserRes["email"].(string),
+		Name:  GitHubUserRes["login"].(string),
+		Photo: GitHubUserRes["avatar_url"].(string),
+	}
+
+	return userBody, nil
+}