Created
April 10, 2022 07:47
-
-
Save u-r-w/028109300fb04772bcc05445e0f3f88f to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package main | |
import ( | |
"encoding/json" | |
"fmt" | |
"io/ioutil" | |
"log" | |
"net/http" | |
"os" | |
"golang.org/x/net/context" | |
"golang.org/x/oauth2" | |
"golang.org/x/oauth2/google" | |
"google.golang.org/api/sheets/v4" | |
) | |
// Retrieve a token, saves the token, then returns the generated client. | |
func getClient(config *oauth2.Config) *http.Client { | |
// The file token.json stores the user's access and refresh tokens, and is | |
// created automatically when the authorization flow completes for the first | |
// time. | |
tokFile := "token.json" | |
tok, err := tokenFromFile(tokFile) | |
if err != nil { | |
tok = getTokenFromWeb(config) | |
saveToken(tokFile, tok) | |
} | |
return config.Client(context.Background(), tok) | |
} | |
// Request a token from the web, then returns the retrieved token. | |
func getTokenFromWeb(config *oauth2.Config) *oauth2.Token { | |
authURL := config.AuthCodeURL("state-token", oauth2.AccessTypeOffline) | |
fmt.Printf("Go to the following link in your browser then type the "+ | |
"authorization code: \n%v\n", authURL) | |
var authCode string | |
if _, err := fmt.Scan(&authCode); err != nil { | |
log.Fatalf("Unable to read authorization code: %v", err) | |
} | |
tok, err := config.Exchange(context.TODO(), authCode) | |
if err != nil { | |
log.Fatalf("Unable to retrieve token from web: %v", err) | |
} | |
return tok | |
} | |
// Retrieves a token from a local file. | |
func tokenFromFile(file string) (*oauth2.Token, error) { | |
f, err := os.Open(file) | |
if err != nil { | |
return nil, err | |
} | |
defer f.Close() | |
tok := &oauth2.Token{} | |
err = json.NewDecoder(f).Decode(tok) | |
return tok, err | |
} | |
// Saves a token to a file path. | |
func saveToken(path string, token *oauth2.Token) { | |
fmt.Printf("Saving credential file to: %s\n", path) | |
f, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0600) | |
if err != nil { | |
log.Fatalf("Unable to cache oauth token: %v", err) | |
} | |
defer f.Close() | |
json.NewEncoder(f).Encode(token) | |
} | |
// Create a struct to store our rows | |
// type Entry struct { | |
// ID int `json:"id"` | |
// Title string `json:"title"` | |
// Date string `json:"date"` | |
// } | |
func main() { | |
// I had to use a Service Account to build my app | |
b, err := ioutil.ReadFile("<CREDENTIALS FROM GCP GSHEET SERVICE ACCOUNT.json>") | |
if err != nil { | |
log.Fatalf("Unable to read client secret file: %v", err) | |
} | |
// If modifying these scopes, delete your previously saved token.json. | |
config, err := google.JWTConfigFromJSON(b, "https://www.googleapis.com/auth/spreadsheets") | |
if err != nil { | |
log.Fatalf("Unable to parse client secret file to config: %v", err) | |
} | |
client := config.Client(oauth2.NoContext) | |
spreadsheetId := "<ID HERE>" | |
//readRange := "Sheet1!A2:B5" | |
srv, err := sheets.New(client) | |
if err != nil { | |
log.Fatalf("Unable to retrieve Sheets client: %v", err) | |
} | |
// Modify this to your Needs | |
rangeData := "sheet1!A1:E" | |
rb := &sheets.BatchUpdateValuesRequest{ | |
ValueInputOption: "USER_ENTERED", | |
} | |
rows := make([][]interface{},0) | |
rows = append(rows,[]interface{}{"Last Updated","Title","Article"}) | |
rb.Data = append(rb.Data, &sheets.ValueRange{ | |
Range: rangeData, | |
Values: rows, | |
}) | |
// Do a batch update at once | |
_, err = srv.Spreadsheets.Values.BatchUpdate(spreadsheetId, rb).Do() | |
if err != nil { | |
log.Fatal(err) | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment