Allow downloading without gob

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
 	}