zathura/document.c

568 lines
12 KiB
C
Raw Normal View History

2010-11-17 22:51:15 +01:00
/* See LICENSE file for license and copyright information */
2010-11-18 13:54:35 +01:00
#define _BSD_SOURCE
#define _XOPEN_SOURCE 700
2010-11-18 13:54:35 +01:00
// TODO: Implement realpath
2012-04-01 21:15:32 +02:00
#include <sys/wait.h>
2010-11-17 22:51:15 +01:00
#include <stdlib.h>
2010-11-17 23:15:08 +01:00
#include <stdio.h>
2010-11-18 03:15:32 +01:00
#include <string.h>
2010-11-18 13:54:35 +01:00
#include <limits.h>
#include <errno.h>
#include <glib.h>
#include <glib/gi18n.h>
2010-11-17 22:51:15 +01:00
2012-03-27 13:30:04 +02:00
#include <girara/datastructures.h>
#include <girara/utils.h>
#include <girara/statusbar.h>
#include <girara/session.h>
#include <girara/settings.h>
2010-11-17 22:51:15 +01:00
#include "document.h"
#include "utils.h"
#include "zathura.h"
2011-04-18 21:22:35 +02:00
#include "render.h"
#include "database.h"
2012-03-26 14:44:56 +02:00
#include "page.h"
2012-03-16 14:37:54 +01:00
#include "page-widget.h"
2012-03-27 13:30:04 +02:00
#include "plugin.h"
/** Read a most GT_MAX_READ bytes before falling back to file. */
static const size_t GT_MAX_READ = 1 << 16;
2012-03-27 13:30:04 +02:00
static const gchar* guess_type(const char* path);
/**
* Document
*/
struct zathura_document_s
{
char* file_path; /**< File path of the document */
const char* password; /**< Password of the document */
unsigned int current_page_number; /**< Current page number */
unsigned int number_of_pages; /**< Number of pages */
double scale; /**< Scale value */
unsigned int rotate; /**< Rotation */
void* data; /**< Custom data */
zathura_adjust_mode_t adjust_mode; /**< Adjust mode (best-fit, width) */
unsigned int page_offset; /**< Page offset */
/**
* Document pages
*/
zathura_page_t** pages;
2012-03-30 18:24:00 +02:00
/**
* Used plugin
*/
zathura_plugin_t* plugin;
};
zathura_document_t*
zathura_document_open(zathura_plugin_manager_t* plugin_manager, const char*
path, const char* password, zathura_error_t* error)
{
if (path == NULL) {
return NULL;
}
if (g_file_test(path, G_FILE_TEST_EXISTS) == FALSE) {
girara_error("File '%s' does not exist", path);
return NULL;
}
const gchar* content_type = guess_type(path);
if (content_type == NULL) {
girara_error("Could not determine file type.");
return NULL;
}
2010-11-18 13:54:35 +01:00
/* determine real path */
long path_max;
2010-11-18 13:54:35 +01:00
#ifdef PATH_MAX
path_max = PATH_MAX;
#else
path_max = pathconf(path,_PC_PATH_MAX);
if (path_max <= 0)
2010-11-18 13:54:35 +01:00
path_max = 4096;
#endif
2010-12-28 00:47:41 +01:00
char* real_path = NULL;
zathura_document_t* document = NULL;
real_path = malloc(sizeof(char) * path_max);
2011-09-21 00:25:41 +02:00
if (real_path == NULL) {
2011-09-29 15:23:13 +02:00
g_free((void*)content_type);
return NULL;
2010-11-18 13:54:35 +01:00
}
2011-09-21 00:25:41 +02:00
if (realpath(path, real_path) == NULL) {
2011-09-29 15:23:13 +02:00
g_free((void*)content_type);
free(real_path);
return NULL;
}
zathura_plugin_t* plugin = zathura_plugin_manager_get_plugin(plugin_manager, content_type);
2011-09-29 15:23:13 +02:00
g_free((void*)content_type);
if (plugin == NULL) {
girara_error("unknown file type\n");
goto error_free;
2010-11-18 13:54:35 +01:00
}
2011-09-29 12:07:07 +02:00
document = g_malloc0(sizeof(zathura_document_t));
2010-11-18 02:35:33 +01:00
2012-03-27 13:30:04 +02:00
document->file_path = real_path;
document->password = password;
document->scale = 1.0;
document->plugin = plugin;
2010-11-18 02:35:33 +01:00
2012-02-08 23:21:27 +01:00
/* open document */
2012-03-27 13:30:04 +02:00
if (plugin->functions.document_open == NULL) {
2012-02-08 23:21:27 +01:00
girara_error("plugin has no open function\n");
goto error_free;
}
zathura_error_t int_error = plugin->functions.document_open(document);
if (int_error != ZATHURA_ERROR_OK) {
if (error != NULL) {
*error = int_error;
2012-02-08 23:21:27 +01:00
}
girara_error("could not open document\n");
goto error_free;
}
2012-02-07 19:25:47 +01:00
/* read all pages */
document->pages = calloc(document->number_of_pages, sizeof(zathura_page_t*));
if (document->pages == NULL) {
goto error_free;
}
2011-09-29 15:23:13 +02:00
2012-02-07 19:25:47 +01:00
for (unsigned int page_id = 0; page_id < document->number_of_pages; page_id++) {
2012-03-26 15:17:01 +02:00
zathura_page_t* page = zathura_page_new(document, page_id, NULL);
2012-02-07 19:25:47 +01:00
if (page == NULL) {
goto error_free;
2010-11-18 03:15:32 +01:00
}
2012-02-07 19:25:47 +01:00
document->pages[page_id] = page;
}
return document;
2010-11-18 03:15:32 +01:00
2010-12-28 00:47:41 +01:00
error_free:
2010-11-18 13:54:35 +01:00
free(real_path);
2010-12-28 00:47:41 +01:00
2011-09-21 00:25:41 +02:00
if (document != NULL && document->pages != NULL) {
for (unsigned int page_id = 0; page_id < document->number_of_pages; page_id++) {
2010-12-28 00:47:41 +01:00
zathura_page_free(document->pages[page_id]);
}
free(document->pages);
}
2011-09-29 12:07:07 +02:00
g_free(document);
2010-11-17 22:51:15 +01:00
return NULL;
}
2012-03-27 13:30:04 +02:00
zathura_error_t
2010-11-17 23:15:08 +01:00
zathura_document_free(zathura_document_t* document)
2010-11-17 22:51:15 +01:00
{
if (document == NULL || document->plugin == NULL) {
2012-03-27 13:30:04 +02:00
return ZATHURA_ERROR_INVALID_ARGUMENTS;
2010-11-17 23:15:08 +01:00
}
2010-12-28 00:47:41 +01:00
/* free pages */
2011-09-21 00:25:41 +02:00
for (unsigned int page_id = 0; page_id < document->number_of_pages; page_id++) {
2010-12-28 00:47:41 +01:00
zathura_page_free(document->pages[page_id]);
document->pages[page_id] = NULL;
2010-12-28 00:47:41 +01:00
}
free(document->pages);
/* free document */
2012-03-27 13:30:04 +02:00
zathura_error_t error = ZATHURA_ERROR_OK;
if (document->plugin->functions.document_free == NULL) {
error = ZATHURA_ERROR_NOT_IMPLEMENTED;
2011-10-21 23:40:16 +02:00
} else {
error = document->plugin->functions.document_free(document, document->data);
2010-11-17 23:15:08 +01:00
}
2011-09-21 00:25:41 +02:00
if (document->file_path != NULL) {
2010-11-18 13:54:35 +01:00
free(document->file_path);
}
2011-09-29 12:07:07 +02:00
g_free(document);
2010-11-18 02:35:33 +01:00
2012-03-05 08:22:16 +01:00
return error;
2010-11-17 23:15:08 +01:00
}
const char*
zathura_document_get_path(zathura_document_t* document)
{
if (document == NULL) {
return NULL;
}
return document->file_path;
}
const char*
zathura_document_get_password(zathura_document_t* document)
{
if (document == NULL) {
return NULL;
}
return document->password;
}
zathura_page_t*
zathura_document_get_page(zathura_document_t* document, unsigned int index)
{
if (document == NULL || document->pages == NULL || (document->number_of_pages <= index)) {
return NULL;
}
return document->pages[index];
}
void*
zathura_document_get_data(zathura_document_t* document)
{
if (document == NULL) {
return NULL;
}
return document->data;
}
void
zathura_document_set_data(zathura_document_t* document, void* data)
{
if (document == NULL) {
return;
}
document->data = data;
}
unsigned int
zathura_document_get_number_of_pages(zathura_document_t* document)
{
if (document == NULL) {
return 0;
}
return document->number_of_pages;
}
void
zathura_document_set_number_of_pages(zathura_document_t* document, unsigned int number_of_pages)
{
if (document == NULL) {
return;
}
document->number_of_pages = number_of_pages;
}
unsigned int
zathura_document_get_current_page_number(zathura_document_t* document)
{
if (document == NULL) {
return 0;
}
return document->current_page_number;
}
void
zathura_document_set_current_page_number(zathura_document_t* document, unsigned int
current_page)
{
if (document == NULL) {
return;
}
document->current_page_number = current_page;
}
double
zathura_document_get_scale(zathura_document_t* document)
{
if (document == NULL) {
return 0;
}
return document->scale;
}
void
zathura_document_set_scale(zathura_document_t* document, double scale)
{
if (document == NULL) {
return;
}
document->scale = scale;
}
unsigned int
zathura_document_get_rotation(zathura_document_t* document)
{
if (document == NULL) {
return 0;
}
return document->rotate;
}
void
zathura_document_set_rotation(zathura_document_t* document, unsigned int rotation)
{
if (document == NULL) {
return;
}
document->rotate = rotation % 360;
2012-03-28 16:44:46 +02:00
if (document->rotate > 0 && document->rotate <= 90) {
document->rotate = 90;
2012-03-28 16:44:46 +02:00
} else if (document->rotate > 0 && document->rotate <= 180) {
document->rotate = 180;
2012-03-28 16:44:46 +02:00
} else if (document->rotate > 0 && document->rotate <= 270) {
document->rotate = 270;
} else {
document->rotate = 0;
}
}
zathura_adjust_mode_t
zathura_document_get_adjust_mode(zathura_document_t* document)
{
if (document == NULL) {
return ZATHURA_ADJUST_NONE;
}
return document->adjust_mode;
}
void
zathura_document_set_adjust_mode(zathura_document_t* document, zathura_adjust_mode_t mode)
{
if (document == NULL) {
return;
}
if (mode == ZATHURA_ADJUST_BESTFIT || mode == ZATHURA_ADJUST_WIDTH) {
document->adjust_mode = mode;
} else {
document->adjust_mode = ZATHURA_ADJUST_NONE;
}
}
unsigned int
zathura_document_get_page_offset(zathura_document_t* document)
{
if (document == NULL) {
return 0;
}
return document->page_offset;
}
void
zathura_document_set_page_offset(zathura_document_t* document, unsigned int page_offset)
{
if (document == NULL) {
return;
}
if (page_offset < document->number_of_pages) {
document->page_offset = page_offset;
}
}
2012-03-27 13:30:04 +02:00
zathura_error_t
2010-11-17 23:15:08 +01:00
zathura_document_save_as(zathura_document_t* document, const char* path)
{
if (document == NULL || document->plugin == NULL || path == NULL) {
2012-03-27 13:30:04 +02:00
return ZATHURA_ERROR_UNKNOWN;
2010-11-17 23:15:08 +01:00
}
2012-03-27 13:30:04 +02:00
if (document->plugin->functions.document_save_as == NULL) {
return ZATHURA_ERROR_NOT_IMPLEMENTED;
2010-11-17 23:15:08 +01:00
}
return document->plugin->functions.document_save_as(document, document->data, path);
2010-11-17 22:51:15 +01:00
}
girara_tree_node_t*
2012-03-27 13:30:04 +02:00
zathura_document_index_generate(zathura_document_t* document, zathura_error_t* error)
2010-11-17 22:51:15 +01:00
{
if (document == NULL || document->plugin == NULL) {
2012-03-05 00:15:09 +01:00
if (error != NULL) {
2012-03-27 13:30:04 +02:00
*error = ZATHURA_ERROR_INVALID_ARGUMENTS;
2012-03-05 00:15:09 +01:00
}
2010-11-17 23:15:08 +01:00
return NULL;
}
2012-03-27 13:30:04 +02:00
if (document->plugin->functions.document_index_generate == NULL) {
2012-03-05 00:15:09 +01:00
if (error != NULL) {
2012-03-27 13:30:04 +02:00
*error = ZATHURA_ERROR_NOT_IMPLEMENTED;
2012-03-05 00:15:09 +01:00
}
2010-11-17 23:15:08 +01:00
return NULL;
}
return document->plugin->functions.document_index_generate(document, document->data, error);
2010-11-17 22:51:15 +01:00
}
2011-10-22 00:11:42 +02:00
girara_list_t*
2012-03-27 13:30:04 +02:00
zathura_document_attachments_get(zathura_document_t* document, zathura_error_t* error)
2010-11-17 22:51:15 +01:00
{
if (document == NULL || document->plugin == NULL) {
2012-03-05 00:15:09 +01:00
if (error != NULL) {
2012-03-27 13:30:04 +02:00
*error = ZATHURA_ERROR_INVALID_ARGUMENTS;
2012-03-05 00:15:09 +01:00
}
2010-11-17 23:15:08 +01:00
return NULL;
}
2012-03-27 13:30:04 +02:00
if (document->plugin->functions.document_attachments_get == NULL) {
2012-03-05 00:15:09 +01:00
if (error != NULL) {
2012-03-27 13:30:04 +02:00
*error = ZATHURA_ERROR_NOT_IMPLEMENTED;
2012-03-05 00:15:09 +01:00
}
2010-11-17 23:15:08 +01:00
return NULL;
}
return document->plugin->functions.document_attachments_get(document, document->data, error);
2010-11-17 22:51:15 +01:00
}
2012-03-27 13:30:04 +02:00
zathura_error_t
2012-02-07 21:01:54 +01:00
zathura_document_attachment_save(zathura_document_t* document, const char* attachment, const char* file)
2010-11-17 22:51:15 +01:00
{
if (document == NULL || document->plugin == NULL) {
2012-03-27 13:30:04 +02:00
return ZATHURA_ERROR_INVALID_ARGUMENTS;
2012-01-13 18:25:17 +01:00
}
2012-03-27 13:30:04 +02:00
if (document->plugin->functions.document_attachment_save == NULL) {
return ZATHURA_ERROR_NOT_IMPLEMENTED;
2012-01-13 18:25:17 +01:00
}
return document->plugin->functions.document_attachment_save(document, document->data, attachment, file);
2010-11-17 22:51:15 +01:00
}
2012-03-30 18:24:00 +02:00
girara_list_t*
zathura_document_get_information(zathura_document_t* document, zathura_error_t* error)
2011-10-01 23:29:40 +02:00
{
if (document == NULL || document->plugin == NULL) {
2012-02-07 21:01:54 +01:00
if (error != NULL) {
2012-03-27 13:30:04 +02:00
*error = ZATHURA_ERROR_INVALID_ARGUMENTS;
2012-02-07 21:01:54 +01:00
}
2011-10-01 23:29:40 +02:00
return NULL;
}
2012-03-30 18:24:00 +02:00
if (document->plugin->functions.document_get_information == NULL) {
2012-02-07 21:01:54 +01:00
if (error != NULL) {
2012-03-27 13:30:04 +02:00
*error = ZATHURA_ERROR_NOT_IMPLEMENTED;
2012-02-07 21:01:54 +01:00
}
2011-10-01 23:29:40 +02:00
return NULL;
}
2012-03-30 18:24:00 +02:00
girara_list_t* result = document->plugin->functions.document_get_information(document, document->data, error);
if (result != NULL) {
girara_list_set_free_function(result, (girara_free_function_t) zathura_document_information_entry_free);
}
return result;
2011-10-01 23:29:40 +02:00
}
2012-03-27 13:30:04 +02:00
static const gchar*
guess_type(const char* path)
2010-12-25 00:47:52 +01:00
{
2012-03-27 13:30:04 +02:00
gboolean uncertain;
const gchar* content_type = g_content_type_guess(path, NULL, 0, &uncertain);
if (content_type == NULL) {
2010-12-25 00:47:52 +01:00
return NULL;
2010-12-26 01:12:20 +01:00
}
2010-12-25 00:47:52 +01:00
2012-03-27 13:30:04 +02:00
if (uncertain == FALSE) {
return content_type;
2010-12-26 01:12:20 +01:00
}
2012-03-27 13:30:04 +02:00
girara_debug("g_content_type is uncertain, guess: %s\n", content_type);
2012-03-27 13:30:04 +02:00
FILE* f = fopen(path, "rb");
if (f == NULL) {
return NULL;
}
2012-03-27 13:30:04 +02:00
const int fd = fileno(f);
guchar* content = NULL;
size_t length = 0u;
while (uncertain == TRUE && length < GT_MAX_READ) {
g_free((void*)content_type);
content_type = NULL;
2012-03-27 13:30:04 +02:00
content = g_realloc(content, length + BUFSIZ);
const ssize_t r = read(fd, content + length, BUFSIZ);
if (r == -1) {
break;
}
2012-03-27 13:30:04 +02:00
length += r;
content_type = g_content_type_guess(NULL, content, length, &uncertain);
girara_debug("new guess: %s uncertain: %d, read: %zu\n", content_type, uncertain, length);
}
2012-03-27 13:30:04 +02:00
fclose(f);
g_free(content);
if (uncertain == FALSE) {
return content_type;
}
2012-03-27 13:30:04 +02:00
g_free((void*)content_type);
content_type = NULL;
2011-09-29 15:23:13 +02:00
2012-03-27 13:30:04 +02:00
girara_debug("falling back to file");
2011-09-29 15:23:13 +02:00
2012-03-27 13:30:04 +02:00
GString* command = g_string_new("file -b --mime-type ");
char* tmp = g_shell_quote(path);
2011-09-29 15:23:13 +02:00
2012-03-27 13:30:04 +02:00
g_string_append(command, tmp);
g_free(tmp);
2011-09-29 15:23:13 +02:00
2012-03-27 13:30:04 +02:00
GError* error = NULL;
char* out = NULL;
int ret = 0;
g_spawn_command_line_sync(command->str, &out, NULL, &ret, &error);
g_string_free(command, TRUE);
if (error != NULL) {
girara_warning("failed to execute command: %s", error->message);
g_error_free(error);
g_free(out);
return NULL;
}
if (WEXITSTATUS(ret) != 0) {
girara_warning("file failed with error code: %d", WEXITSTATUS(ret));
g_free(out);
return NULL;
2011-09-29 15:23:13 +02:00
}
2012-03-27 13:30:04 +02:00
g_strdelimit(out, "\n\r", '\0');
return out;
2011-09-29 15:23:13 +02:00
}
zathura_plugin_t*
zathura_document_get_plugin(zathura_document_t* document)
{
if (document == NULL) {
return NULL;
}
return document->plugin;
}