#include <stdio.h>
#include <stdlib.h>

#include <File.h>
#include <Message.h>
#include <String.h>

#include <package/RepositoryInfo.h>


using namespace BPackageKit;


int
main(int argc, const char** argv)
{
        if (argc < 5) {
                fprintf(stderr, "usage: %s <name> <url> <priority> <pkg-count>\n",
                        argv[0]);
                return 1;
        }

        BRepositoryInfo repoInfo;
        repoInfo.SetName(argv[1]);
        repoInfo.SetURL(argv[2]);
        repoInfo.SetPriority(atoi(argv[3]));

        BMessage repoInfoArchive;
        status_t result = repoInfo.Archive(&repoInfoArchive);
        if (result != B_OK) {
                fprintf(stderr, "couldn't archive repository-header\n");
                return 1;
        }

        BFile output(argv[1], B_READ_WRITE | B_CREATE_FILE | B_ERASE_FILE);
        if ((result = repoInfoArchive.Flatten(&output)) != B_OK) {
                fprintf(stderr, "couldn't flatten repository-header archive\n");
                return 1;
        }

        int pkgCount = atoi(argv[4]);
        for (int i = 0; i < pkgCount; ++i) {
                BMessage pkg('PKGp');

                BString name = BString("pkg") << i + 1;
                pkg.AddString("name", name.String());

                BString majorVersion
                        = BString() << 1 + i % 5 << "." << i % 10;
                BString minorVersion
                        = BString() << i % 100;
                pkg.AddString("version",
                        (BString() << majorVersion << "." << minorVersion).String());

                pkg.AddString("provides",
                        (BString() << name << "-" << majorVersion).String());
                if (i % 2 == 1)
                        pkg.AddString("provides", (BString("lib") << name).String());
                if (i % 3 != 1)
                        pkg.AddString("provides", (BString("cmd:") << name).String());

                if (i > 1) {
                        int requiresCount = rand() % 10 % i;
                        for (int r = 0; r < requiresCount; ++r) {
                                int reqIndex = rand() % i;
                                pkg.AddString("requires",
                                        (BString("pkg") << reqIndex).String());
                        }
                }

                result = pkg.Flatten(&output);
                if (result != B_OK) {
                        fprintf(stderr, "couldn't flatten pkg message #%d\n", i + 1);
                        return 1;
                }
        }

        return 0;
}