Add copy & move & more error checks.

2026-07-03 00:39:02 +00:00 · 2020-03-21 06:48:05 +01:00
parent 3fc7aa93a3
commit 44db400cbc
7 changed files with 210 additions and 11 deletions
@@ -319,4 +319,101 @@ std::string selectFilePath(std::string selectText, const std::vector<std::string
 			}
 		}
 	}
+}
+
+#define copyBufSize 0x8000
+
+u32 copyBuf[copyBufSize];
+
+void dirCopy(DirEntry* entry, int i, const char *destinationPath, const char *sourcePath) {
+	std::vector<DirEntry> dirContents;
+	dirContents.clear();
+	if(entry->isDirectory)	chdir((sourcePath + ("/" + entry->name)).c_str());
+	getDirectoryContents(dirContents);
+	if(((int)dirContents.size()) == 1)	mkdir((destinationPath + ("/" + entry->name)).c_str(), 0777);
+	if(((int)dirContents.size()) != 1)	fcopy((sourcePath + ("/" + entry->name)).c_str(), (destinationPath + ("/" + entry->name)).c_str());
+}
+
+int fcopy(const char *sourcePath, const char *destinationPath) {
+	DIR *isDir = opendir(sourcePath);
+
+	if(isDir != NULL) {
+		closedir(isDir);
+
+		// Source path is a directory
+		chdir(sourcePath);
+		std::vector<DirEntry> dirContents;
+		getDirectoryContents(dirContents);
+		DirEntry* entry = &dirContents.at(1);
+
+		mkdir(destinationPath, 0777);
+		for(int i = 1; i < ((int)dirContents.size()); i++) {
+			chdir(sourcePath);
+			entry = &dirContents.at(i);
+			dirCopy(entry, i, destinationPath, sourcePath);
+		}
+
+		chdir(destinationPath);
+		chdir("..");
+		return 1;
+	} else {
+		closedir(isDir);
+
+		// Source path is a file
+		FILE* sourceFile = fopen(sourcePath, "rb");
+		off_t fsize = 0;
+		if(sourceFile) {
+			fseek(sourceFile, 0, SEEK_END);
+			fsize = ftell(sourceFile);			// Get source file's size
+			fseek(sourceFile, 0, SEEK_SET);
+		} else {
+			fclose(sourceFile);
+			return -1;
+		}
+
+		FILE* destinationFile = fopen(destinationPath, "wb");
+		//if(destinationFile) {
+			fseek(destinationFile, 0, SEEK_SET);
+		/*} else {
+			fclose(sourceFile);
+			fclose(destinationFile);
+			return -1;
+		}*/
+
+		off_t offset = 0;
+		int numr;
+		while(1)
+		{
+			scanKeys();
+			if(keysHeld() & KEY_B) {
+				// Cancel copying
+				fclose(sourceFile);
+				fclose(destinationFile);
+				return -1;
+				break;
+			}
+			printf("\x1b[16;0H");
+			printf("Progress:\n");
+			printf("%i/%i Bytes					   ", (int)offset, (int)fsize);
+
+			// Copy file to destination path
+			numr = fread(copyBuf, 2, copyBufSize, sourceFile);
+			fwrite(copyBuf, 2, numr, destinationFile);
+			offset += copyBufSize;
+
+			if(offset > fsize) {
+				fclose(sourceFile);
+				fclose(destinationFile);
+
+				printf("\x1b[17;0H");
+				printf("%i/%i Bytes					   ", (int)fsize, (int)fsize);
+				for(int i = 0; i < 30; i++) gspWaitForVBlank();
+
+				return 1;
+				break;
+			}
+		}
+
+		return -1;
+	}
 }