// Copyright 2017 Drone.IO Inc. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package main
import (
"flag"
"fmt"
"log"
"net/http"
"os"
"git.awesome-for.me/liuzhiguo/go-login/login"
"git.awesome-for.me/liuzhiguo/go-login/login/bitbucket"
"git.awesome-for.me/liuzhiguo/go-login/login/gitee"
"git.awesome-for.me/liuzhiguo/go-login/login/github"
"git.awesome-for.me/liuzhiguo/go-login/login/gitlab"
"git.awesome-for.me/liuzhiguo/go-login/login/gogs"
"git.awesome-for.me/liuzhiguo/go-login/login/logger"
"git.awesome-for.me/liuzhiguo/go-login/login/stash"
)
var (
provider = flag.String("provider", "github", "")
providerURL = flag.String("provider-url", "", "")
clientID = flag.String("client-id", "", "")
clientSecret = flag.String("client-secret", "", "")
consumerKey = flag.String("consumer-key", "", "")
consumerRsa = flag.String("consumer-private-key", "", "")
redirectURL = flag.String("redirect-url", "http://localhost:8080/login", "")
address = flag.String("address", ":8080", "")
dump = flag.Bool("dump", false, "")
help = flag.Bool("help", false, "")
)
func main() {
flag.Usage = usage
flag.Parse()
if *help {
flag.Usage()
os.Exit(0)
}
dumper := logger.DiscardDumper()
if *dump {
dumper = logger.StandardDumper()
}
var middleware login.Middleware
switch *provider {
case "gogs", "gitea":
middleware = &gogs.Config{
Login: "/login/form",
Server: *providerURL,
}
case "gitlab":
middleware = &gitlab.Config{
ClientID: *clientID,
ClientSecret: *clientSecret,
RedirectURL: *redirectURL,
Scope: []string{"read_user", "api"},
}
case "gitee":
middleware = &gitee.Config{
ClientID: *clientID,
ClientSecret: *clientSecret,
RedirectURL: *redirectURL,
Scope: []string{"user_info", "projects", "pull_requests", "hook"},
}
case "github":
middleware = &github.Config{
ClientID: *clientID,
ClientSecret: *clientSecret,
Server: *providerURL,
Scope: []string{"repo", "user", "read:org"},
Dumper: dumper,
}
case "bitbucket":
middleware = &bitbucket.Config{
ClientID: *clientID,
ClientSecret: *clientSecret,
RedirectURL: *redirectURL,
}
case "stash":
privateKey, err := stash.ParsePrivateKeyFile(*consumerRsa)
if err != nil {
log.Fatalf("Cannot parse Private Key. %s", err)
}
middleware = &stash.Config{
Address: *providerURL,
CallbackURL: *redirectURL,
ConsumerKey: *consumerKey,
PrivateKey: privateKey,
}
}
log.Printf("Staring server at %s", *address)
// handles the authorization flow and displays the
// authorization results at completion.
http.Handle("/login/form", http.HandlerFunc(form))
http.Handle("/login", middleware.Handler(
http.HandlerFunc(details),
))
// redirects the user to the login handler.
http.Handle("/", http.RedirectHandler("/login", http.StatusSeeOther))
http.ListenAndServe(*address, nil)
}
// returns the login credentials.
func details(w http.ResponseWriter, r *http.Request) {
ctx := r.Context()
err := login.ErrorFrom(ctx)
if err != nil {
fmt.Fprintf(w, failure, err)
return
}
token := login.TokenFrom(ctx)
fmt.Fprintf(w, success, token.Access, token.Refresh)
}
// display the login form.
func form(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "text/html")
fmt.Fprint(w, loginForm)
}
// html page displayed to collect credentials.
var loginForm = `
`
// html page displayed on success.
var success = `
Access Token
%s
Refresh / Secret Token
%s
`
// html page displayed on failure.
var failure = `
Error
%s
`
func usage() {
fmt.Println(`Usage: go run main.go [OPTION]...
--provider provider (github, gitlab, gogs, gitea, bitbucket)
--provider-url provider url (gitea, gogs, stash only)
--client-id oauth2 client id
--client-secret oauth2 client secret
--consumer-key oauth1 consumer key
--consumer-private-key oauth1 consumer rsa private key file
--redirect-url oauth redirect url
--address http server address (:8080)
--help display this help and exit`)
}