ferencd@0: //
ferencd@0: // VMime library (http://www.vmime.org)
ferencd@0: // Copyright (C) 2002-2013 Vincent Richard <vincent@vmime.org>
ferencd@0: //
ferencd@0: // This program is free software; you can redistribute it and/or
ferencd@0: // modify it under the terms of the GNU General Public License as
ferencd@0: // published by the Free Software Foundation; either version 3 of
ferencd@0: // the License, or (at your option) any later version.
ferencd@0: //
ferencd@0: // This program is distributed in the hope that it will be useful,
ferencd@0: // but WITHOUT ANY WARRANTY; without even the implied warranty of
ferencd@0: // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
ferencd@0: // General Public License for more details.
ferencd@0: //
ferencd@0: // You should have received a copy of the GNU General Public License along
ferencd@0: // with this program; if not, write to the Free Software Foundation, Inc.,
ferencd@0: // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
ferencd@0: //
ferencd@0: // Linking this library statically or dynamically with other modules is making
ferencd@0: // a combined work based on this library.  Thus, the terms and conditions of
ferencd@0: // the GNU General Public License cover the whole combination.
ferencd@0: //
ferencd@0: 
ferencd@0: //
ferencd@0: // EXAMPLE DESCRIPTION:
ferencd@0: // ====================
ferencd@0: // A simple MIME viewer to show all the components of a message.
ferencd@0: // The user interface is written using GTK+ 2.6.
ferencd@0: //
ferencd@0: // For more information, please visit:
ferencd@0: // http://www.vmime.org/
ferencd@0: //
ferencd@0: 
ferencd@0: #include <iostream>
ferencd@0: #include <fstream>
ferencd@0: #include <vector>
ferencd@0: #include <typeinfo>
ferencd@0: 
ferencd@0: #include <gtk/gtk.h>
ferencd@0: 
ferencd@0: #include "vmime/vmime.hpp"
ferencd@0: #include "vmime/platforms/posix/posixHandler.hpp"
ferencd@0: 
ferencd@0: 
ferencd@0: 
ferencd@0: GtkWidget* window = NULL;
ferencd@0: GtkWidget* treeView = NULL;
ferencd@0: GtkWidget* textArea = NULL;
ferencd@0: 
ferencd@0: GtkTreeStore* treeModel = NULL;
ferencd@0: 
ferencd@0: vmime::shared_ptr <vmime::message> currentMessage;
ferencd@0: 
ferencd@0: 
ferencd@0: 
ferencd@0: void insertRowInModel(GtkTreeStore* model, vmime::shared_ptr <vmime::component> comp, GtkTreeIter* parent = NULL)
ferencd@0: {
ferencd@0: 	GtkTreeIter iter;
ferencd@0: 
ferencd@0: 	gtk_tree_store_append(model, &iter, parent);
ferencd@0: 	gtk_tree_store_set(model, &iter, 0, typeid(*comp).name(), 1, comp.get(), -1);
ferencd@0: 
ferencd@0: 	const std::vector <vmime::shared_ptr <vmime::component> > children = comp->getChildComponents();
ferencd@0: 
ferencd@0: 	for (int i = 0 ; i < children.size() ; ++i)
ferencd@0: 	{
ferencd@0: 		insertRowInModel(model, children[i], &iter);
ferencd@0: 	}
ferencd@0: }
ferencd@0: 
ferencd@0: 
ferencd@0: void updateTreeView()
ferencd@0: {
ferencd@0: 	GtkTreeStore* model = GTK_TREE_STORE(gtk_tree_view_get_model(GTK_TREE_VIEW(treeView)));
ferencd@0: 
ferencd@0: 	g_object_ref(model);
ferencd@0: 	gtk_tree_view_set_model(GTK_TREE_VIEW(treeView), NULL);
ferencd@0: 
ferencd@0: 	gtk_tree_store_clear(model);
ferencd@0: 
ferencd@0: 	insertRowInModel(model, currentMessage);
ferencd@0: 
ferencd@0: 	gtk_tree_view_set_model(GTK_TREE_VIEW(treeView), GTK_TREE_MODEL(model));
ferencd@0: 	g_object_unref(model);
ferencd@0: }
ferencd@0: 
ferencd@0: 
ferencd@0: static void treeViewSelChanged(GtkTreeView* treeView, gpointer userData)
ferencd@0: {
ferencd@0: 	GtkTreePath* path = NULL;
ferencd@0: 	GtkTreeViewColumn* col = NULL;
ferencd@0: 
ferencd@0: 	gtk_tree_view_get_cursor(treeView, &path, &col);
ferencd@0: 
ferencd@0: 	GtkTreeIter iter;
ferencd@0: 	gtk_tree_model_get_iter(GTK_TREE_MODEL(treeModel), &iter, path);
ferencd@0: 
ferencd@0: 	vmime::component* comp = NULL;
ferencd@0: 	gtk_tree_model_get(GTK_TREE_MODEL(treeModel), &iter, 1, &comp, -1);
ferencd@0: 
ferencd@0: 	GtkTextBuffer* textBuffer = gtk_text_view_get_buffer(GTK_TEXT_VIEW(textArea));
ferencd@0: 	GtkTextIter start, end;
ferencd@0: 
ferencd@0: 	gtk_text_buffer_get_iter_at_offset(textBuffer, &start, comp->getParsedOffset());
ferencd@0: 	gtk_text_buffer_get_iter_at_offset(textBuffer, &end, comp->getParsedOffset() + comp->getParsedLength());
ferencd@0: 
ferencd@0: 	gtk_text_buffer_select_range(textBuffer, &start, &end);
ferencd@0: 
ferencd@0: 	gtk_text_view_scroll_to_iter(GTK_TEXT_VIEW(textArea), &start, 0.0, FALSE, 0.0, 0.0);
ferencd@0: 
ferencd@0: 	gtk_tree_path_free(path);
ferencd@0: }
ferencd@0: 
ferencd@0: 
ferencd@0: static void destroy(GtkWidget* widget, gpointer data)
ferencd@0: {
ferencd@0: 	gtk_main_quit();
ferencd@0: }
ferencd@0: 
ferencd@0: 
ferencd@0: void openFile(const std::string& filename)
ferencd@0: {
ferencd@0: 	std::ifstream file;
ferencd@0: 	file.open(filename.c_str(), std::ios::in | std::ios::binary);
ferencd@0: 
ferencd@0: 	if (!file)
ferencd@0: 	{
ferencd@0: 		std::cerr << "Can't open file '" << filename << "'." << std::endl;
ferencd@0: 		return;
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	vmime::string data;
ferencd@0: 	char buffer[16384];
ferencd@0: 
ferencd@0: 	do
ferencd@0: 	{
ferencd@0: 		file.read(buffer, sizeof(buffer));
ferencd@0: 		data += vmime::string(buffer, file.gcount());
ferencd@0: 	}
ferencd@0: 	while (file.gcount());
ferencd@0: 
ferencd@0: 	vmime::shared_ptr <vmime::message> msg = vmime::make_shared <vmime::message>();
ferencd@0: 	msg->parse(data);
ferencd@0: 
ferencd@0: 	currentMessage = msg;
ferencd@0: 
ferencd@0: 	char* convData = g_convert_with_fallback(data.c_str(), data.length(),
ferencd@0: 		"UTF-8", "ISO-8859-1", "?", NULL, NULL, NULL);
ferencd@0: 
ferencd@0: 	if (convData == NULL)
ferencd@0: 	{
ferencd@0: 		gtk_text_buffer_set_text(gtk_text_view_get_buffer(GTK_TEXT_VIEW(textArea)),
ferencd@0: 			"GLib UTF-8 conversion error.", -1);
ferencd@0: 	}
ferencd@0: 	else
ferencd@0: 	{
ferencd@0: 		gtk_text_buffer_set_text(gtk_text_view_get_buffer(GTK_TEXT_VIEW(textArea)),
ferencd@0: 			convData, strlen(convData));
ferencd@0: 
ferencd@0: 		g_free(convData);
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	updateTreeView();
ferencd@0: }
ferencd@0: 
ferencd@0: 
ferencd@0: static void onFileOpen()
ferencd@0: {
ferencd@0: 	GtkWidget* dlg = gtk_file_chooser_dialog_new
ferencd@0: 		("Open Message File", GTK_WINDOW(window), GTK_FILE_CHOOSER_ACTION_OPEN,
ferencd@0: 		 GTK_STOCK_CANCEL, GTK_RESPONSE_CANCEL,
ferencd@0: 		 GTK_STOCK_OPEN, GTK_RESPONSE_ACCEPT,
ferencd@0: 		 NULL);
ferencd@0: 
ferencd@0: 	if (gtk_dialog_run(GTK_DIALOG(dlg)) == GTK_RESPONSE_ACCEPT)
ferencd@0: 	{
ferencd@0: 		char* filename = gtk_file_chooser_get_filename(GTK_FILE_CHOOSER(dlg));
ferencd@0: 
ferencd@0: 		openFile(filename);
ferencd@0: 
ferencd@0: 		g_free(filename);
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	gtk_widget_destroy(dlg);
ferencd@0: }
ferencd@0: 
ferencd@0: 
ferencd@0: 
ferencd@0: // UI definitions
ferencd@0: static const GtkActionEntry uiActions[] =
ferencd@0: {
ferencd@0: 	{ "FileMenu", NULL, "_File" },
ferencd@0: 	{ "FileOpen", GTK_STOCK_OPEN, "_Open...", "<control>O", NULL, G_CALLBACK(onFileOpen) },
ferencd@0: 	{ "FileExit", GTK_STOCK_QUIT, "_Exit", "<control>Q", NULL, G_CALLBACK(gtk_main_quit) }
ferencd@0: };
ferencd@0: 
ferencd@0: static const char* uiDefinition =
ferencd@0: 	"<ui>" \
ferencd@0: 	"	<menubar name=\"MainMenuBar\">" \
ferencd@0: 	"		<menu action=\"FileMenu\">" \
ferencd@0: 	"			<menuitem action=\"FileOpen\"/>" \
ferencd@0: 	"			<menuitem action=\"FileExit\"/>" \
ferencd@0: 	"		</menu>" \
ferencd@0: 	"	</menubar>" \
ferencd@0: 	"</ui>";
ferencd@0: 
ferencd@0: 
ferencd@0: int main(int argc, char* argv[])
ferencd@0: {
ferencd@0: 	// VMime initialization
ferencd@0: 	vmime::platform::setHandler<vmime::platforms::posix::posixHandler>();
ferencd@0: 
ferencd@0: 	// GTK+ initialization
ferencd@0: 	gtk_init(&argc, &argv);
ferencd@0: 
ferencd@0: 	// Create a new window
ferencd@0: 	window = gtk_window_new(GTK_WINDOW_TOPLEVEL);
ferencd@0: 
ferencd@0: 	gtk_window_set_default_size(GTK_WINDOW(window), 800, 550);
ferencd@0: 	gtk_window_set_title(GTK_WINDOW(window), "VMime Viewer Example");
ferencd@0: 
ferencd@0: 	g_signal_connect(G_OBJECT(window), "destroy", G_CALLBACK(destroy), NULL);
ferencd@0: 
ferencd@0: 	GtkWidget* vbox = gtk_vbox_new(FALSE, 0);
ferencd@0: 	gtk_container_add(GTK_CONTAINER(window), vbox);
ferencd@0: 
ferencd@0: 	// Menubar
ferencd@0: 	GtkActionGroup* actionGroup = gtk_action_group_new ("Actions");
ferencd@0: 	gtk_action_group_add_actions(actionGroup, uiActions, G_N_ELEMENTS(uiActions), NULL);
ferencd@0: 
ferencd@0: 	GtkUIManager* uiManager = gtk_ui_manager_new();
ferencd@0: 	gtk_ui_manager_insert_action_group(uiManager, actionGroup, 1);
ferencd@0: 	gtk_ui_manager_add_ui_from_string(uiManager, uiDefinition, -1, NULL);
ferencd@0: 
ferencd@0: 	GtkWidget* menuBar = gtk_ui_manager_get_widget(uiManager, "/MainMenuBar");
ferencd@0: 
ferencd@0: 	gtk_box_pack_start(GTK_BOX(vbox), menuBar, FALSE, FALSE, 0);
ferencd@0: 
ferencd@0: 	// Horizontal Pane
ferencd@0: 	GtkWidget* hpane = gtk_hpaned_new();
ferencd@0: 	gtk_box_pack_start(GTK_BOX(vbox), hpane, TRUE, TRUE, 0);
ferencd@0: 
ferencd@0: 	// Tree View
ferencd@0: 	treeModel = gtk_tree_store_new(2, G_TYPE_STRING, G_TYPE_POINTER);
ferencd@0: 
ferencd@0: 	treeView = gtk_tree_view_new();
ferencd@0: 
ferencd@0: 	g_signal_connect(G_OBJECT(treeView), "cursor-changed", G_CALLBACK(treeViewSelChanged), NULL);
ferencd@0: 
ferencd@0: 	GtkWidget* scroll = gtk_scrolled_window_new(NULL, NULL);
ferencd@0: 
ferencd@0: 	gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(scroll), GTK_POLICY_AUTOMATIC, GTK_POLICY_AUTOMATIC);
ferencd@0: 	gtk_scrolled_window_set_shadow_type(GTK_SCROLLED_WINDOW(scroll), GTK_SHADOW_IN);
ferencd@0: 	gtk_container_add(GTK_CONTAINER(scroll), treeView);
ferencd@0: 
ferencd@0: 	gtk_paned_add1(GTK_PANED(hpane), scroll);
ferencd@0: 
ferencd@0: 	GtkTreeViewColumn* col = gtk_tree_view_column_new();
ferencd@0: 	gtk_tree_view_column_set_title(col, "Component Name");
ferencd@0: 	gtk_tree_view_append_column(GTK_TREE_VIEW(treeView), col);
ferencd@0: 
ferencd@0: 	GtkCellRenderer* renderer = gtk_cell_renderer_text_new();
ferencd@0: 
ferencd@0: 	gtk_tree_view_column_pack_start(col, renderer, TRUE);
ferencd@0: 	gtk_tree_view_column_add_attribute(col, renderer, "text", 0);
ferencd@0: 
ferencd@0: 	gtk_tree_view_set_model(GTK_TREE_VIEW(treeView), GTK_TREE_MODEL(treeModel));
ferencd@0: 	g_object_unref(treeModel);
ferencd@0: 
ferencd@0: 	gtk_widget_set_size_request(treeView, 200, 100);
ferencd@0: 
ferencd@0: 	// Text Area
ferencd@0: 	textArea = gtk_text_view_new();
ferencd@0: 
ferencd@0: 	gtk_text_view_set_editable(GTK_TEXT_VIEW(textArea), FALSE);
ferencd@0: 
ferencd@0: 	scroll = gtk_scrolled_window_new(NULL, NULL);
ferencd@0: 
ferencd@0: 	gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(scroll), GTK_POLICY_AUTOMATIC, GTK_POLICY_AUTOMATIC);
ferencd@0: 	gtk_scrolled_window_set_shadow_type(GTK_SCROLLED_WINDOW(scroll), GTK_SHADOW_IN);
ferencd@0: 	gtk_container_add(GTK_CONTAINER(scroll), textArea);
ferencd@0: 
ferencd@0: 	gtk_paned_add2(GTK_PANED(hpane), scroll);
ferencd@0: 
ferencd@0: 	// Show main window
ferencd@0: 	gtk_widget_show_all(window);
ferencd@0: 
ferencd@0: 	// GTK main loop
ferencd@0: 	gtk_main();
ferencd@0: 
ferencd@0: 	return 0;
ferencd@0: }
ferencd@0: 
ferencd@0: