156 lines
5.1 KiB
C++
156 lines
5.1 KiB
C++
#include "dataManager.h"
|
|
#include <iostream>
|
|
#include <filesystem>
|
|
#include <fstream>
|
|
#include <sstream>
|
|
#include <vector>
|
|
#include <string>
|
|
|
|
using namespace std;
|
|
namespace fs = std::filesystem;
|
|
|
|
void Table::CreateTB(string tb_name, string column, string data_type1, string column2, string data_type2, string data_of_type){
|
|
//string path = program_path /= tb_name;
|
|
if (fs::exists(program_path) && fs::is_regular_file(tb_name + ".csv")) {
|
|
cout << "!Failed to create table " << tb_name << " because it already exists." << endl;
|
|
} else {
|
|
//cout << "program_path: " << program_path << "/" << endl;
|
|
//program_path /= tb_name;
|
|
string pathInital = program_path;
|
|
string path = program_path /= tb_name;
|
|
//cout << "path: " << path << endl;
|
|
ofstream newTable(path + ".csv");
|
|
cout << "Table " << tb_name << " created." << endl;
|
|
newTable << column << "," << column2 << endl;
|
|
newTable << data_type1 << "," << data_type2 << "(" <<data_of_type << ")"<< endl;
|
|
newTable.close();
|
|
program_path = pathInital;
|
|
|
|
|
|
|
|
|
|
}
|
|
}
|
|
|
|
void Table::DropTB(string tb_name){
|
|
//cout << "program_path: " << program_path << tb_name << ".csv"<< endl;
|
|
//cout << program_path/=tb_name + ".csv" << endl;
|
|
string pathInital = program_path;
|
|
if (fs::exists(program_path/=tb_name + ".csv")) {
|
|
fs::remove(program_path);
|
|
|
|
cout << "Table " << tb_name << " deleted." << endl;
|
|
|
|
} else {
|
|
cout << "!Failed to drop database " << tb_name << " because it does not exist." << endl;
|
|
}
|
|
program_path = pathInital;
|
|
}
|
|
|
|
void Table::SelectTB(string tb_name){
|
|
//cout << "program_path: " << program_path << endl;
|
|
string pathInital = program_path;
|
|
if (fs::exists(program_path/=tb_name + ".csv")) {
|
|
string line;
|
|
int length, currentLine = 1, varline = 2, columnline = 1, size = 5;
|
|
|
|
string column[5], data_type[5];
|
|
|
|
int count = 0;
|
|
|
|
ifstream table(program_path);
|
|
if (table.is_open()) {
|
|
|
|
while (getline(table, line)) {
|
|
for (char c : line) {
|
|
// If the character is a comma, increment the count
|
|
if (c == ',') {
|
|
count++;
|
|
}
|
|
}
|
|
if (currentLine == varline){
|
|
stringstream ss(line);
|
|
for (int i = 0; i < count; i++){
|
|
getline(ss, data_type[i], ',');
|
|
}
|
|
|
|
} else if ( currentLine == columnline) {
|
|
//cout << line << endl;
|
|
stringstream ss(line);
|
|
for (int i = 0; i < count; i++){
|
|
getline(ss, column[i], ',');
|
|
}
|
|
}
|
|
|
|
currentLine++;
|
|
}
|
|
}
|
|
for (int i = 0; i < count; i++){
|
|
// Print the parsed values
|
|
cout << column[i] << " " << data_type[i] << " | " ;
|
|
}
|
|
cout << endl;
|
|
|
|
table.close();
|
|
|
|
|
|
} else {
|
|
cout << "!Failed to select table " << tb_name << " because it does not exist." << endl;
|
|
}
|
|
program_path = pathInital;
|
|
}
|
|
|
|
void Table::AlterTB(string tb_name, string columnInput, string data_typeInput){ //adds column to table by coppying from old csv, appending to new csv, then replacing old csv.
|
|
//cout << "program_path: " << program_path << endl;
|
|
string pathInital = program_path;
|
|
program_path_copy = program_path;
|
|
//cout << "program_path_temp: " << pathTemp << endl;
|
|
if (fs::exists(program_path/=tb_name + ".csv")) {
|
|
string line;
|
|
int length, currentLine = 1, varline = 2, columnline = 1, size = 5;
|
|
|
|
string column[5], data_type[5];
|
|
int count = 0;
|
|
|
|
|
|
ifstream table(program_path);
|
|
string pathTemp = program_path_copy/=tb_name + "_temp.csv";
|
|
cout << "program_path_temp: " << pathTemp << endl;
|
|
ofstream outfile(pathTemp, ios::app); //append to file ios::app
|
|
cout << program_path << endl;
|
|
while (getline(table, line)) {
|
|
for (char c : line) {
|
|
// If the character is a comma, increment the count
|
|
if (c == ',') {
|
|
count++;
|
|
}
|
|
}
|
|
if (currentLine == varline){
|
|
stringstream ss(line);
|
|
data_typeInput = "," + data_typeInput;
|
|
line += data_typeInput;
|
|
outfile << line << endl;
|
|
|
|
} else if ( currentLine == columnline) {
|
|
//cout << line << endl;
|
|
stringstream ss(line);
|
|
columnInput = "," + columnInput;
|
|
line += columnInput;
|
|
outfile << line << endl;
|
|
|
|
}
|
|
|
|
currentLine++;
|
|
}
|
|
table.close();
|
|
rename(program_path_copy.c_str(), program_path.c_str()); //replace file
|
|
|
|
|
|
outfile.close();
|
|
//outfile << columnInput << "," << data_typeInput << endl;
|
|
|
|
} else {
|
|
cout << "!Failed to select table " << tb_name << " because it does not exist." << endl;
|
|
}
|
|
program_path = pathInital;
|
|
}
|