exporter.cpp

#include "exporter.h"
#include <fstream>
#include <iostream>
#include <vector>

// Helper to escape CSV strings
static std::string EscapeCsv(const std::string& s) {
    std::string out = "\"";
    for (char c : s) {
        if (c == '"') out += "\"\"";
        else out += c;
    }
    out += "\"";
    return out;
}

// Helper to serialize Star Ratings
static std::string SerializeStars(const std::vector<std::pair<int, double>>& stars) {
    std::string out;
    for (size_t i = 0; i < stars.size(); ++i) {
        out += std::to_string(stars[i].first) + ":" + std::to_string(stars[i].second);
        if (i < stars.size() - 1) out += ";";
    }
    return out;
}

// Helper to serialize Timing Points
static std::string SerializeTiming(const std::vector<TimingPoint>& points) {
    std::string out;
    for (size_t i = 0; i < points.size(); ++i) {
        out += std::to_string(points[i].bpm) + "|" + std::to_string(points[i].offset) + "|" + (points[i].inherited ? "1" : "0");
        if (i < points.size() - 1) out += ";";
    }
    return out;
}

// Helper to get Ranked Status string
static std::string GetRankedStatusString(int status) {
    switch (status) {
        case 0: return "Unknown";
        case 1: return "Unsubmitted";
        case 2: return "Pending/Graveyard";
        case 3: return "Unused";
        case 4: return "Ranked";
        case 5: return "Approved";
        case 6: return "Qualified";
        case 7: return "Loved";
        default: return "Unknown (" + std::to_string(status) + ")";
    }
}

// Helper to get Mode string
static std::string GetModeString(int mode) {
    switch (mode) {
        case 0: return "Standard";
        case 1: return "Taiko";
        case 2: return "CTB";
        case 3: return "Mania";
        default: return "Unknown (" + std::to_string(mode) + ")";
    }
}

void OsuDbExporter::SaveToCsv(const OsuDbInfo& dbInfo, const std::string& path) {
    std::ofstream file(path);
    if (!file.is_open()) {
        std::cerr << "Failed to open " << path << " for writing CSV." << std::endl;
        return;
    }

    // Header
    file << "Artist,ArtistUnicode,Title,TitleUnicode,Creator,Difficulty,AudioFile,MD5,OsuFile,RankedStatus,Mode,"
         << "BeatmapID,DifficultyID,"
         << "CS,AR,OD,HP,Circles,Sliders,Spinners,DrainTime,TotalTime,PreviewTime,"
         << "LocalOffset,StackLeniency,OnlineOffset,Font,IsUnplayed,LastPlayed,IsOsz2,FolderName,"
         << "LastChecked,IgnoreSound,IgnoreSkin,DisableStoryboard,DisableVideo,VisualOverride,"
         << "LastModTime,ManiaScrollSpeed,"
         << "StarsStd,StarsTaiko,StarsCTB,StarsMania,TimingPoints\n";

    for (const auto& entry : dbInfo.beatmaps) {
        file << EscapeCsv(entry.artistName) << ","
             << EscapeCsv(entry.artistNameUnicode) << ","
             << EscapeCsv(entry.songTitle) << ","
             << EscapeCsv(entry.songTitleUnicode) << ","
             << EscapeCsv(entry.creatorName) << ","
             << EscapeCsv(entry.difficulty) << ","
             << EscapeCsv(entry.audioFileName) << ","
             << EscapeCsv(entry.md5Hash) << ","
             << EscapeCsv(entry.osuFileName) << ","
             << EscapeCsv(GetRankedStatusString(entry.rankedStatus)) << ","
             << EscapeCsv(GetModeString(entry.gameplayMode)) << ","
             << entry.beatmapId << ","
             << entry.difficultyId << ","
             << entry.circleSize << ","
             << entry.approachRate << ","
             << entry.overallDifficulty << ","
             << entry.hpDrain << ","
             << entry.hitCircleCount << ","
             << entry.sliderCount << ","
             << entry.spinnerCount << ","
             << entry.drainTime << ","
             << entry.totalTime << ","
             << entry.audioPreviewTime << ","
             << entry.localOffset << ","
             << entry.stackLeniency << ","
             << entry.onlineOffset << ","
             << EscapeCsv(entry.font) << ","
             << (entry.isUnplayed ? "1" : "0") << ","
             << entry.lastPlayed << ","
             << (entry.isOsz2 ? "1" : "0") << ","
             << EscapeCsv(entry.folderName) << ","
             << entry.lastCheckedAgainstOsuRepo << ","
             << (entry.ignoreBeatmapSound ? "1" : "0") << ","
             << (entry.ignoreBeatmapSkin ? "1" : "0") << ","
             << (entry.disableStoryboard ? "1" : "0") << ","
             << (entry.disableVideo ? "1" : "0") << ","
             << (entry.visualOverride ? "1" : "0") << ","
             << entry.lastModificationTime << ","
             << (int)entry.maniaScrollSpeed << ","
             << EscapeCsv(SerializeStars(entry.starRatingStd)) << ","
             << EscapeCsv(SerializeStars(entry.starRatingTaiko)) << ","
             << EscapeCsv(SerializeStars(entry.starRatingCTB)) << ","
             << EscapeCsv(SerializeStars(entry.starRatingMania)) << ","
             << EscapeCsv(SerializeTiming(entry.timingPoints)) << "\n";
    }

    std::cout << "Successfully saved " << dbInfo.beatmaps.size() << " beatmaps to " << path << std::endl;
}