gen/confluence.pc

   1 /* $Id$
   2  *
   3  * (c) Copyright 2007 by the Massachusetts Institute of Technology.
   4  */
   5
   6 #include <mit-copyright.h>
   7 #include <moira.h>
   8 #include <moira_site.h>
   9
  10 #include <sys/stat.h>
  11
  12 #include <ctype.h>
  13 #include <stdio.h>
  14 #include <stdlib.h>
  15 #include <string.h>
  16
  17 #include "util.h"
  18
  19 EXEC SQL INCLUDE sqlca;
  20
  21 char *whoami = "confluence.gen";
  22 char *db = "moira/moira";
  23
  24 struct hash *lists;
  25
  26 void output_list(int id, void *list, void *out);
  27
  28 int main(int argc, char **argv)
  29 {
  30   char filename[MAXPATHLEN], *targetfile, *l;
  31   FILE *out = stdout;
  32   int cnt = 0;
  33   EXEC SQL BEGIN DECLARE SECTION;
  34   int lid;
  35   char lname[LIST_NAME_SIZE];
  36   EXEC SQL END DECLARE SECTION;
  37
  38   EXEC SQL CONNECT :db;
  39
  40   if (argc == 2)
  41     {
  42       targetfile = argv[1];
  43       sprintf(filename, "%s~", targetfile);
  44       if (!(out = fopen(filename, "w")))
  45         {
  46           fprintf(stderr, "unable to open %s for output\n", filename);
  47           exit(MR_OCONFIG);
  48         }
  49     }
  50   else if (argc != 1)
  51     {
  52       fprintf(stderr, "usage: %s [outfile]\n", argv[0]);
  53       exit(MR_ARGS);
  54     }
  55
  56   lists = create_hash(15000);
  57
  58   EXEC SQL DECLARE l_cursor CURSOR FOR
  59     SELECT l.list_id, l.name FROM list l
  60     WHERE l.active = 1 and l.grouplist = 1;
  61   EXEC SQL OPEN l_cursor;
  62   while (1)
  63     {
  64       EXEC SQL FETCH l_cursor INTO :lid, :lname;
  65       if (sqlca.sqlcode)
  66         break;
  67       l = strdup(strtrim(lname));
  68       if (hash_store(lists, lid, l) < 0)
  69         {
  70           fprintf(stderr, "Out of memory!\n");
  71           exit(MR_NO_MEM);
  72         }
  73       cnt++;
  74     }
  75   EXEC SQL CLOSE l_cursor;
  76   fprintf(stderr, "Loaded %d lists\n", cnt);
  77
  78   hash_step(lists, output_list, out);
  79
  80   if (fclose(out))
  81     {
  82       perror("close failed");
  83       exit(MR_CCONFIG);
  84     }
  85
  86   if (argc == 2)
  87     fix_file(targetfile);
  88   exit(MR_SUCCESS);
  89 }
  90
  91 void output_list(int id, void *list, void *out)
  92 {
  93   EXEC SQL BEGIN DECLARE SECTION;
  94   char *l = list;
  95   int lid = id;
  96   int lgid;
  97   char login[USERS_LOGIN_SIZE];
  98   char stringmember[STRINGS_STRING_SIZE];
  99   EXEC SQL END DECLARE SECTION;
 100   char *maybecomma = "";
 101
 102   EXEC SQL SELECT gid INTO :lgid FROM list WHERE list_id = :lid;
 103
 104   fprintf(out, "%s:%d:", list, lgid);
 105
 106   EXEC SQL DECLARE u_cursor CURSOR FOR
 107     SELECT UNIQUE u.login FROM users u, imembers i, list l
 108     WHERE l.list_id = :lid AND l.list_id = i.list_id AND
 109     i.member_type = 'USER' AND i.member_id = u.users_id;
 110   EXEC SQL OPEN u_cursor;
 111   while (1)
 112     {
 113       EXEC SQL FETCH u_cursor INTO :login;
 114       if (sqlca.sqlcode)
 115         break;
 116       fprintf(out, "%s%s%s", maybecomma, strtrim(login), "@mit.edu");
 117       maybecomma = ",";
 118     }
 119   EXEC SQL CLOSE u_cursor;
 120
 121   EXEC SQL DECLARE s_cursor CURSOR FOR
 122     SELECT UNIQUE s.string FROM strings s, imembers i, list l
 123     WHERE l.list_id = :lid AND l.list_id = i.list_id AND
 124     i.member_type = 'STRING' AND i.member_id = s.string_id;
 125   EXEC SQL OPEN s_cursor;
 126   while (1)
 127     {
 128       EXEC SQL FETCH s_cursor INTO :stringmember;
 129       if (sqlca.sqlcode)
 130         break;
 131       fprintf(out, "%s%s", maybecomma, strtrim(stringmember));
 132       maybecomma = ",";
 133     }
 134   EXEC SQL CLOSE s_cursor;
 135
 136   fprintf(out, "\n");
 137 }