/////////////////////////////////////////////////////////////////////////////
// Name:        dynlib.cpp
// Purpose:     Dynamic library management
// Author:      Guilhem Lavaux
// Modified by:
// Created:     20/07/98
// RCS-ID:      $Id$
// Copyright:   (c) Guilhem Lavaux
// Licence:     wxWindows license
/////////////////////////////////////////////////////////////////////////////

#ifdef __GNUG__
#pragma implementation "dynlib.h"
#endif

#include <wx/dynlib.h>
#include <wx/filefn.h>
#include <wx/list.h>
#include <wx/string.h>

// ---------------------------------------------------------------------------
// System dependent include
// ---------------------------------------------------------------------------

#ifdef __UNIX__ 
#include <dlfcn.h>
#endif

#ifdef __WINDOWS__
#include <windows.h>
#endif

#ifdef LoadLibrary
#undef LoadLibrary
#endif

// ---------------------------------------------------------------------------
// Global variables
// ---------------------------------------------------------------------------

wxLibraries wxTheLibraries;

// ---------------------------------------------------------------------------
// wxLibrary (one instance per dynamic library
// ---------------------------------------------------------------------------

wxLibrary::wxLibrary(void *handle)
{
  typedef wxClassInfo **(*t_get_first)(void);
  t_get_first get_first;

  m_handle = handle;
  m_destroy = TRUE;

  // Some system may use a local heap for library.
  get_first = (t_get_first)GetSymbol("wxGetClassFirst");
  // It is a wxWindows DLL.
  if (get_first)
    PrepareClasses(get_first());
}

wxLibrary::~wxLibrary()
{
  if (m_handle && m_destroy) {
#ifdef __UNIX__
    dlclose(m_handle);
#endif
#ifdef __WINDOWS__
    FreeLibrary((HMODULE)m_handle);
#endif
  }
}

wxObject *wxLibrary::CreateObject(const wxString& name)
{
  wxClassInfo *info = (wxClassInfo *)classTable.Get(name);

  if (!info)
    return NULL;

  return info->CreateObject();
}

void wxLibrary::PrepareClasses(wxClassInfo **first)
{
  // Index all class infos by their class name
  wxClassInfo *info = *first;
  while (info)
  {
    if (info->className)
      classTable.Put(info->className, (wxObject *)info);
    info = info->next;
  }

  // Set base pointers for each wxClassInfo
  info = *first;
  while (info)
  {
    if (info->GetBaseClassName1())
      info->baseInfo1 = (wxClassInfo *)classTable.Get(info->GetBaseClassName1());
    if (info->GetBaseClassName2())
      info->baseInfo2 = (wxClassInfo *)classTable.Get(info->GetBaseClassName2());
    info = info->next;
  }
  *first = NULL;
}

void *wxLibrary::GetSymbol(const wxString& symbname)
{
#ifdef __UNIX__
  return dlsym(m_handle, WXSTRINGCAST symbname);
#endif
#ifdef __WINDOWS__
  return GetProcAddress(m_handle, WXSTRINGCAST symbname);
#endif
  return NULL;
}

// ---------------------------------------------------------------------------
// wxLibraries (only one instance should normally exist)
// ---------------------------------------------------------------------------

wxLibraries::wxLibraries()
{
}

wxLibraries::~wxLibraries()
{
  wxNode *node = m_loaded.First();

  while (node) {
    wxLibrary *lib = (wxLibrary *)node->Data();
    delete lib;

    node = node->Next();
  }
}

wxLibrary *wxLibraries::LoadLibrary(const wxString& name)
{
  wxString lib_name = name;
  wxNode *node;
  wxLibrary *lib;

  if ( (node = m_loaded.Find(name.GetData())) )
    return ((wxLibrary *)node->Data());

#if defined(__UNIX__)
  lib_name.Prepend("./lib");
  lib_name += ".so";

  printf("lib_name = %s\n", WXSTRINGCAST lib_name);

  void *handle = dlopen(WXSTRINGCAST lib_name, RTLD_LAZY);

  if (!handle)
    return NULL;
#elif defined(__WINDOWS__)
  lib_name += ".dll";

#ifdef UNICODE
  HMODULE handle = LoadLibraryW(lib_name);
#else
  HMODULE handle = LoadLibraryA(lib_name);
#endif
  if (!handle)
    return NULL;
#else
  return NULL;
#endif

  lib = new wxLibrary((void *)handle);

  m_loaded.Append(name.GetData(), lib);
  return lib;
}

wxObject *wxLibraries::CreateObject(const wxString& path)
{
  wxNode *node = m_loaded.First();
  wxObject *obj;

  while (node) {
    obj = ((wxLibrary *)node->Data())->CreateObject(path);
    if (obj) 
      return obj;

    node = node->Next();
  }
  return NULL;
}