~ubuntu-branches/ubuntu/precise/migration-assistant/precise

#include <libxml/parser.h>

#include <libxml/tree.h>

#include <stdio.h>

#include <string.h>

#include <sys/types.h>

#include <sys/stat.h>

#include <dirent.h>

#include "gaim-import.h"

#include "registry.h"

#include "utils.h"

xmlDoc* gaim_new_accounts_file(void) {

    static xmlDoc* doc = NULL;

    xmlNode* node = NULL;

    char* accounts_file;

    int opts = XML_PARSE_NOBLANKS | XML_PARSE_NOERROR | XML_PARSE_RECOVER;

    if(!doc) {

	asprintf(&accounts_file, "%s/%s/%s/%s", to_location,

	    "home", to_user,

	    ".gaim/accounts.xml");

	create_file(accounts_file);

	doc = xmlReadFile(accounts_file, NULL, opts);

	free(accounts_file);

	node = xmlDocGetRootElement(doc);

	if(!node) {

	    node = xmlNewNode(NULL, (xmlChar*) "accounts");

	    xmlNewProp(node, (xmlChar*) "version", (xmlChar*) "1.0");

	    xmlDocSetRootElement(doc, node);

	}

    }

    return doc;

}

void gaim_save_accounts_file(void) {

    xmlDoc* doc = gaim_new_accounts_file();

    char* accounts_file;

    asprintf(&accounts_file, "%s/%s/%s/%s", to_location,

	    "home", to_user,

	    ".gaim/accounts.xml");

    xmlSaveFormatFile(accounts_file, doc, 1);

}

int gaim_account_exists(xmlNode* node) {

    xmlNode* ptr = NULL;

    xmlDoc* doc = NULL;

    xmlChar* username = NULL;

    xmlChar* protocol = NULL;

    node = node->children;

    while(node != NULL) {

       if(xmlStrcmp(node->name, (xmlChar*) "name") == 0)

	   username = node->children->content;

       else if(xmlStrcmp(node->name, (xmlChar*) "protocol") == 0)

	   protocol = node->children->content;

       node = node->next;

    }

    doc = gaim_new_accounts_file();

    node = xmlDocGetRootElement(doc);

    node = node->children;

    int found = 0;

    // For each account.

    while(node) {

       if(node->type == XML_ELEMENT_NODE &&

	(xmlStrcmp(node->name, (xmlChar*) "account") == 0)) {

	   ptr = node->children;

	   // For each account property.

	   while(ptr != NULL) {

	       if(xmlStrcmp(ptr->name, (xmlChar*) "name") == 0) {

		   if(xmlStrcmp(ptr->children->content,

			       (xmlChar*) username) == 0) {

		       found++;

		   }

	       }

	       if(xmlStrcmp(ptr->name, (xmlChar*) "protocol") == 0) {

		   if(xmlStrcmp(ptr->children->content,

			       (xmlChar*) protocol) == 0) {

		       found++;

		   }

	       }

	       ptr = ptr->next;

	   }

	   if(found == 2) return 1;

	   found = 0;

       }

       node = node->next;

    }

    return 0;

}

void gaim_add_account(xmlNode* node) {

    xmlDoc* doc = gaim_new_accounts_file();

    xmlNode* top = xmlDocGetRootElement(doc);

    xmlAddChild(top, node);

}

void gaim_import_gaim(void) {

    xmlDoc* doc = NULL;

    xmlNode* node, *node_copy = NULL;

    char* accounts_file;

    char* filename, *path;

    char* appdata = NULL;

    if(os_type == LINUX)

	asprintf(&accounts_file, "%s/%s/%s/%s", from_location,

		"home", from_user,

		".gaim/accounts.xml");

    else if(os_type == WINDOWSXP) {

        asprintf(&filename, "%s/%s/%s/%s", from_location,

	        "Documents and Settings", from_user, "NTUSER.DAT");

        appdata = findkey(filename, "\\Software\\Microsoft\\Windows\\"

            "CurrentVersion\\Explorer\\Shell Folders\\Local AppData");

        free(filename);

        if(!appdata) {

            printf("Couldn't find %s\n", appdata);

            return;

        }

        path = reformat_path(appdata);

        free(appdata);

	    asprintf(&accounts_file, "%s/%s/%s", from_location, path,

    		"/.gaim/accounts.xml");

        free(path);

    }

    if(!accounts_file) return;

    doc = xmlReadFile(accounts_file, NULL, XML_PARSE_NOBLANKS);

    free(accounts_file);

    if(!doc) return;

    node = xmlDocGetRootElement(doc);

    node = node->children;

    while(node) {

       if(node->type == XML_ELEMENT_NODE &&

	(xmlStrcmp(node->name, (xmlChar*) "account") == 0)) {

	   if(!gaim_account_exists(node)) {

		node_copy = xmlCopyNode(node, 1);

		gaim_add_account(node_copy);

	   }

       }

       node = node->next;

    }

    gaim_save_accounts_file();

}

void gaim_import_other(const char* proto, const char* username,

	const char* password) {

    xmlNode *account, *name, *protocol;

    account = xmlNewNode(NULL, (xmlChar*) "account");

    // protocol must come before name in the accounts file otherwise Gaim will

    // have a bad time.

    protocol = xmlNewChild(account, NULL, (xmlChar*) "protocol", (xmlChar*) proto);

    name = xmlNewChild(account, NULL, (xmlChar*) "name", (xmlChar*) username);

    if(!gaim_account_exists(account))

	gaim_add_account(account);

}

void gaim_import_yahoo(void) {

    char* username;

    char* filename;

    asprintf(&filename, "%s/%s/%s/%s", from_location,

	    "Documents and Settings", from_user, "NTUSER.DAT");

    username = findkey(filename,

	    "\\Software\\Yahoo\\pager\\Yahoo! User ID");

    free(filename);

    if(username) {

	gaim_import_other("prpl-yahoo", username, NULL);

	free(username);

	gaim_save_accounts_file();

    } else {

	puts("could not get yahoo ID from registry.");

	exit(EXIT_FAILURE);

    }

}

// FIXME: This is a terrible test.  If the user has an AIM account with the

// same username as the one they use for Windows, the test will fail.

void gaim_import_aimtriton(void) {

    DIR *dir, *dir2;

    struct dirent *entry, *entry2;

    struct stat buf;

    char* dirname, *uprofile, *filename, *cls, *path;

    char* appdata = NULL;

    asprintf(&filename, "%s/%s/%s/%s", from_location,

	    "Documents and Settings", from_user, "NTUSER.DAT");

    appdata = findkey(filename, "\\Software\\Microsoft\\Windows\\"

        "CurrentVersion\\Explorer\\Shell Folders\\Local AppData");

    free(filename);

    if(!appdata) {

        printf("Couldn't find %s\n", appdata);

        return;

    }

    path = reformat_path(appdata);

    free(appdata);

    asprintf(&dirname, "%s/%s/%s", from_location, path, "AOL/UserProfiles");

    free(path);

    dir = opendir(dirname);

    if(!dir) {

	    printf("Could not open AIM profile root directory: %s\n", dirname);

	    free(dirname);

	    return;

    }

    while((entry = readdir(dir)) != NULL) {

        if(entry->d_type != DT_DIR)

            continue;

        if(strcmp(entry->d_name,"All Users") == 0 ||

            (strcmp(entry->d_name,".") == 0 ||

             strcmp(entry->d_name,"..") == 0)) {

            continue;

        }

        asprintf(&uprofile, "%s/%s", dirname, entry->d_name);

        dir2 = opendir(uprofile);

        if(!dir2) {

            printf("Could not open user's AIM profile directory: %s\n", uprofile);

            free(uprofile);

            return;

        }

        while((entry2 = readdir(dir2)) != NULL) {

            if((strcmp(entry2->d_name,from_user) != 0) &&

                ((strcmp(entry2->d_name,".") != 0) &&

                 (strcmp(entry2->d_name,"..") != 0))) {

            asprintf(&cls, "%s/%s/cls", uprofile, entry2->d_name);

            if(stat(cls, &buf) == 0)

                gaim_import_other("prpl-oscar", entry2->d_name, NULL);

            free(cls);

            }

        }

        closedir(dir2);

        free(uprofile);

    }

    free(dirname);

    closedir(dir);

    gaim_save_accounts_file();

}