Update from https://github.com/cloudimpl/next-coder/commit/adfc1daaa95b3038b47c92e42614d8b5f8787fda

Author: CloudimplBot

polycode/client.go

@@ -150,6 +150,11 @@ type GetFileRequest struct {
 	Key string `json:"key"`
 }
 
+type GetUploadLinkRequest struct {
+	Key      string `json:"key"`
+	TempFile bool   `json:"tempFile"`
+}
+
 // GetFileResponse represents the JSON structure for get file response
 type GetFileResponse struct {
 	Content string `json:"content"`
@@ -162,6 +167,7 @@ type GetLinkResponse struct {
 // PutFileRequest represents the JSON structure for put file operations
 type PutFileRequest struct {
 	Key      string `json:"key"`
+	TempFile bool   `json:"tempFile"`
 	Content  string `json:"content"`
 	FilePath string `json:"filePath"`
 }
@@ -171,8 +177,9 @@ type DeleteFileRequest struct {
 }
 
 type RenameFileRequest struct {
-	OldKey string `json:"oldKey"`
-	NewKey string `json:"newKey"`
+	OldKey   string `json:"oldKey"`
+	NewKey   string `json:"newKey"`
+	TempFile bool   `json:"tempFile"`
 }
 
 type CreateFolderRequest struct {
@@ -397,7 +404,7 @@ func (sc *ServiceClient) PutFile(sessionId string, req PutFileRequest) error {
 	return executeApiWithoutResponse(sc.httpClient, sc.baseURL, sessionId, "v1/context/file/put", req)
 }
 
-func (sc *ServiceClient) GetFileUploadLink(sessionId string, req GetFileRequest) (GetLinkResponse, error) {
+func (sc *ServiceClient) GetFileUploadLink(sessionId string, req GetUploadLinkRequest) (GetLinkResponse, error) {
 	var res GetLinkResponse
 	err := executeApiWithResponse(sc.httpClient, sc.baseURL, sessionId, "v1/context/file/get-upload-link", req, &res)
 	return res, err

polycode/filestore.go

@@ -88,8 +88,27 @@ func (d FileStore) Save(path string, data []byte) error {
 	// Encode the data as base64
 	base64Data := base64.StdEncoding.EncodeToString(data)
 	req := PutFileRequest{
-		Key:     path,
-		Content: base64Data,
+		Key:      path,
+		TempFile: false,
+		Content:  base64Data,
+	}
+
+	err := d.client.PutFile(d.sessionId, req)
+	if err != nil {
+		fmt.Printf("failed to put file: %s\n", err.Error())
+		return err
+	}
+
+	return nil
+}
+
+func (d FileStore) SaveTemp(path string, data []byte) error {
+	// Encode the data as base64
+	base64Data := base64.StdEncoding.EncodeToString(data)
+	req := PutFileRequest{
+		Key:      path,
+		TempFile: true,
+		Content:  base64Data,
 	}
 
 	err := d.client.PutFile(d.sessionId, req)
@@ -104,6 +123,23 @@ func (d FileStore) Save(path string, data []byte) error {
 func (d FileStore) Upload(path string, filePath string) error {
 	req := PutFileRequest{
 		Key:      path,
+		TempFile: false,
+		FilePath: filePath,
+	}
+
+	err := d.client.PutFile(d.sessionId, req)
+	if err != nil {
+		fmt.Printf("failed to put file: %s\n", err.Error())
+		return err
+	}
+
+	return nil
+}
+
+func (d FileStore) UploadTemp(path string, filePath string) error {
+	req := PutFileRequest{
+		Key:      path,
+		TempFile: true,
 		FilePath: filePath,
 	}
 
@@ -117,8 +153,28 @@ func (d FileStore) Upload(path string, filePath string) error {
 }
 
 func (d FileStore) GetUploadLink(path string) (string, error) {
-	req := GetFileRequest{
-		Key: path,
+	req := GetUploadLinkRequest{
+		Key:      path,
+		TempFile: false,
+	}
+
+	res, err := d.client.GetFileUploadLink(d.sessionId, req)
+	if err != nil {
+		fmt.Printf("failed to get file link: %s\n", err.Error())
+		return "", err
+	}
+
+	if res.Link == "" {
+		return "", errors.New("empty link")
+	}
+
+	return res.Link, nil
+}
+
+func (d FileStore) GetTempUploadLink(path string) (string, error) {
+	req := GetUploadLinkRequest{
+		Key:      path,
+		TempFile: true,
 	}
 
 	res, err := d.client.GetFileUploadLink(d.sessionId, req)
@@ -206,8 +262,27 @@ func (f Folder) Save(name string, data []byte) error {
 	// Encode the data as base64
 	base64Data := base64.StdEncoding.EncodeToString(data)
 	req := PutFileRequest{
-		Key:     f.name + "/" + name,
-		Content: base64Data,
+		Key:      f.name + "/" + name,
+		TempFile: false,
+		Content:  base64Data,
+	}
+
+	err := f.client.PutFile(f.sessionId, req)
+	if err != nil {
+		fmt.Printf("failed to put file: %s\n", err.Error())
+		return err
+	}
+
+	return nil
+}
+
+func (f Folder) SaveTemp(name string, data []byte) error {
+	// Encode the data as base64
+	base64Data := base64.StdEncoding.EncodeToString(data)
+	req := PutFileRequest{
+		Key:      f.name + "/" + name,
+		TempFile: true,
+		Content:  base64Data,
 	}
 
 	err := f.client.PutFile(f.sessionId, req)
@@ -222,6 +297,23 @@ func (f Folder) Save(name string, data []byte) error {
 func (f Folder) Upload(name string, filePath string) error {
 	req := PutFileRequest{
 		Key:      f.name + "/" + name,
+		TempFile: false,
+		FilePath: filePath,
+	}
+
+	err := f.client.PutFile(f.sessionId, req)
+	if err != nil {
+		fmt.Printf("failed to put file: %s\n", err.Error())
+		return err
+	}
+
+	return nil
+}
+
+func (f Folder) UploadTemp(name string, filePath string) error {
+	req := PutFileRequest{
+		Key:      f.name + "/" + name,
+		TempFile: true,
 		FilePath: filePath,
 	}