#include "mwcc_decomp.h"

// TODO: figure out who defines this
extern char STSbuf[256];

int GetFileCount() {
    SInt32 num;
    CWGetProjectFileCount(parseopts.context, &num);
    return num;
}

void SetFileOutputName(SInt32 position, SInt16 which, char *outfilename) {
    CWResult result;
    if (outfilename && outfilename[0]) {
        result = CWParserSetFileOutputName(parseopts.context, position, which ? which : 1, outfilename);
        if (result) {
            failedCallback = "CWParserSetFileOutputName";
            longjmp(exit_plugin, result);
        }
    }
}

int AddFileToProject(OSSpec *oss, SInt16 which, char *outfilename, Boolean exists, SInt32 position) {
    CWFileSpec cws;
    CWNewProjectEntryInfo pei;
    CWResult result;
    int err;

    err = OS_OSSpec_To_FSSpec(oss, &cws);
    if (err) {
        CLPOSAlert(44, err, OS_SpecToStringRelative(oss, 0, STSbuf, sizeof(STSbuf)));
        return 0;
    }

    if (position == -2) {
        pei.position = 0;
    } else if (position == -1) {
        CWGetProjectFileCount(parseopts.context, &pei.position);
    } else if (position == 0) {
        pei.position = -1;
    } else {
        pei.position = position;
    }

    pei.segment = parseopts.currentSegment;
    pei.overlayGroup = parseopts.currentOverlayGroup;
    pei.overlay = parseopts.currentOverlay;
    pei.groupPath = NULL;
    pei.mergeintooutput = parseopts.mergeIntoOutput;
    pei.weakimport = parseopts.weakImport;
    pei.initbefore = parseopts.initBefore;
    result = CWAddProjectEntry(parseopts.context, &cws, exists == 0, &pei, &position);
    if (result) {
        failedCallback = "CWAddProjectEntry";
        longjmp(exit_plugin, result);
    }
    parseopts.initBefore = 0;
    parseopts.weakImport = 0;
    parseopts.mergeIntoOutput = 0;
    SetFileOutputName(position, which, outfilename);
    return 1;
}

Boolean GetFileInfo(SInt32 position, OSSpec *spec, char *plugin) {
    CWProjectFileInfo pfi;
    if (CWGetFileInfo(parseopts.context, position, 1, &pfi))
        return 0;

    OS_FSSpec_To_OSSpec(&pfi.filespec, spec);
    strncpy(plugin, pfi.dropinname, 32);
    plugin[31] = 0;
    return 1;
}

int AddAccessPath(OSPathSpec *oss, SInt16 type, SInt32 position, Boolean recursive) {
    CWNewAccessPathInfo api;
    CWResult result;
    OSSpec spec;
    int err;
    CWAccessPathListInfo apli;

    if ((err = OS_MakeSpecWithPath(oss, NULL, 0, &spec)) || (err = OS_OSSpec_To_FSSpec(&spec, &api.pathSpec))) {
        CLPOSAlert(45, err, OS_PathSpecToString(&spec.path, STSbuf, sizeof(STSbuf)));
        return 0;
    }

    if (position == -2) {
        api.position = 0;
    } else if (position == -1 || position == 0) {
        api.position = -1;
    } else if (position == -1) {
        result = CWGetAccessPathListInfo(parseopts.context, &apli);
        if (result) {
            failedCallback = "CWGetAccessPathListInfo";
            longjmp(exit_plugin, result);
        }
        if ((type & 1) == 1) {
            api.position = apli.systemPathCount;
        } else {
            api.position = apli.userPathCount;
        }
    } else {
        api.position = position;
    }

    api.type = (type & 1) ? cwSystemPath : cwUserPath;
    api.type = api.type | (CWAccessPathType) ((type & 2) ? cwAccessPathTypeFlag2 : 0);
    api.recursive = recursive;

    result = CWParserAddAccessPath(parseopts.context, &api);
    if (result) {
        failedCallback = "CWParserAddAccessPath";
        longjmp(exit_plugin, result);
    }

    return 1;
}

int MoveSystemPathsIntoUserList() {
    CWResult result = CWParserSwapAccessPaths(parseopts.context);
    return result == cwNoErr;
}

void AddVirtualFile(const char *filename, Handle *text) {
    CWResult result;
    CWMemHandle mh;

    if (*text) {
        CWSecretAttachHandle(parseopts.context, *text, &mh);
        result = CWParserCreateVirtualFile(parseopts.context, filename, mh);
        if (result) {
            failedCallback = "CWParserCreateVirtualFile";
            longjmp(exit_plugin, result);
        }
        DisposeHandle(*text);
        *text = NULL;
    }
}

void GetOutputFileDirectory(OSPathSpec *dir) {
    CWResult result;
    CWFileSpec idefss;
    OSSpec spec;

    result = CWGetOutputFileDirectory(parseopts.context, &idefss);
    if (result) {
        failedCallback = "CWGetOutputFileDirectory";
        longjmp(exit_plugin, result);
    }

    OS_FSSpec_To_OSSpec(&idefss, &spec);
    *dir = spec.path;
}

void SetOutputFileDirectory(OSPathSpec *dir) {
    CWResult result;
    CWFileSpec idefss;
    OSSpec spec;

    OS_MakeSpecWithPath(dir, NULL, 0, &spec);
    OS_OSSpec_To_FSSpec(&spec, &idefss);

    result = CWParserSetOutputFileDirectory(parseopts.context, &idefss);
    if (result) {
        failedCallback = "CWParserSetOutputFileDirectory";
        longjmp(exit_plugin, result);
    }
}

void AddOverlayGroup(const char *name, CWAddr64 *addr, SInt32 *groupnum, SInt32 *overlaynum) {
    CWResult result;

    *overlaynum = -1;
    result = CWParserAddOverlay1Group(parseopts.context, name, addr, groupnum);
    if (result) {
        if (result == cwErrInvalidCallback) {
            CLPReportError(72);
        } else {
            failedCallback = "CWParserAddOverlay1Group";
            longjmp(exit_plugin, result);
        }
    }
}

void AddOverlay(SInt32 groupnum, const char *name, SInt32 *overlaynum) {
    CWResult result;

    result = CWParserAddOverlay1(parseopts.context, name, groupnum, overlaynum);
    if (result) {
        if (result == cwErrInvalidCallback) {
            CLPReportError(72);
        } else {
            failedCallback = "CWParserAddOverlay1";
            longjmp(exit_plugin, result);
        }
    }
}

void AddSegment(const char *name, SInt16 attrs, SInt32 *segmentnum) {
    CWResult result;

    result = CWParserAddSegment(parseopts.context, name, attrs, segmentnum);
    if (result) {
        if (result == cwErrInvalidCallback) {
            CLPReportError(73);
        } else {
            failedCallback = "CWParserAddSegment";
            longjmp(exit_plugin, result);
        }
    }
}

void ChangeSegment(SInt32 segmentnum, const char *name, SInt16 attrs) {
    CWResult result;

    result = CWParserSetSegment(parseopts.context, segmentnum, name, attrs);
    if (result) {
        if (result == cwErrInvalidCallback) {
            CLPReportError(73);
        } else {
            failedCallback = "CWParserSetSegment";
            longjmp(exit_plugin, result);
        }
    }
}

int GetSegment(SInt32 segmentnum, char *name, SInt16 *attrs) {
    CWResult result;
    CWProjectSegmentInfo psi;

    result = CWGetSegmentInfo(parseopts.context, segmentnum, &psi);
    if (result) {
        if (result == cwErrInvalidCallback) {
            CLPReportError(73);
        }
        return 0;
    }

    strcpy(name, psi.name);
    *attrs = psi.attributes;
    return 1;
}