79 lines
1.7 KiB
Markdown
79 lines
1.7 KiB
Markdown
|
```c title="C"
|
||
|
|
#include <kreuzberg.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <string.h>
|
||
|
|
|
||
|
|
static char *dup_cstr(const char *s) {
|
||
|
|
size_t len = strlen(s);
|
||
|
|
char *out = (char *)malloc(len + 1);
|
||
|
|
if (out) {
|
||
|
|
memcpy(out, s, len + 1);
|
||
|
|
}
|
||
|
|
return out;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int32_t process_fn(
|
||
|
|
const void *user_data,
|
||
|
|
const char *result,
|
||
|
|
const char *config,
|
||
|
|
char **out_error
|
||
|
|
) {
|
||
|
|
(void)user_data;
|
||
|
|
(void)config;
|
||
|
|
(void)out_error;
|
||
|
|
|
||
|
|
printf("pdf-only-processor: handling result of length %zu\n", strlen(result));
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int32_t processing_stage_fn(
|
||
|
|
const void *user_data,
|
||
|
|
char **out_result
|
||
|
|
) {
|
||
|
|
(void)user_data;
|
||
|
|
*out_result = dup_cstr("\"Middle\"");
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int32_t should_process_fn(
|
||
|
|
const void *user_data,
|
||
|
|
const char *result,
|
||
|
|
const char *config
|
||
|
|
) {
|
||
|
|
(void)user_data;
|
||
|
|
(void)config;
|
||
|
|
/* Only process PDF mime types. */
|
||
|
|
return strstr(result, "\"mime_type\":\"application/pdf\"") != NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int32_t priority_fn(const void *user_data) {
|
||
|
|
(void)user_data;
|
||
|
|
return 50;
|
||
|
|
}
|
||
|
|
|
||
|
|
int main(void) {
|
||
|
|
KREUZBERGKreuzbergPostProcessorVTable vtable = {0};
|
||
|
|
vtable.process = process_fn;
|
||
|
|
vtable.processing_stage = processing_stage_fn;
|
||
|
|
vtable.should_process = should_process_fn;
|
||
|
|
vtable.priority = priority_fn;
|
||
|
|
|
||
|
|
char *err = NULL;
|
||
|
|
int32_t rc = kreuzberg_register_post_processor(
|
||
|
|
"pdf-only-processor",
|
||
|
|
vtable,
|
||
|
|
NULL,
|
||
|
|
&err
|
||
|
|
);
|
||
|
|
if (rc != 0) {
|
||
|
|
fprintf(stderr, "register post-processor failed: %s\n", err ? err : "(no detail)");
|
||
|
|
kreuzberg_free_string(err);
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
printf("pdf-only-processor registered\n");
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
```
|