186 lines
3.2 KiB
C
186 lines
3.2 KiB
C
/*
|
|
* ISpell interface
|
|
* Teodor Sigaev <teodor@sigaev.ru>
|
|
*/
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <ctype.h>
|
|
#include <errno.h>
|
|
|
|
#include "postgres.h"
|
|
|
|
#include "dict.h"
|
|
#include "common.h"
|
|
|
|
#define SYNBUFLEN 4096
|
|
typedef struct
|
|
{
|
|
char *in;
|
|
char *out;
|
|
} Syn;
|
|
|
|
typedef struct
|
|
{
|
|
int len;
|
|
Syn *syn;
|
|
} DictSyn;
|
|
|
|
PG_FUNCTION_INFO_V1(syn_init);
|
|
Datum syn_init(PG_FUNCTION_ARGS);
|
|
|
|
PG_FUNCTION_INFO_V1(syn_lexize);
|
|
Datum syn_lexize(PG_FUNCTION_ARGS);
|
|
|
|
static char *
|
|
findwrd(char *in, char **end)
|
|
{
|
|
char *start;
|
|
|
|
*end = NULL;
|
|
while (*in && isspace(*in))
|
|
in++;
|
|
|
|
if (!in)
|
|
return NULL;
|
|
start = in;
|
|
|
|
while (*in && !isspace(*in))
|
|
in++;
|
|
|
|
*end = in;
|
|
return start;
|
|
}
|
|
|
|
static int
|
|
compareSyn(const void *a, const void *b)
|
|
{
|
|
return strcmp(((Syn *) a)->in, ((Syn *) b)->in);
|
|
}
|
|
|
|
|
|
Datum
|
|
syn_init(PG_FUNCTION_ARGS)
|
|
{
|
|
text *in;
|
|
DictSyn *d;
|
|
int cur = 0;
|
|
FILE *fin;
|
|
char *filename;
|
|
char buf[SYNBUFLEN];
|
|
char *starti,
|
|
*starto,
|
|
*end = NULL;
|
|
int slen;
|
|
|
|
if (PG_ARGISNULL(0) || PG_GETARG_POINTER(0) == NULL)
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_INVALID_PARAMETER_VALUE),
|
|
errmsg("NULL config")));
|
|
|
|
in = PG_GETARG_TEXT_P(0);
|
|
if (VARSIZE(in) - VARHDRSZ == 0)
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_INVALID_PARAMETER_VALUE),
|
|
errmsg("VOID config")));
|
|
|
|
filename = text2char(in);
|
|
PG_FREE_IF_COPY(in, 0);
|
|
if ((fin = fopen(filename, "r")) == NULL)
|
|
ereport(ERROR,
|
|
(errcode_for_file_access(),
|
|
errmsg("could not open file \"%s\": %m",
|
|
filename)));
|
|
|
|
d = (DictSyn *) malloc(sizeof(DictSyn));
|
|
if (!d)
|
|
{
|
|
fclose(fin);
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_OUT_OF_MEMORY),
|
|
errmsg("out of memory")));
|
|
}
|
|
memset(d, 0, sizeof(DictSyn));
|
|
|
|
while (fgets(buf, SYNBUFLEN, fin))
|
|
{
|
|
slen = strlen(buf) - 1;
|
|
buf[slen] = '\0';
|
|
if (*buf == '\0')
|
|
continue;
|
|
if (cur == d->len)
|
|
{
|
|
d->len = (d->len) ? 2 * d->len : 16;
|
|
d->syn = (Syn *) realloc(d->syn, sizeof(Syn) * d->len);
|
|
if (!d->syn)
|
|
{
|
|
fclose(fin);
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_OUT_OF_MEMORY),
|
|
errmsg("out of memory")));
|
|
}
|
|
}
|
|
|
|
starti = findwrd(buf, &end);
|
|
if (!starti)
|
|
continue;
|
|
*end = '\0';
|
|
if (end >= buf + slen)
|
|
continue;
|
|
|
|
starto = findwrd(end + 1, &end);
|
|
if (!starto)
|
|
continue;
|
|
*end = '\0';
|
|
|
|
d->syn[cur].in = strdup(lowerstr(starti));
|
|
d->syn[cur].out = strdup(lowerstr(starto));
|
|
if (!(d->syn[cur].in && d->syn[cur].out))
|
|
{
|
|
fclose(fin);
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_OUT_OF_MEMORY),
|
|
errmsg("out of memory")));
|
|
}
|
|
|
|
cur++;
|
|
}
|
|
|
|
fclose(fin);
|
|
|
|
d->len = cur;
|
|
if (cur > 1)
|
|
qsort(d->syn, d->len, sizeof(Syn), compareSyn);
|
|
|
|
pfree(filename);
|
|
PG_RETURN_POINTER(d);
|
|
}
|
|
|
|
Datum
|
|
syn_lexize(PG_FUNCTION_ARGS)
|
|
{
|
|
DictSyn *d = (DictSyn *) PG_GETARG_POINTER(0);
|
|
char *in = (char *) PG_GETARG_POINTER(1);
|
|
Syn key,
|
|
*found;
|
|
char **res = NULL;
|
|
|
|
if (!PG_GETARG_INT32(2))
|
|
PG_RETURN_POINTER(NULL);
|
|
|
|
key.out = NULL;
|
|
key.in = lowerstr(pnstrdup(in, PG_GETARG_INT32(2)));
|
|
|
|
found = (Syn *) bsearch(&key, d->syn, d->len, sizeof(Syn), compareSyn);
|
|
pfree(key.in);
|
|
|
|
if (!found)
|
|
PG_RETURN_POINTER(NULL);
|
|
|
|
res = palloc(sizeof(char *) * 2);
|
|
|
|
res[0] = pstrdup(found->out);
|
|
res[1] = NULL;
|
|
|
|
PG_RETURN_POINTER(res);
|
|
}
|