Use moira_schema.h

[moira.git] / gen / warehouse.pc

/* $Id$
 *
 * This generates a database extract from the users table for the MIT
 * Warehouse.
 *
 * Copyright (C) 1996-1998 by the Massachusetts Institute of Technology.
 * For copying and distribution information, please see the file
 * <mit-copyright.h>.
 */

#include <mit-copyright.h>
#include <moira.h>
#include <moira_site.h>

#include <sys/stat.h>
#include <sys/types.h>

#include <fcntl.h>
#include <ctype.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include <krb.h>
#include <gdss.h>

#include "util.h"

EXEC SQL INCLUDE sqlca;

RCSID("$Header$");

char *whoami = "warehouse.gen";
char *db = "moira/moira";

#ifndef WAREHOUSE_SUBDIR
#define WAREHOUSE_SUBDIR "warehouse"
#endif

char warehouse_dir[MAXPATHLEN];

int main(int argc, char **argv)
{
  int out = STDOUT_FILENO;
  char *outf = NULL, outft[MAXPATHLEN];
  char control_fn[MAXPATHLEN], cmd[256];
  FILE *f;
  struct stat sb;
  int flag1;
  SigInfo si;
  int records = 0;
  EXEC SQL BEGIN DECLARE SECTION;
  char login[USERS_LOGIN_SIZE], sig[USERS_SIGNATURE_SIZE];
  char id[USERS_CLEARID_SIZE], kname[MAX_K_NAME_SZ];
  char fname[USERS_FIRST_SIZE], lname[USERS_LAST_SIZE];
  char middle[USERS_MIDDLE_SIZE];
  EXEC SQL VAR sig IS STRING(USERS_SIGNATURE_SIZE);
  int timestamp, sigwho;
  struct {
    char login[12];
    char id[12];
    char fname[20];
    char lname[20];
    char middle[20];
    char sig[256];
  } outrec;
  EXEC SQL END DECLARE SECTION;

  initialize_sms_error_table();

  sprintf(warehouse_dir, "%s/%s", DCM_DIR, WAREHOUSE_SUBDIR);

  EXEC SQL CONNECT :db;

  if (argc == 2)
    {
      if (stat(argv[1], &sb) == 0)
	{
	  if (ModDiff(&flag1, "users", sb.st_mtime) == 0 && flag1 < 0)
	    {
	      fprintf(stderr, "File %s does not need to be rebuilt.\n",
		      argv[1]);
	      exit(MR_NO_CHANGE);
	    }
	}
      outf = argv[1];
      sprintf(outft, "%s/username_id.map", warehouse_dir);
      if ((out = open(outft, O_WRONLY|O_APPEND|O_CREAT|O_TRUNC, 0666)) < 0)
	{
	  fprintf(stderr, "unable to open %s for output\n", outf);
	  exit(MR_OCONFIG);
	}
    }
  else if (argc != 1)
    {
      fprintf(stderr, "usage: %s [outfile]\n", argv[0]);
      exit(MR_ARGS);
    }
  else
    outf = NULL;

  /* The following is declarative, not executed,
   * and so is dependent on where it is in the file,
   * not in the order of execution of statements.
   */
  EXEC SQL WHENEVER SQLERROR GOTO sqlerr;

  EXEC SQL DECLARE x CURSOR FOR
    SELECT u.login, u.clearid, u.signature, u.sigdate, s.string, u.sigwho,
    u.first, u.last, u.middle
    FROM users u, strings s
    WHERE (u.status = 1 or u.status = 2 or u.status = 5 OR
	   u.status = 6 or u.status = 7)
    AND u.sigwho = s.string_id;
  EXEC SQL OPEN x;
  while (1)
    {
      EXEC SQL FETCH x INTO :login, :id, :sig, :timestamp, :kname, :sigwho,
	:fname, :lname, :middle;
      if (sqlca.sqlcode)
	break;
      if (id == 0)
	continue;
      if (!isdigit(id[1]))
	continue;
      strtrim(login);
      strtrim(id);
      strtrim(kname);
      strtrim(fname);
      strtrim(lname);
      strtrim(middle);
      memset(&outrec, 0, sizeof(outrec));
      strcpy(outrec.login, login);
      strcpy(outrec.id, id);
      strcpy(outrec.fname, fname);
      strcpy(outrec.lname, lname);
      strcpy(outrec.middle, middle);
      if (sigwho)
	{
	  si.timestamp = timestamp;
	  si.SigInfoVersion = 0; /* XXXXXX this isn't used */
	  kname_parse(si.pname, si.pinst, si.prealm, kname);
	  si.rawsig = (unsigned char *)sig;
	  GDSS_Recompose(&si, outrec.sig);
	}
      write(out, &outrec, sizeof(outrec));
      records++;
    }
  EXEC SQL CLOSE x;

  EXEC SQL COMMIT;

  if (close(out))
    {
      perror("close failed");
      exit(MR_CCONFIG);
    }
  sprintf(control_fn, "%s/username_id.ctl", warehouse_dir);

  f = fopen(control_fn, "w");
  if (!f)
    {
      perror("creating username_id control file");
      exit(MR_OCONFIG);
    }
  fprintf(f, "username_id.map --- Moira feed for MIT Data Warehouse\n");
  fprintf(f, "%d records\n", records);
  fprintf(f, "%d bytes\n", records * (12 + 12 + 20 + 20 + 20 + 256));
  fprintf(f, "Full feed\n");
  fprintf(f, "Fixed format -- binary\n");
  fclose(f);
  if (outf)
    {
      fprintf(stderr, "Building tar file.\n");
      sprintf(cmd, "(cd %s; tar cf - . ) | compress > %s",
	      warehouse_dir, outf);
      if (system(cmd))
	exit(MR_TAR_FAIL);
    }

  exit(MR_SUCCESS);

sqlerr:
  db_error(sqlca.sqlcode);
  exit(MR_DBMS_ERR);
}