src/subscribe.c

   1 /* Copyright (c) 2012, Bastien Dejean
   2  * All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions are met:
   6  *
   7  * 1. Redistributions of source code must retain the above copyright notice, this
   8  *    list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright notice,
  10  *    this list of conditions and the following disclaimer in the documentation
  11  *    and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
  14  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  15  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  16  * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
  17  * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  18  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  19  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  20  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  21  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  22  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  23  */
  24
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27 #include <unistd.h>
  28 #include <ctype.h>
  29 #include <stdarg.h>
  30 #include <fcntl.h>
  31 #include "bspwm.h"
  32 #include "desktop.h"
  33 #include "settings.h"
  34 #include "subscribe.h"
  35 #include "tree.h"
  36
  37 subscriber_list_t *make_subscriber(FILE *stream, char *fifo_path, int field, int count)
  38 {
  39         subscriber_list_t *sb = calloc(1, sizeof(subscriber_list_t));
  40         sb->prev = sb->next = NULL;
  41         sb->stream = stream;
  42         sb->fifo_path = fifo_path;
  43         sb->field = field;
  44         sb->count = count;
  45         return sb;
  46 }
  47
  48 void remove_subscriber(subscriber_list_t *sb)
  49 {
  50         if (sb == NULL) {
  51                 return;
  52         }
  53         subscriber_list_t *a = sb->prev;
  54         subscriber_list_t *b = sb->next;
  55         if (a != NULL) {
  56                 a->next = b;
  57         }
  58         if (b != NULL) {
  59                 b->prev = a;
  60         }
  61         if (sb == subscribe_head) {
  62                 subscribe_head = b;
  63         }
  64         if (sb == subscribe_tail) {
  65                 subscribe_tail = a;
  66         }
  67         if (restart) {
  68                 int cli_fd = fileno(sb->stream);
  69                 fcntl(cli_fd, F_SETFD, ~FD_CLOEXEC & fcntl(cli_fd, F_GETFD));
  70         } else {
  71                 fclose(sb->stream);
  72                 unlink(sb->fifo_path);
  73         }
  74         free(sb->fifo_path);
  75         free(sb);
  76 }
  77
  78 void add_subscriber(subscriber_list_t *sb)
  79 {
  80         if (subscribe_head == NULL) {
  81                 subscribe_head = subscribe_tail = sb;
  82         } else {
  83                 subscribe_tail->next = sb;
  84                 sb->prev = subscribe_tail;
  85                 subscribe_tail = sb;
  86         }
  87         int cli_fd = fileno(sb->stream);
  88         fcntl(cli_fd, F_SETFD, FD_CLOEXEC | fcntl(cli_fd, F_GETFD));
  89         if (sb->field & SBSC_MASK_REPORT) {
  90                 print_report(sb->stream);
  91                 if (sb->count-- == 1) {
  92                         remove_subscriber(sb);
  93                 }
  94         }
  95 }
  96
  97 int print_report(FILE *stream)
  98 {
  99         fprintf(stream, "%s", status_prefix);
 100         for (monitor_t *m = mon_head; m != NULL; m = m->next) {
 101                 fprintf(stream, "%c%s", (mon == m ? 'M' : 'm'), m->name);
 102                 for (desktop_t *d = m->desk_head; d != NULL; d = d->next) {
 103                         char c = (is_urgent(d) ? 'u' : (d->root == NULL ? 'f' : 'o'));
 104                         if (m->desk == d) {
 105                                 c = toupper(c);
 106                         }
 107                         fprintf(stream, ":%c%s", c, d->name);
 108                 }
 109                 if (m->desk != NULL) {
 110                         fprintf(stream, ":L%c", LAYOUT_CHR(m->desk->layout));
 111                         if (m->desk->focus != NULL) {
 112                                 node_t *n = m->desk->focus;
 113                                 if (n->client != NULL) {
 114                                         fprintf(stream, ":T%c", STATE_CHR(n->client->state));
 115                                 } else {
 116                                         fprintf(stream, ":T@");
 117                                 }
 118                                 int i = 0;
 119                                 char flags[5];
 120                                 if (n->sticky) {
 121                                         flags[i++] = 'S';
 122                                 }
 123                                 if (n->private) {
 124                                         flags[i++] = 'P';
 125                                 }
 126                                 if (n->locked) {
 127                                         flags[i++] = 'L';
 128                                 }
 129                                 if (n->marked) {
 130                                         flags[i++] = 'M';
 131                                 }
 132                                 flags[i] = '\0';
 133                                 fprintf(stream, ":G%s", flags);
 134                         }
 135                 }
 136                 if (m != mon_tail) {
 137                         fprintf(stream, "%s", ":");
 138                 }
 139         }
 140         fprintf(stream, "%s", "\n");
 141         return fflush(stream);
 142 }
 143
 144 void put_status(subscriber_mask_t mask, ...)
 145 {
 146         subscriber_list_t *sb = subscribe_head;
 147         int ret;
 148         while (sb != NULL) {
 149                 subscriber_list_t *next = sb->next;
 150                 if (sb->field & mask) {
 151                         if (sb->count > 0) {
 152                                 sb->count--;
 153                         }
 154                         if (mask == SBSC_MASK_REPORT) {
 155                                 ret = print_report(sb->stream);
 156                         } else {
 157                                 char *fmt;
 158                                 va_list args;
 159                                 va_start(args, mask);
 160                                 fmt = va_arg(args, char *);
 161                                 vfprintf(sb->stream, fmt, args);
 162                                 va_end(args);
 163                                 ret = fflush(sb->stream);
 164                         }
 165                         if (ret != 0 || sb->count == 0) {
 166                                 remove_subscriber(sb);
 167                         }
 168                 }
 169                 sb = next;
 170         }
 171 }
 172
 173 void prune_dead_subscribers(void)
 174 {
 175         subscriber_list_t *sb = subscribe_head;
 176         while (sb != NULL) {
 177                 subscriber_list_t *next = sb->next;
 178                 // Is the subscriber's stream closed?
 179                 if (write(fileno(sb->stream), NULL, 0) == -1) {
 180                         remove_subscriber(sb);
 181                 }
 182                 sb = next;
 183         }
 184 }