1
This commit is contained in:
Alex Yatskov 2016-10-31 22:24:20 -07:00
parent 0b56451536
commit d35f25b366
4 changed files with 88 additions and 42 deletions

View File

@ -109,6 +109,6 @@ static char * convert (const char *from_charset, const char *to_charset, const c
return output; return output;
} }
const char * eucjp_to_utf8(const char src[]) { char* eucjp_to_utf8(const char src[]) {
return convert("EUC-JP", "UTF-8", src); return convert("EUC-JP", "UTF-8", src);
} }

View File

@ -19,6 +19,6 @@
#ifndef CONVERT_H #ifndef CONVERT_H
#define CONVERT_H #define CONVERT_H
const char * eucjp_to_utf8(const char src[]); char* eucjp_to_utf8(const char src[]);
#endif #endif

115
main.c
View File

@ -30,8 +30,8 @@
#include "eb/eb/text.h" #include "eb/eb/text.h"
#define MAX_ENTRY_HITS 128 #define MAX_ENTRY_HITS 128
#define MAX_ENTRY_TEXT 1024 #define MAX_TEXT 1024
#define MAX_ENTRY_HEADER 512 #define MAX_HEADING 512
static void dump_hits(EB_Book* book) { static void dump_hits(EB_Book* book) {
EB_Hit hits[MAX_ENTRY_HITS]; EB_Hit hits[MAX_ENTRY_HITS];
@ -44,11 +44,11 @@ static void dump_hits(EB_Book* book) {
} }
for (int i = 0; i < hit_count; ++i) { for (int i = 0; i < hit_count; ++i) {
char text[MAX_ENTRY_TEXT]; char text[MAX_TEXT];
ssize_t text_length = 0; ssize_t text_length = 0;
eb_seek_text(book, &hits[i].text); eb_seek_text(book, &hits[i].text);
eb_read_text(book, NULL, NULL, NULL, MAX_ENTRY_TEXT, text, &text_length); eb_read_text(book, NULL, NULL, NULL, MAX_TEXT, text, &text_length);
const char* text_utf8 = eucjp_to_utf8(text); const char* text_utf8 = eucjp_to_utf8(text);
if (text_utf8 != NULL) { if (text_utf8 != NULL) {
@ -84,12 +84,50 @@ static void dump_book(EB_Book* book) {
} }
} }
static int export(const char path[]) { static char* read_book_text(EB_Book* book, const EB_Position* position) {
Book book_data = {}; if (eb_seek_text(book, position) != EB_SUCCESS) {
return NULL;
}
char text[MAX_TEXT + 1];
ssize_t text_length = 0;
if (eb_read_text(book, NULL, NULL, NULL, MAX_TEXT, text, &text_length) != EB_SUCCESS) {
return NULL;
}
return eucjp_to_utf8(text);
}
static char* read_book_heading(EB_Book* book, const EB_Position* position) {
if (eb_seek_text(book, position) != EB_SUCCESS) {
return NULL;
}
char heading[MAX_TEXT + 1];
ssize_t heading_length = 0;
if (eb_read_heading(book, NULL, NULL, NULL, MAX_TEXT, heading, &heading_length) != EB_SUCCESS) {
return NULL;
}
return eucjp_to_utf8(heading);
}
static void export_subbook(EB_Book* book, Subbook* subbook_data) {
if (eb_have_copyright(book)) {
EB_Position position;
if (eb_copyright(book, &position) == EB_SUCCESS) {
subbook_data->copyright = read_book_text(book, &position);
puts(subbook_data->copyright);
}
}
}
static void export_book(const char path[], Book* book_data) {
do { do {
if (eb_initialize_library() != EB_SUCCESS) { if (eb_initialize_library() != EB_SUCCESS) {
strcpy(book_data.error, "failed to initialize library"); strcpy(book_data->error, "failed to initialize library");
break; break;
} }
@ -97,7 +135,8 @@ static int export(const char path[]) {
eb_initialize_book(&book); eb_initialize_book(&book);
if (eb_bind(&book, path) != EB_SUCCESS) { if (eb_bind(&book, path) != EB_SUCCESS) {
strcpy(book_data.error, "failed to bind book to path"); strcpy(book_data->error, "failed to bind book to path");
eb_finalize_book(&book);
eb_finalize_library(); eb_finalize_library();
break; break;
} }
@ -106,65 +145,64 @@ static int export(const char path[]) {
if (eb_character_code(&book, &character_code) == EB_SUCCESS) { if (eb_character_code(&book, &character_code) == EB_SUCCESS) {
switch (character_code) { switch (character_code) {
case EB_CHARCODE_ISO8859_1: case EB_CHARCODE_ISO8859_1:
strcpy(book_data.character_code, "iso 8859-1"); strcpy(book_data->character_code, "iso 8859-1");
break; break;
case EB_CHARCODE_JISX0208: case EB_CHARCODE_JISX0208:
strcpy(book_data.character_code, "jis x 0208"); strcpy(book_data->character_code, "jis x 0208");
break; break;
case EB_CHARCODE_JISX0208_GB2312: case EB_CHARCODE_JISX0208_GB2312:
strcpy(book_data.character_code, "jis x 0208 / gb 2312"); strcpy(book_data->character_code, "jis x 0208 / gb 2312");
break; break;
default: default:
strcpy(book_data.character_code, "invalid"); strcpy(book_data->character_code, "invalid");
break; break;
} }
} }
else {
strcpy(book_data.character_code, "error");
}
EB_Disc_Code disc_code; EB_Disc_Code disc_code;
if (eb_disc_type(&book, &disc_code) == EB_SUCCESS) { if (eb_disc_type(&book, &disc_code) == EB_SUCCESS) {
switch (disc_code) { switch (disc_code) {
case EB_DISC_EB: case EB_DISC_EB:
strcpy(book_data.disc_code, "eb"); strcpy(book_data->disc_code, "eb");
break; break;
case EB_DISC_EPWING: case EB_DISC_EPWING:
strcpy(book_data.disc_code, "epwing"); strcpy(book_data->disc_code, "epwing");
break; break;
default: default:
strcpy(book_data.disc_code, "invalid"); strcpy(book_data->disc_code, "invalid");
break; break;
} }
} }
else {
strcpy(book_data.disc_code, "error");
}
EB_Subbook_Code sub_codes[EB_MAX_SUBBOOKS]; EB_Subbook_Code sub_codes[EB_MAX_SUBBOOKS];
int sub_count = 0; if (eb_subbook_list(&book, sub_codes, &book_data->subbook_count) != EB_SUCCESS) {
eb_finalize_book(&book);
if (eb_subbook_list(&book, sub_codes, &sub_count) != EB_SUCCESS) { eb_finalize_library();
fprintf(stderr, "error: failed to get sub-book list\n");
break; break;
} }
for (int i = 0; i < sub_count; ++i) {
if (eb_set_subbook(&book, sub_codes[i]) == EB_SUCCESS) {
dump_book(&book); /* int sub_count = 0; */
}
else { /* if (eb_subbook_list(&book, sub_codes, &sub_count) != EB_SUCCESS) { */
fprintf(stderr, "error: failed to set sub-book\n"); /* fprintf(stderr, "error: failed to get sub-book list\n"); */
} /* break; */
} /* } */
/* for (int i = 0; i < sub_count; ++i) { */
/* if (eb_set_subbook(&book, sub_codes[i]) == EB_SUCCESS) { */
/* dump_book(&book); */
/* } */
/* else { */
/* fprintf(stderr, "error: failed to set sub-book\n"); */
/* } */
/* } */
eb_finalize_book(&book); eb_finalize_book(&book);
eb_finalize_library(); eb_finalize_library();
} }
while(0); while(0);
return 0;
} }
int main(int argc, char *argv[]) { int main(int argc, char *argv[]) {
@ -173,5 +211,8 @@ int main(int argc, char *argv[]) {
return 2; return 2;
} }
return export(argv[1]); Book book_data = {};
export_book(argv[1], &book_data);
return 1;
} }

7
util.h
View File

@ -25,6 +25,11 @@
Dictionary structures Dictionary structures
*/ */
typedef struct {
char* heading;
char* text;
} TextBlock;
typedef struct { typedef struct {
char* heading; char* heading;
char* text; char* text;
@ -36,6 +41,7 @@ typedef struct {
typedef struct { typedef struct {
char* title; char* title;
char* copyright;
char* error; char* error;
Entry* entries; Entry* entries;
@ -45,7 +51,6 @@ typedef struct {
typedef struct { typedef struct {
char character_code[256]; char character_code[256];
char disc_code[256]; char disc_code[256];
char copyright;
Subbook* subbooks; Subbook* subbooks;
int subbook_count; int subbook_count;