TrackGit.cpp

Go to the documentation of this file.

#include <Alert.h>
#include <Application.h>
#include <AppKit.h>
#include <Catalog.h>
#include <InterfaceKit.h>
#include <StorageKit.h>
#include <SupportKit.h>
#include <MenuItem.h>

#include <stdio.h>
#include <strings.h>
#include <vector>
#include <add-ons/tracker/TrackerAddOn.h>

#include "Utils.h"
#include "TrackGitApp.h"

#include <git2.h>

#define B_TRANSLATION_CONTEXT "TrackGit"

extern "C" {
    void populate_menu (BMessage* msg, BMenu* menu, BHandler* handler);
    void message_received (BMessage* msg);
}


void 
process_refs(entry_ref dir_ref, BMessage* msg, void*)
{
    BString buffer("TrackGit: A Tracker Addon for Git Version Control System");
    BAlert *alert = new BAlert("", buffer.String(), "Cancel", 
            0, 0, B_WIDTH_AS_USUAL, B_WARNING_ALERT);
    alert->Go();
}


void
populate_menu(BMessage* msg, BMenu* menu, BHandler* handler)
{
    if (menu == NULL){
        BString buffer("Null menu");
        BAlert *alert = new BAlert("", buffer.String(), "Cancel", 
                0, 0, B_WIDTH_AS_USUAL, B_WARNING_ALERT);
        alert->Go();
        return;
    }

    // Remove Menu item if already exists.
    BMenuItem* item = menu->FindItem(ADDON_NAME);
    if (item != NULL)
        menu->RemoveItem(item);

    BMenu* submenu = new BMenu(ADDON_NAME);

    vector<char*> selected;
    extract_selected_paths(msg, selected);
    BString dirPath = extract_current_directory(msg);

    git_libgit2_init();
    git_buf buf = GIT_BUF_INIT_CONST(NULL, 0);

    // Check if current directory is in git repo.
    if (git_repository_discover(&buf, dirPath.String(), 0, NULL) == 0) {
        // buf is git repo
        git_repository* repo = NULL;
        git_index* index;
        // Init repo
        git_repository_open_ext(&repo, dirPath.String(), 0, NULL);
        // Init index
        git_repository_index(&index, repo);
        if (git_index_has_conflicts(index)) {
            // Show conflicts menu item
            BMessage* showConflictsMsg = new BMessage(*msg);
            showConflictsMsg->AddInt32("addon_item_id", kShowConflicts);
            BMenuItem* showConflictsItem = new BMenuItem(
                    B_TRANSLATE("Show conflicts" B_UTF8_ELLIPSIS),
                    showConflictsMsg);
            submenu->AddItem(showConflictsItem);    
        }

        // Add Status menu item
        BMessage* statusMsg = new BMessage(*msg);
        statusMsg->AddInt32("addon_item_id", kStatus);
        BMenuItem* statusItem = new BMenuItem(
                B_TRANSLATE("Status" B_UTF8_ELLIPSIS), statusMsg);
        submenu->AddItem(statusItem);   

        // Add Commit menu item
        BMessage* commitMsg = new BMessage(*msg);
        commitMsg->AddInt32("addon_item_id", kCommit);
        BMenuItem* commitItem = new BMenuItem(
                B_TRANSLATE("Commit" B_UTF8_ELLIPSIS), commitMsg);
        submenu->AddItem(commitItem);

        // Add Pull menu item
        BMessage* pullMsg = new BMessage(*msg);
        pullMsg->AddInt32("addon_item_id", kPull);
        BMenuItem* pullItem = new BMenuItem(
                B_TRANSLATE("Pull" B_UTF8_ELLIPSIS), pullMsg);
        submenu->AddItem(pullItem);

        // Add Push menu item
        BMessage* pushMsg = new BMessage(*msg);
        pushMsg->AddInt32("addon_item_id", kPush);
        BMenuItem* pushItem = new BMenuItem(
                B_TRANSLATE("Push" B_UTF8_ELLIPSIS), pushMsg);
        submenu->AddItem(pushItem);

        // Add Create Branch menu item
        BMessage* createBranchMsg = new BMessage(*msg);
        createBranchMsg->AddInt32("addon_item_id", kCreateBranch);
        BMenuItem* createBranchItem = new BMenuItem(
                B_TRANSLATE("Create branch" B_UTF8_ELLIPSIS), createBranchMsg);
        submenu->AddItem(createBranchItem);

        // Add Switch Branch menu item
        BMessage* switchBranchMsg = new BMessage(*msg);
        switchBranchMsg->AddInt32("addon_item_id", kSwitchBranch);
        BMenuItem* switchBranchItem = new BMenuItem(
                B_TRANSLATE("Switch branch" B_UTF8_ELLIPSIS), switchBranchMsg);
        submenu->AddItem(switchBranchItem);

        // Add Log menu item
        BMessage* logMsg = new BMessage(*msg);
        logMsg->AddInt32("addon_item_id", kLog);
        BMenuItem* logItem = new BMenuItem(
                B_TRANSLATE("Log" B_UTF8_ELLIPSIS), logMsg);
        submenu->AddItem(logItem);

        if (selected.size() > 0) {
            // Add "Add Files" menu item
            BMessage* addMsg = new BMessage(*msg);
            addMsg->AddInt32("addon_item_id", kAdd);
            BMenuItem* addItem = new BMenuItem(
                    B_TRANSLATE("Add files"), addMsg);
            submenu->AddItem(addItem);
        } else {
            // Add "Add All Files" menu item
            BMessage* addMsg = new BMessage(*msg);
            addMsg->AddInt32("addon_item_id", kAddAll);
            BMenuItem* addItem = new BMenuItem(
                    B_TRANSLATE("Add all files"), addMsg);
            submenu->AddItem(addItem);
        }
    } else {
        // dirPath does not belong to git repo
        // Add Clone menu item
        BMessage* cloneMsg = new BMessage(*msg);
        cloneMsg->AddInt32("addon_item_id", kClone);
        BMenuItem* cloneItem = new BMenuItem(
                B_TRANSLATE("Clone" B_UTF8_ELLIPSIS), cloneMsg);
        submenu->AddItem(cloneItem);

        // Add "Init Here" only if no files are selected.
        if (selected.size() == 0) {
            // Add Init here
            BMessage* initMsg = new BMessage(*msg);
            initMsg->AddInt32("addon_item_id", kInitHere);
            BMenuItem* initItem = new BMenuItem(B_TRANSLATE("Init Here"), 
                initMsg);
            submenu->AddItem(initItem);
        }
    }

    menu->AddItem(submenu);
    submenu->SetTargetForItems(handler);

    git_buf_free(&buf);
    git_libgit2_shutdown();
}


void 
message_received(BMessage* msg)
{
    int32 itemId;

    if (msg->FindInt32("addon_item_id", &itemId) != B_OK)
        return;

    BMessenger messenger(APP_SIGN);

    if (!messenger.IsValid()) {
        be_roster->Launch(APP_SIGN);
        messenger = BMessenger(APP_SIGN);
    }
    messenger.SendMessage(msg);
}


int 
main()
{
    TrackGitApp app;
    app.Run();
    return 0;
}