forked from speedie/spmenu
661 lines
15 KiB
C
661 lines
15 KiB
C
/* spmenu - fancy dynamic menu
|
|
* See LICENSE file for copyright and license details.
|
|
*/
|
|
#include <ctype.h>
|
|
#include <locale.h>
|
|
#include <math.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <strings.h>
|
|
#include <time.h>
|
|
#include <unistd.h>
|
|
#include <signal.h>
|
|
#include <fcntl.h>
|
|
|
|
// set version
|
|
#ifndef VERSION
|
|
#define VERSION "unknown"
|
|
#endif
|
|
|
|
// check if we should enable right to left language support
|
|
#ifndef RTL
|
|
#define USERTL 0
|
|
#else
|
|
#define USERTL 1
|
|
#endif
|
|
|
|
// check if we should enable image support
|
|
#ifndef IMAGE
|
|
#define USEIMAGE 0
|
|
#else
|
|
#define USEIMAGE 1
|
|
#endif
|
|
|
|
// check if we should enable multimonitor support using libXinerama
|
|
#ifndef XINERAMA
|
|
#define USEXINERAMA 0
|
|
#else
|
|
#define USEXINERAMA 1
|
|
#endif
|
|
|
|
// check if we should enable config file support using libconfig
|
|
#ifndef CONFIG
|
|
#define USECONFIG 0
|
|
#else
|
|
#define USECONFIG 1
|
|
#endif
|
|
|
|
// check if we should enable .Xresources support
|
|
#ifndef XRESOURCES
|
|
#define USEXRESOURCES 0
|
|
#else
|
|
#define USEXRESOURCES 1
|
|
#endif
|
|
|
|
// check if we should enable Wayland support
|
|
#ifndef WAYLAND
|
|
#define USEWAYLAND 0
|
|
#else
|
|
#define USEWAYLAND 1
|
|
#endif
|
|
|
|
// check if we should enable X11 support
|
|
#ifndef X11
|
|
#define USEX 0
|
|
#else
|
|
#define USEX 1
|
|
#endif
|
|
|
|
// check if we should enable regex matching
|
|
#ifndef REGEX
|
|
#define USEREGEX 0
|
|
#else
|
|
#define USEREGEX 1
|
|
#endif
|
|
|
|
// include fribidi used for right to left language support
|
|
#if USERTL
|
|
#include <fribidi.h>
|
|
#endif
|
|
|
|
// include libraries used for image support
|
|
#if USEIMAGE
|
|
#include <errno.h>
|
|
#include <pwd.h>
|
|
#include <Imlib2.h>
|
|
#include <openssl/md5.h>
|
|
#endif
|
|
|
|
// include Xlib
|
|
#if USEX
|
|
#include <X11/Xlib.h>
|
|
#endif
|
|
|
|
// include macros and other defines
|
|
#include "libs/define.c"
|
|
|
|
enum { // click enum
|
|
ClickWindow,
|
|
ClickPrompt,
|
|
ClickInput,
|
|
ClickLArrow,
|
|
ClickItem,
|
|
ClickRArrow,
|
|
ClickNumber,
|
|
ClickCaps,
|
|
ClickMode,
|
|
ClickImage,
|
|
ClickNone,
|
|
};
|
|
|
|
struct item {
|
|
char *text;
|
|
char *nsgrtext;
|
|
char *image;
|
|
char *ex;
|
|
struct item *left, *right;
|
|
int hp;
|
|
int index;
|
|
double distance;
|
|
};
|
|
|
|
struct sp {
|
|
int bh; // height of each menu item
|
|
int mw; // width
|
|
int mh; // height
|
|
int vp; // vertical padding for bar
|
|
int sp; // side padding for bar
|
|
int lrpad; // sum of left and right padding
|
|
|
|
int mode; // current mode
|
|
int allowkeys; // interpret a keypress as an insertion?
|
|
int capslockstate; // caps lock state
|
|
int isdrawing;
|
|
|
|
int inputw; // input width
|
|
int promptw; // prompt width
|
|
int plw; // powerline width
|
|
|
|
int itemnumber; // item number
|
|
int listcount;
|
|
int listchanged;
|
|
|
|
size_t cursor; // cursor width
|
|
|
|
int ignoreconfkeys; // can be set globally if you don't want to override keybinds with config file keys
|
|
int ignoreglobalkeys; // should be set in the config file, if 1, the Keys keys array is ignored
|
|
int ignoreconfmouse; // same for mouse
|
|
int ignoreglobalmouse; // same for mouse
|
|
int forceinsertmode;
|
|
};
|
|
|
|
struct mo {
|
|
int output_width; // output width
|
|
int output_height; // output height
|
|
};
|
|
|
|
#if USEIMAGE
|
|
struct img {
|
|
int setlines; // actual lines
|
|
int flip; // %=
|
|
int longestedge; // MAX(imagewidth, imageheight)
|
|
int imagewidth; // current image width
|
|
int imageheight; // current image height
|
|
int imagegaps; // current image gaps
|
|
};
|
|
#endif
|
|
|
|
struct tx {
|
|
char modetext[64]; // mode text
|
|
char text[BUFSIZ]; // input text
|
|
char numbers[NUMBERSBUFSIZE]; // number text
|
|
char capstext[64]; // caps lock text
|
|
};
|
|
|
|
#if USEX
|
|
struct x11 {
|
|
int numlockmask;
|
|
int useargb;
|
|
int depth;
|
|
char *embed;
|
|
int screen;
|
|
Visual *visual;
|
|
Colormap cmap;
|
|
};
|
|
#endif
|
|
|
|
static struct sp sp = {0};
|
|
static struct tx tx = {0};
|
|
static struct mo mo = {0};
|
|
#if USEIMAGE
|
|
static struct img img = {0};
|
|
#endif
|
|
#if USEX
|
|
static struct x11 x11 = {0};
|
|
#endif
|
|
|
|
static struct item *items = NULL;
|
|
static struct item *history_items;
|
|
static struct item *list_items;
|
|
static struct item *matches;
|
|
static struct item *matchend;
|
|
|
|
static struct item *previousitem; // previous item
|
|
static struct item *currentitem; // current item
|
|
static struct item *nextitem; // next item
|
|
static struct item *selecteditem; // selected item
|
|
static struct item *mouseitem; // clicked item
|
|
|
|
// various headers
|
|
#include "libs/draw/draw.h"
|
|
#include "libs/main.h"
|
|
#include "libs/draw.h"
|
|
#include "libs/stream.h"
|
|
#include "libs/schemes.h"
|
|
#include "libs/arg.h"
|
|
#include "libs/x11/inc.h"
|
|
#include "libs/wl/inc.h"
|
|
#include "libs/sort.h"
|
|
#include "libs/history.h"
|
|
|
|
static Draw_t *draw; // Draw_t type, see libs/draw/draw.c
|
|
|
|
// high priority
|
|
static int hplength = 0;
|
|
static char **hpitems = NULL;
|
|
|
|
static int *sel_index = NULL;
|
|
static unsigned int sel_size = 0;
|
|
static int protocol_override = 0;
|
|
static int itemn = 0;
|
|
|
|
#if USERTL
|
|
static int isrtl = 1;
|
|
#else
|
|
static int isrtl = 0;
|
|
#endif
|
|
|
|
// declare functions
|
|
static int is_selected(size_t index);
|
|
static void calcoffsets(void);
|
|
static void recalculatenumbers(void);
|
|
static void insert(const char *str, ssize_t n);
|
|
static void cleanup(void);
|
|
static void navigatehistfile(int dir);
|
|
static void resizeclient(void);
|
|
static void get_width(void);
|
|
static void get_mh(void);
|
|
static void set_mode(void);
|
|
static void handle(void);
|
|
static void appenditem(struct item *item, struct item **list, struct item **last);
|
|
static int max_textw(void);
|
|
static size_t nextrune(int inc);
|
|
|
|
// matching
|
|
static char * cistrstr(const char *s, const char *sub);
|
|
static int (*fstrncmp)(const char *, const char *, size_t) = strncasecmp;
|
|
static char *(*fstrstr)(const char *, const char *) = cistrstr;
|
|
|
|
#if USEX
|
|
static void pastesel(void);
|
|
static void grabfocus(void); // focus doesn't need to be grabbed on wayland
|
|
#endif
|
|
|
|
static char **list;
|
|
static size_t listsize;
|
|
|
|
// user configuration
|
|
#include "libs/options.h"
|
|
#include "libs/keybinds.h"
|
|
#include "libs/mouse.h"
|
|
|
|
static char *fonts[] = { font };
|
|
|
|
// color array
|
|
#include "libs/colors.h"
|
|
|
|
// config file
|
|
#if USECONFIG
|
|
#include "libs/conf/config.h"
|
|
#include "libs/conf/config.c"
|
|
#endif
|
|
|
|
// include functions
|
|
#include "libs/img.h"
|
|
#include "libs/img.c"
|
|
#include "libs/icon.h"
|
|
#include "libs/icon.c"
|
|
#include "libs/rtl.h"
|
|
#include "libs/rtl.c"
|
|
#include "libs/sort.c"
|
|
#include "libs/match.h"
|
|
#include "libs/match.c"
|
|
#include "libs/draw.c"
|
|
#include "libs/schemes.c"
|
|
#include "libs/argv.h"
|
|
#include "libs/argv.c"
|
|
|
|
// include x11 code
|
|
#include "libs/x11/inc.c"
|
|
#include "libs/wl/inc.c"
|
|
|
|
// include more functions
|
|
#include "libs/history.c"
|
|
#include "libs/arg.c"
|
|
#include "libs/stream.c"
|
|
|
|
int is_selected(size_t index) {
|
|
for (int i = 0; i < sel_size; i++) {
|
|
if (sel_index[i] == index) {
|
|
return 1; // selected item index is size_t index
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
void appenditem(struct item *item, struct item **list, struct item **last) {
|
|
if (*last)
|
|
(*last)->right = item;
|
|
else
|
|
*list = item;
|
|
|
|
item->left = *last;
|
|
item->right = NULL;
|
|
*last = item;
|
|
}
|
|
|
|
void recalculatenumbers(void) {
|
|
unsigned int numer = 0, denom = 0, selected = 0;
|
|
struct item *item;
|
|
|
|
if (matchend) {
|
|
numer++;
|
|
|
|
// walk through items that match and add to numer
|
|
for (item = matchend; item && item->left; item = item->left)
|
|
numer++;
|
|
}
|
|
|
|
// walk through all items, matching or not and add to denom
|
|
for (item = items; item && item->text; item++) {
|
|
denom++;
|
|
}
|
|
|
|
for (int i = 0; i < sel_size; i++) {
|
|
if (sel_index[i] == -1) {
|
|
break;
|
|
}
|
|
|
|
selected++;
|
|
}
|
|
|
|
if (selected) {
|
|
snprintf(tx.numbers, NUMBERSBUFSIZE, "%d/%d/%d", numer, denom, selected);
|
|
} else {
|
|
snprintf(tx.numbers, NUMBERSBUFSIZE, "%d/%d", numer, denom);
|
|
}
|
|
}
|
|
|
|
void calcoffsets(void) {
|
|
int i, offset;
|
|
|
|
if (lines > 0) {
|
|
offset = lines * columns * sp.bh;
|
|
} else { // no lines, therefore the size of items must be decreased to fit the menu elements
|
|
int numberw = 0;
|
|
int modew = 0;
|
|
int larroww = 0;
|
|
int rarroww = 0;
|
|
int capsw = 0;
|
|
|
|
if (!hidematchcount) numberw = pango_numbers ? TEXTWM(tx.numbers) : TEXTW(tx.numbers);
|
|
if (!hidemode) modew = pango_mode ? TEXTWM(tx.modetext) : TEXTW(tx.modetext);
|
|
if (!hidelarrow) larroww = pango_leftarrow ? TEXTWM(leftarrow) : TEXTW(leftarrow);
|
|
if (!hiderarrow) rarroww = pango_rightarrow ? TEXTWM(rightarrow) : TEXTW(rightarrow);
|
|
if (!hidecaps) capsw = pango_caps ? TEXTWM(tx.capstext) : TEXTW(tx.capstext);
|
|
|
|
if (!strcmp(tx.capstext, "")) {
|
|
capsw = 0;
|
|
}
|
|
|
|
offset = sp.mw - (sp.promptw + sp.inputw + larroww + rarroww + modew + numberw + capsw + menumarginh);
|
|
}
|
|
|
|
// calculate which items will begin the next page
|
|
for (i = 0, nextitem = currentitem; nextitem; nextitem = nextitem->right) {
|
|
nextitem->nsgrtext = get_text_n_sgr(nextitem);
|
|
|
|
if ((i += (lines > 0) ? sp.bh : MIN(TEXTWM(nextitem->nsgrtext) + (powerlineitems ? !lines ? 3 * sp.plw : 0 : 0), offset)) > offset)
|
|
break;
|
|
}
|
|
|
|
// calculate which items will begin the previous page
|
|
for (i = 0, previousitem = currentitem; previousitem && previousitem->left; previousitem = previousitem->left) {
|
|
previousitem->nsgrtext = get_text_n_sgr(previousitem);
|
|
|
|
if ((i += (lines > 0) ? sp.bh : MIN(TEXTWM(previousitem->left->nsgrtext) + (powerlineitems ? !lines ? 3 * sp.plw : 0 : 0), offset)) > offset)
|
|
break;
|
|
}
|
|
}
|
|
|
|
int max_textw(void) {
|
|
int len = 0;
|
|
|
|
for (struct item *item = items; item && item->text; item++)
|
|
len = MAX(TEXTW(item->text), len);
|
|
|
|
return len;
|
|
}
|
|
|
|
void cleanup(void) {
|
|
size_t i;
|
|
|
|
#if USEIMAGE
|
|
cleanupimage(); // function frees images
|
|
#endif
|
|
|
|
// free high priority items
|
|
for (i = 0; i < hplength; ++i)
|
|
free(hpitems[i]);
|
|
|
|
// free drawing and close the display
|
|
draw_free(draw);
|
|
|
|
#if USEX
|
|
if (!protocol) {
|
|
cleanup_x11(dpy);
|
|
}
|
|
#endif
|
|
|
|
free(sel_index);
|
|
}
|
|
|
|
// This function handles case insensitive matching
|
|
char * cistrstr(const char *h, const char *n) {
|
|
size_t i;
|
|
|
|
if (!n[0])
|
|
return (char *)h;
|
|
|
|
for (; *h; ++h) {
|
|
for (i = 0; n[i] && tolower((unsigned char)n[i]) ==
|
|
tolower((unsigned char)h[i]); ++i);
|
|
|
|
if (n[i] == '\0')
|
|
return (char *)h;
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
#if USEX
|
|
void grabfocus(void) {
|
|
grabfocus_x11();
|
|
}
|
|
#endif
|
|
|
|
void insert(const char *str, ssize_t n) {
|
|
if (strlen(tx.text) + n > sizeof tx.text - 1)
|
|
return; // length of text should not exceed size
|
|
|
|
static char l[BUFSIZ] = "";
|
|
|
|
if (requirematch) {
|
|
memcpy(l, tx.text, BUFSIZ);
|
|
}
|
|
|
|
// move existing text out of the way, insert new text, and update cursor
|
|
memmove(
|
|
&tx.text[sp.cursor + n],
|
|
&tx.text[sp.cursor],
|
|
sizeof tx.text - sp.cursor - MAX(n, 0)
|
|
);
|
|
|
|
// update cursor
|
|
if (n > 0 && str && n) {
|
|
memcpy(&tx.text[sp.cursor], str, n);
|
|
}
|
|
|
|
// add to cursor position and continue matching
|
|
sp.cursor += n;
|
|
match();
|
|
|
|
if (!matches && requirematch) {
|
|
memcpy(tx.text, l, BUFSIZ);
|
|
sp.cursor -= n;
|
|
match();
|
|
}
|
|
|
|
// output on insertion
|
|
if (incremental) {
|
|
puts(tx.text);
|
|
fflush(stdout);
|
|
}
|
|
}
|
|
|
|
size_t nextrune(int inc) {
|
|
ssize_t rune;
|
|
|
|
// return location of next utf8 rune in the given direction (+1 or -1)
|
|
for (rune = sp.cursor + inc; rune + inc >= 0 && (tx.text[rune] & 0xc0) == 0x80; rune += inc)
|
|
;
|
|
|
|
return rune;
|
|
}
|
|
|
|
#if USEX
|
|
void pastesel(void) {
|
|
if (!protocol) {
|
|
pastesel_x11();
|
|
}
|
|
}
|
|
#endif
|
|
|
|
void resizeclient(void) {
|
|
#if USEWAYLAND
|
|
if (protocol) {
|
|
resizeclient_wl(&state);
|
|
} else {
|
|
#if USEX
|
|
resizeclient_x11();
|
|
#endif
|
|
}
|
|
#elif USEX
|
|
resizeclient_x11();
|
|
#endif
|
|
}
|
|
|
|
/* Width reserved for input when !lines is a fixed size of the menu width / 3
|
|
* This is reasonable, but in rare cases may cause input text to overlap
|
|
* items.
|
|
*/
|
|
void get_width(void) {
|
|
sp.inputw = sp.mw / 3;
|
|
}
|
|
|
|
void get_mh(void) {
|
|
int epad;
|
|
|
|
sp.mh = (lines + 1) * sp.bh;
|
|
sp.mh += 2 * menumarginv;
|
|
|
|
// subtract 1 line if there's nothing to draw on the top line
|
|
if ((hideprompt && hideinput && hidemode && hidematchcount && hidecaps) && lines) {
|
|
sp.mh -= sp.bh;
|
|
}
|
|
|
|
epad = 2 * menupaddingv;
|
|
|
|
// the spmenu window should not exceed the screen resolution height
|
|
if (mo.output_height && !xpos && !ypos) {
|
|
sp.mh = MIN(sp.mh, mo.output_height - epad);
|
|
|
|
if (sp.mh == mo.output_height - epad) {
|
|
lines = ((mo.output_height - epad) / sp.bh) - 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
void set_mode(void) {
|
|
if (!type) { // no typing allowed, require normal mode
|
|
sp.mode = 0;
|
|
}
|
|
|
|
// set default mode, must be done before the event loop or keybindings will not work
|
|
if (mode) {
|
|
sp.mode = 1;
|
|
sp.allowkeys = 1;
|
|
|
|
sp_strncpy(tx.modetext, instext, sizeof(tx.modetext));
|
|
} else {
|
|
sp.mode = 0;
|
|
sp.allowkeys = !sp.mode;
|
|
|
|
sp_strncpy(tx.modetext, normtext, sizeof(tx.modetext));
|
|
}
|
|
|
|
// normal mode disabled
|
|
if (sp.forceinsertmode) {
|
|
sp.mode = 1;
|
|
sp.allowkeys = 1;
|
|
hidemode = 1;
|
|
}
|
|
}
|
|
|
|
void handle(void) {
|
|
if (!protocol) {
|
|
#if USEX
|
|
handle_x11();
|
|
|
|
if (!draw_font_create(draw, fonts, LENGTH(fonts))) {
|
|
die("no fonts could be loaded.");
|
|
}
|
|
|
|
loadhistory(); // read history entries
|
|
#if USEIMAGE
|
|
store_image_vars();
|
|
#endif
|
|
|
|
// fast (-f) means we grab keyboard before reading standard input
|
|
if (fast && !isatty(0)) {
|
|
grabkeyboard_x11();
|
|
readstdin();
|
|
} else {
|
|
readstdin();
|
|
grabkeyboard_x11();
|
|
}
|
|
|
|
set_mode();
|
|
|
|
init_appearance(); // init colorschemes by reading arrays
|
|
|
|
setupdisplay_x11(); // set up display and create window
|
|
eventloop_x11(); // function is a loop which checks X11 events and calls other functions accordingly
|
|
#endif
|
|
#if USEWAYLAND
|
|
} else {
|
|
loadhistory();
|
|
#if USEIMAGE
|
|
store_image_vars();
|
|
setimageopts();
|
|
#endif
|
|
|
|
// Disable some X11 only features
|
|
menupaddingv = menupaddingh = 0;
|
|
xpos = ypos = 0;
|
|
borderwidth = 0;
|
|
managed = 0;
|
|
|
|
draw = draw_create_wl(protocol);
|
|
|
|
if (!draw_font_create(draw, fonts, LENGTH(fonts))) {
|
|
die("no fonts could be loaded.");
|
|
}
|
|
|
|
readstdin();
|
|
set_mode();
|
|
init_appearance();
|
|
|
|
handle_wl();
|
|
#endif
|
|
}
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
readargs(argc, argv); // start by reading arguments
|
|
|
|
/* pledge limits what programs can do, so here we specify what spmenu should be allowed to do
|
|
* TODO: test this on a openbsd operating system
|
|
*/
|
|
#ifdef __OpenBSD__
|
|
if (pledge("stdio rpath wpath cpath", NULL) == -1)
|
|
die("pledge");
|
|
#endif
|
|
|
|
handle();
|
|
|
|
return 1; // should be unreachable
|
|
}
|