Be Sweet - a set of visual source code browsers

#include "stdafx.h"
#include "Resource.h"

#include "ESBServer.h"
#include "VSWorkspaceLoader.h"

#include <Model/Simulators/HierarchyLoaderSimulator.h>
#include <Model/CTags/CTagsHierarchyLoader.h>
#include <Renderer/MFC/Browsers/OutlineBrowserRenderer.h>

using namespace std;
using namespace CTags;

ESBServer::ESBServer()
: mRenderer(this), mMethodBrowser("Method/Function"), mTypeBrowser("Type") 
{  
  getOutlineBrowser().setRenderer(make_smart_ptr(new OutlineBrowserDialog()));
  getOutlineBrowser().setESBServer(this);
  
  smart_ptr<SingleSelectionListener> listener = make_smart_ptr(new DefaultSingleSelectionListener());
  getOutlineBrowser().addSelectionListener(listener);
  getHierarchyBrowser().addSelectionListener(listener);
}

void ESBServer::run() 
{ 
	mRenderer.DoModal(); 
}

void ESBServer::showTypes() 
{
 	smart_ptr<TypeBrowserModel> model(new TypeBrowserModel(getTypeInfoLoader()->load("")));

  getTypeBrowser().setModel(model);
	getTypeBrowser().show();	
}

void ESBServer::showMethods()
{
 	smart_ptr<MethodBrowserModel> model(new MethodBrowserModel(getMethodLoader()->load()));

  getMethodBrowser().setModel(model);
	getMethodBrowser().show();	
}


void ESBServer::showWorkspace()
{
  smart_ptr<ScopedNameBrowserModel> model(new WorkspaceBrowserModel(*getWorkspace()));

  getWorkspaceBrowser().setModel(model);
  getWorkspaceBrowser().show();
}

void ESBServer::showFileOutline(const std::string &file, bool enforceModelReload)
{
  if(!file.empty()) {
    string x = file_path_is::pathof(file) + file_name_is::nameof(file);
    string y = file_path_is::pathof(getOutlineBrowser().getModelName()) + file_name_is::nameof(getOutlineBrowser().getModelName());
    
    bool reload = enforceModelReload || x != y;

    if(reload) {
      getOutlineBrowser().setModel(getFileOutlineLoader()->load(file), file);
      getOutlineBrowser().show();
    }  
  }
}

void ESBServer::showHierarchy(const Inheritable &type)
{
  TagCloner cloner;
  const_cast<Inheritable&>(type).acceptVisitor(&cloner);

  smart_ptr<Inheritable> clone = Dynamic_cast<Inheritable>(cloner.getClone());
  clone->setHierarchyLoader(make_smart_ptr(new CTags::HierarchyLoader(getWorkspace())));
  
  getHierarchyBrowser().setModel(clone);
  getHierarchyBrowser().show();
}

void ESBServer::reparseFile(const std::string &file)
{
  try {
    DatabaseManager::instance().getDatabase(getWorkspace())->refreshTagsReferencingFile(file);
    showFileOutline(file, true);

  } catch(std::exception &e) {
    AfxMessageBox(e.what());
  }
}

void ESBServer::openWorkspace(const Workspace& w)
{
  try {
    VisualStudioWorkspaceLoader workspaceLoader(w.getName());
    setWorkspace(workspaceLoader.load(w.getPackages()));

  } catch(std::exception &e) {
    AfxMessageBox(e.what());
  }
}

void ESBServer::reparseWorkspace()
{
  openWorkspace(*getWorkspace());
}

void ESBServer::showBrowser(const std::string &browser)
{
  if(browser == "outline") {
    getOutlineBrowser().show();
    getOutlineBrowser().moveToForeground();

  } else if(browser == "hierarchy") {
    if(getHierarchyBrowser().getSelection() == NULL) {
      showHierarchy(Class("dummy", ePublic, TagLocation()));
      //getHierarchyBrowser().moveToForeground();
    } else {
      getHierarchyBrowser().getRenderer().moveToForeground();
    }

  } else if(browser == "types") {
    showTypes();
    getTypeBrowser().moveToForeground();

  } else if(browser == "methods") {
    showMethods();
    getMethodBrowser().moveToForeground();

  } else if(browser == "workspace") {
    showWorkspace();
    getWorkspaceBrowser().moveToForeground();

  } else {
    AfxMessageBox(("ESBServer::showBrowser():unknown browser type " + browser).c_str());
  }
}

#include "Model/Simulators/TypeInfoLoaderSimulator.h"
smart_ptr<::TypeInfoLoader> ESBServer::getTypeInfoLoader()
{
  //return smart_ptr<TypeInfoLoaderSimulator>(new TypeInfoLoaderSimulator());
  return smart_ptr<::TypeInfoLoader>(new CTags::TypeInfoLoader(getWorkspace())); 
}

smart_ptr<::MethodLoader> ESBServer::getMethodLoader()
{
  return smart_ptr<::MethodLoader>(new CTags::MethodLoader(getWorkspace()));
}

smart_ptr<::FileOutlineLoader> ESBServer::getFileOutlineLoader()
{
  //return smart_ptr<FileOutlineLoaderSimulator>(new FileOutlineLoaderSimulator());
  return smart_ptr<::FileOutlineLoader>(new CTags::FileOutlineLoader(getWorkspace()));
}