Merge branch 'master' of https://github.com/dudik/herbe into master
[herbe.git] / herbe.c
diff --git a/herbe.c b/herbe.c
index 3c65b5eed773132609222c1bc791de9c8d476029..42e7d73328437a0361184b316f01d69834a575f1 100644 (file)
--- a/herbe.c
+++ b/herbe.c
 #include <stdlib.h>
 #include <signal.h>
 #include <unistd.h>
+#include <string.h>
+#include <stdarg.h>
+#include <fcntl.h>
+#include <semaphore.h>
 
 #include "config.h"
 
+#define EXIT_ACTION 3
+
 Display *display;
 Window window;
+int exit_code = EXIT_SUCCESS;
 
-int get_eol(char *body, XftFont *font)
+static void die(const char *format, ...)
 {
-       int body_len = strlen(body);
+       va_list ap;
+       va_start(ap, format);
+       vfprintf(stderr, format, ap);
+       fprintf(stderr, "\n");
+       va_end(ap);
+       exit(EXIT_FAILURE);
+}
+
+int get_max_len(char *body, XftFont *font, int max_text_width)
+{
+       int eol = strlen(body);
        XGlyphInfo info;
-       XftTextExtentsUtf8(display, font, body, body_len, &info);
+       XftTextExtentsUtf8(display, font, (FcChar8 *)body, eol, &info);
 
-       int max_text_width = width - 2 * padding;
+       if (info.width > max_text_width)
+       {
+               eol = max_text_width / font->max_advance_width;
+               info.width = 0;
+
+               while (info.width < max_text_width)
+               {
+                       eol++;
+                       XftTextExtentsUtf8(display, font, (FcChar8 *)body, eol, &info);
+               }
+
+               eol--;
+       }
+
+       for (int i = 0; i < eol; i++)
+               if (body[i] == '\n')
+               {
+                       body[i] = ' ';
+                       return ++i;
+               }
 
        if (info.width < max_text_width)
-               return body_len;
+               return eol;
 
-       int eol = max_text_width / font->max_advance_width;
-       info.width = 0;
+       int temp = eol;
 
-       while (info.width < max_text_width)
-       {
-               eol++;
-               XftTextExtentsUtf8(display, font, body, eol, &info);
-       }
+       while (body[eol] != ' ' && eol)
+               --eol;
 
-       return --eol;
+       if (eol == 0)
+               return temp;
+       else
+               return ++eol;
 }
 
-void expire()
+void expire(int sig)
 {
        XEvent event;
        event.type = ButtonPress;
+       event.xbutton.button = (sig == SIGUSR2) ? (ACTION_BUTTON) : (DISMISS_BUTTON);
        XSendEvent(display, window, 0, 0, &event);
        XFlush(display);
 }
 
 int main(int argc, char *argv[])
 {
-       if (argc != 2)
+       if (argc == 1)
        {
-               fprintf(stderr, "Usage: herbe message\n");
-               exit(EXIT_FAILURE);
+               sem_unlink("/herbe");
+               die("Usage: %s body", argv[0]);
        }
 
-       char *body = argv[1];
+       struct sigaction act_expire, act_ignore;
+       act_expire.sa_handler = expire;
+       act_expire.sa_flags = SA_RESTART;
+       sigemptyset(&act_expire.sa_mask);
 
-       signal(SIGALRM, expire);
-       alarm(duration);
+       act_ignore.sa_handler = SIG_IGN;
+       act_ignore.sa_flags = 0;
+       sigemptyset(&act_ignore.sa_mask);
 
-       display = XOpenDisplay(NULL);
+       sigaction(SIGALRM, &act_expire, 0);
+       sigaction(SIGTERM, &act_expire, 0);
+       sigaction(SIGINT, &act_expire, 0);
 
-       if (display == NULL)
-       {
-               fprintf(stderr, "Cannot open display\n");
-               exit(EXIT_FAILURE);
-       }
+       sigaction(SIGUSR1, &act_ignore, 0);
+       sigaction(SIGUSR2, &act_ignore, 0);
+
+       display = XOpenDisplay(0);
+
+       if (display == 0)
+               die("Cannot open display");
 
        int screen = DefaultScreen(display);
        Visual *visual = DefaultVisual(display, screen);
        Colormap colormap = DefaultColormap(display, screen);
 
-       int window_width = DisplayWidth(display, screen);
-       int window_height = DisplayHeight(display, screen);
+       int screen_width = DisplayWidth(display, screen);
+       int screen_height = DisplayHeight(display, screen);
 
        XftColor color;
 
@@ -80,27 +124,46 @@ int main(int argc, char *argv[])
 
        XftFont *font = XftFontOpenName(display, screen, font_pattern);
 
-       unsigned int x = pos_x;
-       unsigned int y = pos_y;
-       unsigned int height = font->ascent - font->descent + padding * 2;
+       int num_of_lines = 0;
+       int max_text_width = width - 2 * padding;
+       int words_size = 5;
+       char **words = malloc(words_size * sizeof(char *));
+       if (!words)
+               die("malloc failed");
 
-       switch (corner)
+       for (int i = 1; i < argc; i++)
        {
-       case BOTTOM_RIGHT:
-               y = window_height - height - border_size * 2 - pos_y;
-       case TOP_RIGHT:
-               x = window_width - width - border_size * 2 - pos_x;
-               break;
-       case BOTTOM_LEFT:
-               y = window_height - height - border_size * 2 - pos_y;
+               char *body = argv[i];
+
+               for (unsigned int eol = get_max_len(body, font, max_text_width); eol <= strlen(body) && eol; body += eol, num_of_lines++, eol = get_max_len(body, font, max_text_width))
+               {
+                       if (words_size <= num_of_lines)
+                       {
+                               words = realloc(words, (words_size += 5) * sizeof(char *));
+                               if (!words)
+                                       die("malloc failed");
+                       }
+                       words[num_of_lines] = malloc((eol + 1) * sizeof(char));
+                       if (!words[num_of_lines])
+                               die("malloc failed");
+                       strncpy(words[num_of_lines], body, eol);
+                       words[num_of_lines][eol] = '\0';
+               }
        }
 
-       window = XCreateWindow(
-               display, RootWindow(display, screen), x,
-               y, width, height, border_size,
-               DefaultDepth(display, screen), CopyFromParent,
-               visual,
-               CWOverrideRedirect | CWBackPixel | CWBorderPixel, &attributes);
+       unsigned int x = pos_x;
+       unsigned int y = pos_y;
+       unsigned int text_height = font->ascent - font->descent;
+       unsigned int height = (num_of_lines - 1) * line_spacing + num_of_lines * text_height + 2 * padding;
+
+       if (corner == TOP_RIGHT || corner == BOTTOM_RIGHT)
+               x = screen_width - width - border_size * 2 - pos_x;
+
+       if (corner == BOTTOM_LEFT || corner == BOTTOM_RIGHT)
+               y = screen_height - height - border_size * 2 - pos_y;
+
+       window = XCreateWindow(display, RootWindow(display, screen), x, y, width, height, border_size, DefaultDepth(display, screen), CopyFromParent, visual,
+                                                  CWOverrideRedirect | CWBackPixel | CWBorderPixel, &attributes);
 
        XftDraw *draw = XftDrawCreate(display, window, visual, colormap);
        XftColorAllocName(display, visual, colormap, font_color, &color);
@@ -109,27 +172,49 @@ int main(int argc, char *argv[])
 
        XMapWindow(display, window);
 
-       int eol = get_eol(body, font);
+       sem_t *mutex = sem_open("/herbe", O_CREAT, 0644, 1);
+       sem_wait(mutex);
 
-       XEvent event;
+       sigaction(SIGUSR1, &act_expire, 0);
+       sigaction(SIGUSR2, &act_expire, 0);
+
+       if (duration != 0)
+               alarm(duration);
 
        while (1)
        {
+               XEvent event;
                XNextEvent(display, &event);
 
                if (event.type == Expose)
                {
                        XClearWindow(display, window);
-                       XftDrawStringUtf8(draw, &color, font, padding, height - padding, body, eol);
+                       for (int i = 0; i < num_of_lines; i++)
+                               XftDrawStringUtf8(draw, &color, font, padding, line_spacing * i + text_height * (i + 1) + padding, (FcChar8 *)words[i], strlen(words[i]));
                }
                if (event.type == ButtonPress)
-                       break;
+               {
+                       if (event.xbutton.button == DISMISS_BUTTON)
+                               break;
+                       else if (event.xbutton.button == ACTION_BUTTON)
+                       {
+                               exit_code = EXIT_ACTION;
+                               break;
+                       }
+               }
        }
 
+       sem_post(mutex);
+       sem_close(mutex);
+
+       for (int i = 0; i < num_of_lines; i++)
+               free(words[i]);
+
+       free(words);
        XftDrawDestroy(draw);
        XftColorFree(display, visual, colormap, &color);
        XftFontClose(display, font);
-
        XCloseDisplay(display);
-       return EXIT_SUCCESS;
+
+       exit(exit_code);
 }
\ No newline at end of file