Skip to content

Commit

Permalink
Go fmt the code and use proper imports
Browse files Browse the repository at this point in the history
This allows people to be able to go get your project.
  • Loading branch information
crosbymichael committed Mar 15, 2014
1 parent d9b1ad5 commit 019930d
Show file tree
Hide file tree
Showing 7 changed files with 670 additions and 672 deletions.
96 changes: 48 additions & 48 deletions auth/auth.go
Original file line number Diff line number Diff line change
@@ -1,68 +1,68 @@
package auth

import (
"net/http"
"fmt"
"code.google.com/p/goauth2/oauth"
"../util"
"code.google.com/p/goauth2/oauth"
"fmt"
"github.com/prasmussen/gdrive/util"
"net/http"
)

// Get auth code from user
func promptUserForAuthCode(config *oauth.Config) string {
authUrl := config.AuthCodeURL("state")
fmt.Println("Go to the following link in your browser:")
fmt.Printf("%v\n\n", authUrl)
return util.Prompt("Enter verification code: ")
authUrl := config.AuthCodeURL("state")
fmt.Println("Go to the following link in your browser:")
fmt.Printf("%v\n\n", authUrl)
return util.Prompt("Enter verification code: ")
}

// Returns true if we have a valid cached token
func hasValidToken(cacheFile oauth.CacheFile, transport *oauth.Transport) bool {
// Check if we have a cached token
token, err := cacheFile.Token()
if err != nil {
return false
}
// Check if we have a cached token
token, err := cacheFile.Token()
if err != nil {
return false
}

// Refresh token if its expired
if token.Expired() {
transport.Token = token
err = transport.Refresh()
if err != nil {
fmt.Println(err)
return false
}
}
return true
// Refresh token if its expired
if token.Expired() {
transport.Token = token
err = transport.Refresh()
if err != nil {
fmt.Println(err)
return false
}
}
return true
}

func GetOauth2Client(clientId, clientSecret, cachePath string) (*http.Client, error) {
cacheFile := oauth.CacheFile(cachePath)
cacheFile := oauth.CacheFile(cachePath)

config := &oauth.Config{
ClientId: clientId,
ClientSecret: clientSecret,
Scope: "https://www.googleapis.com/auth/drive",
RedirectURL: "urn:ietf:wg:oauth:2.0:oob",
AuthURL: "https://accounts.google.com/o/oauth2/auth",
TokenURL: "https://accounts.google.com/o/oauth2/token",
TokenCache: cacheFile,
}
config := &oauth.Config{
ClientId: clientId,
ClientSecret: clientSecret,
Scope: "https://www.googleapis.com/auth/drive",
RedirectURL: "urn:ietf:wg:oauth:2.0:oob",
AuthURL: "https://accounts.google.com/o/oauth2/auth",
TokenURL: "https://accounts.google.com/o/oauth2/token",
TokenCache: cacheFile,
}

transport := &oauth.Transport{
Config: config,
Transport: http.DefaultTransport,
}
transport := &oauth.Transport{
Config: config,
Transport: http.DefaultTransport,
}

// Return client if we have a valid token
if hasValidToken(cacheFile, transport) {
return transport.Client(), nil
}
// Return client if we have a valid token
if hasValidToken(cacheFile, transport) {
return transport.Client(), nil
}

// Get auth code from user and request a new token
code := promptUserForAuthCode(config)
_, err := transport.Exchange(code)
if err != nil {
return nil, err
}
return transport.Client(), nil
// Get auth code from user and request a new token
code := promptUserForAuthCode(config)
_, err := transport.Exchange(code)
if err != nil {
return nil, err
}
return transport.Client(), nil
}
Loading

0 comments on commit 019930d

Please sign in to comment.