Simplify drive wrapper, s/client/auth/

AnggaR96s · Jan 18, 2016 · 4f4152c · 4f4152c
1 parent e60833f
commit 4f4152c
Show file tree

Hide file tree

Showing 6 changed files with 13 additions and 45 deletions.
diff --git a/client/auth.go → auth/oauth.go b/client/auth.go → auth/oauth.go
@@ -1,4 +1,4 @@
-package client
+package auth
 
 import (
     "net/http"

diff --git a/client/token.go → auth/token.go b/client/token.go → auth/token.go
@@ -1,4 +1,4 @@
-package client
+package auth
 
 import (
     "golang.org/x/oauth2"

diff --git a/client/util.go → auth/util.go b/client/util.go → auth/util.go
@@ -1,4 +1,4 @@
-package client
+package auth
 
 import (
     "os"

diff --git a/client/client.go b/client/client.go
diff --git a/drive/drive.go b/drive/drive.go
@@ -5,19 +5,15 @@ import (
     "google.golang.org/api/drive/v3"
 )
 
-type Client interface {
-    Service() *drive.Service
-    Http() *http.Client
-}
-
 type Drive struct {
     service *drive.Service
-    http *http.Client
 }
 
-func NewDrive(client Client) *Drive {
-    return &Drive{
-        service: client.Service(),
-        http: client.Http(),
+func New(client *http.Client) (*Drive, error) {
+    service, err := drive.New(client)
+    if err != nil {
+        return nil, err
     }
+
+    return &Drive{service}, nil
 }
diff --git a/handlers.go b/handlers.go
@@ -5,7 +5,7 @@ import (
 	"os"
 	"strings"
     "./cli"
-	"./client"
+	"./auth"
 	"./drive"
 )
 
@@ -155,17 +155,17 @@ func printCommandHelp(ctx cli.Context) {
 func newDrive(args cli.Arguments) *drive.Drive {
     configDir := args.String("configDir")
     tokenPath := ConfigFilePath(configDir, TokenFilename)
-    oauth, err := client.NewOauthClient(ClientId, ClientSecret, tokenPath, authCodePrompt)
+    oauth, err := auth.NewOauthClient(ClientId, ClientSecret, tokenPath, authCodePrompt)
     if err != nil {
         ExitF("Failed getting oauth client: %s", err.Error())
     }
 
-    client, err := client.NewClient(oauth)
+    client, err := drive.New(oauth)
     if err != nil {
         ExitF("Failed getting drive: %s", err.Error())
     }
 
-    return drive.NewDrive(client)
+    return client
 }
 
 func authCodePrompt(url string) func() string {