postgresql/contrib/tsearch2/dict_ispell.c

/* 
 * ISpell interface
 * Teodor Sigaev <teodor@sigaev.ru>
 */
#include <stdlib.h>
#include <string.h>
#include <ctype.h>

#include "postgres.h"

#include "dict.h"
#include "common.h"
#include "ispell/spell.h"

typedef struct {
	StopList	stoplist;
	IspellDict	obj;
} DictISpell;

PG_FUNCTION_INFO_V1(spell_init);
Datum spell_init(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(spell_lexize);
Datum spell_lexize(PG_FUNCTION_ARGS);

static void
freeDictISpell(DictISpell  *d) {
	FreeIspell(&(d->obj));
	freestoplist(&(d->stoplist));
	free(d);
}

Datum 
spell_init(PG_FUNCTION_ARGS) {
	DictISpell	*d;
	Map *cfg, *pcfg;
	text *in;
	bool affloaded=false, dictloaded=false, stoploaded=false;

	if ( PG_ARGISNULL(0) || PG_GETARG_POINTER(0)==NULL )
		ereport(ERROR,
				(errcode(ERRCODE_CONFIG_FILE_ERROR),
				 errmsg("ISpell confguration error")));
 
	d = (DictISpell*)malloc( sizeof(DictISpell) );
	if ( !d )
		ereport(ERROR,
				(errcode(ERRCODE_OUT_OF_MEMORY),
				 errmsg("out of memory")));
	memset(d,0,sizeof(DictISpell));
	d->stoplist.wordop=lowerstr;

	in = PG_GETARG_TEXT_P(0);
	parse_cfgdict(in,&cfg);
	PG_FREE_IF_COPY(in, 0);
	pcfg=cfg;
	while(pcfg->key) {
		if ( strcasecmp("DictFile", pcfg->key) == 0 ) {
			if ( dictloaded ) {
				freeDictISpell(d);
				ereport(ERROR,
						(errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
						 errmsg("dictionary already loaded")));
			}
			if ( ImportDictionary(&(d->obj), pcfg->value) ) {
				freeDictISpell(d);
				ereport(ERROR,
						(errcode(ERRCODE_CONFIG_FILE_ERROR),
						 errmsg("could not load dictionary file \"%s\"",
								pcfg->value)));
			}
			dictloaded=true;
		} else if ( strcasecmp("AffFile", pcfg->key) == 0 ) {
			if ( affloaded ) {
				freeDictISpell(d);
				ereport(ERROR,
						(errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
						 errmsg("affixes already loaded")));
			}
			if ( ImportAffixes(&(d->obj), pcfg->value) ) {
				freeDictISpell(d);
				ereport(ERROR,
						(errcode(ERRCODE_CONFIG_FILE_ERROR),
						 errmsg("could not load affix file \"%s\"",
								pcfg->value)));
			}
			affloaded=true;
		} else if ( strcasecmp("StopFile", pcfg->key) == 0 ) {
			text *tmp=char2text(pcfg->value);
			if ( stoploaded ) {
				freeDictISpell(d);
				ereport(ERROR,
						(errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
						 errmsg("stop words already loaded")));
			}
			readstoplist(tmp, &(d->stoplist));
			sortstoplist(&(d->stoplist));
			pfree(tmp);
			stoploaded=true;
		} else {
			freeDictISpell(d);
			ereport(ERROR,
					(errcode(ERRCODE_SYNTAX_ERROR),
					 errmsg("unrecognized option: %s => %s",
							 pcfg->key, pcfg->value)));
		}
		pfree(pcfg->key);
		pfree(pcfg->value);
		pcfg++;
	}
	pfree(cfg);

	if ( affloaded && dictloaded ) {
		SortDictionary(&(d->obj));
		SortAffixes(&(d->obj));
	} else if ( !affloaded ) {
		freeDictISpell(d);
		ereport(ERROR,
				(errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
				 errmsg("no affixes")));
	} else {
		freeDictISpell(d);
		ereport(ERROR,
				(errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
				 errmsg("no dictionary")));
	}

	PG_RETURN_POINTER(d);
}

Datum
spell_lexize(PG_FUNCTION_ARGS) {
	DictISpell *d = (DictISpell*)PG_GETARG_POINTER(0);
	char       *in = (char*)PG_GETARG_POINTER(1);
	char *txt;
	char	**res;
	char 	**ptr, **cptr;

	if ( !PG_GETARG_INT32(2) )
		PG_RETURN_POINTER(NULL);

	res=palloc(sizeof(char*)*2);
	txt = pnstrdup(in, PG_GETARG_INT32(2));
	res=NormalizeWord(&(d->obj), txt);
	pfree(txt);

	if ( res==NULL ) 
		PG_RETURN_POINTER(NULL);

	ptr=cptr=res;
	while(*ptr) {
		if ( searchstoplist(&(d->stoplist),*ptr) ) {
			pfree(*ptr);
			*ptr=NULL;
			ptr++;
		} else {
			*cptr=*ptr;
			cptr++; ptr++;
		}
	}
	*cptr=NULL;

	PG_RETURN_POINTER(res);
}