libmobi 0.1.0

Sign up to get free protection for your applications and to get access to all the features.
Files changed (20) hide show
  1. checksums.yaml +7 -0
  2. data/.clang-format +10 -0
  3. data/.gitignore +10 -0
  4. data/.rubocop.yml +26 -0
  5. data/.travis.yml +5 -0
  6. data/Gemfile +19 -0
  7. data/LICENSE +674 -0
  8. data/README.md +38 -0
  9. data/Rakefile +53 -0
  10. data/bin/console +23 -0
  11. data/bin/setup +23 -0
  12. data/ext/mobi_ext/extconf.rb +33 -0
  13. data/ext/mobi_ext/mobi_ext.c +909 -0
  14. data/lib/libmobi.rb +23 -0
  15. data/lib/mobi.rb +17 -0
  16. data/lib/mobi/book.rb +20 -0
  17. data/lib/mobi/error.rb +21 -0
  18. data/lib/mobi/version.rb +19 -0
  19. data/libmobi.gemspec +41 -0
  20. metadata +103 -0
data/README.md ADDED
@@ -0,0 +1,38 @@
1
+ # libmobi.rb
2
+
3
+ This is a ruby binding for library for handling Mobipocket/Kindle (MOBI) ebook format documents: https://github.com/bfabiszewski/libmobi.
4
+
5
+ ## Installation
6
+
7
+ Add this line to your application's Gemfile:
8
+
9
+ ```ruby
10
+ gem 'libmobi'
11
+ ```
12
+
13
+ And then execute:
14
+
15
+ $ bundle
16
+
17
+ Or install it yourself as:
18
+
19
+ $ gem install libmobi
20
+
21
+ ## Usage
22
+
23
+ TODO: Write usage instructions here
24
+
25
+ ## Development
26
+
27
+ After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test` to run
28
+ the tests. You can also run `bin/console` for an interactive prompt that will allow you to
29
+ experiment.
30
+
31
+ To install this gem onto your local machine, run `bundle exec rake install`. To release a new
32
+ version, update the version number in `version.rb`, and then run `bundle exec rake release`, which
33
+ will create a git tag for the version, push git commits and tags, and push the `.gem` file to
34
+ [rubygems.org](https://rubygems.org).
35
+
36
+ ## Contributing
37
+
38
+ Bug reports and pull requests are welcome on GitHub at https://github.com/avsej/libmobi.rb.
data/Rakefile ADDED
@@ -0,0 +1,53 @@
1
+ # libmobi - library for handling Kindle (MOBI) formats of ebook documents
2
+ # Copyright (C) 2018 Sergey Avseyev <sergey.avseyev@gmail.com>
3
+ #
4
+ # This program is free software: you can redistribute it and/or modify
5
+ # it under the terms of the GNU General Public License as published by
6
+ # the Free Software Foundation, either version 3 of the License, or
7
+ # (at your option) any later version.
8
+ #
9
+ # This program is distributed in the hope that it will be useful,
10
+ # but WITHOUT ANY WARRANTY; without even the implied warranty of
11
+ # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12
+ # GNU General Public License for more details.
13
+ #
14
+ # You should have received a copy of the GNU General Public License
15
+ # along with this program. If not, see <https://www.gnu.org/licenses/>.
16
+
17
+ require 'bundler/gem_tasks'
18
+ require 'rake/testtask'
19
+
20
+ module Bundler
21
+ class GemHelper
22
+
23
+ def tag_version
24
+ sh "git tag -s -m \"Version #{version}\" #{version_tag}"
25
+ Bundler.ui.confirm "Tagged #{version_tag}."
26
+ yield if block_given?
27
+ rescue
28
+ Bundler.ui.error "Untagging #{version_tag} due to error."
29
+ sh_with_code "git tag -d #{version_tag}"
30
+ raise
31
+ end
32
+ end
33
+ end
34
+
35
+ Rake::TestTask.new(:test) do |t|
36
+ t.libs << 'test'
37
+ t.libs << 'lib'
38
+ t.test_files = FileList['test/**/*_test.rb']
39
+ end
40
+
41
+ require "rake/extensiontask"
42
+
43
+ def gemspec
44
+ @clean_gemspec ||= eval(File.read(File.expand_path('libmobi.gemspec', File.dirname(__FILE__))))
45
+ end
46
+
47
+ Rake::ExtensionTask.new("mobi_ext", gemspec) do |ext|
48
+ ext.ext_dir = "ext/mobi_ext"
49
+ CLEAN.include "#{ext.lib_dir}/*.#{RbConfig::CONFIG['DLEXT']}"
50
+ end
51
+ Rake::Task['test'].prerequisites.unshift('compile')
52
+
53
+ task :default => :test
data/bin/console ADDED
@@ -0,0 +1,23 @@
1
+ #!/usr/bin/env ruby
2
+
3
+ # libmobi - library for handling Kindle (MOBI) formats of ebook documents
4
+ # Copyright (C) 2018 Sergey Avseyev <sergey.avseyev@gmail.com>
5
+ #
6
+ # This program is free software: you can redistribute it and/or modify
7
+ # it under the terms of the GNU General Public License as published by
8
+ # the Free Software Foundation, either version 3 of the License, or
9
+ # (at your option) any later version.
10
+ #
11
+ # This program is distributed in the hope that it will be useful,
12
+ # but WITHOUT ANY WARRANTY; without even the implied warranty of
13
+ # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14
+ # GNU General Public License for more details.
15
+ #
16
+ # You should have received a copy of the GNU General Public License
17
+ # along with this program. If not, see <https://www.gnu.org/licenses/>.
18
+
19
+ require 'bundler/setup'
20
+ require 'libmobi'
21
+
22
+ require 'irb'
23
+ IRB.start(__FILE__)
data/bin/setup ADDED
@@ -0,0 +1,23 @@
1
+ #!/usr/bin/env bash
2
+
3
+ # libmobi - library for handling Kindle (MOBI) formats of ebook documents
4
+ # Copyright (C) 2018 Sergey Avseyev <sergey.avseyev@gmail.com>
5
+ #
6
+ # This program is free software: you can redistribute it and/or modify
7
+ # it under the terms of the GNU General Public License as published by
8
+ # the Free Software Foundation, either version 3 of the License, or
9
+ # (at your option) any later version.
10
+ #
11
+ # This program is distributed in the hope that it will be useful,
12
+ # but WITHOUT ANY WARRANTY; without even the implied warranty of
13
+ # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14
+ # GNU General Public License for more details.
15
+ #
16
+ # You should have received a copy of the GNU General Public License
17
+ # along with this program. If not, see <https://www.gnu.org/licenses/>.
18
+
19
+ set -euo pipefail
20
+ IFS=$'\n\t'
21
+ set -vx
22
+
23
+ bundle install
data/ext/mobi_ext/extconf.rb ADDED
@@ -0,0 +1,33 @@
1
+ # libmobi - library for handling Kindle (MOBI) formats of ebook documents
2
+ # Copyright (C) 2018 Sergey Avseyev <sergey.avseyev@gmail.com>
3
+ #
4
+ # This program is free software: you can redistribute it and/or modify
5
+ # it under the terms of the GNU General Public License as published by
6
+ # the Free Software Foundation, either version 3 of the License, or
7
+ # (at your option) any later version.
8
+ #
9
+ # This program is distributed in the hope that it will be useful,
10
+ # but WITHOUT ANY WARRANTY; without even the implied warranty of
11
+ # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12
+ # GNU General Public License for more details.
13
+ #
14
+ # You should have received a copy of the GNU General Public License
15
+ # along with this program. If not, see <https://www.gnu.org/licenses/>.
16
+
17
+ require 'rbconfig'
18
+ require 'mkmf'
19
+
20
+ def define(macro, value = nil)
21
+ $defs.push("-D #{[macro.upcase, Shellwords.shellescape(value)].compact.join('=')}")
22
+ end
23
+
24
+ pkg_config('libmobi') || abort('libmobi headers not found. (dnf install libmobi-devel on Fedora)')
25
+
26
+ $CFLAGS << ' -pedantic -Wall -Wextra -Werror '
27
+ if ENV['DEBUG_BUILD']
28
+ $CFLAGS.gsub!(/\W-Wp,-D_FORTIFY_SOURCE=\d+\W/, ' ')
29
+ $CFLAGS << ' -ggdb3 -O0 '
30
+ end
31
+
32
+ create_header('mobi_config.h')
33
+ create_makefile('mobi_ext')
data/ext/mobi_ext/mobi_ext.c ADDED
@@ -0,0 +1,909 @@
1
+ /* -*- Mode: C; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
2
+
3
+ #include <mobi.h>
4
+
5
+ #include <ruby.h>
6
+
7
+ VALUE mb_mMOBI;
8
+ VALUE mb_eError;
9
+ VALUE mb_cBook;
10
+
11
+ static const char *mb_libmobi_strerror(MOBI_RET code)
12
+ {
13
+ switch (code) {
14
+ case MOBI_ERROR:
15
+ return "generic error return value";
16
+ case MOBI_PARAM_ERR:
17
+ return "wrong function parameter";
18
+ case MOBI_DATA_CORRUPT:
19
+ return "corrupted data";
20
+ case MOBI_FILE_NOT_FOUND:
21
+ return "file not found";
22
+ case MOBI_FILE_ENCRYPTED:
23
+ return "unsupported encrypted data";
24
+ case MOBI_FILE_UNSUPPORTED:
25
+ return "unsupported document type";
26
+ case MOBI_MALLOC_FAILED:
27
+ return "memory allocation error";
28
+ case MOBI_INIT_FAILED:
29
+ return "initialization error";
30
+ case MOBI_BUFFER_END:
31
+ return "out of buffer error";
32
+ case MOBI_XML_ERR:
33
+ return "XMLwriter error";
34
+ case MOBI_DRM_PIDINV:
35
+ return "invalid DRM PID";
36
+ case MOBI_DRM_KEYNOTFOUND:
37
+ return "key not found";
38
+ case MOBI_DRM_UNSUPPORTED:
39
+ return "DRM support not included";
40
+ case MOBI_WRITE_FAILED:
41
+ return "writing to file failed";
42
+ default:
43
+ return "FIXME: unkwnown code";
44
+ }
45
+ }
46
+
47
+ static void mb_raise_at(MOBI_RET code, const char *message, const char *file, int line)
48
+ {
49
+ VALUE exc, str;
50
+
51
+ str = rb_str_new_cstr(message);
52
+ if (code > 0) {
53
+ rb_str_catf(str, ": (0x%02x) \"%s\"", (int)code, mb_libmobi_strerror(code));
54
+ }
55
+ rb_str_buf_cat_ascii(str, " at ");
56
+ rb_str_buf_cat_ascii(str, file);
57
+ rb_str_catf(str, ":%d", line);
58
+ exc = rb_exc_new3(mb_eError, str);
59
+ rb_ivar_set(exc, rb_intern("@code"), INT2FIX(code));
60
+ rb_exc_raise(exc);
61
+ }
62
+
63
+ #define mb_raise(code, message) mb_raise_at(code, message, __FILE__, __LINE__)
64
+ #define mb_raise_msg(message) mb_raise_at(0, message, __FILE__, __LINE__)
65
+
66
+ typedef struct mb_BOOK {
67
+ MOBIData *data;
68
+ } mb_BOOK;
69
+
70
+ static void mb_book_mark(void *ptr)
71
+ {
72
+ mb_BOOK *book = ptr;
73
+ (void)book;
74
+ }
75
+
76
+ static void mb_book_free(void *ptr)
77
+ {
78
+ mb_BOOK *book = ptr;
79
+ if (book) {
80
+ if (book->data) {
81
+ mobi_free(book->data);
82
+ }
83
+ book->data = NULL;
84
+ }
85
+ }
86
+
87
+ static VALUE mb_book_alloc(VALUE klass)
88
+ {
89
+ VALUE obj;
90
+ mb_BOOK *book;
91
+
92
+ obj = Data_Make_Struct(klass, mb_BOOK, mb_book_mark, mb_book_free, book);
93
+ return obj;
94
+ }
95
+
96
+ static VALUE mb_book_init(VALUE self, VALUE path)
97
+ {
98
+ mb_BOOK *book = DATA_PTR(self);
99
+ MOBI_RET rc;
100
+
101
+ Check_Type(path, T_STRING);
102
+
103
+ book->data = mobi_init();
104
+ if (book->data == NULL) {
105
+ mb_raise(MOBI_MALLOC_FAILED, "unable to allocate MOBIData struct");
106
+ }
107
+
108
+ rc = mobi_load_filename(book->data, RSTRING_PTR(path));
109
+ if (rc != MOBI_SUCCESS) {
110
+ mb_raise(rc, "unable to load book from path");
111
+ }
112
+ return self;
113
+ }
114
+
115
+ static VALUE mb_book_next(VALUE self)
116
+ {
117
+ mb_BOOK *book = DATA_PTR(self);
118
+
119
+ if (book == NULL || book->data == NULL) {
120
+ mb_raise(MOBI_INIT_FAILED, "the MOBI data is not loaded");
121
+ }
122
+ if (mobi_is_hybrid(book->data) && book->data->next != NULL) {
123
+ VALUE obj;
124
+ mb_BOOK *next;
125
+
126
+ obj = Data_Make_Struct(mb_cBook, mb_BOOK, mb_book_mark, mb_book_free, next);
127
+ next->data = book->data->next;
128
+ return obj;
129
+ }
130
+
131
+ return Qnil;
132
+ }
133
+
134
+ #define COPY_HEADER_INT(NAME) \
135
+ if (hdr->NAME) { \
136
+ rb_hash_aset(res, ID2SYM(rb_intern(#NAME)), INT2FIX(*hdr->NAME)); \
137
+ }
138
+ static VALUE mb_book_mobi_header(VALUE self)
139
+ {
140
+ mb_BOOK *book = DATA_PTR(self);
141
+ VALUE res;
142
+ MOBIMobiHeader *hdr;
143
+
144
+ if (book == NULL || book->data == NULL) {
145
+ mb_raise(MOBI_INIT_FAILED, "the MOBI data is not loaded");
146
+ }
147
+ hdr = book->data->mh;
148
+ if (hdr == NULL) {
149
+ return Qnil;
150
+ }
151
+
152
+ res = rb_hash_new();
153
+ rb_hash_aset(res, ID2SYM(rb_intern("magic")), rb_str_new_cstr(hdr->mobi_magic));
154
+ COPY_HEADER_INT(header_length);
155
+ COPY_HEADER_INT(mobi_type);
156
+ if (hdr->text_encoding) {
157
+ rb_hash_aset(res, ID2SYM(rb_intern("text_encoding")), INT2FIX(*hdr->text_encoding));
158
+ switch (*hdr->text_encoding) {
159
+ case MOBI_CP1252:
160
+ rb_hash_aset(res, ID2SYM(rb_intern("text_encoding_sym")), ID2SYM(rb_intern("cp1252")));
161
+ break;
162
+ case MOBI_UTF8:
163
+ rb_hash_aset(res, ID2SYM(rb_intern("text_encoding_sym")), ID2SYM(rb_intern("utf8")));
164
+ break;
165
+ case MOBI_UTF16:
166
+ rb_hash_aset(res, ID2SYM(rb_intern("text_encoding_sym")), ID2SYM(rb_intern("utf16")));
167
+ break;
168
+ default:
169
+ rb_hash_aset(res, ID2SYM(rb_intern("text_encoding_sym")), ID2SYM(rb_intern("unknown")));
170
+ break;
171
+ }
172
+ }
173
+ if (hdr->locale) {
174
+ const char *locale_string = mobi_get_locale_string(*hdr->locale);
175
+ if (locale_string) {
176
+ rb_hash_aset(res, ID2SYM(rb_intern("locale_str")), rb_str_new_cstr(locale_string));
177
+ }
178
+ rb_hash_aset(res, ID2SYM(rb_intern("locale")), INT2FIX(*hdr->locale));
179
+ }
180
+ if (hdr->dict_input_lang) {
181
+ const char *locale_string = mobi_get_locale_string(*hdr->dict_input_lang);
182
+ if (locale_string) {
183
+ rb_hash_aset(res, ID2SYM(rb_intern("dict_input_lang_str")), rb_str_new_cstr(locale_string));
184
+ }
185
+ rb_hash_aset(res, ID2SYM(rb_intern("dict_input_lang")), INT2FIX(*hdr->dict_input_lang));
186
+ }
187
+ if (hdr->dict_output_lang) {
188
+ const char *locale_string = mobi_get_locale_string(*hdr->dict_output_lang);
189
+ if (locale_string) {
190
+ rb_hash_aset(res, ID2SYM(rb_intern("dict_output_lang_str")), rb_str_new_cstr(locale_string));
191
+ }
192
+ rb_hash_aset(res, ID2SYM(rb_intern("dict_output_lang")), INT2FIX(*hdr->dict_output_lang));
193
+ }
194
+ COPY_HEADER_INT(uid);
195
+ COPY_HEADER_INT(version);
196
+ COPY_HEADER_INT(orth_index);
197
+ COPY_HEADER_INT(infl_index);
198
+ COPY_HEADER_INT(names_index);
199
+ COPY_HEADER_INT(keys_index);
200
+ COPY_HEADER_INT(keys_index);
201
+ COPY_HEADER_INT(extra0_index);
202
+ COPY_HEADER_INT(extra1_index);
203
+ COPY_HEADER_INT(extra2_index);
204
+ COPY_HEADER_INT(extra3_index);
205
+ COPY_HEADER_INT(extra4_index);
206
+ COPY_HEADER_INT(extra5_index);
207
+ COPY_HEADER_INT(non_text_index);
208
+ COPY_HEADER_INT(full_name_offset);
209
+ COPY_HEADER_INT(full_name_length);
210
+ COPY_HEADER_INT(min_version);
211
+ COPY_HEADER_INT(image_index);
212
+ COPY_HEADER_INT(huff_rec_index);
213
+ COPY_HEADER_INT(huff_rec_count);
214
+ COPY_HEADER_INT(datp_rec_index);
215
+ COPY_HEADER_INT(datp_rec_count);
216
+ COPY_HEADER_INT(exth_flags);
217
+ COPY_HEADER_INT(unknown6);
218
+ COPY_HEADER_INT(drm_offset);
219
+ COPY_HEADER_INT(drm_count);
220
+ COPY_HEADER_INT(drm_size);
221
+ COPY_HEADER_INT(drm_flags);
222
+ COPY_HEADER_INT(first_text_index);
223
+ COPY_HEADER_INT(last_text_index);
224
+ COPY_HEADER_INT(fdst_index);
225
+ COPY_HEADER_INT(fdst_section_count);
226
+ COPY_HEADER_INT(fcis_index);
227
+ COPY_HEADER_INT(fcis_count);
228
+ COPY_HEADER_INT(flis_index);
229
+ COPY_HEADER_INT(flis_count);
230
+ COPY_HEADER_INT(unknown10);
231
+ COPY_HEADER_INT(unknown11);
232
+ COPY_HEADER_INT(srcs_index);
233
+ COPY_HEADER_INT(srcs_count);
234
+ COPY_HEADER_INT(unknown12);
235
+ COPY_HEADER_INT(unknown13);
236
+ COPY_HEADER_INT(extra_flags);
237
+ COPY_HEADER_INT(ncx_index);
238
+ COPY_HEADER_INT(unknown14);
239
+ COPY_HEADER_INT(unknown15);
240
+ COPY_HEADER_INT(fragment_index);
241
+ COPY_HEADER_INT(skeleton_index);
242
+ COPY_HEADER_INT(datp_index);
243
+ COPY_HEADER_INT(unknown16);
244
+ COPY_HEADER_INT(guide_index);
245
+ COPY_HEADER_INT(unknown17);
246
+ COPY_HEADER_INT(unknown18);
247
+ COPY_HEADER_INT(unknown19);
248
+ COPY_HEADER_INT(unknown20);
249
+
250
+ return res;
251
+ }
252
+ #undef COPY_HEADER_INT
253
+
254
+ static VALUE mb_book_pdb_header(VALUE self)
255
+ {
256
+ mb_BOOK *book = DATA_PTR(self);
257
+ VALUE res;
258
+ MOBIPdbHeader *hdr;
259
+
260
+ if (book == NULL || book->data == NULL) {
261
+ mb_raise(MOBI_INIT_FAILED, "the MOBI data is not loaded");
262
+ }
263
+ hdr = book->data->ph;
264
+ if (hdr == NULL) {
265
+ return Qnil;
266
+ }
267
+ res = rb_hash_new();
268
+ rb_hash_aset(res, ID2SYM(rb_intern("name")), rb_str_new_cstr(hdr->name));
269
+ rb_hash_aset(res, ID2SYM(rb_intern("attributes")), INT2FIX(hdr->attributes));
270
+ rb_hash_aset(res, ID2SYM(rb_intern("version")), INT2FIX(hdr->version));
271
+ rb_hash_aset(res, ID2SYM(rb_intern("ctime")), INT2FIX(hdr->ctime));
272
+ if (hdr->ctime) {
273
+ rb_hash_aset(res, ID2SYM(rb_intern("ctime_time")), rb_time_new(mktime(mobi_pdbtime_to_time(hdr->ctime)), 0));
274
+ }
275
+ rb_hash_aset(res, ID2SYM(rb_intern("mtime")), INT2FIX(hdr->mtime));
276
+ if (hdr->mtime) {
277
+ rb_hash_aset(res, ID2SYM(rb_intern("mtime_time")), rb_time_new(mktime(mobi_pdbtime_to_time(hdr->mtime)), 0));
278
+ }
279
+ rb_hash_aset(res, ID2SYM(rb_intern("btime")), INT2FIX(hdr->btime));
280
+ if (hdr->btime) {
281
+ rb_hash_aset(res, ID2SYM(rb_intern("btime_time")), rb_time_new(mktime(mobi_pdbtime_to_time(hdr->btime)), 0));
282
+ }
283
+ rb_hash_aset(res, ID2SYM(rb_intern("mod_num")), INT2FIX(hdr->mod_num));
284
+ rb_hash_aset(res, ID2SYM(rb_intern("appinfo_offset")), INT2FIX(hdr->appinfo_offset));
285
+ rb_hash_aset(res, ID2SYM(rb_intern("sortinfo_offset")), INT2FIX(hdr->sortinfo_offset));
286
+ rb_hash_aset(res, ID2SYM(rb_intern("type")), rb_str_new_cstr(hdr->type));
287
+ rb_hash_aset(res, ID2SYM(rb_intern("creator")), rb_str_new_cstr(hdr->creator));
288
+ rb_hash_aset(res, ID2SYM(rb_intern("uid")), INT2FIX(hdr->uid));
289
+ rb_hash_aset(res, ID2SYM(rb_intern("next_rec")), INT2FIX(hdr->next_rec));
290
+ rb_hash_aset(res, ID2SYM(rb_intern("rec_count")), INT2FIX(hdr->rec_count));
291
+ return res;
292
+ }
293
+
294
+ static VALUE mb_book_record0_header(VALUE self)
295
+ {
296
+ mb_BOOK *book = DATA_PTR(self);
297
+ VALUE res;
298
+ MOBIRecord0Header *hdr;
299
+
300
+ if (book == NULL || book->data == NULL) {
301
+ mb_raise(MOBI_INIT_FAILED, "the MOBI data is not loaded");
302
+ }
303
+ hdr = book->data->rh;
304
+ if (hdr == NULL) {
305
+ return Qnil;
306
+ }
307
+
308
+ res = rb_hash_new();
309
+ rb_hash_aset(res, ID2SYM(rb_intern("compression_type")), INT2FIX(hdr->compression_type));
310
+ switch (hdr->compression_type) {
311
+ case 1:
312
+ rb_hash_aset(res, ID2SYM(rb_intern("compression_type_sym")), ID2SYM(rb_intern("none")));
313
+ break;
314
+ case 2:
315
+ rb_hash_aset(res, ID2SYM(rb_intern("compression_type_sym")), ID2SYM(rb_intern("palm_doc")));
316
+ break;
317
+ case 17480:
318
+ rb_hash_aset(res, ID2SYM(rb_intern("compression_type_sym")), ID2SYM(rb_intern("huff_cdic")));
319
+ break;
320
+ }
321
+ rb_hash_aset(res, ID2SYM(rb_intern("text_length")), INT2FIX(hdr->text_length));
322
+ rb_hash_aset(res, ID2SYM(rb_intern("text_record_count")), INT2FIX(hdr->text_record_count));
323
+ rb_hash_aset(res, ID2SYM(rb_intern("text_record_size")), INT2FIX(hdr->text_record_size));
324
+ rb_hash_aset(res, ID2SYM(rb_intern("encryption_type")), INT2FIX(hdr->encryption_type));
325
+ switch (hdr->encryption_type) {
326
+ case 0:
327
+ rb_hash_aset(res, ID2SYM(rb_intern("encryption_type_sym")), ID2SYM(rb_intern("none")));
328
+ break;
329
+ case 1:
330
+ rb_hash_aset(res, ID2SYM(rb_intern("encryption_type_sym")), ID2SYM(rb_intern("old")));
331
+ break;
332
+ case 2:
333
+ rb_hash_aset(res, ID2SYM(rb_intern("encryption_type_sym")), ID2SYM(rb_intern("mobi")));
334
+ break;
335
+ }
336
+ rb_hash_aset(res, ID2SYM(rb_intern("unknown1")), INT2FIX(hdr->unknown1));
337
+ return res;
338
+ }
339
+
340
+ static VALUE mb_book_exth_header(VALUE self)
341
+ {
342
+ mb_BOOK *book = DATA_PTR(self);
343
+ VALUE res;
344
+ MOBIExthHeader *hdr;
345
+
346
+ if (book == NULL || book->data == NULL) {
347
+ mb_raise(MOBI_INIT_FAILED, "the MOBI data is not loaded");
348
+ }
349
+ hdr = book->data->eh;
350
+ if (hdr == NULL) {
351
+ return Qnil;
352
+ }
353
+
354
+ res = rb_ary_new();
355
+ while (hdr != NULL) {
356
+ MOBIExthMeta tag = mobi_get_exthtagmeta_by_tag(hdr->tag);
357
+ uint32_t val32;
358
+ VALUE item = rb_hash_new();
359
+
360
+ rb_hash_aset(item, ID2SYM(rb_intern("code")), INT2FIX(tag.tag));
361
+ switch (hdr->tag) {
362
+ case EXTH_SAMPLE:
363
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("sample")));
364
+ break;
365
+ case EXTH_STARTREADING:
366
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("start_reading")));
367
+ break;
368
+ case EXTH_KF8BOUNDARY:
369
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("kf8_boundary")));
370
+ break;
371
+ case EXTH_COUNTRESOURCES:
372
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("count_resources")));
373
+ break;
374
+ case EXTH_RESCOFFSET:
375
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("resc_offset")));
376
+ break;
377
+ case EXTH_COVEROFFSET:
378
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("cover_offset")));
379
+ break;
380
+ case EXTH_THUMBOFFSET:
381
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("thumb_offset")));
382
+ break;
383
+ case EXTH_HASFAKECOVER:
384
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("has_fake_cover")));
385
+ break;
386
+ case EXTH_CREATORSOFT:
387
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("creator_soft")));
388
+ break;
389
+ case EXTH_CREATORMAJOR:
390
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("creator_major")));
391
+ break;
392
+ case EXTH_CREATORMINOR:
393
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("creator_minor")));
394
+ break;
395
+ case EXTH_CREATORBUILD:
396
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("creator_build")));
397
+ break;
398
+ case EXTH_CLIPPINGLIMIT:
399
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("clipping_limit")));
400
+ break;
401
+ case EXTH_PUBLISHERLIMIT:
402
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("publisher_limit")));
403
+ break;
404
+ case EXTH_TTSDISABLE:
405
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("tts_disabled")));
406
+ break;
407
+ case EXTH_RENTAL:
408
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("rental")));
409
+ break;
410
+ case EXTH_DRMSERVER:
411
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("drm_server")));
412
+ break;
413
+ case EXTH_DRMCOMMERCE:
414
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("drm_commerce")));
415
+ break;
416
+ case EXTH_DRMEBOOKBASE:
417
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("drm_ebookbase")));
418
+ break;
419
+ case EXTH_TITLE:
420
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("title")));
421
+ break;
422
+ case EXTH_AUTHOR:
423
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("creator")));
424
+ break;
425
+ case EXTH_PUBLISHER:
426
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("publisher")));
427
+ break;
428
+ case EXTH_IMPRINT:
429
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("imprint")));
430
+ break;
431
+ case EXTH_DESCRIPTION:
432
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("description")));
433
+ break;
434
+ case EXTH_ISBN:
435
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("isbn")));
436
+ break;
437
+ case EXTH_SUBJECT:
438
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("subject")));
439
+ break;
440
+ case EXTH_PUBLISHINGDATE:
441
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("published")));
442
+ break;
443
+ case EXTH_REVIEW:
444
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("review")));
445
+ break;
446
+ case EXTH_CONTRIBUTOR:
447
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("contributor")));
448
+ break;
449
+ case EXTH_RIGHTS:
450
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("rights")));
451
+ break;
452
+ case EXTH_SUBJECTCODE:
453
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("subject_code")));
454
+ break;
455
+ case EXTH_TYPE:
456
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("type")));
457
+ break;
458
+ case EXTH_SOURCE:
459
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("source")));
460
+ break;
461
+ case EXTH_ASIN:
462
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("asin")));
463
+ break;
464
+ case EXTH_VERSION:
465
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("version")));
466
+ break;
467
+ case EXTH_ADULT:
468
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("adult")));
469
+ break;
470
+ case EXTH_PRICE:
471
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("price")));
472
+ break;
473
+ case EXTH_CURRENCY:
474
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("currency")));
475
+ break;
476
+ case EXTH_FIXEDLAYOUT:
477
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("fixed_layout")));
478
+ break;
479
+ case EXTH_BOOKTYPE:
480
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("book_type")));
481
+ break;
482
+ case EXTH_ORIENTATIONLOCK:
483
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("orientation_lock")));
484
+ break;
485
+ case EXTH_ORIGRESOLUTION:
486
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("orig_resolution")));
487
+ break;
488
+ case EXTH_ZEROGUTTER:
489
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("zero_gutter")));
490
+ break;
491
+ case EXTH_ZEROMARGIN:
492
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("zero_margin")));
493
+ break;
494
+ case EXTH_KF8COVERURI:
495
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("kf8_cover_uri")));
496
+ break;
497
+ case EXTH_REGIONMAGNI:
498
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("region_magnification")));
499
+ break;
500
+ case EXTH_DICTNAME:
501
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("dict_name")));
502
+ break;
503
+ case EXTH_WATERMARK:
504
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("watermark")));
505
+ break;
506
+ case EXTH_DOCTYPE:
507
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("doc_type")));
508
+ break;
509
+ case EXTH_LASTUPDATE:
510
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("last_update")));
511
+ break;
512
+ case EXTH_UPDATEDTITLE:
513
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("updated_title")));
514
+ break;
515
+ case EXTH_ASIN504:
516
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("asin_504")));
517
+ break;
518
+ case EXTH_TITLEFILEAS:
519
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("title_file_as")));
520
+ break;
521
+ case EXTH_CREATORFILEAS:
522
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("creator_file_as")));
523
+ break;
524
+ case EXTH_PUBLISHERFILEAS:
525
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("publisher_file_as")));
526
+ break;
527
+ case EXTH_LANGUAGE:
528
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("language")));
529
+ break;
530
+ case EXTH_ALIGNMENT:
531
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("alignment")));
532
+ break;
533
+ case EXTH_PAGEDIR:
534
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("page_dir")));
535
+ break;
536
+ case EXTH_OVERRIDEFONTS:
537
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("override_fonts")));
538
+ break;
539
+ case EXTH_SORCEDESC:
540
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("source_desc")));
541
+ break;
542
+ case EXTH_DICTLANGIN:
543
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("dict_lang_in")));
544
+ break;
545
+ case EXTH_DICTLANGOUT:
546
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("dict_lang_out")));
547
+ break;
548
+ case EXTH_INPUTSOURCE:
549
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("input_source")));
550
+ break;
551
+ case EXTH_CREATORBUILDREV:
552
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("creator_build_rev")));
553
+ break;
554
+ case EXTH_CREATORSTRING:
555
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("creator_string")));
556
+ break;
557
+ case EXTH_TAMPERKEYS:
558
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("tamper_keys")));
559
+ break;
560
+ case EXTH_FONTSIGNATURE:
561
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("font_signature")));
562
+ break;
563
+ case EXTH_UNK403:
564
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("unknown_403")));
565
+ break;
566
+ case EXTH_UNK405:
567
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("unknown_405")));
568
+ break;
569
+ case EXTH_UNK407:
570
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("unknown_407")));
571
+ break;
572
+ case EXTH_UNK450:
573
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("unknown_450")));
574
+ break;
575
+ case EXTH_UNK451:
576
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("unknown_451")));
577
+ break;
578
+ case EXTH_UNK452:
579
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("unknown_452")));
580
+ break;
581
+ case EXTH_UNK453:
582
+ rb_hash_aset(item, ID2SYM(rb_intern("id")), ID2SYM(rb_intern("unknown_453")));
583
+ break;
584
+ }
585
+ if (tag.tag == 0) {
586
+ rb_hash_aset(item, ID2SYM(rb_intern("val_bin")), rb_str_new((const char *)hdr->data, hdr->size));
587
+ val32 = mobi_decode_exthvalue(hdr->data, hdr->size);
588
+ rb_hash_aset(item, ID2SYM(rb_intern("val_num")), INT2FIX(val32));
589
+ } else {
590
+ char *str;
591
+ rb_hash_aset(item, ID2SYM(rb_intern("name")), rb_str_new_cstr(tag.name));
592
+ switch (tag.type) {
593
+ case EXTH_NUMERIC:
594
+ val32 = mobi_decode_exthvalue(hdr->data, hdr->size);
595
+ rb_hash_aset(item, ID2SYM(rb_intern("val_num")), INT2FIX(val32));
596
+ break;
597
+ case EXTH_STRING:
598
+ str = mobi_decode_exthstring(book->data, hdr->data, hdr->size);
599
+ if (str) {
600
+ rb_hash_aset(item, ID2SYM(rb_intern("val_str")), rb_str_new_cstr(str));
601
+ free(str);
602
+ }
603
+ break;
604
+ case EXTH_BINARY:
605
+ rb_hash_aset(item, ID2SYM(rb_intern("val_bin")), rb_str_new((const char *)hdr->data, hdr->size));
606
+ break;
607
+ default:
608
+ break;
609
+ }
610
+ }
611
+ rb_ary_push(res, item);
612
+ hdr = hdr->next;
613
+ }
614
+ return res;
615
+ }
616
+
617
+ #define FULL_NAME_MAX 1024
618
+ static VALUE mb_book_full_name(VALUE self)
619
+ {
620
+ mb_BOOK *book = DATA_PTR(self);
621
+ MOBI_RET rc;
622
+ char full_name[FULL_NAME_MAX + 1] = {0};
623
+
624
+ if (book == NULL || book->data == NULL) {
625
+ mb_raise(MOBI_INIT_FAILED, "the MOBI data is not loaded");
626
+ }
627
+ rc = mobi_get_fullname(book->data, full_name, FULL_NAME_MAX);
628
+ if (rc != MOBI_SUCCESS) {
629
+ mb_raise(rc, "unable to fetch book full name");
630
+ }
631
+ return rb_str_new_cstr(full_name);
632
+ }
633
+ #undef FULL_NAME_MAX
634
+
635
+ #define DEFINE_META_GETTER_STR(ATTR) \
636
+ static VALUE mb_book_##ATTR(VALUE self) \
637
+ { \
638
+ mb_BOOK *book = DATA_PTR(self); \
639
+ const char *str; \
640
+ \
641
+ if (book == NULL || book->data == NULL) { \
642
+ mb_raise(MOBI_INIT_FAILED, "the MOBI data is not loaded"); \
643
+ } \
644
+ str = mobi_meta_get_##ATTR(book->data); \
645
+ if (str) { \
646
+ return rb_str_new_cstr(str); \
647
+ } \
648
+ return Qnil; \
649
+ }
650
+
651
+ DEFINE_META_GETTER_STR(title)
652
+ DEFINE_META_GETTER_STR(author)
653
+ DEFINE_META_GETTER_STR(publisher)
654
+ DEFINE_META_GETTER_STR(imprint)
655
+ DEFINE_META_GETTER_STR(description)
656
+ DEFINE_META_GETTER_STR(isbn)
657
+ DEFINE_META_GETTER_STR(subject)
658
+ DEFINE_META_GETTER_STR(publishdate)
659
+ DEFINE_META_GETTER_STR(review)
660
+ DEFINE_META_GETTER_STR(contributor)
661
+ DEFINE_META_GETTER_STR(copyright)
662
+ DEFINE_META_GETTER_STR(asin)
663
+ DEFINE_META_GETTER_STR(language)
664
+
665
+ #define DEFINE_PREDICATE(METHOD, FUNCTION) \
666
+ static VALUE mb_book_p_##METHOD(VALUE self) \
667
+ { \
668
+ mb_BOOK *book = DATA_PTR(self); \
669
+ \
670
+ if (book == NULL || book->data == NULL) { \
671
+ mb_raise(MOBI_INIT_FAILED, "the MOBI data is not loaded"); \
672
+ } \
673
+ if (FUNCTION(book->data)) { \
674
+ return Qtrue; \
675
+ } \
676
+ return Qfalse; \
677
+ }
678
+
679
+ DEFINE_PREDICATE(has_mobi_header, mobi_exists_mobiheader)
680
+ DEFINE_PREDICATE(has_fdst, mobi_exists_fdst)
681
+ DEFINE_PREDICATE(has_skeleton_index, mobi_exists_skel_indx)
682
+ DEFINE_PREDICATE(has_fragments_index, mobi_exists_frag_indx)
683
+ DEFINE_PREDICATE(has_guide_index, mobi_exists_guide_indx)
684
+ DEFINE_PREDICATE(has_ncx_index, mobi_exists_ncx)
685
+ DEFINE_PREDICATE(has_orth_index, mobi_exists_orth)
686
+ DEFINE_PREDICATE(has_infl_index, mobi_exists_infl)
687
+ DEFINE_PREDICATE(is_hybrid, mobi_is_hybrid)
688
+ DEFINE_PREDICATE(is_encrypted, mobi_is_encrypted)
689
+ DEFINE_PREDICATE(is_mobipocket, mobi_is_mobipocket)
690
+ DEFINE_PREDICATE(is_dictionary, mobi_is_dictionary)
691
+ DEFINE_PREDICATE(is_kf8, mobi_is_kf8)
692
+
693
+ static VALUE mb_book_records(VALUE self)
694
+ {
695
+ mb_BOOK *book = DATA_PTR(self);
696
+ VALUE res;
697
+ const MOBIPdbRecord *rec;
698
+
699
+ if (book == NULL || book->data == NULL) {
700
+ mb_raise(MOBI_INIT_FAILED, "the MOBI data is not loaded");
701
+ }
702
+ rec = book->data->rec;
703
+ if (rec == NULL) {
704
+ return Qnil;
705
+ }
706
+
707
+ res = rb_ary_new();
708
+ while (rec != NULL) {
709
+ VALUE item = rb_hash_new();
710
+ rb_hash_aset(item, ID2SYM(rb_intern("offset")), INT2FIX(rec->offset));
711
+ rb_hash_aset(item, ID2SYM(rb_intern("size")), INT2FIX(rec->size));
712
+ rb_hash_aset(item, ID2SYM(rb_intern("attributes")), INT2FIX(rec->attributes));
713
+ rb_hash_aset(item, ID2SYM(rb_intern("uid")), INT2FIX(rec->uid));
714
+ rb_hash_aset(item, ID2SYM(rb_intern("data")), rb_str_new((const char *)rec->data, rec->size));
715
+ rb_ary_push(res, item);
716
+ rec = rec->next;
717
+ }
718
+ return res;
719
+ }
720
+
721
+ static VALUE mb_book_rawml(VALUE self)
722
+ {
723
+ mb_BOOK *book = DATA_PTR(self);
724
+ MOBI_RET rc;
725
+ char *text = NULL;
726
+ size_t size;
727
+ VALUE res;
728
+
729
+ if (book == NULL || book->data == NULL) {
730
+ mb_raise(MOBI_INIT_FAILED, "the MOBI data is not loaded");
731
+ }
732
+ size = mobi_get_text_maxsize(book->data);
733
+ if (size == MOBI_NOTSET) {
734
+ mb_raise(MOBI_DATA_CORRUPT, "unable to determine size for rawml");
735
+ }
736
+ text = calloc(size + 1, sizeof(char));
737
+ if (text == NULL) {
738
+ mb_raise(MOBI_MALLOC_FAILED, "unable to allocate memory for rawml buffer");
739
+ }
740
+ rc = mobi_get_rawml(book->data, text, &size);
741
+ if (rc != MOBI_SUCCESS) {
742
+ free(text);
743
+ mb_raise(rc, "unable to generate rawml");
744
+ return Qnil;
745
+ }
746
+ res = rb_str_new(text, size);
747
+ free(text);
748
+ return res;
749
+ }
750
+
751
+ static VALUE mb_extract_mobiparts(const MOBIPart *part)
752
+ {
753
+ VALUE items = rb_ary_new();
754
+ while (part != NULL) {
755
+ VALUE item = rb_hash_new();
756
+ rb_hash_aset(item, ID2SYM(rb_intern("type")), INT2FIX(part->type));
757
+ switch (part->type) {
758
+ case T_UNKNOWN:
759
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("unknown")));
760
+ break;
761
+ case T_HTML:
762
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("html")));
763
+ break;
764
+ case T_CSS:
765
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("css")));
766
+ break;
767
+ case T_SVG:
768
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("svg")));
769
+ break;
770
+ case T_OPF:
771
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("opf")));
772
+ break;
773
+ case T_NCX:
774
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("ncx")));
775
+ break;
776
+ case T_JPG:
777
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("jpg")));
778
+ break;
779
+ case T_GIF:
780
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("gif")));
781
+ break;
782
+ case T_PNG:
783
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("png")));
784
+ break;
785
+ case T_BMP:
786
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("bmp")));
787
+ break;
788
+ case T_OTF:
789
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("otf")));
790
+ break;
791
+ case T_TTF:
792
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("ttf")));
793
+ break;
794
+ case T_MP3:
795
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("mp3")));
796
+ break;
797
+ case T_MPG:
798
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("mpg")));
799
+ break;
800
+ case T_PDF:
801
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("pdf")));
802
+ break;
803
+ case T_FONT:
804
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("font")));
805
+ break;
806
+ case T_AUDIO:
807
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("audio")));
808
+ break;
809
+ case T_VIDEO:
810
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("video")));
811
+ break;
812
+ case T_BREAK:
813
+ rb_hash_aset(item, ID2SYM(rb_intern("type_sym")), ID2SYM(rb_intern("break")));
814
+ break;
815
+ }
816
+ rb_hash_aset(item, ID2SYM(rb_intern("uid")), INT2FIX(part->uid));
817
+ rb_hash_aset(item, ID2SYM(rb_intern("size")), INT2FIX(part->size));
818
+ rb_hash_aset(item, ID2SYM(rb_intern("data")), rb_str_new((const char *)part->data, part->size));
819
+ part = part->next;
820
+ rb_ary_push(items, item);
821
+ }
822
+ return items;
823
+ }
824
+
825
+ static VALUE mb_book_rawml_parts(VALUE self)
826
+ {
827
+ mb_BOOK *book = DATA_PTR(self);
828
+ MOBI_RET rc;
829
+ MOBIRawml *rawml;
830
+ VALUE res;
831
+
832
+ if (book == NULL || book->data == NULL) {
833
+ mb_raise(MOBI_INIT_FAILED, "the MOBI data is not loaded");
834
+ }
835
+ rawml = mobi_init_rawml(book->data);
836
+ if (rawml == NULL) {
837
+ mb_raise(MOBI_MALLOC_FAILED, "unable to allocate memory for rawml structure");
838
+ }
839
+ rc = mobi_parse_rawml(rawml, book->data);
840
+ if (rc != MOBI_SUCCESS) {
841
+ mobi_free_rawml(rawml);
842
+ mb_raise(rc, "unable to generate rawml");
843
+ return Qnil;
844
+ }
845
+ res = rb_hash_new();
846
+ rb_hash_aset(res, ID2SYM(rb_intern("version")), INT2FIX(rawml->version));
847
+ if (rawml->markup != NULL) {
848
+ rb_hash_aset(res, ID2SYM(rb_intern("markup")), mb_extract_mobiparts(rawml->markup));
849
+ }
850
+ if (rawml->flow != NULL) {
851
+ rb_hash_aset(res, ID2SYM(rb_intern("flow")), mb_extract_mobiparts(rawml->flow));
852
+ }
853
+ if (rawml->resources != NULL) {
854
+ rb_hash_aset(res, ID2SYM(rb_intern("resources")), mb_extract_mobiparts(rawml->resources));
855
+ }
856
+ mobi_free_rawml(rawml);
857
+ return res;
858
+ }
859
+
860
+ static void init_mobi_book()
861
+ {
862
+ mb_cBook = rb_define_class_under(mb_mMOBI, "Book", rb_cObject);
863
+ rb_define_alloc_func(mb_cBook, mb_book_alloc);
864
+ rb_define_method(mb_cBook, "initialize", mb_book_init, 1);
865
+ rb_define_method(mb_cBook, "next", mb_book_next, 0);
866
+ rb_define_method(mb_cBook, "mobi_header", mb_book_mobi_header, 0);
867
+ rb_define_method(mb_cBook, "pdb_header", mb_book_pdb_header, 0);
868
+ rb_define_method(mb_cBook, "record0_header", mb_book_record0_header, 0);
869
+ rb_define_method(mb_cBook, "exth_header", mb_book_exth_header, 0);
870
+ rb_define_method(mb_cBook, "full_name", mb_book_full_name, 0);
871
+ rb_define_method(mb_cBook, "title", mb_book_title, 0);
872
+ rb_define_method(mb_cBook, "author", mb_book_author, 0);
873
+ rb_define_method(mb_cBook, "publisher", mb_book_publisher, 0);
874
+ rb_define_method(mb_cBook, "imprint", mb_book_imprint, 0);
875
+ rb_define_method(mb_cBook, "description", mb_book_description, 0);
876
+ rb_define_method(mb_cBook, "isbn", mb_book_isbn, 0);
877
+ rb_define_method(mb_cBook, "subject", mb_book_subject, 0);
878
+ rb_define_method(mb_cBook, "publishdate", mb_book_publishdate, 0);
879
+ rb_define_method(mb_cBook, "review", mb_book_review, 0);
880
+ rb_define_method(mb_cBook, "contributor", mb_book_contributor, 0);
881
+ rb_define_method(mb_cBook, "copyright", mb_book_copyright, 0);
882
+ rb_define_method(mb_cBook, "asin", mb_book_asin, 0);
883
+ rb_define_method(mb_cBook, "language", mb_book_language, 0);
884
+ rb_define_method(mb_cBook, "has_mobi_header?", mb_book_p_has_mobi_header, 0);
885
+ rb_define_method(mb_cBook, "has_fdst?", mb_book_p_has_fdst, 0);
886
+ rb_define_method(mb_cBook, "has_skeleton_index?", mb_book_p_has_skeleton_index, 0);
887
+ rb_define_method(mb_cBook, "has_fragments_index?", mb_book_p_has_fragments_index, 0);
888
+ rb_define_method(mb_cBook, "has_guide_index?", mb_book_p_has_guide_index, 0);
889
+ rb_define_method(mb_cBook, "has_ncx_index?", mb_book_p_has_ncx_index, 0);
890
+ rb_define_method(mb_cBook, "has_orth_index?", mb_book_p_has_orth_index, 0);
891
+ rb_define_method(mb_cBook, "has_infl_index?", mb_book_p_has_infl_index, 0);
892
+ rb_define_method(mb_cBook, "is_hybrid?", mb_book_p_is_hybrid, 0);
893
+ rb_define_method(mb_cBook, "is_encrypted?", mb_book_p_is_encrypted, 0);
894
+ rb_define_method(mb_cBook, "is_mobipocket?", mb_book_p_is_mobipocket, 0);
895
+ rb_define_method(mb_cBook, "is_dictionary?", mb_book_p_is_dictionary, 0);
896
+ rb_define_method(mb_cBook, "is_kf8?", mb_book_p_is_kf8, 0);
897
+ rb_define_method(mb_cBook, "records", mb_book_records, 0);
898
+ rb_define_method(mb_cBook, "rawml", mb_book_rawml, 0);
899
+ rb_define_method(mb_cBook, "rawml_parts", mb_book_rawml_parts, 0);
900
+ }
901
+
902
+ void Init_mobi_ext()
903
+ {
904
+ mb_mMOBI = rb_define_module("MOBI");
905
+ rb_define_const(mb_mMOBI, "LIB_VERSION", rb_str_freeze(rb_external_str_new_cstr(mobi_version())));
906
+ mb_eError = rb_const_get(mb_mMOBI, rb_intern("Error"));
907
+
908
+ init_mobi_book();
909
+ }