mirror of
https://github.com/lbonn/rofi
synced 2024-12-13 05:52:35 +00:00
485 lines
17 KiB
C
485 lines
17 KiB
C
/**
|
|
* rofi-file_browser
|
|
*
|
|
* MIT/X11 License
|
|
* Copyright (c) 2017 Qball Cow <qball@gmpclient.org>
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining
|
|
* a copy of this software and associated documentation files (the
|
|
* "Software"), to deal in the Software without restriction, including
|
|
* without limitation the rights to use, copy, modify, merge, publish,
|
|
* distribute, sublicense, and/or sell copies of the Software, and to
|
|
* permit persons to whom the Software is furnished to do so, subject to
|
|
* the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be
|
|
* included in all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
* OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
|
|
* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
|
|
* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
|
|
* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
|
|
* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
*/
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <gmodule.h>
|
|
#include <gio/gio.h>
|
|
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <dirent.h>
|
|
|
|
#include "mode.h"
|
|
#include "helper.h"
|
|
#include "mode-private.h"
|
|
#include "dialogs/filebrowser.h"
|
|
#include "rofi.h"
|
|
#include "history.h"
|
|
|
|
#include <stdint.h>
|
|
|
|
#include "rofi-icon-fetcher.h"
|
|
|
|
#define FILEBROWSER_CACHE_FILE "rofi3.filebrowsercache"
|
|
|
|
/**
|
|
* The internal data structure holding the private data of the TEST Mode.
|
|
*/
|
|
enum FBFileType
|
|
{
|
|
UP,
|
|
DIRECTORY,
|
|
RFILE,
|
|
NUM_FILE_TYPES,
|
|
};
|
|
/** Icons to use for the file type */
|
|
const char *icon_name[NUM_FILE_TYPES] =
|
|
{
|
|
"go-up",
|
|
"folder",
|
|
"gtk-file"
|
|
};
|
|
typedef struct
|
|
{
|
|
char *name;
|
|
char *path;
|
|
enum FBFileType type;
|
|
uint32_t icon_fetch_uid;
|
|
gboolean link;
|
|
} FBFile;
|
|
|
|
typedef struct
|
|
{
|
|
GFile *current_dir;
|
|
FBFile *array;
|
|
unsigned int array_length;
|
|
} FileBrowserModePrivateData;
|
|
|
|
static void free_list ( FileBrowserModePrivateData *pd )
|
|
{
|
|
for ( unsigned int i = 0; i < pd->array_length; i++ ) {
|
|
FBFile *fb = &( pd->array[i] );
|
|
g_free ( fb->name );
|
|
g_free ( fb->path );
|
|
}
|
|
g_free ( pd->array );
|
|
pd->array = NULL;
|
|
pd->array_length = 0;
|
|
}
|
|
#include <sys/types.h>
|
|
#include <dirent.h>
|
|
|
|
static gint compare ( gconstpointer a, gconstpointer b, G_GNUC_UNUSED gpointer data )
|
|
{
|
|
FBFile *fa = (FBFile*) a;
|
|
FBFile *fb = (FBFile*) b;
|
|
if ( fa->type != fb->type ) {
|
|
return fa->type - fb->type;
|
|
}
|
|
|
|
return g_strcmp0 ( fa->name, fb->name );
|
|
}
|
|
|
|
static void get_file_browser ( Mode *sw )
|
|
{
|
|
FileBrowserModePrivateData *pd = (FileBrowserModePrivateData *) mode_get_private_data ( sw );
|
|
/**
|
|
* Get the entries to display.
|
|
* this gets called on plugin initialization.
|
|
*/
|
|
char *cdir = g_file_get_path ( pd->current_dir );
|
|
DIR *dir = opendir ( cdir );
|
|
if ( dir ) {
|
|
struct dirent *rd = NULL;
|
|
while ( ( rd = readdir ( dir ) ) != NULL ) {
|
|
if ( g_strcmp0 ( rd->d_name, ".." ) == 0 ) {
|
|
pd->array = g_realloc ( pd->array, ( pd->array_length + 1 ) * sizeof ( FBFile ) );
|
|
// Rofi expects utf-8, so lets convert the filename.
|
|
pd->array[pd->array_length].name = g_strdup ( ".." );
|
|
pd->array[pd->array_length].path = NULL;
|
|
pd->array[pd->array_length].type = UP;
|
|
pd->array[pd->array_length].icon_fetch_uid = 0;
|
|
pd->array[pd->array_length].link = FALSE;
|
|
pd->array_length++;
|
|
continue;
|
|
}
|
|
else if ( rd->d_name[0] == '.' ) {
|
|
continue;
|
|
}
|
|
|
|
switch ( rd->d_type )
|
|
{
|
|
case DT_BLK:
|
|
case DT_CHR:
|
|
case DT_FIFO:
|
|
case DT_UNKNOWN:
|
|
case DT_SOCK:
|
|
default:
|
|
break;
|
|
case DT_REG:
|
|
case DT_DIR:
|
|
pd->array = g_realloc ( pd->array, ( pd->array_length + 1 ) * sizeof ( FBFile ) );
|
|
// Rofi expects utf-8, so lets convert the filename.
|
|
pd->array[pd->array_length].name = g_filename_to_utf8 ( rd->d_name, -1, NULL, NULL, NULL );
|
|
pd->array[pd->array_length].path = g_build_filename ( cdir, rd->d_name, NULL );
|
|
pd->array[pd->array_length].type = ( rd->d_type == DT_DIR ) ? DIRECTORY : RFILE;
|
|
pd->array[pd->array_length].icon_fetch_uid = 0;
|
|
pd->array[pd->array_length].link = FALSE;
|
|
pd->array_length++;
|
|
break;
|
|
case DT_LNK:
|
|
pd->array = g_realloc ( pd->array, ( pd->array_length + 1 ) * sizeof ( FBFile ) );
|
|
// Rofi expects utf-8, so lets convert the filename.
|
|
pd->array[pd->array_length].name = g_filename_to_utf8 ( rd->d_name, -1, NULL, NULL, NULL );
|
|
pd->array[pd->array_length].path = g_build_filename ( cdir, rd->d_name, NULL );
|
|
pd->array[pd->array_length].icon_fetch_uid = 0;
|
|
pd->array[pd->array_length].link = TRUE;
|
|
// Default to file.
|
|
pd->array[pd->array_length].type = RFILE;
|
|
{
|
|
// If we have link, use a stat to fine out what it is, if we fail, we mark it as file.
|
|
// TODO have a 'broken link' mode?
|
|
// Convert full path to right encoding.
|
|
char *file = g_filename_from_utf8 ( pd->array[pd->array_length].path, -1, NULL, NULL, NULL );
|
|
if ( file ) {
|
|
struct stat statbuf;
|
|
if ( stat ( file, &statbuf ) == 0 ) {
|
|
if ( S_ISDIR ( statbuf.st_mode ) ) {
|
|
pd->array[pd->array_length].type = DIRECTORY;
|
|
}
|
|
else if ( S_ISREG ( statbuf.st_mode ) ) {
|
|
pd->array[pd->array_length].type = RFILE;
|
|
}
|
|
}
|
|
else {
|
|
g_warning ( "Failed to stat file: %s, %s", file, strerror ( errno ) );
|
|
}
|
|
|
|
g_free ( file );
|
|
}
|
|
}
|
|
pd->array_length++;
|
|
break;
|
|
}
|
|
}
|
|
closedir ( dir );
|
|
}
|
|
g_qsort_with_data ( pd->array, pd->array_length, sizeof ( FBFile ), compare, NULL );
|
|
}
|
|
|
|
static int file_browser_mode_init ( Mode *sw )
|
|
{
|
|
/**
|
|
* Called on startup when enabled (in modi list)
|
|
*/
|
|
if ( mode_get_private_data ( sw ) == NULL ) {
|
|
FileBrowserModePrivateData *pd = g_malloc0 ( sizeof ( *pd ) );
|
|
mode_set_private_data ( sw, (void *) pd );
|
|
{
|
|
char *path = g_build_filename ( cache_dir, FILEBROWSER_CACHE_FILE, NULL );
|
|
char *file = NULL;
|
|
if ( g_file_get_contents ( path, &file, NULL, NULL ) ) {
|
|
if ( g_file_test ( file, G_FILE_TEST_IS_DIR ) ) {
|
|
pd->current_dir = g_file_new_for_path ( file );
|
|
}
|
|
g_free ( file );
|
|
}
|
|
// Store it based on the unique identifiers (desktop_id).
|
|
g_free ( path );
|
|
if ( pd->current_dir == NULL ) {
|
|
pd->current_dir = g_file_new_for_path ( g_get_home_dir () );
|
|
}
|
|
}
|
|
// Load content.
|
|
get_file_browser ( sw );
|
|
}
|
|
return TRUE;
|
|
}
|
|
static unsigned int file_browser_mode_get_num_entries ( const Mode *sw )
|
|
{
|
|
const FileBrowserModePrivateData *pd = (const FileBrowserModePrivateData *) mode_get_private_data ( sw );
|
|
return pd->array_length;
|
|
}
|
|
|
|
static ModeMode file_browser_mode_result ( Mode *sw, int mretv, char **input, unsigned int selected_line )
|
|
{
|
|
ModeMode retv = MODE_EXIT;
|
|
FileBrowserModePrivateData *pd = (FileBrowserModePrivateData *) mode_get_private_data ( sw );
|
|
if ( mretv & MENU_NEXT ) {
|
|
retv = NEXT_DIALOG;
|
|
}
|
|
else if ( mretv & MENU_PREVIOUS ) {
|
|
retv = PREVIOUS_DIALOG;
|
|
}
|
|
else if ( mretv & MENU_QUICK_SWITCH ) {
|
|
retv = ( mretv & MENU_LOWER_MASK );
|
|
} else if ( mretv & MENU_CUSTOM_COMMAND ) {
|
|
retv = ( mretv & MENU_LOWER_MASK );
|
|
} else if ( ( mretv & MENU_OK ) ) {
|
|
if ( selected_line < pd->array_length ) {
|
|
if ( pd->array[selected_line].type == UP ) {
|
|
GFile *new = g_file_get_parent ( pd->current_dir );
|
|
if ( new ) {
|
|
g_object_unref ( pd->current_dir );
|
|
pd->current_dir = new;
|
|
free_list ( pd );
|
|
get_file_browser ( sw );
|
|
return RESET_DIALOG;
|
|
}
|
|
}
|
|
else if ( pd->array[selected_line].type == DIRECTORY ) {
|
|
char *path = g_build_filename ( cache_dir, FILEBROWSER_CACHE_FILE, NULL );
|
|
g_file_set_contents ( path, pd->array[selected_line].path, -1, NULL );
|
|
g_free ( path );
|
|
GFile *new = g_file_new_for_path ( pd->array[selected_line].path );
|
|
g_object_unref ( pd->current_dir );
|
|
pd->current_dir = new;
|
|
free_list ( pd );
|
|
get_file_browser ( sw );
|
|
return RESET_DIALOG;
|
|
}
|
|
else if ( pd->array[selected_line].type == RFILE ) {
|
|
char *d = g_filename_from_utf8 ( pd->array[selected_line].path, -1, NULL, NULL, NULL );
|
|
char *d_esc = g_shell_quote(d);
|
|
char *cmd = g_strdup_printf ( "xdg-open %s", d_esc );
|
|
g_free(d_esc);
|
|
g_free ( d );
|
|
char *cdir = g_file_get_path ( pd->current_dir );
|
|
helper_execute_command ( cdir, cmd, FALSE, NULL );
|
|
g_free ( cdir );
|
|
g_free ( cmd );
|
|
return MODE_EXIT;
|
|
}
|
|
}
|
|
retv = RELOAD_DIALOG;
|
|
}
|
|
else if ( ( mretv & MENU_CUSTOM_INPUT ) && *input ) {
|
|
char *p = rofi_expand_path ( *input );
|
|
char *dir = g_filename_from_utf8 ( p, -1, NULL, NULL, NULL );
|
|
g_free ( p );
|
|
if ( g_file_test ( dir, G_FILE_TEST_EXISTS ) ) {
|
|
if ( g_file_test ( dir, G_FILE_TEST_IS_DIR ) ) {
|
|
g_object_unref ( pd->current_dir );
|
|
pd->current_dir = g_file_new_for_path ( dir );
|
|
g_free ( dir );
|
|
free_list ( pd );
|
|
get_file_browser ( sw );
|
|
return RESET_DIALOG;
|
|
}
|
|
}
|
|
g_free ( dir );
|
|
retv = RELOAD_DIALOG;
|
|
}
|
|
else if ( ( mretv & MENU_ENTRY_DELETE ) == MENU_ENTRY_DELETE ) {
|
|
retv = RELOAD_DIALOG;
|
|
}
|
|
return retv;
|
|
}
|
|
|
|
static void file_browser_mode_destroy ( Mode *sw )
|
|
{
|
|
FileBrowserModePrivateData *pd = (FileBrowserModePrivateData *) mode_get_private_data ( sw );
|
|
if ( pd != NULL ) {
|
|
g_object_unref ( pd->current_dir );
|
|
free_list ( pd );
|
|
g_free ( pd );
|
|
mode_set_private_data ( sw, NULL );
|
|
}
|
|
}
|
|
|
|
static char *_get_display_value ( const Mode *sw, unsigned int selected_line, G_GNUC_UNUSED int *state, G_GNUC_UNUSED GList **attr_list, int get_entry )
|
|
{
|
|
FileBrowserModePrivateData *pd = (FileBrowserModePrivateData *) mode_get_private_data ( sw );
|
|
|
|
// Only return the string if requested, otherwise only set state.
|
|
if ( !get_entry ) {
|
|
return NULL;
|
|
}
|
|
if ( pd->array[selected_line].type == UP ) {
|
|
return g_strdup ( " .." );
|
|
}
|
|
else {
|
|
if ( pd->array[selected_line].link ) {
|
|
return g_strconcat ( "@", pd->array[selected_line].name, NULL );
|
|
}
|
|
else {
|
|
return g_strdup ( pd->array[selected_line].name );
|
|
}
|
|
}
|
|
return g_strdup ( "n/a" );
|
|
}
|
|
|
|
/**
|
|
* @param sw The mode object.
|
|
* @param tokens The tokens to match against.
|
|
* @param index The index in this plugin to match against.
|
|
*
|
|
* Match the entry.
|
|
*
|
|
* @returns try when a match.
|
|
*/
|
|
static int file_browser_token_match ( const Mode *sw, rofi_int_matcher **tokens, unsigned int index )
|
|
{
|
|
FileBrowserModePrivateData *pd = (FileBrowserModePrivateData *) mode_get_private_data ( sw );
|
|
|
|
// Call default matching function.
|
|
return helper_token_match ( tokens, pd->array[index].name );
|
|
}
|
|
|
|
static cairo_surface_t *_get_icon ( const Mode *sw, unsigned int selected_line, int height )
|
|
{
|
|
FileBrowserModePrivateData *pd = (FileBrowserModePrivateData *) mode_get_private_data ( sw );
|
|
g_return_val_if_fail ( pd->array != NULL, NULL );
|
|
FBFile *dr = &( pd->array[selected_line] );
|
|
if ( dr->icon_fetch_uid > 0 ) {
|
|
return rofi_icon_fetcher_get ( dr->icon_fetch_uid );
|
|
}
|
|
if ( rofi_icon_fetcher_file_is_image ( dr->path ) ) {
|
|
dr->icon_fetch_uid = rofi_icon_fetcher_query ( dr->path, height );
|
|
}
|
|
else {
|
|
dr->icon_fetch_uid = rofi_icon_fetcher_query ( icon_name[dr->type], height );
|
|
}
|
|
return rofi_icon_fetcher_get ( dr->icon_fetch_uid );
|
|
}
|
|
|
|
static char * _get_message ( const Mode *sw )
|
|
{
|
|
FileBrowserModePrivateData *pd = (FileBrowserModePrivateData *) mode_get_private_data ( sw );
|
|
if ( pd->current_dir ) {
|
|
char *dirname = g_file_get_parse_name ( pd->current_dir );
|
|
char *str = g_markup_printf_escaped ( "<b>Current directory:</b> %s", dirname );
|
|
g_free ( dirname );
|
|
return str;
|
|
}
|
|
return "n/a";
|
|
}
|
|
|
|
static char *_get_completion ( const Mode *sw, unsigned int index )
|
|
{
|
|
FileBrowserModePrivateData *pd = (FileBrowserModePrivateData *) mode_get_private_data ( sw );
|
|
|
|
char *d = g_strescape ( pd->array[index].path, NULL );
|
|
return d;
|
|
}
|
|
|
|
Mode *create_new_file_browser ( void )
|
|
{
|
|
Mode *sw = g_malloc0 ( sizeof ( Mode ) );
|
|
|
|
*sw = file_browser_mode;
|
|
|
|
sw->private_data = NULL;
|
|
return sw;
|
|
}
|
|
|
|
#if 0
|
|
ModeMode file_browser_mode_completer ( Mode *sw, int mretv, char **input, unsigned int selected_line, char **path )
|
|
{
|
|
ModeMode retv = MODE_EXIT;
|
|
FileBrowserModePrivateData *pd = (FileBrowserModePrivateData *) mode_get_private_data ( sw );
|
|
if ( mretv & MENU_NEXT ) {
|
|
retv = NEXT_DIALOG;
|
|
}
|
|
else if ( mretv & MENU_PREVIOUS ) {
|
|
retv = PREVIOUS_DIALOG;
|
|
}
|
|
else if ( mretv & MENU_QUICK_SWITCH ) {
|
|
retv = ( mretv & MENU_LOWER_MASK );
|
|
}
|
|
else if ( ( mretv & MENU_OK ) ) {
|
|
if ( selected_line < pd->array_length ) {
|
|
if ( pd->array[selected_line].type == UP ) {
|
|
GFile *new = g_file_get_parent ( pd->current_dir );
|
|
if ( new ) {
|
|
g_object_unref ( pd->current_dir );
|
|
pd->current_dir = new;
|
|
free_list ( pd );
|
|
get_file_browser ( sw );
|
|
return RESET_DIALOG;
|
|
}
|
|
}
|
|
else if ( pd->array[selected_line].type == DIRECTORY ) {
|
|
GFile *new = g_file_new_for_path ( pd->array[selected_line].path );
|
|
g_object_unref ( pd->current_dir );
|
|
pd->current_dir = new;
|
|
free_list ( pd );
|
|
get_file_browser ( sw );
|
|
return RESET_DIALOG;
|
|
}
|
|
else if ( pd->array[selected_line].type == RFILE ) {
|
|
*path = g_strescape ( pd->array[selected_line].path, NULL );
|
|
return MODE_EXIT;
|
|
}
|
|
}
|
|
retv = RELOAD_DIALOG;
|
|
}
|
|
else if ( ( mretv & MENU_CUSTOM_INPUT ) && *input ) {
|
|
char *p = rofi_expand_path ( *input );
|
|
char *dir = g_filename_from_utf8 ( p, -1, NULL, NULL, NULL );
|
|
g_free ( p );
|
|
if ( g_file_test ( dir, G_FILE_TEST_EXISTS ) ) {
|
|
if ( g_file_test ( dir, G_FILE_TEST_IS_DIR ) ) {
|
|
g_object_unref ( pd->current_dir );
|
|
pd->current_dir = g_file_new_for_path ( dir );
|
|
g_free ( dir );
|
|
free_list ( pd );
|
|
get_file_browser ( sw );
|
|
return RESET_DIALOG;
|
|
}
|
|
}
|
|
g_free ( dir );
|
|
retv = RELOAD_DIALOG;
|
|
}
|
|
else if ( ( mretv & MENU_ENTRY_DELETE ) == MENU_ENTRY_DELETE ) {
|
|
retv = RELOAD_DIALOG;
|
|
}
|
|
return retv;
|
|
}
|
|
#endif
|
|
|
|
Mode file_browser_mode =
|
|
{
|
|
.display_name = NULL,
|
|
.abi_version = ABI_VERSION,
|
|
.name = "file-browser",
|
|
.cfg_name_key = "display-file_browser",
|
|
._init = file_browser_mode_init,
|
|
._get_num_entries = file_browser_mode_get_num_entries,
|
|
._result = file_browser_mode_result,
|
|
._destroy = file_browser_mode_destroy,
|
|
._token_match = file_browser_token_match,
|
|
._get_display_value = _get_display_value,
|
|
._get_icon = _get_icon,
|
|
._get_message = _get_message,
|
|
._get_completion = _get_completion,
|
|
._preprocess_input = NULL,
|
|
.private_data = NULL,
|
|
.free = NULL,
|
|
};
|