89 lines
2.5 KiB
Go
89 lines
2.5 KiB
Go
|
// Command createtabs creates the directory sheet's local-layer tabs with their header rows.
|
||
|
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"flag"
|
||
|
|
"log"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"heliosian/internal/data"
|
||
|
|
"google.golang.org/api/option"
|
||
|
|
"google.golang.org/api/sheets/v4"
|
||
|
|
)
|
||
|
|
|
||
|
|
var tabs = []struct {
|
||
|
|
title string
|
||
|
|
header []string
|
||
|
|
}{
|
||
|
|
{"Name to Email", []string{"Name", "Email"}},
|
||
|
|
{"Overrides", []string{
|
||
|
|
"Email", "Added",
|
||
|
|
"Full Name", "Legal Name", "Preferred Name",
|
||
|
|
"Is Student", "Is Parent", "Is Staff",
|
||
|
|
"New to Helios", "Pronouns", "Facts",
|
||
|
|
"Grade", "Classroom", "Crew",
|
||
|
|
"Phone", "Job Title", "Department", "Grade Band", "Room Parent",
|
||
|
|
"Address", "Family Phone", "Family Photo Caption",
|
||
|
|
}},
|
||
|
|
{"Change Log", []string{
|
||
|
|
"Timestamp", "Actor",
|
||
|
|
"Email", "Added",
|
||
|
|
"Full Name", "Legal Name", "Preferred Name",
|
||
|
|
"Is Student", "Is Parent", "Is Staff",
|
||
|
|
"New to Helios", "Pronouns", "Facts",
|
||
|
|
"Grade", "Classroom", "Crew",
|
||
|
|
"Phone", "Job Title", "Department", "Grade Band", "Room Parent",
|
||
|
|
"Address", "Family Phone", "Family Photo Caption",
|
||
|
|
}},
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
sheet := flag.String("sheet", "", "spreadsheet id")
|
||
|
|
flag.Parse()
|
||
|
|
if *sheet == "" {
|
||
|
|
log.Fatal("[ERROR] -sheet <spreadsheet id> is required")
|
||
|
|
}
|
||
|
|
svc, err := sheets.NewService(context.Background(),
|
||
|
|
option.WithCredentialsFile(data.KeyFile),
|
||
|
|
option.WithScopes(sheets.SpreadsheetsScope))
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("[ERROR] create sheets client: %v", err)
|
||
|
|
}
|
||
|
|
meta, err := svc.Spreadsheets.Get(*sheet).Fields("sheets(properties(title))").Do()
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("[ERROR] get spreadsheet: %v", err)
|
||
|
|
}
|
||
|
|
existing := map[string]bool{}
|
||
|
|
for _, s := range meta.Sheets {
|
||
|
|
existing[s.Properties.Title] = true
|
||
|
|
}
|
||
|
|
for _, t := range tabs {
|
||
|
|
if existing[t.title] {
|
||
|
|
log.Fatalf("[ERROR] tab %q already exists", t.title)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
for _, t := range tabs {
|
||
|
|
_, err := svc.Spreadsheets.BatchUpdate(*sheet, &sheets.BatchUpdateSpreadsheetRequest{
|
||
|
|
Requests: []*sheets.Request{{AddSheet: &sheets.AddSheetRequest{
|
||
|
|
Properties: &sheets.SheetProperties{Title: t.title},
|
||
|
|
}}},
|
||
|
|
}).Do()
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("[ERROR] create tab %q: %v", t.title, err)
|
||
|
|
}
|
||
|
|
values := make([]interface{}, len(t.header))
|
||
|
|
for i, h := range t.header {
|
||
|
|
values[i] = h
|
||
|
|
}
|
||
|
|
quoted := "'" + strings.ReplaceAll(t.title, "'", "''") + "'"
|
||
|
|
_, err = svc.Spreadsheets.Values.Update(*sheet, quoted+"!1:1", &sheets.ValueRange{
|
||
|
|
Values: [][]interface{}{values},
|
||
|
|
}).ValueInputOption("RAW").Do()
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("[ERROR] write header of %q: %v", t.title, err)
|
||
|
|
}
|
||
|
|
log.Printf("created tab %q with %d columns", t.title, len(t.header))
|
||
|
|
}
|
||
|
|
}
|