Xinqi Bao's Git
1 /* See LICENSE file for copyright and license details. */
11 #include <X11/Xatom.h>
12 #include <X11/Xutil.h>
14 #include <X11/extensions/Xinerama.h>
16 #include <X11/Xft/Xft.h>
22 #define INTERSECT(x,y,w,h,r) (MAX(0, MIN((x)+(w),(r).x_org+(r).width) - MAX((x),(r).x_org)) \
23 * MAX(0, MIN((y)+(h),(r).y_org+(r).height) - MAX((y),(r).y_org)))
24 #define LENGTH(X) (sizeof X / sizeof X[0])
25 #define TEXTNW(X,N) (drw_font_getexts_width(drw->fonts[0], (X), (N)))
26 #define TEXTW(X) (drw_text(drw, 0, 0, 0, 0, (X), 0) + drw->fonts[0]->h)
29 enum { SchemeNorm
, SchemeSel
, SchemeOut
, SchemeLast
}; /* color schemes */
33 struct item
*left
, *right
;
37 static char text
[BUFSIZ
] = "";
38 static int bh
, mw
, mh
;
39 static int sw
, sh
; /* X display screen geometry width, height */
40 static int inputw
, promptw
;
42 static struct item
*items
= NULL
;
43 static struct item
*matches
, *matchend
;
44 static struct item
*prev
, *curr
, *next
, *sel
;
45 static int mon
= -1, screen
;
47 static Atom clip
, utf8
;
49 static Window root
, win
;
52 static ClrScheme scheme
[SchemeLast
];
57 static int (*fstrncmp
)(const char *, const char *, size_t) = strncmp
;
58 static char *(*fstrstr
)(const char *, const char *) = strstr
;
61 appenditem(struct item
*item
, struct item
**list
, struct item
**last
)
64 (*last
)->right
= item
;
81 n
= mw
- (promptw
+ inputw
+ TEXTW("<") + TEXTW(">"));
82 /* calculate which items will begin the next page and previous page */
83 for (i
= 0, next
= curr
; next
; next
= next
->right
)
84 if ((i
+= (lines
> 0) ? bh
: MIN(TEXTW(next
->text
), n
)) > n
)
86 for (i
= 0, prev
= curr
; prev
&& prev
->left
; prev
= prev
->left
)
87 if ((i
+= (lines
> 0) ? bh
: MIN(TEXTW(prev
->left
->text
), n
)) > n
)
96 XUngrabKey(dpy
, AnyKey
, AnyModifier
, root
);
97 for (i
= 0; i
< SchemeLast
; i
++) {
98 drw_clr_free(scheme
[i
].bg
);
99 drw_clr_free(scheme
[i
].fg
);
107 cistrstr(const char *s
, const char *sub
)
111 for (len
= strlen(sub
); *s
; s
++)
112 if (!strncasecmp(s
, sub
, len
))
122 int x
= 0, y
= 0, h
= bh
, w
;
124 drw_setscheme(drw
, &scheme
[SchemeNorm
]);
125 drw_rect(drw
, 0, 0, mw
, mh
, 1, 1, 1);
127 if (prompt
&& *prompt
) {
128 drw_setscheme(drw
, &scheme
[SchemeSel
]);
129 drw_text(drw
, x
, 0, promptw
, bh
, prompt
, 0);
132 /* draw input field */
133 w
= (lines
> 0 || !matches
) ? mw
- x
: inputw
;
134 drw_setscheme(drw
, &scheme
[SchemeNorm
]);
135 drw_text(drw
, x
, 0, w
, bh
, text
, 0);
137 if ((curpos
= TEXTNW(text
, cursor
) + bh
/ 2 - 2) < w
) {
138 drw_setscheme(drw
, &scheme
[SchemeNorm
]);
139 drw_rect(drw
, x
+ curpos
+ 2, 2, 1, bh
- 4, 1, 1, 0);
143 /* draw vertical list */
145 for (item
= curr
; item
!= next
; item
= item
->right
) {
148 drw_setscheme(drw
, &scheme
[SchemeSel
]);
150 drw_setscheme(drw
, &scheme
[SchemeOut
]);
152 drw_setscheme(drw
, &scheme
[SchemeNorm
]);
154 drw_text(drw
, x
, y
, w
, bh
, item
->text
, 0);
156 } else if (matches
) {
157 /* draw horizontal list */
161 drw_setscheme(drw
, &scheme
[SchemeNorm
]);
162 drw_text(drw
, x
, 0, w
, bh
, "<", 0);
164 for (item
= curr
; item
!= next
; item
= item
->right
) {
166 w
= MIN(TEXTW(item
->text
), mw
- x
- TEXTW(">"));
169 drw_setscheme(drw
, &scheme
[SchemeSel
]);
171 drw_setscheme(drw
, &scheme
[SchemeOut
]);
173 drw_setscheme(drw
, &scheme
[SchemeNorm
]);
174 drw_text(drw
, x
, 0, w
, bh
, item
->text
, 0);
179 drw_setscheme(drw
, &scheme
[SchemeNorm
]);
180 drw_text(drw
, x
, 0, w
, bh
, ">", 0);
183 drw_map(drw
, win
, 0, 0, mw
, mh
);
189 struct timespec ts
= { .tv_sec
= 0, .tv_nsec
= 1000000 };
192 /* try to grab keyboard, we may have to wait for another process to ungrab */
193 for (i
= 0; i
< 1000; i
++) {
194 if (XGrabKeyboard(dpy
, DefaultRootWindow(dpy
), True
,
195 GrabModeAsync
, GrabModeAsync
, CurrentTime
) == GrabSuccess
)
197 nanosleep(&ts
, NULL
);
199 die("cannot grab keyboard\n");
205 static char **tokv
= NULL
;
208 char buf
[sizeof text
], *s
;
210 size_t len
, textsize
;
211 struct item
*item
, *lprefix
, *lsubstr
, *prefixend
, *substrend
;
214 /* separate input text into tokens to be matched individually */
215 for (s
= strtok(buf
, " "); s
; tokv
[tokc
- 1] = s
, s
= strtok(NULL
, " "))
216 if (++tokc
> tokn
&& !(tokv
= realloc(tokv
, ++tokn
* sizeof *tokv
)))
217 die("cannot realloc %u bytes\n", tokn
* sizeof *tokv
);
218 len
= tokc
? strlen(tokv
[0]) : 0;
220 matches
= lprefix
= lsubstr
= matchend
= prefixend
= substrend
= NULL
;
221 textsize
= strlen(text
);
222 for (item
= items
; item
&& item
->text
; item
++) {
223 for (i
= 0; i
< tokc
; i
++)
224 if (!fstrstr(item
->text
, tokv
[i
]))
226 if (i
!= tokc
) /* not all tokens match */
228 /* exact matches go first, then prefixes, then substrings */
229 if (!tokc
|| !fstrncmp(text
, item
->text
, textsize
))
230 appenditem(item
, &matches
, &matchend
);
231 else if (!fstrncmp(tokv
[0], item
->text
, len
))
232 appenditem(item
, &lprefix
, &prefixend
);
234 appenditem(item
, &lsubstr
, &substrend
);
238 matchend
->right
= lprefix
;
239 lprefix
->left
= matchend
;
242 matchend
= prefixend
;
246 matchend
->right
= lsubstr
;
247 lsubstr
->left
= matchend
;
250 matchend
= substrend
;
252 curr
= sel
= matches
;
257 insert(const char *str
, ssize_t n
)
259 if (strlen(text
) + n
> sizeof text
- 1)
261 /* move existing text out of the way, insert new text, and update cursor */
262 memmove(&text
[cursor
+ n
], &text
[cursor
], sizeof text
- cursor
- MAX(n
, 0));
264 memcpy(&text
[cursor
], str
, n
);
274 /* return location of next utf8 rune in the given direction (+1 or -1) */
275 for (n
= cursor
+ inc
; n
+ inc
>= 0 && (text
[n
] & 0xc0) == 0x80; n
+= inc
)
281 keypress(XKeyEvent
*ev
)
285 KeySym ksym
= NoSymbol
;
288 len
= XmbLookupString(xic
, ev
, buf
, sizeof buf
, &ksym
, &status
);
289 if (status
== XBufferOverflow
)
291 if (ev
->state
& ControlMask
)
293 case XK_a
: ksym
= XK_Home
; break;
294 case XK_b
: ksym
= XK_Left
; break;
295 case XK_c
: ksym
= XK_Escape
; break;
296 case XK_d
: ksym
= XK_Delete
; break;
297 case XK_e
: ksym
= XK_End
; break;
298 case XK_f
: ksym
= XK_Right
; break;
299 case XK_g
: ksym
= XK_Escape
; break;
300 case XK_h
: ksym
= XK_BackSpace
; break;
301 case XK_i
: ksym
= XK_Tab
; break;
302 case XK_j
: /* fallthrough */
303 case XK_J
: /* fallthrough */
304 case XK_m
: /* fallthrough */
305 case XK_M
: ksym
= XK_Return
; ev
->state
&= ~ControlMask
; break;
306 case XK_n
: ksym
= XK_Down
; break;
307 case XK_p
: ksym
= XK_Up
; break;
309 case XK_k
: /* delete right */
313 case XK_u
: /* delete left */
314 insert(NULL
, 0 - cursor
);
316 case XK_w
: /* delete word */
317 while (cursor
> 0 && strchr(worddelimiters
,
319 insert(NULL
, nextrune(-1) - cursor
);
320 while (cursor
> 0 && !strchr(worddelimiters
,
322 insert(NULL
, nextrune(-1) - cursor
);
324 case XK_y
: /* paste selection */
326 XConvertSelection(dpy
, (ev
->state
& ShiftMask
) ? clip
: XA_PRIMARY
,
327 utf8
, utf8
, win
, CurrentTime
);
338 else if (ev
->state
& Mod1Mask
)
340 case XK_g
: ksym
= XK_Home
; break;
341 case XK_G
: ksym
= XK_End
; break;
342 case XK_h
: ksym
= XK_Up
; break;
343 case XK_j
: ksym
= XK_Next
; break;
344 case XK_k
: ksym
= XK_Prior
; break;
345 case XK_l
: ksym
= XK_Down
; break;
355 if (text
[cursor
] == '\0')
357 cursor
= nextrune(+1);
362 insert(NULL
, nextrune(-1) - cursor
);
365 if (text
[cursor
] != '\0') {
366 cursor
= strlen(text
);
370 /* jump to end of list and position items in reverse */
375 while (next
&& (curr
= curr
->right
))
384 if (sel
== matches
) {
388 sel
= curr
= matches
;
392 if (cursor
> 0 && (!sel
|| !sel
->left
|| lines
> 0)) {
393 cursor
= nextrune(-1);
400 if (sel
&& sel
->left
&& (sel
= sel
->left
)->right
== curr
) {
419 puts((sel
&& !(ev
->state
& ShiftMask
)) ? sel
->text
: text
);
420 if (!(ev
->state
& ControlMask
)) {
428 if (text
[cursor
] != '\0') {
429 cursor
= nextrune(+1);
436 if (sel
&& sel
->right
&& (sel
= sel
->right
) == next
) {
444 strncpy(text
, sel
->text
, sizeof text
- 1);
445 text
[sizeof text
- 1] = '\0';
446 cursor
= strlen(text
);
461 /* we have been given the current selection, now insert it into input */
462 XGetWindowProperty(dpy
, win
, utf8
, 0, (sizeof text
/ 4) + 1, False
,
463 utf8
, &da
, &di
, &dl
, &dl
, (unsigned char **)&p
);
464 insert(p
, (q
= strchr(p
, '\n')) ? q
- p
: (ssize_t
)strlen(p
));
472 char buf
[sizeof text
], *p
, *maxstr
= NULL
;
473 size_t i
, max
= 0, size
= 0;
475 /* read each line from stdin and add it to the item list */
476 for (i
= 0; fgets(buf
, sizeof buf
, stdin
); i
++) {
477 if (i
+ 1 >= size
/ sizeof *items
)
478 if (!(items
= realloc(items
, (size
+= BUFSIZ
))))
479 die("cannot realloc %u bytes:", size
);
480 if ((p
= strchr(buf
, '\n')))
482 if (!(items
[i
].text
= strdup(buf
)))
483 die("cannot strdup %u bytes:", strlen(buf
) + 1);
485 if (strlen(items
[i
].text
) > max
)
486 max
= strlen(maxstr
= items
[i
].text
);
489 items
[i
].text
= NULL
;
490 inputw
= maxstr
? TEXTW(maxstr
) : 0;
491 lines
= MIN(lines
, i
);
499 while (!XNextEvent(dpy
, &ev
)) {
500 if (XFilterEvent(&ev
, win
))
504 if (ev
.xexpose
.count
== 0)
505 drw_map(drw
, win
, 0, 0, mw
, mh
);
510 case SelectionNotify
:
511 if (ev
.xselection
.property
== utf8
)
514 case VisibilityNotify
:
515 if (ev
.xvisibility
.state
!= VisibilityUnobscured
)
516 XRaiseWindow(dpy
, win
);
526 XSetWindowAttributes swa
;
529 XineramaScreenInfo
*info
;
530 Window w
, pw
, dw
, *dws
;
531 XWindowAttributes wa
;
532 int a
, j
, di
, n
, i
= 0, area
= 0;
536 /* init appearance */
537 scheme
[SchemeNorm
].bg
= drw_clr_create(drw
, normbgcolor
);
538 scheme
[SchemeNorm
].fg
= drw_clr_create(drw
, normfgcolor
);
539 scheme
[SchemeSel
].bg
= drw_clr_create(drw
, selbgcolor
);
540 scheme
[SchemeSel
].fg
= drw_clr_create(drw
, selfgcolor
);
541 scheme
[SchemeOut
].bg
= drw_clr_create(drw
, outbgcolor
);
542 scheme
[SchemeOut
].fg
= drw_clr_create(drw
, outfgcolor
);
544 clip
= XInternAtom(dpy
, "CLIPBOARD", False
);
545 utf8
= XInternAtom(dpy
, "UTF8_STRING", False
);
547 /* calculate menu geometry */
548 bh
= drw
->fonts
[0]->h
+ 2;
549 lines
= MAX(lines
, 0);
550 mh
= (lines
+ 1) * bh
;
552 if ((info
= XineramaQueryScreens(dpy
, &n
))) {
553 XGetInputFocus(dpy
, &w
, &di
);
554 if (mon
!= -1 && mon
< n
)
556 else if (w
!= root
&& w
!= PointerRoot
&& w
!= None
) {
557 /* find top-level window containing current input focus */
559 if (XQueryTree(dpy
, (pw
= w
), &dw
, &w
, &dws
, &du
) && dws
)
561 } while (w
!= root
&& w
!= pw
);
562 /* find xinerama screen with which the window intersects most */
563 if (XGetWindowAttributes(dpy
, pw
, &wa
))
564 for (j
= 0; j
< n
; j
++)
565 if ((a
= INTERSECT(wa
.x
, wa
.y
, wa
.width
, wa
.height
, info
[j
])) > area
) {
570 /* no focused window is on screen, so use pointer location instead */
571 if (mon
== -1 && !area
&& XQueryPointer(dpy
, root
, &dw
, &dw
, &x
, &y
, &di
, &di
, &du
))
572 for (i
= 0; i
< n
; i
++)
573 if (INTERSECT(x
, y
, 1, 1, info
[i
]))
577 y
= info
[i
].y_org
+ (topbar
? 0 : info
[i
].height
- mh
);
584 y
= topbar
? 0 : sh
- mh
;
587 promptw
= (prompt
&& *prompt
) ? TEXTW(prompt
) : 0;
588 inputw
= MIN(inputw
, mw
/3);
591 /* create menu window */
592 swa
.override_redirect
= True
;
593 swa
.background_pixel
= scheme
[SchemeNorm
].bg
->pix
;
594 swa
.event_mask
= ExposureMask
| KeyPressMask
| VisibilityChangeMask
;
595 win
= XCreateWindow(dpy
, root
, x
, y
, mw
, mh
, 0,
596 DefaultDepth(dpy
, screen
), CopyFromParent
,
597 DefaultVisual(dpy
, screen
),
598 CWOverrideRedirect
| CWBackPixel
| CWEventMask
, &swa
);
600 /* open input methods */
601 xim
= XOpenIM(dpy
, NULL
, NULL
, NULL
);
602 xic
= XCreateIC(xim
, XNInputStyle
, XIMPreeditNothing
| XIMStatusNothing
,
603 XNClientWindow
, win
, XNFocusWindow
, win
, NULL
);
605 XMapRaised(dpy
, win
);
606 drw_resize(drw
, mw
, mh
);
613 fputs("usage: dmenu [-b] [-f] [-i] [-l lines] [-p prompt] [-fn font] [-m monitor]\n"
614 " [-nb color] [-nf color] [-sb color] [-sf color] [-v]\n", stderr
);
619 main(int argc
, char *argv
[])
623 for (i
= 1; i
< argc
; i
++)
624 /* these options take no arguments */
625 if (!strcmp(argv
[i
], "-v")) { /* prints version information */
626 puts("dmenu-"VERSION
);
628 } else if (!strcmp(argv
[i
], "-b")) /* appears at the bottom of the screen */
630 else if (!strcmp(argv
[i
], "-f")) /* grabs keyboard before reading stdin */
632 else if (!strcmp(argv
[i
], "-i")) { /* case-insensitive item matching */
633 fstrncmp
= strncasecmp
;
635 } else if (i
+ 1 == argc
)
637 /* these options take one argument */
638 else if (!strcmp(argv
[i
], "-l")) /* number of lines in vertical list */
639 lines
= atoi(argv
[++i
]);
640 else if (!strcmp(argv
[i
], "-m"))
641 mon
= atoi(argv
[++i
]);
642 else if (!strcmp(argv
[i
], "-p")) /* adds prompt to left of input field */
644 else if (!strcmp(argv
[i
], "-fn")) /* font or font set */
645 fonts
[0] = argv
[++i
];
646 else if (!strcmp(argv
[i
], "-nb")) /* normal background color */
647 normbgcolor
= argv
[++i
];
648 else if (!strcmp(argv
[i
], "-nf")) /* normal foreground color */
649 normfgcolor
= argv
[++i
];
650 else if (!strcmp(argv
[i
], "-sb")) /* selected background color */
651 selbgcolor
= argv
[++i
];
652 else if (!strcmp(argv
[i
], "-sf")) /* selected foreground color */
653 selfgcolor
= argv
[++i
];
657 if (!setlocale(LC_CTYPE
, "") || !XSupportsLocale())
658 fputs("warning: no locale support\n", stderr
);
659 if (!(dpy
= XOpenDisplay(NULL
)))
660 die("cannot open display\n");
661 screen
= DefaultScreen(dpy
);
662 root
= RootWindow(dpy
, screen
);
663 sw
= DisplayWidth(dpy
, screen
);
664 sh
= DisplayHeight(dpy
, screen
);
665 drw
= drw_create(dpy
, screen
, root
, sw
, sh
);
666 drw_load_fonts(drw
, fonts
, LENGTH(fonts
));
668 die("no fonts could be loaded.\n");
669 drw_setscheme(drw
, &scheme
[SchemeNorm
]);
681 return 1; /* unreachable */