diff options
Diffstat (limited to 'tools')
| -rw-r--r-- | tools/notes/notes.go | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tools/notes/notes.go b/tools/notes/notes.go index 9f190f6..7714a89 100644 --- a/tools/notes/notes.go +++ b/tools/notes/notes.go @@ -10,6 +10,7 @@ import ( "io" "net/http" "os" + "path/filepath" "time" "github.com/google/subcommands" @@ -213,6 +214,7 @@ func (*newCommand) SetFlags(*flag.FlagSet) {} type createNoteRequestStreamWriter struct { ctx context.Context + fileName string continuationToken []byte } @@ -225,6 +227,7 @@ func (w *createNoteRequestStreamWriter) Write(buf []byte) (int, error) { for i := 0; i < 3; i++ { resp, err = createNote(w.ctx, &api.CreateNoteRequest{ ContinuationToken: w.continuationToken, + FileName: w.fileName, Chunk: buf, More: true, }) @@ -242,6 +245,7 @@ func (w *createNoteRequestStreamWriter) Write(buf []byte) (int, error) { if err != nil { return 0, err } + w.fileName = "" w.continuationToken = resp.ContinuationToken return len(buf), nil } @@ -252,7 +256,7 @@ func (*newCommand) new(ctx context.Context, fileName string) error { return err } defer f.Close() - streamWriter := &createNoteRequestStreamWriter{ctx: ctx} + streamWriter := &createNoteRequestStreamWriter{ctx: ctx, fileName: filepath.Base(fileName)} if _, err := io.Copy(streamWriter, f); err != nil { return err } |