[be485e]: src / index / recollindex.cpp History

Parent: [f773ef] (diff)

Child: [dac569] (diff)

Download this file

recollindex.cpp    139 lines (118 with data), 3.1 kB 

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
#ifndef lint

static char rcsid[] = "@(#$Id: recollindex.cpp,v 1.13 2005-12-14 11:00:48 dockes Exp $ (C) 2004 J.F.Dockes";

#endif



#include <stdio.h>

#include <signal.h>

#include <sys/stat.h>



#include <iostream>

#include <list>

#include <string>



using namespace std;



#include "debuglog.h"

#include "rclinit.h"

#include "indexer.h"

#include "smallut.h"

#include "pathut.h"





ConfIndexer *confindexer;

DbIndexer *dbindexer;



bool indexfiles(RclConfig *config, const list<string> &filenames)

{

    if (filenames.empty())

	return true;



    // Note that we do not bother to check for multiple databases,

    // which are currently a fiction anyway. 

    config->setKeyDir(path_getfather(*filenames.begin()));

    string dbdir;

    if (!config->getConfParam("dbdir", dbdir)) {

	LOGERR(("indexfiles: no database directory in "

		"configuration for %s\n", filenames.begin()->c_str()));

	return false;

    }

    dbdir = path_tildexpand(dbdir);



    dbindexer = new DbIndexer(config, dbdir);

    return dbindexer->indexFiles(filenames);

}



static void cleanup()

{

    delete confindexer;

    confindexer = 0;

    delete dbindexer;

    dbindexer = 0;

}



static void sigcleanup(int sig)

{

    fprintf(stderr, "sigcleanup\n");

    cleanup();

    exit(1);

}



static const char *thisprog;

static int     op_flags;

#define OPT_MOINS 0x1

#define OPT_z     0x2 

#define OPT_h     0x4 

#define OPT_i     0x8



static const char usage [] =

"  recollindex [-hz] \n"

"  recollindex -i <filename [filename ...]>\n"

"Options:\n"

" -h : print this message\n"

" -z : reset database before starting indexation\n\n"

" -i <filename [filename ...]> : index individual files. No db purge or stem\n"

"           database updates in this case\n"

;



static void

Usage(void)

{

    FILE *fp = (op_flags & OPT_h) ? stdout : stderr;

    fprintf(fp, "%s: usage: %s", thisprog, usage);

    exit((op_flags & OPT_h)==0);

}





int main(int argc, const char **argv)

{

    thisprog = argv[0];

    argc--; argv++;



    while (argc > 0 && **argv == '-') {

	(*argv)++;

	if (!(**argv))

	    Usage();

	while (**argv)

	    switch (*(*argv)++) {

	    case 'z': op_flags |= OPT_z; break;

	    case 'h': op_flags |= OPT_h; break;

	    case 'i': op_flags |= OPT_i; break;

	    default: Usage(); break;

	    }

    b1: argc--; argv++;

    }

    if (op_flags & OPT_h)

	Usage();

    if ((op_flags & OPT_z) && (op_flags & OPT_i))

	Usage();



    string reason;

    RclConfig *config = recollinit(cleanup, sigcleanup, reason);



    if (config == 0 || !config->ok()) {

	cerr << "Configuration problem: " << reason << endl;

	exit(1);

    }

    

    if (op_flags & OPT_i) {

	list<string> filenames;

	if (argc == 0) {

	    // Read from stdin

	    char line[1024];

	    while (fgets(line, 1023, stdin)) {

		string sl(line);

		trimstring(sl, "\n\r");

		filenames.push_back(sl);

	    }

	} else {

	    while (argc--) {

		filenames.push_back(*argv++);

	    }

	}

	exit(!indexfiles(config, filenames));

    } else {

	confindexer = new ConfIndexer(config);

	bool rezero(op_flags & OPT_z);

	exit(!confindexer->index(rezero));

    }

}