/* 
   Unix SMB/Netbios implementation.
   Version 2.0
   SMB client GTK+ tree-based application
   Copyright (C) Andrew Tridgell 1998
   Copyright (C) Richard Sharpe 2001
   Copyright (C) John Terpstra 2001
   
   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 2 of the License, or
   (at your option) any later version.
   
   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.
   
   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/

/* example-gtk+ application, ripped off from the gtk+ tree.c sample */

#include <stdio.h>
#include <errno.h>
#include <gtk/gtk.h>
#include "libsmbclient.h"

struct tree_data {

  guint32 type;    /* Type of tree item, an SMBC_TYPE */
  char name[256];  /* May need to change this later   */

};

/*
 * We are given a widget, and we want to retrieve its URL so we 
 * can do a directory listing.
 *
 * We walk back up the tree, picking up pieces until we hit a server or
 * workgroup type and return a path from there
 */

static char path_string[1024];

char *get_path(GtkWidget *item)
{
  GtkWidget *p = item;
  struct tree_data *pd;
  char *comps[1024];  /* We keep pointers to the components here */
  int i = 0, j, level,type;

  /* Walk back up the tree, getting the private data */

  level = GTK_TREE(item->parent)->level;

  /* Pick up this item's component info */

  pd = (struct tree_data *)gtk_object_get_user_data(GTK_OBJECT(item));

  comps[i++] = pd->name;
  type = pd->type;

  while (level > 0 && type != SMBC_SERVER && type != SMBC_WORKGROUP) {

    /* Find the parent and extract the data etc ... */

    p = GTK_WIDGET(p->parent);    
    p = GTK_WIDGET(GTK_TREE(p)->tree_owner);

    pd = (struct tree_data *)gtk_object_get_user_data(GTK_OBJECT(p));

    level = GTK_TREE(item->parent)->level;

    comps[i++] = pd->name;
    type = pd->type;

  }

  /* 
   * Got a list of comps now, should check that we did not hit a workgroup
   * when we got other things as well ... Later
   *
   * Now, build the path
   */

  snprintf(path_string, sizeof(path_string), "smb:/");

  for (j = i - 1; j >= 0; j--) {

    strncat(path_string, "/", sizeof(path_string) - strlen(path_string));
    strncat(path_string, comps[j], sizeof(path_string) - strlen(path_string));

  }
  
  fprintf(stdout, "Path string = %s\n", path_string);

  return path_string;

}

struct tree_data *make_tree_data(guint32 type, const char *name)
{
  struct tree_data *p = (struct tree_data *)malloc(sizeof(struct tree_data));

  if (p) {

    p->type = type;
    strncpy(p->name, name, sizeof(p->name));

  }

  return p;

}

/* for all the GtkItem:: and GtkTreeItem:: signals */
static void cb_itemsignal( GtkWidget *item,
                           gchar     *signame )
{
  GtkWidget *real_tree, *aitem, *subtree;
  gchar *name;
  GtkLabel *label;
  gint dh, err, dirlen, level;
  char dirbuf[512];
  struct smbc_dirent *dirp;
  
  label = GTK_LABEL (GTK_BIN (item)->child);
  /* Get the text of the label */
  gtk_label_get (label, &name);

  level = GTK_TREE(item->parent)->level;

  /* Get the level of the tree which the item is in */
  g_print ("%s called for item %s->%p, level %d\n", signame, name,
	   item, GTK_TREE (item->parent)->level);

  if (strncmp(signame, "expand", 6) == 0) { /* Expand called */
    char server[128];

    if ((dh = smbc_opendir(get_path(item))) < 0) { /* Handle error */

      g_print("cb_wholenet: Could not open dir %s, %s\n", server, 
	      strerror(errno));

      gtk_main_quit();

      return;

    }

    real_tree = GTK_TREE_ITEM_SUBTREE(item);  /* Get the subtree */

    while ((err = smbc_getdents(dh, (struct smbc_dirent *)dirbuf, 
				sizeof(dirbuf))) != 0) {

      if (err < 0) { /* An error, report it */

	g_print("cb_wholenet: Could not read dir smbc://, %s\n",
		strerror(errno));

	gtk_main_quit();

	return;

      }

      dirp = (struct smbc_dirent *)dirbuf;

      while (err > 0) {
	struct tree_data *my_data;

	dirlen = dirp->dirlen;

	my_data = make_tree_data(dirp->smbc_type, dirp->name);

	if (!my_data) {

	  g_print("Could not allocate space for tree_data: %s\n",
		  dirp->name);

	  gtk_main_quit();
	  return;

	}

	aitem = gtk_tree_item_new_with_label(dirp->name);

	/* Connect all GtkItem:: and GtkTreeItem:: signals */
	gtk_signal_connect (GTK_OBJECT(aitem), "select",
			    GTK_SIGNAL_FUNC(cb_itemsignal), "select");
	gtk_signal_connect (GTK_OBJECT(aitem), "deselect",
			    GTK_SIGNAL_FUNC(cb_itemsignal), "deselect");
	gtk_signal_connect (GTK_OBJECT(aitem), "toggle",
			    GTK_SIGNAL_FUNC(cb_itemsignal), "toggle");
	gtk_signal_connect (GTK_OBJECT(aitem), "expand",
			    GTK_SIGNAL_FUNC(cb_itemsignal), "expand");
	gtk_signal_connect (GTK_OBJECT(aitem), "collapse",
			    GTK_SIGNAL_FUNC(cb_itemsignal), "collapse");
	/* Add it to the parent tree */
	gtk_tree_append (GTK_TREE(real_tree), aitem);

	gtk_widget_show (aitem);

	gtk_object_set_user_data(GTK_OBJECT(aitem), (gpointer)my_data);

	fprintf(stdout, "Added: %s, len: %u\n", dirp->name, dirlen);

	if (dirp->smbc_type != SMBC_FILE && dirp->smbc_type != SMBC_IPC_SHARE){
	  
	  subtree = gtk_tree_new();
	  gtk_tree_item_set_subtree(GTK_TREE_ITEM(aitem), subtree);

	}

	(char *)dirp += dirlen;
	err -= dirlen;

      }

    }

    smbc_closedir(dh);   

  }

}

/* Note that this is never called */
static void cb_unselect_child( GtkWidget *root_tree,
                               GtkWidget *child,
                               GtkWidget *subtree )
{
  g_print ("unselect_child called for root tree %p, subtree %p, child %p\n",
	   root_tree, subtree, child);
}

/* Note that this is called every time the user clicks on an item,
   whether it is already selected or not. */
static void cb_select_child (GtkWidget *root_tree, GtkWidget *child,
			     GtkWidget *subtree)
{
  g_print ("select_child called for root tree %p, subtree %p, child %p\n",
	   root_tree, subtree, child);
}

static void cb_selection_changed( GtkWidget *tree )
{
  GList *i;
  
  g_print ("selection_change called for tree %p\n", tree);
  g_print ("selected objects are:\n");

  i = GTK_TREE_SELECTION(tree);
  while (i){
    gchar *name;
    GtkLabel *label;
    GtkWidget *item;

    /* Get a GtkWidget pointer from the list node */
    item = GTK_WIDGET (i->data);
    label = GTK_LABEL (GTK_BIN (item)->child);
    gtk_label_get (label, &name);
    g_print ("\t%s on level %d\n", name, GTK_TREE
	     (item->parent)->level);
    i = i->next;
  }
}

/*
 * Expand or collapse the whole network ...
 */
static void cb_wholenet(GtkWidget *item, gchar *signame)
{
  GtkWidget *real_tree, *aitem, *subtree;
  gchar *name;
  GtkLabel *label;
  gint dh, err, dirlen;
  char dirbuf[512];
  struct smbc_dirent *dirp;
  
  label = GTK_LABEL (GTK_BIN (item)->child);
  gtk_label_get (label, &name);
  g_print ("%s called for item %s->%p, level %d\n", signame, name,
	   item, GTK_TREE (item->parent)->level);

  if (strncmp(signame, "expand", 6) == 0) { /* Expand called */

    if ((dh = smbc_opendir("smb://")) < 0) { /* Handle error */

      g_print("cb_wholenet: Could not open dir smbc://, %s\n",
	      strerror(errno));

      gtk_main_quit();

      return;

    }

    real_tree = GTK_TREE_ITEM_SUBTREE(item);  /* Get the subtree */

    while ((err = smbc_getdents(dh, (struct smbc_dirent *)dirbuf, 
				sizeof(dirbuf))) != 0) {

      if (err < 0) { /* An error, report it */

	g_print("cb_wholenet: Could not read dir smbc://, %s\n",
		strerror(errno));

	gtk_main_quit();

	return;

      }

      dirp = (struct smbc_dirent *)dirbuf;

      while (err > 0) {
	struct tree_data *my_data;

	dirlen = dirp->dirlen;

	my_data = make_tree_data(dirp->smbc_type, dirp->name);

	aitem = gtk_tree_item_new_with_label(dirp->name);

	/* Connect all GtkItem:: and GtkTreeItem:: signals */
	gtk_signal_connect (GTK_OBJECT(aitem), "select",
			    GTK_SIGNAL_FUNC(cb_itemsignal), "select");
	gtk_signal_connect (GTK_OBJECT(aitem), "deselect",
			    GTK_SIGNAL_FUNC(cb_itemsignal), "deselect");
	gtk_signal_connect (GTK_OBJECT(aitem), "toggle",
			    GTK_SIGNAL_FUNC(cb_itemsignal), "toggle");
	gtk_signal_connect (GTK_OBJECT(aitem), "expand",
			    GTK_SIGNAL_FUNC(cb_itemsignal), "expand");
	gtk_signal_connect (GTK_OBJECT(aitem), "collapse",
			    GTK_SIGNAL_FUNC(cb_itemsignal), "collapse");

	gtk_tree_append (GTK_TREE(real_tree), aitem);
	/* Show it - this can be done at any time */
	gtk_widget_show (aitem);

	gtk_object_set_user_data(GTK_OBJECT(aitem), (gpointer)my_data);

	fprintf(stdout, "Added: %s, len: %u\n", dirp->name, dirlen);

	subtree = gtk_tree_new();

	gtk_tree_item_set_subtree(GTK_TREE_ITEM(aitem), subtree);

	(char *)dirp += dirlen;
	err -= dirlen;

      }

    }

    smbc_closedir(dh);   

  }

}

static void 
auth_fn(char *server, char *share,
	     char **workgroup, char **username, char **password)
{

  *workgroup = "";
  *username = "test";
  *password = "test";

}

int main( int   argc,
          char *argv[] )
{
  GtkWidget *window, *scrolled_win, *tree;
  GtkWidget *subtree, *item;
  gint err, dh;
  gint i;
  char dirbuf[512];
  struct smbc_dirent *dirp;

  gtk_init (&argc, &argv);

  /* Init the smbclient library */

  err = smbc_init(auth_fn, "", 10);

  /* a generic toplevel window */
  window = gtk_window_new (GTK_WINDOW_TOPLEVEL);
  gtk_signal_connect (GTK_OBJECT(window), "delete_event",
		      GTK_SIGNAL_FUNC (gtk_main_quit), NULL);
  gtk_container_set_border_width (GTK_CONTAINER(window), 5);

  /* A generic scrolled window */
  scrolled_win = gtk_scrolled_window_new (NULL, NULL);
  gtk_scrolled_window_set_policy (GTK_SCROLLED_WINDOW (scrolled_win),
				  GTK_POLICY_AUTOMATIC,
				  GTK_POLICY_AUTOMATIC);
  gtk_widget_set_usize (scrolled_win, 150, 200);
  gtk_container_add (GTK_CONTAINER(window), scrolled_win);
  gtk_widget_show (scrolled_win);
  
  /* Create the root tree */
  tree = gtk_tree_new();
  g_print ("root tree is %p\n", tree);
  /* connect all GtkTree:: signals */
  gtk_signal_connect (GTK_OBJECT(tree), "select_child",
		      GTK_SIGNAL_FUNC(cb_select_child), tree);
  gtk_signal_connect (GTK_OBJECT(tree), "unselect_child",
		      GTK_SIGNAL_FUNC(cb_unselect_child), tree);
  gtk_signal_connect (GTK_OBJECT(tree), "selection_changed",
		      GTK_SIGNAL_FUNC(cb_selection_changed), tree);
  /* Add it to the scrolled window */
  gtk_scrolled_window_add_with_viewport (GTK_SCROLLED_WINDOW(scrolled_win),
                                         tree);
  /* Set the selection mode */
  gtk_tree_set_selection_mode (GTK_TREE(tree),
			       GTK_SELECTION_MULTIPLE);
  /* Show it */
  gtk_widget_show (tree);

  /* Now, build the top level display ... */

  if ((dh = smbc_opendir("smb:///")) < 0) {

    fprintf(stderr, "Could not list default workgroup: smb:///: %s\n",
	    strerror(errno));

    exit(1);

  }

  /* Create a tree item for Whole Network */

  item = gtk_tree_item_new_with_label ("Whole Network");
  /* Connect all GtkItem:: and GtkTreeItem:: signals */
  gtk_signal_connect (GTK_OBJECT(item), "select",
		      GTK_SIGNAL_FUNC(cb_itemsignal), "select");
  gtk_signal_connect (GTK_OBJECT(item), "deselect",
		      GTK_SIGNAL_FUNC(cb_itemsignal), "deselect");
  gtk_signal_connect (GTK_OBJECT(item), "toggle",
		      GTK_SIGNAL_FUNC(cb_itemsignal), "toggle");
  gtk_signal_connect (GTK_OBJECT(item), "expand",
		      GTK_SIGNAL_FUNC(cb_wholenet), "expand");
  gtk_signal_connect (GTK_OBJECT(item), "collapse",
		      GTK_SIGNAL_FUNC(cb_wholenet), "collapse");
  /* Add it to the parent tree */
  gtk_tree_append (GTK_TREE(tree), item);
  /* Show it - this can be done at any time */
  gtk_widget_show (item);

  subtree = gtk_tree_new();  /* A subtree for Whole Network */

  gtk_tree_item_set_subtree(GTK_TREE_ITEM(item), subtree);

  /* Now, get the items in smb:/// and add them to the tree */

  dirp = (struct smbc_dirent *)dirbuf;

  while ((err = smbc_getdents(dh, (struct smbc_dirent *)dirbuf, 
			      sizeof(dirbuf))) != 0) {

    if (err < 0) { /* Handle the error */

      fprintf(stderr, "Could not read directory for smbc:///: %s\n",
	      strerror(errno));

      exit(1);

    }

    fprintf(stdout, "Dir len: %u\n", err);

    while (err > 0) { /* Extract each entry and make a sub-tree */
      struct tree_data *my_data;
      int dirlen = dirp->dirlen;

      my_data = make_tree_data(dirp->smbc_type, dirp->name);

      item = gtk_tree_item_new_with_label(dirp->name);
      /* Connect all GtkItem:: and GtkTreeItem:: signals */
      gtk_signal_connect (GTK_OBJECT(item), "select",
			  GTK_SIGNAL_FUNC(cb_itemsignal), "select");
      gtk_signal_connect (GTK_OBJECT(item), "deselect",
			  GTK_SIGNAL_FUNC(cb_itemsignal), "deselect");
      gtk_signal_connect (GTK_OBJECT(item), "toggle",
			  GTK_SIGNAL_FUNC(cb_itemsignal), "toggle");
      gtk_signal_connect (GTK_OBJECT(item), "expand",
			  GTK_SIGNAL_FUNC(cb_itemsignal), "expand");
      gtk_signal_connect (GTK_OBJECT(item), "collapse",
			  GTK_SIGNAL_FUNC(cb_itemsignal), "collapse");
      /* Add it to the parent tree */
      gtk_tree_append (GTK_TREE(tree), item);
      /* Show it - this can be done at any time */
      gtk_widget_show (item);

      gtk_object_set_user_data(GTK_OBJECT(item), (gpointer)my_data);

      fprintf(stdout, "Added: %s, len: %u\n", dirp->name, dirlen);

      subtree = gtk_tree_new();

      gtk_tree_item_set_subtree(GTK_TREE_ITEM(item), subtree);

      (char *)dirp += dirlen;
      err -= dirlen;

    }

  }

  smbc_closedir(dh); /* FIXME, check for error :-) */

  /* Show the window and loop endlessly */
  gtk_widget_show (window);
  gtk_main();
  return 0;
}
/* example-end */