Merge pull request #25 from Alex-Monahan/delete-before-write

Delete sheet1 content before writing

src/gsheets_copy.cpp

@@ -72,6 +72,10 @@ namespace duckdb
         std::string sheet_id = extract_sheet_id(file_path);
         std::string sheet_name = "Sheet1"; // TODO: make this configurable
 
+        // If writing, clear out the entire sheet first.
+        // Do this here in the initialization so that it only happens once
+        std::string response = delete_sheet_data(sheet_id, token, sheet_name);
+
         return make_uniq<GSheetCopyGlobalState>(context, sheet_id, token, sheet_name);
     }
 
@@ -92,6 +96,8 @@ namespace duckdb
         sheet_data["range"] = "Sheet1";
         sheet_data["majorDimension"] = "ROWS";
 
+        // TODO: Add column headers
+
         vector<vector<string>> values;
         for (idx_t r = 0; r < input.size(); r++)
         {
@@ -129,9 +135,6 @@ namespace duckdb
 
         // Make the API call to write data to the Google Sheet
         // Today, this is only append.
-        // To overwrite, need to detect if we are the first data chunk and clear out the sheet. 
-        // Is there a way to force that to happen only once, even in the presence of multithreading?
-        // Maybe this? https://github.com/duckdb/duckdb/pull/7368
         std::string response = fetch_sheet_data(gstate.sheet_id, gstate.token, gstate.sheet_name, HttpMethod::POST, request_body);
 
         // Check for errors in the response

src/gsheets_requests.cpp

@@ -3,7 +3,6 @@
 #include <openssl/ssl.h>
 #include <openssl/err.h>
 #include <openssl/bio.h>
-
 namespace duckdb
 {
 
@@ -102,4 +101,12 @@ namespace duckdb
 
         return perform_https_request(host, path, token, method, body);
     }
+
+    std::string delete_sheet_data(const std::string &sheet_id, const std::string &token, const std::string &sheet_name)
+    {
+        std::string host = "sheets.googleapis.com";
+        std::string path = "/v4/spreadsheets/" + sheet_id + "/values/" + sheet_name + ":clear";
+
+        return perform_https_request(host, path, token, HttpMethod::POST, "{}");
+    }
 }

src/include/gsheets_requests.hpp

@@ -14,4 +14,7 @@ std::string perform_https_request(const std::string& host, const std::string& pa
                                   HttpMethod method = HttpMethod::GET, const std::string& body = "", const std::string& content_type = "application/json");
 
 std::string fetch_sheet_data(const std::string& sheet_id, const std::string& token, const std::string& sheet_name, HttpMethod method = HttpMethod::GET, const std::string& body = "");
+
+std::string delete_sheet_data(const std::string& sheet_id, const std::string& token, const std::string& sheet_name);
+
 }