pcdmutils.cpp

是一本关于

// PCDMUtils.cpp : Implementation of CPCDMUtils
#include "stdafx.h"
#include "PCDMDLL.h"
#include "PCDMUtils.h"

/////////////////////////////////////////////////////////////////////////////
// CPCDMUtils
#include "PCDMAccessDB.h"

STDMETHODIMP CPCDMUtils::AnalyzeURL(BSTR bstrURL)
{
int i=0,idx=0;
long CmdClass=0;
BSTR strURL,strCMD;
HRESULT hr;

	while (GetURL(idx,bstrURL,&strURL)==S_OK)
	{
		hr=GetClass(strURL,&CmdClass);
		if (hr==S_OK && CmdClass==PCDMCLASS_ACCESSDB)
		{
			if (GetCommand(strURL,&strCMD)==S_OK)
			{
				hr=ExecuteADBCmd(strCMD);
				if (hr==ERROR_DUP_NAME)
					return hr;
			}
		}
		idx++;
	}
	return S_OK;
}

STDMETHODIMP CPCDMUtils::GetURL(LONG Index, BSTR InURL, BSTR *OutURL)
{
int pos,oldPos,j;

	pos=0;
	for (j=0;j<Index+1;j++)
	{
		oldPos=pos;
		pos=InstrPos(InURL,pos,TEXT(";"));
		if (pos<0)	break;
	}
	if (pos>0)
		MidStr(InURL,oldPos,pos-1,OutURL);
	else
		return S_FALSE;
	return S_OK;
}

STDMETHODIMP CPCDMUtils::GetClass(BSTR InURL, LONG *ClassID)
{
BSTR ClassStr;
long pos=-1,startpos=-1,endpos=-1;

	pos=InstrPos(InURL,0,TEXT("://"));
	if (pos>=0)
		startpos=InstrPos(InURL,pos,TEXT("/"));
	if (startpos>=0)
		endpos=InstrPos(InURL,startpos,TEXT("?"));
	if (endpos-1>0)
	{
		*ClassID=0L;
		MidStr(InURL,startpos,endpos-1,&ClassStr);
		if (!CompareString(LOCALE_USER_DEFAULT,NORM_IGNORECASE,ClassStr,-1,TEXT("PCDMAccessDB"), -1)-2)
			*ClassID=PCDMCLASS_ACCESSDB;
		else if (!CompareString(LOCALE_USER_DEFAULT,NORM_IGNORECASE,ClassStr,-1,TEXT("PCDMAccessReg"), -1)-2)
			*ClassID=PCDMCLASS_ACCESSREG;
		else if (!CompareString(LOCALE_USER_DEFAULT,NORM_IGNORECASE,ClassStr,-1,TEXT("PCDMUtils"), -1)-2)
			*ClassID=PCDMCLASS_UTILS;
		else if (!CompareString(LOCALE_USER_DEFAULT,NORM_IGNORECASE,ClassStr,-1,TEXT("PCDMInput"), -1)-2)
			*ClassID=PCDMCLASS_INPUT;
		else if (!CompareString(LOCALE_USER_DEFAULT,NORM_IGNORECASE,ClassStr,-1,TEXT("PCDMTransfer"), -1)-2)
			*ClassID=PCDMCLASS_TRANSFER;
	}
	else
		return S_FALSE;
	return S_OK;
}

STDMETHODIMP CPCDMUtils::GetCommand(BSTR InURL, BSTR *Command)
{
int pos=-1;
	pos=InstrPos(InURL,0,TEXT("?"));
	if (pos>=0)
		MidStr(InURL,pos,lstrlen(InURL),Command);
	else
		return S_FALSE;
	return S_OK;
}

STDMETHODIMP CPCDMUtils::ExecuteADBCmd(BSTR Command)
{
BSTR Cmd,Table,Value,IDX;
VARIANT	tValue,tlValue;
TCHAR tStr[20];
HRESULT hre,hret=S_OK;

	VariantInit(&tValue);
	VariantInit(&tlValue);
	HRESULT hr=GetTagValue(Command,TEXT("Command"),&Cmd);
	if (hr==S_OK)
	{
		tValue.vt=VT_BSTR;
		tValue.bstrVal=SysAllocString(Cmd);
		VariantChangeType(&tlValue,&tValue,0,VT_I4);
		switch (tlValue.lVal)
		{
			case ADBCMD_INSERT:
				{
					int i=0;
					long DBHandle;
					hr=GetTagValue(Command,TEXT("Table"),&Table);
					if (hr==S_OK)
					{
						CComObject<CPCDMAccessDB>* comDB = new CComObject<CPCDMAccessDB>();
						HRESULT hr = comDB->FinalConstruct();
						hr=GetTagValue(Command,TEXT("IDX"),&IDX);
						if (hr!=S_OK)
							IDX=SysAllocString(TEXT("0"));
						hre=comDB->OpenTable(TEXT("\\My Documents\\PCDM.cdb"),Table,_variant_t(IDX),&DBHandle);
						if (FAILED(hre)) break;
						if (hr==S_OK)
						{
							wsprintf(tStr,TEXT("Fld%s"),IDX);
							hr=GetTagValue(Command,tStr,&Value);
							if (hr!=S_OK) 
							{
								comDB->CloseTable(DBHandle);
								hret=ERROR_INVALID_PARAMETER;
								break;
							}
							hr=comDB->FindRecord(DBHandle,_variant_t(IDX),_variant_t(Value));
							if (hr==S_OK) 
							{
								comDB->CloseTable(DBHandle);
								hret=ERROR_DUP_NAME;
								break;
							}
						}
						hr=comDB->AddNewRecord(DBHandle);
						if (FAILED(hr)) break;
						do
						{
							wsprintf(tStr,TEXT("Fld%d"),i);
							hr=GetTagValue(Command,tStr,&Value);
							if (hr==S_OK) 
								hr=comDB->SetValue(_variant_t((LONG)i),_variant_t(Value));
							i++;
						} while (hr==S_OK);
						hr=comDB->WriteRecord(DBHandle);
						comDB->CloseTable(DBHandle);
					}
				}
				break;
			default:	// Add more commands here...
				break;
		}
	}

	return hret;
}

long CPCDMUtils::InstrPos(TCHAR *InSTR, LONG Start, TCHAR *tChar)
{
unsigned int i;

	if (Start>(long)lstrlen(InSTR)) return -1;
	i=Start;
	while (i<=(lstrlen(InSTR)-lstrlen(tChar)) && CompareString(LOCALE_USER_DEFAULT,NORM_IGNORECASE,&InSTR[i],lstrlen(tChar),tChar, lstrlen(tChar))-2) 
		i++;
	i+=lstrlen(tChar);
	if (i>lstrlen(InSTR)) i=-1;
	return (long)i;
}

BOOL CPCDMUtils::MidStr(TCHAR *InSTR, long StartPos, long EndPos, BSTR *OutSTR)
{
TCHAR tmpStr[500];
long j=0,i;

	for (i=StartPos;i<EndPos;i++) 
	{
		tmpStr[j]=InSTR[i];
		j++;
	}
	tmpStr[j]=0;
	*OutSTR=SysAllocString(tmpStr);	
	return TRUE;
}

STDMETHODIMP CPCDMUtils::GetTagValue(BSTR InCMD, BSTR TagName, BSTR *TagValue)
{
int pos=0,endpos=-1;
TCHAR tTagName[255];

	wsprintf(tTagName,TEXT("%s="),TagName);
	pos=InstrPos(InCMD,0,tTagName);
	if (pos>=0)
		endpos=InstrPos(InCMD,pos,TEXT("&"));
	if (endpos-1>0 || (pos<lstrlen(InCMD) && pos>=0))
	{
		if (endpos<pos) endpos=lstrlen(InCMD)+1;
		MidStr(InCMD,pos,endpos-1,TagValue);
	}
	else
		return S_FALSE;
	return S_OK;
}