For text filter, return only header if elementSetName=H . elementSetName=R
[idzebra-moved-to-github.git] / recctrl / rectext.c
index 9993d70..1de6567 100644 (file)
@@ -1,55 +1,26 @@
-/*
- * Copyright (C) 1994-1998, Index Data I/S 
- * All rights reserved.
- * Sebastian Hammer, Adam Dickmeiss
- *
- * $Log: rectext.c,v $
- * Revision 1.6  1998-02-10 12:03:06  adam
- * Implemented Sort.
- *
- * Revision 1.5  1997/10/27 14:33:06  adam
- * Moved towards generic character mapping depending on "structure"
- * field in abstract syntax file. Fixed a few memory leaks. Fixed
- * bug with negative integers when doing searches with relational
- * operators.
- *
- * Revision 1.4  1996/11/04 14:09:16  adam
- * Minor changes.
- *
- * Revision 1.3  1996/11/01 09:00:33  adam
- * This simple "text" format now supports element specs B and M.
- *
- * Revision 1.2  1996/10/29 14:02:45  adam
- * Uses buffered read to speed up things.
- *
- * Revision 1.1  1996/10/11 10:57:28  adam
- * New module recctrl. Used to manage records (extract/retrieval).
- *
- * Revision 1.7  1996/01/17 14:57:55  adam
- * Prototype changed for reader functions in extract/retrieve. File
- *  is identified by 'void *' instead of 'int.
- *
- * Revision 1.6  1995/10/10  13:59:24  adam
- * Function rset_open changed its wflag parameter to general flags.
- *
- * Revision 1.5  1995/10/02  16:24:39  adam
- * Use attribute actually used in search requests.
- *
- * Revision 1.4  1995/10/02  15:42:55  adam
- * Extract uses file descriptors instead of FILE pointers.
- *
- * Revision 1.3  1995/09/28  09:19:45  adam
- * xfree/xmalloc used everywhere.
- * Extract/retrieve method seems to work for text records.
- *
- * Revision 1.2  1995/09/15  14:45:21  adam
- * Retrieve control.
- * Work on truncation.
- *
- * Revision 1.1  1995/09/14  07:48:25  adam
- * Record control management.
- *
- */
+/* $Id: rectext.c,v 1.16 2004-03-22 20:52:11 adam Exp $
+   Copyright (C) 1995,1996,1997,1998,1999,2000,2001,2002,2003,2004
+   Index Data Aps
+
+This file is part of the Zebra server.
+
+Zebra 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, or (at your option) any later
+version.
+
+Zebra 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 Zebra; see the file LICENSE.zebra.  If not, write to the
+Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
+02111-1307, USA.
+*/
+
+
 #include <stdio.h>
 #include <assert.h>
 #include <ctype.h>
 #include <zebrautl.h>
 #include "rectext.h"
 
-static void text_init (void)
+static void *text_init (RecType recType)
+{
+    return 0;
+}
+
+static void text_destroy (void *clientData)
 {
 }
 
@@ -70,10 +46,10 @@ struct buf_info {
 
 struct buf_info *buf_open (struct recExtractCtrl *p)
 {
-    struct buf_info *fi = xmalloc (sizeof(*fi));
+    struct buf_info *fi = (struct buf_info *) xmalloc (sizeof(*fi));
 
     fi->p = p;
-    fi->buf = xmalloc (4096);
+    fi->buf = (char *) xmalloc (4096);
     fi->offset = 1;
     fi->max = 1;
     return fi;
@@ -100,11 +76,11 @@ void buf_close (struct buf_info *fi)
     xfree (fi);
 }
 
-static int text_extract (struct recExtractCtrl *p)
+static int text_extract (void *clientData, struct recExtractCtrl *p)
 {
     char w[512];
     RecWord recWord;
-    int r, seqno = 1;
+    int r;
     struct buf_info *fi = buf_open (p);
 
     (*p->init)(p, &recWord);
@@ -121,22 +97,22 @@ static int text_extract (struct recExtractCtrl *p)
         }
         if (i)
         {
-            recWord.seqno = seqno++;
             recWord.string = w;
            recWord.length = i;
-            (*p->add)(&recWord);
+            (*p->tokenAdd)(&recWord);
         }
     } while (r > 0);
     buf_close (fi);
-    return 0;
+    return RECCTRL_EXTRACT_OK;
 }
 
-static int text_retrieve (struct recRetrieveCtrl *p)
+static int text_retrieve (void *clientData, struct recRetrieveCtrl *p)
 {
     int r, text_ptr = 0;
     static char *text_buf = NULL;
     static int text_size = 0;
-    int start_flag = 1;
+    int make_header = 1;
+    int make_body = 1;
     const char *elementSetName = NULL;
     int no_lines = 0;
 
@@ -144,6 +120,21 @@ static int text_retrieve (struct recRetrieveCtrl *p)
         p->comp->u.simple->which == Z_ElementSetNames_generic)
         elementSetName = p->comp->u.simple->u.generic;
 
+    if (elementSetName)
+    {
+       /* don't make header for the R(aw) element set name */
+       if (!strcmp(elementSetName, "R"))
+       {
+           make_header = 0;
+           make_body = 1;
+       }
+       /* only make header for the H(eader) element set name */
+       else if (!strcmp(elementSetName, "H"))
+       {
+           make_header = 1;
+           make_body = 0;
+       }
+    }
     while (1)
     {
         if (text_ptr + 4096 >= text_size)
@@ -151,7 +142,7 @@ static int text_retrieve (struct recRetrieveCtrl *p)
             char *nb;
 
             text_size = 2*text_size + 8192;
-            nb = xmalloc (text_size);
+            nb = (char *) xmalloc (text_size);
             if (text_buf)
             {
                 memcpy (nb, text_buf, text_ptr);
@@ -159,9 +150,8 @@ static int text_retrieve (struct recRetrieveCtrl *p)
             }
             text_buf = nb;
         }
-        if (start_flag)
+        if (make_header && text_ptr == 0)
         {
-            start_flag = 0;
             if (p->score >= 0)
             {
                 sprintf (text_buf, "Rank: %d\n", p->score);
@@ -169,7 +159,15 @@ static int text_retrieve (struct recRetrieveCtrl *p)
             }
             sprintf (text_buf + text_ptr, "Local Number: %d\n", p->localno);
             text_ptr = strlen(text_buf);
+           if (p->fname)
+           {
+               sprintf (text_buf + text_ptr, "Filename: %s\n", p->fname);
+               text_ptr = strlen(text_buf);
+           }
+           strcpy(text_buf+text_ptr++, "\n");
         }
+       if (!make_body)
+           break;
         r = (*p->readf)(p->fh, text_buf + text_ptr, 4096);
         if (r <= 0)
             break;
@@ -205,6 +203,7 @@ static int text_retrieve (struct recRetrieveCtrl *p)
 static struct recType text_type = {
     "text",
     text_init,
+    text_destroy,
     text_extract,
     text_retrieve
 };