path: root/plugins/Import/src/stdafx.h

/*

Import plugin for Miranda NG

Copyright (C) 2012-24 Miranda NG team (https://miranda-ng.org)

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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

*/

#pragma once

#define _CRT_NONSTDC_NO_DEPRECATE

#include <windows.h>
#include <commctrl.h> // datetimepicker
#include <ShlObj.h>

#include <io.h>
#include <malloc.h>
#include <time.h>

#include <algorithm>
#include <memory>

#include <newpluginapi.h>
#include <m_langpack.h>
#include <m_clist.h>
#include <m_contacts.h>
#include <m_db_int.h>
#include <m_database.h>
#include <m_gui.h>
#include <m_icolib.h>
#include <m_import.h>
#include <m_metacontacts.h>
#include <m_netlib.h>
#include <m_protosvc.h>
#include <m_timezones.h>
#include <pcre.h>

#include "version.h"
#include "resource.h"

// dbrw
#include "dbrw\dbrw.h"
#include "dbrw\dbintf.h"

// Global constants

#define IMPORT_MODULE  "MIMImport"

#define MS_IMPORT_SERVICE "MIMImport/Import"        // Service for main menu item
#define MS_IMPORT_CONTACT "MIMImport/ImportContact" // Service for contact menu item

struct CImportPattern : public MZeroedObject
{
	struct CRegexp
	{
		pcre16 *pattern;
		pcre16_extra *extra;
	};

	CMStringW wszName, wszExt;
	int iType = 1;
	int iCodePage = CP_UTF8;
	int iUseHeader, iUsePreMsg, iUseFilename;

	CRegexp regMessage;
	CMStringW wszIncoming, wszOutgoing;
	int iDirection, iDay, iMonth, iYear, iHours, iMinutes, iSeconds;

	CRegexp regFilename;
	int iInNick, iInUID, iOutNick, iOutUID;

	CRegexp regHeader;
	int iHdrIncoming, iHdrOutgoing, iHdrInNick, iHdrOutNick, iHdrInUID, iHdrOutUID;

	// symbols pre & after a messages
	int preRN, afterRN, preSP, afterSP;
};

struct CMPlugin : public PLUGIN<CMPlugin>
{
private:
	friend class CMirandaPageDlg;
	friend class CContactImportDlg;

	void LoadPattern(const wchar_t *pwszFileName);
	void LoadPatterns();

	OBJLIST<CImportPattern> m_patterns;

public:
	CMPlugin();

	int Load() override;
	int Unload() override;
};

// Keys
#define IMP_KEY_FR     "FirstRun"         // First run

#define WIZM_GOTOPAGE       (WM_USER+10)  // wParam=0, lParam=page class
#define WIZM_DISABLEBUTTON  (WM_USER+11)  // wParam=0:back, 1:next, 2:cancel
#define WIZM_SETCANCELTEXT  (WM_USER+12)  // lParam=(char*)newText
#define WIZM_ENABLEBUTTON   (WM_USER+13)  // wParam=0:back, 1:next, 2:cancel

class CWizardPageDlg : public CDlgBase
{
	CCtrlButton btnOk, btnCancel;

protected:
	virtual void OnNext() PURE;
	virtual void OnCancel();

public:
	CWizardPageDlg(int dlgId);

	void onClick_Ok(CCtrlButton*) { OnNext(); }
	void onClick_Cancel(CCtrlButton*) { OnCancel(); }
};

void    AddMessage(const wchar_t* fmt, ...);
LRESULT RunWizard(CWizardPageDlg*, bool bModal);
void    SetProgress(int);

class CIntroPageDlg : public CWizardPageDlg
{
public:
	CIntroPageDlg();
	int Resizer(UTILRESIZECONTROL *urc) override;

	bool OnInitDialog() override;
	void OnNext() override;
};

class CProgressPageDlg : public CWizardPageDlg
{
	CCtrlListBox m_list;
	CTimer m_timer;

public:
	CProgressPageDlg();
	int Resizer(UTILRESIZECONTROL *urc) override;

	bool OnInitDialog() override;
	void OnDestroy() override;
	void OnNext() override;
	
	void OnTimer(CTimer*);
	void OnContextMenu(CCtrlBase*);

	void AddMessage(const wchar_t *pMsg);
	void SetProgress(int);
};

class CMirandaPageDlg : public CWizardPageDlg
{
	CCtrlButton btnBack, btnOther, btnPath;
	CCtrlListBox m_list;
	CCtrlCombo m_cmbFileType;
	int m_iFileType = -1;

public:
	CMirandaPageDlg();
	int Resizer(UTILRESIZECONTROL *urc) override;

	bool OnInitDialog() override;
	void OnDestroy() override;
	void OnNext() override;
	
	void onClick_Back(CCtrlButton*);
	void onClick_Path(CCtrlButton*);
	void onClick_Other(CCtrlButton*);

	void onChange_Pattern(CCtrlCombo*);
};

class CMirandaOptionsPageDlg : public CWizardPageDlg
{
	CCtrlButton btnBack;
	CCtrlCheck chkDups;

public:
	CMirandaOptionsPageDlg();
	int Resizer(UTILRESIZECONTROL *urc) override;

	bool OnInitDialog() override;
	void OnNext() override;

	void onClick_Back(CCtrlButton*);
};

class CMirandaAdvOptionsPageDlg : public CWizardPageDlg
{
	CCtrlButton btnBack;
	CCtrlCheck chkSince, chkAll, chkOutgoing, chkIncoming, chkMsg, chkFT, chkOther;

public:
	CMirandaAdvOptionsPageDlg();

	bool OnInitDialog() override;
	void OnNext() override;

	void onClick_Back(CCtrlButton*);

	void onChange_Since(CCtrlCheck*);
	void onChange_All(CCtrlCheck*);
	void onChange_Msg(CCtrlCheck*);
	void onChange_FT(CCtrlCheck*);
	void onChange_Other(CCtrlCheck*);
};

class CFinishedPageDlg : public CWizardPageDlg
{
	typedef CWizardPageDlg CSuper;

public:
	CFinishedPageDlg();
	int Resizer(UTILRESIZECONTROL *urc) override;

	bool OnInitDialog() override;
	void OnNext() override;
	void OnCancel() override;
};

/////////////////////////////////////////////////////////////////////////////////////////

struct AccountMap
{
	AccountMap(const char *_src, int _origIdx, const wchar_t *_srcName) :
		szSrcAcc(mir_strdup(_src)),
		iSrcIndex(_origIdx),
		tszSrcName(mir_wstrdup(_srcName))
	{
	}

	ptrA szSrcAcc, szBaseProto;
	ptrW tszSrcName;
	int iSrcIndex;
	int iOrder = 0;
	PROTOACCOUNT *pa = nullptr;
};

struct ContactMap
{
	ContactMap(MCONTACT _src, MCONTACT _dst) :
		srcID(_src),
		dstID(_dst)
	{
	}

	MCONTACT srcID, dstID;
};

class CImportBatch : public MZeroedObject
{
	OBJLIST<AccountMap> m_accounts;
	OBJLIST<ContactMap> m_contacts;
	LIST<DBCachedContact> m_metas;

	uint32_t nDupes, nContactsCount, nMessagesCount, nGroupsCount, nSkippedEvents, nSkippedContacts;
	MDatabaseCommon *srcDb, *dstDb;

	bool     ImportAccounts(OBJLIST<char> &arSkippedModules);
	MCONTACT ImportContact(MCONTACT hSrc);
	void     ImportHistory(MCONTACT hContact, PROTOACCOUNT **protocol, int protoCount);
	void     ImportMeta(DBCachedContact *ccSrc);

	MCONTACT MapContact(MCONTACT hSrc);

	bool     FindDestAccount(const char *szProto);
	DBCachedContact* FindDestMeta(DBCachedContact *ccSrc);
	MCONTACT FindExistingMeta(DBCachedContact *ccSrc);
	PROTOACCOUNT* FindMyAccount(const char *szProto, const char *szBaseProto, const wchar_t *ptszName, bool bStrict);

	MCONTACT HContactFromID(const char *pszProtoName, const char *pszSetting, wchar_t *pwszID);
	MCONTACT HContactFromBlobID(const char *pszProtoName, const char *pszSetting, const DBVARIANT &dbv);
	MCONTACT HContactFromNumericID(const char *pszProtoName, const char *pszSetting, uint32_t dwID);

public:
	CImportBatch();

	__forceinline const OBJLIST<AccountMap> &AccountsMap() const { return m_accounts; }

	void CopySettings(MCONTACT srcID, const char *szSrcModule, MCONTACT dstID, const char *szDstModule);
	void DoImport();

	int myGet(MCONTACT hContact, const char *szModule, const char *szSetting, DBVARIANT *dbv);
	int myGetD(MCONTACT hContact, const char *szModule, const char *szSetting, int iDefault);
	BOOL myGetS(MCONTACT hContact, const char *szModule, const char *szSetting, char *dest);
	wchar_t* myGetWs(MCONTACT hContact, const char *szModule, const char *szSetting);

	wchar_t m_wszFileName[MAX_PATH];
	time_t m_dwSinceDate;
	int m_iOptions; // set of IOPT_* flags
	bool m_bSendQuit;
	MCONTACT m_hContact;
	CImportPattern *m_pPattern;
};

int CreateGroup(const wchar_t *name, MCONTACT hContact);

uint32_t RLInteger(const uint8_t *p);
uint32_t RLWord(const uint8_t *p);

extern bool g_bServiceMode, g_bSendQuit;
extern DATABASELINK g_patternDbLink;
extern CImportBatch *g_pBatch;
extern HWND g_hwndWizard, g_hwndAccMerge;

void RegisterIcons(void);

void RegisterMContacts();
void UnregisterMContacts();

void RegisterJson();
void UnregisterJson();

INT_PTR ImportContact(WPARAM hContact, LPARAM);