Compare commits

...

39 Commits

Author SHA1 Message Date
ae8652d86e UI tweaks, search syntax (#25) 2020-02-16 15:24:29 -05:00
849beb09d8 hotfix 2020-02-15 19:33:18 -05:00
e1aaaee617 UI tweak 2020-02-15 09:30:14 -05:00
c02b940945 (I forgot to commit this) 2020-02-14 20:58:10 -05:00
2934ddb07f Add image viewer (#2) 2020-02-14 18:28:55 -05:00
7f6f3c02fa OCR tweaks 2020-02-11 21:13:47 -05:00
7f98d5a682 Fix buffer overflow (whoops) 2020-02-09 18:11:29 -05:00
7eb9c5d7d5 Fix web/index issue with NULL mime types 2020-02-09 17:23:49 -05:00
184439aa38 increase minimum image size for OCR 2020-02-09 14:06:59 -05:00
1ce8b298a1 Display EXIF tags on document info panel, remove march=native on openjp 2020-02-09 13:21:19 -05:00
75f99025d9 add exif dateTime, allow some special characters in text meta 2020-02-09 08:47:13 -05:00
ebe852bd5a Fix rewrite-url arg 2020-02-09 08:23:17 -05:00
402b103c49 Fix total count for ES 7.5 2020-02-08 09:25:00 -05:00
e9b6e1cdc2 Turn off auto optimisation in libtesseract build 2020-02-08 08:32:04 -05:00
ed1ce8ab5e Handle XML errors #18 2020-02-07 10:08:01 -05:00
d1fa4febc4 Improve scroll feature, UI fix 2020-02-07 10:08:01 -05:00
048c55df7b Update README.md 2020-02-06 19:56:29 -05:00
f77bc6a025 Update README.md 2020-02-06 19:55:32 -05:00
efdde2734e version bump 2020-02-06 19:28:05 -05:00
66658fa8f7 Remove trailing/leading white space in text meta fields 2020-02-06 19:27:30 -05:00
df41c251e4 (Breaking!) Add some exif tags 2020-02-06 19:21:50 -05:00
3282ab56ba Version bump 2020-02-02 09:26:54 -05:00
8300838d30 Suppress XML parsing errors (#18) 2020-02-02 09:26:03 -05:00
c9870a6d3d Remove -march=native for release build... 2020-02-02 09:03:06 -05:00
a143cc4fcf bundle openssl... 2020-02-02 08:39:20 -05:00
9ef1f3781d fix attempt for #11 2020-02-01 20:04:26 -05:00
bbee8aa721 tesseract ocr path fix 2020-02-01 20:03:59 -05:00
d22f83c797 curl fix 2020-02-01 15:22:43 -05:00
50615486a4 curl fix attempt 2020-02-01 14:42:42 -05:00
ca79e4f797 add /status endpoint 2020-01-28 10:18:37 -05:00
6a9fd08a80 Merge pull request #21 from simon987/wip-20
Fixes #20
2020-01-27 09:16:00 -05:00
cab890dc9b #20 wip 2020-01-27 09:09:42 -05:00
b3c4faf2df Update README.md 2020-01-26 12:37:13 -05:00
353937171a Update README.md 2020-01-20 15:54:53 -05:00
c80002bea4 Bundle libcurl attempt 2 2020-01-18 11:53:12 -05:00
56adee9d81 Bundle libcurl, libopc bugfix #18 2020-01-18 10:25:02 -05:00
d6493d6d5f Bundle libpng 2020-01-16 16:21:38 -05:00
0967e9676d remove static build in CI... 2020-01-16 15:45:18 -05:00
487e998ea0 Display error message on /d/ error 2020-01-16 15:04:50 -05:00
40 changed files with 625 additions and 184 deletions

3
.gitmodules vendored
View File

@@ -40,3 +40,6 @@
[submodule "lib/libtiff"] [submodule "lib/libtiff"]
path = lib/libtiff path = lib/libtiff
url = https://gitlab.com/libtiff/libtiff url = https://gitlab.com/libtiff/libtiff
[submodule "lib/libpng"]
path = lib/libpng
url = https://github.com/glennrp/libpng

View File

@@ -45,7 +45,6 @@ add_executable(
find_package(PkgConfig REQUIRED) find_package(PkgConfig REQUIRED)
set(ENV{PKG_CONFIG_PATH} "$ENV{PKG_CONFIG_PATH}:/usr/local/lib/pkgconfig/") set(ENV{PKG_CONFIG_PATH} "$ENV{PKG_CONFIG_PATH}:/usr/local/lib/pkgconfig/")
#find_package(OpenSSL REQUIRED)
find_package(Freetype REQUIRED) find_package(Freetype REQUIRED)
pkg_check_modules(GLIB REQUIRED glib-2.0) pkg_check_modules(GLIB REQUIRED glib-2.0)
@@ -84,7 +83,8 @@ target_link_directories(
target_compile_options(sist2 target_compile_options(sist2
PRIVATE PRIVATE
-Ofast -Ofast
# -march=native # -march=native
-fPIC
-fno-stack-protector -fno-stack-protector
-fomit-frame-pointer -fomit-frame-pointer
) )
@@ -111,7 +111,7 @@ TARGET_LINK_LIBRARIES(
${PROJECT_SOURCE_DIR}/lib/libonion_static.a ${PROJECT_SOURCE_DIR}/lib/libonion_static.a
pthread pthread
curl
m m
bz2 bz2
${PROJECT_SOURCE_DIR}/lib/libmagic.a ${PROJECT_SOURCE_DIR}/lib/libmagic.a
@@ -128,8 +128,14 @@ TARGET_LINK_LIBRARIES(
${PROJECT_SOURCE_DIR}/lib/libtesseract.a ${PROJECT_SOURCE_DIR}/lib/libtesseract.a
${PROJECT_SOURCE_DIR}/lib/liblept.a ${PROJECT_SOURCE_DIR}/lib/liblept.a
${PROJECT_SOURCE_DIR}/lib/libtiff.a ${PROJECT_SOURCE_DIR}/lib/libtiff.a
png ${PROJECT_SOURCE_DIR}/lib/libpng16.a
stdc++ stdc++
# curl
${PROJECT_SOURCE_DIR}/lib/libcurl.a
${PROJECT_SOURCE_DIR}/lib/libcrypto.a
${PROJECT_SOURCE_DIR}/lib/libssl.a
dl
) )
add_custom_target( add_custom_target(

View File

@@ -16,4 +16,7 @@ RUN mkdir -p /usr/share/tessdata && \
ADD sist2 /root/sist2 ADD sist2 /root/sist2
ENV LANG C.UTF-8
ENV LC_ALL C.UTF-8
ENTRYPOINT ["/root/sist2"] ENTRYPOINT ["/root/sist2"]

View File

@@ -5,6 +5,11 @@ strip sist2
version=$(./sist2 --version) version=$(./sist2 --version)
echo "Version ${version}" echo "Version ${version}"
docker build . -t simon987/sist2:${version} -t simon987/sist2:latest docker build . -t simon987/sist2:${version} -t simon987/sist2:latest \
-t docker.pkg.github.com/simon987/sist2/sist2:latest -t docker.pkg.github.com/simon987/sist2/sist2:${version}
docker push simon987/sist2:${version} docker push simon987/sist2:${version}
docker push simon987/sist2:latest docker push simon987/sist2:latest
docker push docker.pkg.github.com/simon987/sist2/sist2:latest
docker push docker.pkg.github.com/simon987/sist2/sist2:${version}
docker run --rm -it simon987/sist2 -v

View File

@@ -29,7 +29,7 @@ sist2 (Simple incremental search tool)
1. Have an [Elasticsearch](https://www.elastic.co/downloads/elasticsearch) instance running 1. Have an [Elasticsearch](https://www.elastic.co/downloads/elasticsearch) instance running
1. 1.
1. Download the [latest sist2 release](https://github.com/simon987/sist2/releases) * 1. Download the [latest sist2 release](https://github.com/simon987/sist2/releases) *
1. *(or)* Download an [development snapshot](https://files.simon987.net/artifacts/Sist2/Build/) *(Not recommended!)* 1. *(or)* Download a [development snapshot](https://files.simon987.net/artifacts/Sist2/Build/) *(Not recommended!)*
1. *(or)* `docker pull simon987/sist2:latest` 1. *(or)* `docker pull simon987/sist2:latest`
@@ -39,6 +39,9 @@ sist2 (Simple incremental search tool)
## Example usage ## Example usage
![demo](demo.gif)
See help page `sist2 --help` for more details. See help page `sist2 --help` for more details.
**Scan a directory** **Scan a directory**
@@ -92,7 +95,7 @@ File type | Library | Content | Thumbnail | Metadata
pdf,xps,cbz,fb2,epub | MuPDF | text+ocr | yes, `png` | title | pdf,xps,cbz,fb2,epub | MuPDF | text+ocr | yes, `png` | title |
`audio/*` | ffmpeg | - | yes, `jpeg` | ID3 tags | `audio/*` | ffmpeg | - | yes, `jpeg` | ID3 tags |
`video/*` | ffmpeg | - | yes, `jpeg` | title, comment, artist | `video/*` | ffmpeg | - | yes, `jpeg` | title, comment, artist |
`image/*` | ffmpeg | - | yes, `jpeg` | `EXIF:Artist`, `EXIF:ImageDescription` | `image/*` | ffmpeg | - | yes, `jpeg` | [Common EXIF tags](https://github.com/simon987/sist2/blob/efdde2734eca9b14a54f84568863b7ffd59bdba3/src/parsing/media.c#L190) |
ttf,ttc,cff,woff,fnt,otf | Freetype2 | - | yes, `bmp` | Name & style | ttf,ttc,cff,woff,fnt,otf | Freetype2 | - | yes, `bmp` | Name & style |
`text/plain` | *(none)* | yes | no | - | `text/plain` | *(none)* | yes | no | - |
tar, zip, rar, 7z, ar ... | Libarchive | yes\* | - | no | tar, zip, rar, 7z, ar ... | Libarchive | yes\* | - | no |

View File

@@ -5,4 +5,3 @@
cmake . cmake .
make make
strip sist2 strip sist2
strip sist2_scan

Binary file not shown.

Binary file not shown.

1
lib/libpng Submodule

Submodule lib/libpng added at 301f7a1429

View File

@@ -320,7 +320,7 @@ video/x-dv, dif|dv
video/x-fli, fli video/x-fli, fli
video/x-isvideo, isu video/x-isvideo, isu
video/x-motion-jpeg, mjpg video/x-motion-jpeg, mjpg
video/x-ms-asf, asf|asx video/x-ms-asf, asf|asx|wmv
video/x-qtc, qtc video/x-qtc, qtc
video/x-sgi-movie, movie|mv video/x-sgi-movie, movie|mv
application/x-7z-compressed, 7z application/x-7z-compressed, 7z
1 application/arj arj
320 video/x-fli fli
321 video/x-isvideo isu
322 video/x-motion-jpeg mjpg
323 video/x-ms-asf asf|asx asf|asx|wmv
324 video/x-qtc qtc
325 video/x-sgi-movie movie|mv
326 application/x-7z-compressed 7z

View File

@@ -1,5 +1,9 @@
{ {
"properties": { "properties": {
"_tie": {
"type": "keyword",
"doc_values": true
},
"path": { "path": {
"type": "text", "type": "text",
"analyzer": "path_analyzer", "analyzer": "path_analyzer",
@@ -7,7 +11,7 @@
}, },
"suggest-path": { "suggest-path": {
"type": "completion", "type": "completion",
"analyzer": "keyword" "analyzer": "case_insensitive_kw_analyzer"
}, },
"mime": { "mime": {
"type": "keyword" "type": "keyword"
@@ -105,6 +109,30 @@
}, },
"tag": { "tag": {
"type": "keyword" "type": "keyword"
},
"exif_make": {
"type": "text"
},
"exif_model": {
"type": "text"
},
"exif:software": {
"type": "text"
},
"exif_exposure_time": {
"type": "keyword"
},
"exif_fnumber": {
"type": "keyword"
},
"exif_iso_speed_ratings": {
"type": "keyword"
},
"exif_focal_length": {
"type": "keyword"
},
"exif_user_comment": {
"type": "text"
} }
} }
} }

10
schema/pipeline.json Normal file
View File

@@ -0,0 +1,10 @@
{
"description": "Copy _id to _tie",
"processors": [
{
"script": {
"source": "ctx._tie = ctx._id;"
}
}
]
}

View File

@@ -21,6 +21,12 @@
"lowercase" "lowercase"
] ]
}, },
"case_insensitive_kw_analyzer": {
"tokenizer": "keyword",
"filter": [
"lowercase"
]
},
"my_nGram": { "my_nGram": {
"tokenizer": "my_nGram_tokenizer", "tokenizer": "my_nGram_tokenizer",
"filter": [ "filter": [

View File

@@ -13,8 +13,7 @@ mv mupdf/build/release/libmupdf-third.a .
# openjp2 # openjp2
cd openjpeg cd openjpeg
#cmake . -DCMAKE_BUILD_TYPE=Release -DCMAKE_C_FLAGS="-O3 -march=native -DNDEBUG" cmake . -DCMAKE_BUILD_TYPE=Release -DCMAKE_C_FLAGS="-O3 -DNDEBUG -fPIC"
cmake . -DCMAKE_BUILD_TYPE=Release -DCMAKE_C_FLAGS="-O3"
make -j $THREADS make -j $THREADS
cd .. cd ..
mv openjpeg/bin/libopenjp2.a . mv openjpeg/bin/libopenjp2.a .
@@ -22,7 +21,7 @@ mv openjpeg/bin/libopenjp2.a .
# harfbuzz # harfbuzz
cd harfbuzz cd harfbuzz
./autogen.sh ./autogen.sh
./configure --disable-shared --enable-static CFLAGS=-fPIC ./configure --disable-shared --enable-static
make -j $THREADS make -j $THREADS
cd .. cd ..
mv harfbuzz/src/.libs/libharfbuzz.a . mv harfbuzz/src/.libs/libharfbuzz.a .
@@ -75,7 +74,8 @@ mv libmagic/src/.libs/libmagic.a .
cd tesseract cd tesseract
mkdir build mkdir build
cd build cd build
cmake -DSTATIC=on -DBUILD_TRAINING_TOOLS=off .. cmake -DSTATIC=on -DBUILD_TRAINING_TOOLS=off -DBUILD_TESTS=off -DCMAKE_BUILD_TYPE=Release \
-DCMAKE_CXX_FLAGS="-fPIC" -DAUTO_OPTIMIZE=off ..
make -j $THREADS make -j $THREADS
cd ../.. cd ../..
mv tesseract/build/libtesseract.a . mv tesseract/build/libtesseract.a .
@@ -83,7 +83,7 @@ mv tesseract/build/libtesseract.a .
# leptonica # leptonica
cd leptonica cd leptonica
./autogen.sh ./autogen.sh
./configure --without-zlib --without-jpeg --without-giflib \ CFLAGS="-fPIC" ./configure --without-zlib --without-jpeg --without-giflib \
--without-giflib --without-libwebp --without-libwebpmux --without-libopenjpeg \ --without-giflib --without-libwebp --without-libwebpmux --without-libopenjpeg \
--enable-static --disable-shared --enable-static --disable-shared
make -j $THREADS make -j $THREADS
@@ -93,8 +93,36 @@ mv leptonica/src/.libs/liblept.a .
# tiff # tiff
cd libtiff cd libtiff
./autogen.sh ./autogen.sh
./configure --enable-static --disable-shared --disable-lzw --disable-jpeg --disable-webp \ CFLAGS="-fPIC" CXXFLAGS="-fPIC" CXX_FLAGS="-fPIC" ./configure --enable-static --disable-shared --disable-lzw --disable-jpeg --disable-webp \
--disable-lzma --disable-zstd --disable-jbig --disable-lzma --disable-zstd --disable-jbig
make -j $THREADS make -j $THREADS
cd .. cd ..
mv libtiff/libtiff/.libs/libtiff.a . mv libtiff/libtiff/.libs/libtiff.a .
# png
cd libpng
CFLAGS="-fPIC" ./configure --enable-static --disable-shared
make -j $THREADS
cd ..
mv libpng/.libs/libpng16.a .
# openssl...
git clone --depth 1 -b OpenSSL_1_1_0-stable https://github.com/openssl/openssl
cd openssl
./config --prefix=$(pwd)/../ssl
make depend
make -j $THREADS
make install
cd ..
mv ./openssl/libcrypto.a ./openssl/libssl.a .
# curl
wget -nc https://curl.haxx.se/download/curl-7.68.0.tar.gz
tar -xzf curl-7.68.0.tar.gz
cd curl-7.68.0
./configure --disable-ldap --disable-ldaps --without-librtmp --disable-rtsp --disable-crypto-auth \
--disable-smtp --without-libidn2 --without-nghttp2 --without-brotli --enable-static --disable-shared \
--without-libpsl --with-ssl=$(pwd)/../ssl
make -j $THREADS
cd ..
mv curl-7.68.0/lib/.libs/libcurl.a .

View File

@@ -1,6 +1,9 @@
import json
files = [ files = [
"schema/mappings.json", "schema/mappings.json",
"schema/settings.json", "schema/settings.json",
"schema/pipeline.json",
] ]
@@ -9,6 +12,6 @@ def clean(filepath):
for file in files: for file in files:
with open(file, "rb") as f: with open(file, "r") as f:
data = f.read() data = json.dumps(json.load(f), separators=(",", ":")).encode()
print("char %s[%d] = {%s};" % (clean(file), len(data), ",".join(str(int(b)) for b in data))) print("char %s[%d] = {%s};" % (clean(file), len(data), ",".join(str(int(b)) for b in data)))

View File

@@ -15,6 +15,13 @@
#define DEFAULT_BIND_ADDR "localhost" #define DEFAULT_BIND_ADDR "localhost"
#define DEFAULT_PORT "4090" #define DEFAULT_PORT "4090"
const char* TESS_DATAPATHS[] = {
"/usr/share/tessdata/",
"/usr/share/tesseract-ocr/tessdata/",
"./",
NULL
};
scan_args_t *scan_args_create() { scan_args_t *scan_args_create() {
scan_args_t *args = calloc(sizeof(scan_args_t), 1); scan_args_t *args = calloc(sizeof(scan_args_t), 1);
@@ -136,13 +143,23 @@ int scan_args_validate(scan_args_t *args, int argc, const char **argv) {
if (args->tesseract_lang != NULL) { if (args->tesseract_lang != NULL) {
TessBaseAPI *api = TessBaseAPICreate(); TessBaseAPI *api = TessBaseAPICreate();
ret = TessBaseAPIInit3(api, TESS_DATAPATH, args->tesseract_lang);
char filename[128];
sprintf(filename, "%s.traineddata", args->tesseract_lang);
const char * path = find_file_in_paths(TESS_DATAPATHS, filename);
if (path == NULL) {
LOG_FATAL("cli.c", "Could not find tesseract language file!");
}
ret = TessBaseAPIInit3(api, path, args->tesseract_lang);
if (ret != 0) { if (ret != 0) {
fprintf(stderr, "Could not initialize tesseract with lang '%s'\n", args->tesseract_lang); fprintf(stderr, "Could not initialize tesseract with lang '%s'\n", args->tesseract_lang);
return 1; return 1;
} }
TessBaseAPIEnd(api); TessBaseAPIEnd(api);
TessBaseAPIDelete(api); TessBaseAPIDelete(api);
args->tesseract_path = path;
} }
LOG_DEBUGF("cli.c", "arg quality=%f", args->quality) LOG_DEBUGF("cli.c", "arg quality=%f", args->quality)
@@ -156,7 +173,8 @@ int scan_args_validate(scan_args_t *args, int argc, const char **argv) {
LOG_DEBUGF("cli.c", "arg depth=%d", args->depth) LOG_DEBUGF("cli.c", "arg depth=%d", args->depth)
LOG_DEBUGF("cli.c", "arg path=%s", args->path) LOG_DEBUGF("cli.c", "arg path=%s", args->path)
LOG_DEBUGF("cli.c", "arg archive=%s", args->archive) LOG_DEBUGF("cli.c", "arg archive=%s", args->archive)
LOG_DEBUGF("cli.c", "arg ocr=%s", args->tesseract_lang) LOG_DEBUGF("cli.c", "arg tesseract_lang=%s", args->tesseract_lang)
LOG_DEBUGF("cli.c", "arg tesseract_path=%s", args->tesseract_path)
return 0; return 0;
} }

View File

@@ -17,6 +17,7 @@ typedef struct scan_args {
char *archive; char *archive;
archive_mode_t archive_mode; archive_mode_t archive_mode;
char *tesseract_lang; char *tesseract_lang;
const char *tesseract_path;
} scan_args_t; } scan_args_t;
scan_args_t *scan_args_create(); scan_args_t *scan_args_create();

View File

@@ -28,6 +28,7 @@ struct {
pthread_mutex_t mupdf_mu; pthread_mutex_t mupdf_mu;
char * tesseract_lang; char * tesseract_lang;
const char * tesseract_path;
} ScanCtx; } ScanCtx;
struct { struct {

View File

@@ -129,7 +129,7 @@ void elastic_flush() {
Indexer->queued = 0; Indexer->queued = 0;
char bulk_url[4096]; char bulk_url[4096];
snprintf(bulk_url, 4096, "%s/sist2/_bulk", Indexer->es_url); snprintf(bulk_url, 4096, "%s/sist2/_bulk?pipeline=tie", Indexer->es_url);
response_t *r = web_post(bulk_url, buf, "Content-Type: application/x-ndjson"); response_t *r = web_post(bulk_url, buf, "Content-Type: application/x-ndjson");
if (r->status_code == 0) { if (r->status_code == 0) {
@@ -138,19 +138,21 @@ void elastic_flush() {
LOG_INFOF("elastic.c", "Indexed %d documents (%zukB) <%d>", count, buf_cur / 1024, r->status_code); LOG_INFOF("elastic.c", "Indexed %d documents (%zukB) <%d>", count, buf_cur / 1024, r->status_code);
cJSON *ret_json = cJSON_Parse(r->body); if (r->status_code != 200 && r->status_code != 413) {
if (cJSON_GetObjectItem(ret_json, "errors")->valueint != 0) { cJSON *ret_json = cJSON_Parse(r->body);
cJSON *err; if (cJSON_GetObjectItem(ret_json, "errors")->valueint != 0) {
cJSON_ArrayForEach(err, cJSON_GetObjectItem(ret_json, "items")) { cJSON *err;
if (cJSON_GetObjectItem(cJSON_GetObjectItem(err, "index"), "status")->valueint != 201) { cJSON_ArrayForEach(err, cJSON_GetObjectItem(ret_json, "items")) {
char* str = cJSON_Print(err); if (cJSON_GetObjectItem(cJSON_GetObjectItem(err, "index"), "status")->valueint != 201) {
LOG_ERRORF("elastic.c", "%s\n", str); char* str = cJSON_Print(err);
cJSON_free(str); LOG_ERRORF("elastic.c", "%s\n", str);
cJSON_free(str);
}
} }
} }
}
cJSON_Delete(ret_json); cJSON_Delete(ret_json);
}
free_response(r); free_response(r);
free(buf); free(buf);
@@ -245,6 +247,11 @@ void elastic_init(int force_reset) {
LOG_INFOF("elastic.c", "Close index <%d>", r->status_code); LOG_INFOF("elastic.c", "Close index <%d>", r->status_code);
free_response(r); free_response(r);
snprintf(url, 4096, "%s/_ingest/pipeline/tie", IndexCtx.es_url);
r = web_put(url, pipeline_json, "Content-Type: application/json");
LOG_INFOF("elastic.c", "Create pipeline <%d>", r->status_code);
free_response(r);
snprintf(url, 4096, "%s/sist2/_settings", IndexCtx.es_url); snprintf(url, 4096, "%s/sist2/_settings", IndexCtx.es_url);
r = web_put(url, settings_json, "Content-Type: application/json"); r = web_put(url, settings_json, "Content-Type: application/json");
LOG_INFOF("elastic.c", "Update settings <%d>", r->status_code); LOG_INFOF("elastic.c", "Update settings <%d>", r->status_code);
@@ -274,3 +281,28 @@ cJSON *elastic_get_document(const char *uuid_str) {
free_response(r); free_response(r);
return json; return json;
} }
char *elastic_get_status() {
char url[4096];
snprintf(url, 4096,
"%s/_cluster/state/metadata/sist2?filter_path=metadata.indices.*.state", WebCtx.es_url);
response_t *r = web_get(url);
cJSON *json = NULL;
char *status = malloc(128 * sizeof(char));
status[0] = '\0';
if (r->status_code == 200) {
json = cJSON_Parse(r->body);
const cJSON *metadata = cJSON_GetObjectItem(json, "metadata");
if (metadata != NULL) {
const cJSON *indices = cJSON_GetObjectItem(metadata, "indices");
const cJSON *sist2 = cJSON_GetObjectItem(indices, "sist2");
const cJSON *state = cJSON_GetObjectItem(sist2, "state");
strcpy(status, state->valuestring);
}
}
free_response(r);
cJSON_Delete(json);
return status;
}

View File

@@ -30,4 +30,6 @@ void elastic_init(int force_reset);
cJSON *elastic_get_document(const char *uuid_str); cJSON *elastic_get_document(const char *uuid_str);
char *elastic_get_status();
#endif #endif

File diff suppressed because one or more lines are too long

View File

@@ -118,6 +118,24 @@ char *get_meta_key_text(enum metakey meta_key) {
return "font_name"; return "font_name";
case MetaParent: case MetaParent:
return "parent"; return "parent";
case MetaExifMake:
return "exif_make";
case MetaExifSoftware:
return "exif_software";
case MetaExifExposureTime:
return "exif_exposure_time";
case MetaExifFNumber:
return "exif_fnumber";
case MetaExifFocalLength:
return "exif_focal_length";
case MetaExifUserComment:
return "exif_user_comment";
case MetaExifIsoSpeedRatings:
return "exif_iso_speed_ratings";
case MetaExifModel:
return "exif_model";
case MetaExifDateTime:
return "exif_datetime";
default: default:
return NULL; return NULL;
} }
@@ -196,7 +214,12 @@ void read_index_bin(const char *path, const char *index_id, index_func func) {
char uuid_str[UUID_STR_LEN]; char uuid_str[UUID_STR_LEN];
uuid_unparse(line.uuid, uuid_str); uuid_unparse(line.uuid, uuid_str);
cJSON_AddStringToObject(document, "mime", mime_get_mime_text(line.mime)); const char* mime_text = mime_get_mime_text(line.mime);
if (mime_text == NULL) {
cJSON_AddNullToObject(document, "mime");
} else {
cJSON_AddStringToObject(document, "mime", mime_get_mime_text(line.mime));
}
cJSON_AddNumberToObject(document, "size", (double) line.size); cJSON_AddNumberToObject(document, "size", (double) line.size);
cJSON_AddNumberToObject(document, "mtime", line.mtime); cJSON_AddNumberToObject(document, "mtime", line.mtime);
@@ -255,6 +278,15 @@ void read_index_bin(const char *path, const char *index_id, index_func func) {
case MetaGenre: case MetaGenre:
case MetaFontName: case MetaFontName:
case MetaParent: case MetaParent:
case MetaExifMake:
case MetaExifSoftware:
case MetaExifExposureTime:
case MetaExifFNumber:
case MetaExifFocalLength:
case MetaExifUserComment:
case MetaExifIsoSpeedRatings:
case MetaExifDateTime:
case MetaExifModel:
case MetaTitle: { case MetaTitle: {
buf.cur = 0; buf.cur = 0;
while ((c = getc(file)) != 0) { while ((c = getc(file)) != 0) {

View File

@@ -6,7 +6,7 @@
#define EPILOG "Made by simon987 <me@simon987.net>. Released under GPL-3.0" #define EPILOG "Made by simon987 <me@simon987.net>. Released under GPL-3.0"
static const char *const Version = "1.2.2"; static const char *const Version = "1.2.12";
static const char *const usage[] = { static const char *const usage[] = {
"sist2 scan [OPTION]... PATH", "sist2 scan [OPTION]... PATH",
"sist2 index [OPTION]... INDEX", "sist2 index [OPTION]... INDEX",
@@ -49,8 +49,10 @@ void sist2_scan(scan_args_t *args) {
strncpy(ScanCtx.index.path, args->output, sizeof(ScanCtx.index.path)); strncpy(ScanCtx.index.path, args->output, sizeof(ScanCtx.index.path));
strncpy(ScanCtx.index.desc.name, args->name, sizeof(ScanCtx.index.desc.name)); strncpy(ScanCtx.index.desc.name, args->name, sizeof(ScanCtx.index.desc.name));
strncpy(ScanCtx.index.desc.root, args->path, sizeof(ScanCtx.index.desc.root)); strncpy(ScanCtx.index.desc.root, args->path, sizeof(ScanCtx.index.desc.root));
strncpy(ScanCtx.index.desc.rewrite_url, args->rewrite_url, sizeof(ScanCtx.index.desc.rewrite_url));
ScanCtx.index.desc.root_len = (short) strlen(ScanCtx.index.desc.root); ScanCtx.index.desc.root_len = (short) strlen(ScanCtx.index.desc.root);
ScanCtx.tesseract_lang = args->tesseract_lang; ScanCtx.tesseract_lang = args->tesseract_lang;
ScanCtx.tesseract_path = args->tesseract_path;
init_dir(ScanCtx.index.path); init_dir(ScanCtx.index.path);

View File

@@ -1,10 +1,20 @@
#include "doc.h" #include "doc.h"
#include "src/ctx.h" #include "src/ctx.h"
void dump_text(mceTextReader_t *reader, dyn_buffer_t *buf) { int dump_text(mceTextReader_t *reader, dyn_buffer_t *buf) {
mce_skip_attributes(reader); mce_skip_attributes(reader);
xmlErrorPtr err = xmlGetLastError();
if (err != NULL) {
if (err->level == XML_ERR_FATAL) {
LOG_ERRORF("doc.c", "Got fatal XML error while parsing document: %s", err->message)
return -1;
} else {
LOG_ERRORF("doc.c", "Got recoverable XML error while parsing document: %s", err->message)
}
}
mce_start_children(reader) { mce_start_children(reader) {
mce_start_element(reader, NULL, _X("t")) { mce_start_element(reader, NULL, _X("t")) {
mce_skip_attributes(reader); mce_skip_attributes(reader);
@@ -18,10 +28,14 @@ void dump_text(mceTextReader_t *reader, dyn_buffer_t *buf) {
} mce_end_element(reader); } mce_end_element(reader);
mce_start_element(reader, NULL, NULL) { mce_start_element(reader, NULL, NULL) {
dump_text(reader, buf); int ret = dump_text(reader, buf);
if (ret != 0) {
return ret;
}
} mce_end_element(reader); } mce_end_element(reader);
} mce_end_children(reader) } mce_end_children(reader)
return 0;
} }
__always_inline __always_inline
@@ -52,23 +66,28 @@ int should_read_part(opcPart part) {
} }
__always_inline __always_inline
void read_part(opcContainer *c, dyn_buffer_t *buf, opcPart part, document_t *doc) { int read_part(opcContainer *c, dyn_buffer_t *buf, opcPart part, document_t *doc) {
mceTextReader_t reader; mceTextReader_t reader;
int ret = opcXmlReaderOpen(c, &reader, part, NULL, "UTF-8", 0); int ret = opcXmlReaderOpen(c, &reader, part, NULL, "UTF-8", XML_PARSE_NOWARNING | XML_PARSE_NOERROR | XML_PARSE_NONET);
if (ret != OPC_ERROR_NONE) { if (ret != OPC_ERROR_NONE) {
LOG_ERRORF(doc->filepath, "(doc.c) opcXmlReaderOpen() returned error code %d", ret); LOG_ERRORF(doc->filepath, "(doc.c) opcXmlReaderOpen() returned error code %d", ret);
return; return -1;
} }
mce_start_document(&reader) { mce_start_document(&reader) {
mce_start_element(&reader, NULL, NULL) { mce_start_element(&reader, NULL, NULL) {
dump_text(&reader, buf); ret = dump_text(&reader, buf);
if (ret != 0) {
mceTextReaderCleanup(&reader);
return -1;
}
} mce_end_element(&reader); } mce_end_element(&reader);
} mce_end_document(&reader); } mce_end_document(&reader);
mceTextReaderCleanup(&reader); mceTextReaderCleanup(&reader);
return 0;
} }
void parse_doc(void *mem, size_t mem_len, document_t *doc) { void parse_doc(void *mem, size_t mem_len, document_t *doc) {
@@ -88,7 +107,10 @@ void parse_doc(void *mem, size_t mem_len, document_t *doc) {
opcPart part = opcPartGetFirst(c); opcPart part = opcPartGetFirst(c);
do { do {
if (should_read_part(part)) { if (should_read_part(part)) {
read_part(c, &buf, part, doc); int ret = read_part(c, &buf, part, doc);
if (ret != 0) {
break;
}
} }
} while ((part = opcPartGetNext(c, part))); } while ((part = opcPartGetNext(c, part)));

View File

@@ -193,6 +193,24 @@ append_video_meta(AVFormatContext *pFormatCtx, AVFrame *frame, document_t *doc,
APPEND_TAG_META(doc, tag, MetaArtist) APPEND_TAG_META(doc, tag, MetaArtist)
} else if (strcmp(tag->key, "ImageDescription") == 0) { } else if (strcmp(tag->key, "ImageDescription") == 0) {
APPEND_TAG_META(doc, tag, MetaContent) APPEND_TAG_META(doc, tag, MetaContent)
} else if (strcmp(tag->key, "Make") == 0) {
APPEND_TAG_META(doc, tag, MetaExifMake)
} else if (strcmp(tag->key, "Model") == 0) {
APPEND_TAG_META(doc, tag, MetaExifModel)
} else if (strcmp(tag->key, "Software") == 0) {
APPEND_TAG_META(doc, tag, MetaExifSoftware)
} else if (strcmp(tag->key, "FNumber") == 0) {
APPEND_TAG_META(doc, tag, MetaExifFNumber)
} else if (strcmp(tag->key, "FocalLength") == 0) {
APPEND_TAG_META(doc, tag, MetaExifFocalLength)
} else if (strcmp(tag->key, "UserComment") == 0) {
APPEND_TAG_META(doc, tag, MetaExifUserComment)
} else if (strcmp(tag->key, "ISOSpeedRatings") == 0) {
APPEND_TAG_META(doc, tag, MetaExifIsoSpeedRatings)
} else if (strcmp(tag->key, "ExposureTime") == 0) {
APPEND_TAG_META(doc, tag, MetaExifExposureTime)
} else if (strcmp(tag->key, "DateTime") == 0) {
APPEND_TAG_META(doc, tag, MetaExifDateTime)
} }
} }
} }

View File

@@ -1293,6 +1293,7 @@ g_hash_table_insert(ext_table, "isu", (gpointer)video_x_isvideo);
g_hash_table_insert(ext_table, "mjpg", (gpointer)video_x_motion_jpeg); g_hash_table_insert(ext_table, "mjpg", (gpointer)video_x_motion_jpeg);
g_hash_table_insert(ext_table, "asf", (gpointer)video_x_ms_asf); g_hash_table_insert(ext_table, "asf", (gpointer)video_x_ms_asf);
g_hash_table_insert(ext_table, "asx", (gpointer)video_x_ms_asf); g_hash_table_insert(ext_table, "asx", (gpointer)video_x_ms_asf);
g_hash_table_insert(ext_table, "wmv", (gpointer)video_x_ms_asf);
g_hash_table_insert(ext_table, "qtc", (gpointer)video_x_qtc); g_hash_table_insert(ext_table, "qtc", (gpointer)video_x_qtc);
g_hash_table_insert(ext_table, "movie", (gpointer)video_x_sgi_movie); g_hash_table_insert(ext_table, "movie", (gpointer)video_x_sgi_movie);
g_hash_table_insert(ext_table, "mv", (gpointer)video_x_sgi_movie); g_hash_table_insert(ext_table, "mv", (gpointer)video_x_sgi_movie);

View File

@@ -1,7 +1,8 @@
#include "pdf.h" #include "pdf.h"
#include "src/ctx.h" #include "src/ctx.h"
#define MIN_OCR_SIZE 128 #define MIN_OCR_SIZE 350
#define MIN_OCR_LEN 10
__thread text_buffer_t thread_buffer; __thread text_buffer_t thread_buffer;
@@ -128,6 +129,7 @@ int read_stext_block(fz_stext_block *block, text_buffer_t *tex) {
return 0; return 0;
} }
#define IS_VALID_BPP(d) (d==1 || d==2 || d==4 || d==8 || d==16 || d==24 || d==32)
void fill_image(fz_context *ctx, UNUSED(fz_device *dev), void fill_image(fz_context *ctx, UNUSED(fz_device *dev),
fz_image *img, UNUSED(fz_matrix ctm), UNUSED(float alpha), fz_image *img, UNUSED(fz_matrix ctm), UNUSED(float alpha),
@@ -135,30 +137,32 @@ void fill_image(fz_context *ctx, UNUSED(fz_device *dev),
int l2factor = 0; int l2factor = 0;
if (img->w > MIN_OCR_SIZE && img->h > MIN_OCR_SIZE) { if (img->w > MIN_OCR_SIZE && img->h > MIN_OCR_SIZE && IS_VALID_BPP(img->n)) {
fz_pixmap *pix = img->get_pixmap(ctx, img, NULL, img->w, img->h, &l2factor); fz_pixmap *pix = img->get_pixmap(ctx, img, NULL, img->w, img->h, &l2factor);
if (pix->h > MIN_OCR_SIZE && img->h > MIN_OCR_SIZE && img->xres != 0) { if (pix->h > MIN_OCR_SIZE && img->h > MIN_OCR_SIZE && img->xres != 0) {
TessBaseAPI *api = TessBaseAPICreate(); TessBaseAPI *api = TessBaseAPICreate();
TessBaseAPIInit3(api, TESS_DATAPATH, ScanCtx.tesseract_lang); TessBaseAPIInit3(api, ScanCtx.tesseract_path, ScanCtx.tesseract_lang);
TessBaseAPISetImage(api, pix->samples, pix->w, pix->h, pix->n, pix->stride); TessBaseAPISetImage(api, pix->samples, pix->w, pix->h, pix->n, pix->stride);
TessBaseAPISetSourceResolution(api, pix->xres); TessBaseAPISetSourceResolution(api, pix->xres);
char *text = TessBaseAPIGetUTF8Text(api); char *text = TessBaseAPIGetUTF8Text(api);
size_t len = strlen(text); size_t len = strlen(text);
text_buffer_append_string(&thread_buffer, text, len - 1); if (len >= MIN_OCR_LEN) {
LOG_DEBUGF( text_buffer_append_string(&thread_buffer, text, len - 1);
"pdf.c", LOG_DEBUGF(
"(OCR) %dx%d got %dB from tesseract (%s), buffer:%dB", "pdf.c",
pix->w, pix->h, len, ScanCtx.tesseract_lang, thread_buffer.dyn_buffer.cur "(OCR) %dx%d got %dB from tesseract (%s), buffer:%dB",
) pix->w, pix->h, len, ScanCtx.tesseract_lang, thread_buffer.dyn_buffer.cur
)
}
TessBaseAPIEnd(api); TessBaseAPIEnd(api);
TessBaseAPIDelete(api); TessBaseAPIDelete(api);
fz_drop_pixmap(ctx, pix);
} }
fz_drop_pixmap(ctx, pix);
} }
} }

View File

@@ -3,7 +3,6 @@
#define UUID_STR_LEN 37 #define UUID_STR_LEN 37
#define UNUSED(x) __attribute__((__unused__)) x #define UNUSED(x) __attribute__((__unused__)) x
#define TESS_DATAPATH "/usr/share/tessdata/"
#include <glib-2.0/glib.h> #include <glib-2.0/glib.h>
#include <unistd.h> #include <unistd.h>

View File

@@ -2,9 +2,9 @@
#define SIST2_TYPES_H #define SIST2_TYPES_H
#define META_INT_MASK 0xF0 #define META_INT_MASK 0x80
#define META_STR_MASK 0xE0 #define META_STR_MASK 0x40
#define META_LONG_MASK 0xD0 #define META_LONG_MASK 0x20
#define IS_META_INT(key) (key & META_INT_MASK) == META_INT_MASK #define IS_META_INT(key) (key & META_INT_MASK) == META_INT_MASK
#define IS_META_LONG(key) (key & META_LONG_MASK) == META_LONG_MASK #define IS_META_LONG(key) (key & META_LONG_MASK) == META_LONG_MASK
#define IS_META_STR(meta) (meta->key & META_STR_MASK) == META_STR_MASK #define IS_META_STR(meta) (meta->key & META_STR_MASK) == META_STR_MASK
@@ -31,6 +31,16 @@ enum metakey {
MetaTitle = 12 | META_STR_MASK, MetaTitle = 12 | META_STR_MASK,
MetaFontName = 13 | META_STR_MASK, MetaFontName = 13 | META_STR_MASK,
MetaParent = 14 | META_STR_MASK, MetaParent = 14 | META_STR_MASK,
MetaExifMake = 15 | META_STR_MASK,
MetaExifSoftware = 16 | META_STR_MASK,
MetaExifExposureTime = 17 | META_STR_MASK,
MetaExifFNumber = 18 | META_STR_MASK,
MetaExifFocalLength = 19 | META_STR_MASK,
MetaExifUserComment = 20 | META_STR_MASK,
MetaExifModel = 21 | META_STR_MASK,
MetaExifIsoSpeedRatings = 22 | META_STR_MASK,
MetaExifDateTime = 23 | META_STR_MASK,
//Note to self: this will break after 31 entries
}; };
#define INDEX_TYPE_BIN "binary" #define INDEX_TYPE_BIN "binary"
@@ -39,7 +49,7 @@ enum metakey {
typedef struct index_descriptor { typedef struct index_descriptor {
char uuid[UUID_STR_LEN]; char uuid[UUID_STR_LEN];
char version[6]; char version[64];
long timestamp; long timestamp;
char root[PATH_MAX]; char root[PATH_MAX];
char rewrite_url[8196]; char rewrite_url[8196];

View File

@@ -1,4 +1,5 @@
#include "util.h" #include "util.h"
#include "src/ctx.h"
dyn_buffer_t dyn_buffer_create() { dyn_buffer_t dyn_buffer_create() {
dyn_buffer_t buf; dyn_buffer_t buf;
@@ -90,7 +91,11 @@ text_buffer_t text_buffer_create(int max_size) {
} }
void text_buffer_terminate_string(text_buffer_t *buf) { void text_buffer_terminate_string(text_buffer_t *buf) {
dyn_buffer_write_char(&buf->dyn_buffer, '\0'); if (*(buf->dyn_buffer.buf + buf->dyn_buffer.cur - 1) == ' ') {
*(buf->dyn_buffer.buf + buf->dyn_buffer.cur - 1) = '\0';
} else {
dyn_buffer_write_char(&buf->dyn_buffer, '\0');
}
} }
__always_inline __always_inline
@@ -171,8 +176,8 @@ int text_buffer_append_string0(text_buffer_t *buf, char *str) {
int text_buffer_append_char(text_buffer_t *buf, int c) { int text_buffer_append_char(text_buffer_t *buf, int c) {
if (SHOULD_IGNORE_CHAR(c)) { if (SHOULD_IGNORE_CHAR(c) || c == ' ') {
if (!buf->last_char_was_whitespace) { if (!buf->last_char_was_whitespace && buf->dyn_buffer.cur != 0) {
dyn_buffer_write_char(&buf->dyn_buffer, ' '); dyn_buffer_write_char(&buf->dyn_buffer, ' ');
buf->last_char_was_whitespace = TRUE; buf->last_char_was_whitespace = TRUE;
@@ -317,4 +322,29 @@ GHashTable *incremental_get_table() {
return file_table; return file_table;
} }
const char *find_file_in_paths(const char *paths[], const char *filename) {
for (int i = 0; paths[i] != NULL; i++) {
char *apath = abspath(paths[i]);
if (apath == NULL) {
continue;
}
char path[PATH_MAX];
snprintf(path, sizeof(path), "%s%s", apath, filename);
LOG_DEBUGF("util.c", "Looking for '%s' in folder '%s'", filename, apath)
free(apath);
struct stat info;
int ret = stat(path, &info);
if (ret != -1) {
return paths[i];
}
}
return NULL;
}

View File

@@ -7,7 +7,7 @@
#define INITIAL_BUF_SIZE 1024 * 16 #define INITIAL_BUF_SIZE 1024 * 16
#define SHOULD_IGNORE_CHAR(c) !(SHOULD_KEEP_CHAR(c)) #define SHOULD_IGNORE_CHAR(c) !(SHOULD_KEEP_CHAR(c))
#define SHOULD_KEEP_CHAR(c) ((c >= '0' && c <= '9') || (c >= 'A' && c <= 'z') || (c > 127)) #define SHOULD_KEEP_CHAR(c) ((c >= '\'' && c <= ';') || (c >= 'A' && c <= 'z') || (c > 127))
typedef struct dyn_buffer { typedef struct dyn_buffer {
@@ -74,5 +74,6 @@ int incremental_get(GHashTable *table, unsigned long inode_no);
int incremental_mark_file_for_copy(GHashTable *table, unsigned long inode_no); int incremental_mark_file_for_copy(GHashTable *table, unsigned long inode_no);
const char *find_file_in_paths(const char **paths, const char *filename);
#endif #endif

View File

@@ -181,7 +181,12 @@ int chunked_response_file(const char *filename, const char *mime,
} }
} }
onion_response_set_length(res, length); onion_response_set_length(res, length);
onion_response_set_header(res, "Content-Type", mime); if (mime != NULL) {
onion_response_set_header(res, "Content-Type", mime);
} else {
onion_response_set_header(res, "Content-Type", "application/octet-stream");
}
onion_response_write_headers(res); onion_response_write_headers(res);
if ((onion_request_get_flags(request) & OR_HEAD) == OR_HEAD) { if ((onion_request_get_flags(request) & OR_HEAD) == OR_HEAD) {
length = 0; length = 0;
@@ -221,14 +226,6 @@ int search(UNUSED(void *p), onion_request *req, onion_response *res) {
return OCS_NOT_PROCESSED; return OCS_NOT_PROCESSED;
} }
char *scroll_param;
const char *scroll = onion_request_get_query(req, "scroll");
if (scroll != NULL) {
scroll_param = "?scroll=3m";
} else {
scroll_param = "";
}
const struct onion_block_t *block = onion_request_get_data(req); const struct onion_block_t *block = onion_request_get_data(req);
if (block == NULL) { if (block == NULL) {
@@ -236,7 +233,7 @@ int search(UNUSED(void *p), onion_request *req, onion_response *res) {
} }
char url[4096]; char url[4096];
snprintf(url, 4096, "%s/sist2/_search%s", WebCtx.es_url, scroll_param); snprintf(url, 4096, "%s/sist2/_search", WebCtx.es_url);
response_t *r = web_post(url, onion_block_data(block), "Content-Type: application/json"); response_t *r = web_post(url, onion_block_data(block), "Content-Type: application/json");
set_default_headers(res); set_default_headers(res);
@@ -254,43 +251,6 @@ int search(UNUSED(void *p), onion_request *req, onion_response *res) {
return OCS_PROCESSED; return OCS_PROCESSED;
} }
int scroll(UNUSED(void *p), onion_request *req, onion_response *res) {
int flags = onion_request_get_flags(req);
if ((flags & OR_METHODS) != OR_GET) {
return OCS_NOT_PROCESSED;
}
char url[4096];
snprintf(url, 4096, "%s/_search/scroll", WebCtx.es_url);
const char *scroll_id = onion_request_get_query(req, "scroll_id");
cJSON *json = cJSON_CreateObject();
cJSON_AddStringToObject(json, "scroll_id", scroll_id);
cJSON_AddStringToObject(json, "scroll", "3m");
char *json_str = cJSON_PrintUnformatted(json);
response_t *r = web_post(url, json_str, "Content-Type: application/json");
cJSON_Delete(json);
cJSON_free(json_str);
if (r->status_code != 200) {
free_response(r);
return OCS_NOT_PROCESSED;
}
set_default_headers(res);
onion_response_set_header(res, "Content-Type", "application/json");
onion_response_set_header(res, "Content-Disposition", "application/json");
onion_response_set_length(res, r->size);
onion_response_write(res, r->body, r->size);
free_response(r);
return OCS_PROCESSED;
}
int serve_file_from_url(cJSON *json, index_t *idx, onion_request *req, onion_response *res) { int serve_file_from_url(cJSON *json, index_t *idx, onion_request *req, onion_response *res) {
const char *path = cJSON_GetObjectItem(json, "path")->valuestring; const char *path = cJSON_GetObjectItem(json, "path")->valuestring;
@@ -431,6 +391,23 @@ int file(UNUSED(void *p), onion_request *req, onion_response *res) {
return ret; return ret;
} }
int status(UNUSED(void *p), UNUSED(onion_request *req), onion_response *res) {
set_default_headers(res);
onion_response_set_header(res, "Content-Type", "application/x-empty");
char *status = elastic_get_status();
if (strcmp(status, "open") == 0) {
onion_response_set_code(res, 204);
} else {
onion_response_set_code(res, 500);
}
free(status);
return OCS_PROCESSED;
}
void serve(const char *hostname, const char *port) { void serve(const char *hostname, const char *port) {
onion *o = onion_new(O_POOL); onion *o = onion_new(O_POOL);
onion_set_timeout(o, 3500); onion_set_timeout(o, 3500);
@@ -449,7 +426,7 @@ void serve(const char *hostname, const char *port) {
onion_url_add(urls, "img/sprite-skin-flat.png", img_sprite_skin_flag); onion_url_add(urls, "img/sprite-skin-flat.png", img_sprite_skin_flag);
onion_url_add(urls, "es", search); onion_url_add(urls, "es", search);
onion_url_add(urls, "scroll", scroll); onion_url_add(urls, "status", status);
onion_url_add( onion_url_add(
urls, urls,
"^t/([a-fA-F0-9]{8}-[a-fA-F0-9]{4}-[a-fA-F0-9]{4}-[a-fA-F0-9]{4}-[a-fA-F0-9]{12})/" "^t/([a-fA-F0-9]{8}-[a-fA-F0-9]{4}-[a-fA-F0-9]{4}-[a-fA-F0-9]{4}-[a-fA-F0-9]{12})/"

File diff suppressed because one or more lines are too long

View File

@@ -423,3 +423,21 @@ option {
margin-top: -14px; margin-top: -14px;
font-size: 11px; font-size: 11px;
} }
@media (min-width: 800px) {
.small-btn {
display: none;
}
.large-btn {
display: inherit;
}
}
@media (max-width: 801px) {
.small-btn {
display: inherit;
}
.large-btn {
display: none;
}
}

View File

@@ -287,3 +287,21 @@ mark {
margin-top: -14px; margin-top: -14px;
font-size: 11px; font-size: 11px;
} }
@media (min-width: 800px) {
.small-btn {
display: none;
}
.large-btn {
display: inherit;
}
}
@media (max-width: 801px) {
.small-btn {
display: inherit;
}
.large-btn {
display: none;
}
}

1
web/css/smartphoto.min.css vendored Normal file

File diff suppressed because one or more lines are too long

View File

@@ -75,6 +75,10 @@ function shouldPlayVideo(hit) {
return videoc !== "hevc" && videoc !== "mpeg2video" && videoc !== "wmv3"; return videoc !== "hevc" && videoc !== "mpeg2video" && videoc !== "wmv3";
} }
function shouldDisplayRawImage(hit) {
return hit["_source"]["mime"] && hit["_source"]["mime"].startsWith("image/") && hit["_source"]["videoc"] !== "tiff";
}
function makePlaceholder(w, h, small) { function makePlaceholder(w, h, small) {
let calc; let calc;
if (small) { if (small) {
@@ -96,10 +100,14 @@ function makePlaceholder(w, h, small) {
return el; return el;
} }
function ext(hit) {
return hit["_source"].hasOwnProperty("extension") && hit["_source"]["extension"] !== "" ? "." + hit["_source"]["extension"] : "";
}
function makeTitle(hit) { function makeTitle(hit) {
let title = document.createElement("div"); let title = document.createElement("div");
title.setAttribute("class", "file-title"); title.setAttribute("class", "file-title");
let extension = hit["_source"].hasOwnProperty("extension") && hit["_source"]["extension"] !== "" ? "." + hit["_source"]["extension"] : ""; let extension = ext(hit);
applyNameToTitle(hit, title, extension); applyNameToTitle(hit, title, extension);
@@ -113,7 +121,7 @@ function getTags(hit, mimeCategory) {
switch (mimeCategory) { switch (mimeCategory) {
case "video": case "video":
case "image": case "image":
if (hit["_source"].hasOwnProperty("videoc")) { if (hit["_source"].hasOwnProperty("videoc") && hit["_source"]["videoc"]) {
const formatTag = document.createElement("span"); const formatTag = document.createElement("span");
formatTag.setAttribute("class", "badge badge-pill badge-video"); formatTag.setAttribute("class", "badge badge-pill badge-video");
formatTag.appendChild(document.createTextNode(hit["_source"]["videoc"].replace(" ", ""))); formatTag.appendChild(document.createTextNode(hit["_source"]["videoc"].replace(" ", "")));
@@ -121,7 +129,7 @@ function getTags(hit, mimeCategory) {
} }
break; break;
case "audio": { case "audio": {
if (hit["_source"].hasOwnProperty("audioc")) { if (hit["_source"].hasOwnProperty("audioc") && hit["_source"]["audioc"]) {
let formatTag = document.createElement("span"); let formatTag = document.createElement("span");
formatTag.setAttribute("class", "badge badge-pill badge-audio"); formatTag.setAttribute("class", "badge badge-pill badge-audio");
formatTag.appendChild(document.createTextNode(hit["_source"]["audioc"])); formatTag.appendChild(document.createTextNode(hit["_source"]["audioc"]));
@@ -156,7 +164,7 @@ function getTags(hit, mimeCategory) {
function infoButtonCb(hit) { function infoButtonCb(hit) {
return () => { return () => {
getDocumentInfo(hit["_id"]).then(doc => { getDocumentInfo(hit["_id"]).then(doc => {
$("#modal-title").text(doc["name"] + (doc["extension"] ? "." + doc["extension"] : "")); $("#modal-title").text(doc["name"] + ext(hit));
const tbody = $("<tbody>"); const tbody = $("<tbody>");
$("#modal-body").empty() $("#modal-body").empty()
@@ -175,7 +183,7 @@ function infoButtonCb(hit) {
"bitrate", "artist", "album", "album_artist", "genre", "title", "font_name", "tag" "bitrate", "artist", "album", "album_artist", "genre", "title", "font_name", "tag"
]); ]);
Object.keys(doc) Object.keys(doc)
.filter(key => key.startsWith("_keyword.") || key.startsWith("_text.") || displayFields.has(key)) .filter(key => key.startsWith("_keyword.") || key.startsWith("_text.") || displayFields.has(key) || key.startsWith("exif_"))
.forEach(key => { .forEach(key => {
tbody.append($("<tr>") tbody.append($("<tr>")
.append($("<td>").text(key)) .append($("<td>").text(key))
@@ -379,6 +387,15 @@ function makeThumbnail(mimeCategory, hit, imgWrapper, small) {
} }
thumbnail.setAttribute("src", `t/${hit["_source"]["index"]}/${hit["_id"]}`); thumbnail.setAttribute("src", `t/${hit["_source"]["index"]}/${hit["_id"]}`);
if (!hit["_source"]["parent"] && shouldDisplayRawImage(hit)) {
imgWrapper.setAttribute("id", "sp" + hit["_id"]);
imgWrapper.setAttribute("data-src", `t/${hit["_source"]["index"]}/${hit["_id"]}`);
imgWrapper.setAttribute("href", `f/${hit["_id"]}`);
imgWrapper.setAttribute("data-caption", hit["_source"]["path"] + "/" + hit["_source"]["name"] + ext(hit));
imgWrapper.setAttribute("data-group", "p" + Math.floor(docCount / SIZE));
imgWrapper.classList.add("sp");
}
const placeholder = makePlaceholder(hit["_source"]["width"], hit["_source"]["height"], small); const placeholder = makePlaceholder(hit["_source"]["width"], hit["_source"]["height"], small);
imgWrapper.appendChild(placeholder); imgWrapper.appendChild(placeholder);
@@ -499,8 +516,7 @@ function makePreloader() {
function makePageIndicator(searchResult) { function makePageIndicator(searchResult) {
let pageIndicator = document.createElement("div"); let pageIndicator = document.createElement("div");
pageIndicator.setAttribute("class", "page-indicator font-weight-light"); pageIndicator.setAttribute("class", "page-indicator font-weight-light");
const totalHits = searchResult["hits"]["total"].hasOwnProperty("value") const totalHits = searchResult["aggregations"]["total_count"]["value"];
? searchResult["hits"]["total"]["value"] : searchResult["hits"]["total"];
pageIndicator.appendChild(document.createTextNode(docCount + " / " + totalHits)); pageIndicator.appendChild(document.createTextNode(docCount + " / " + totalHits));
return pageIndicator; return pageIndicator;
} }
@@ -547,8 +563,7 @@ function makeStatsCard(searchResult) {
}); });
let stat = document.createElement("span"); let stat = document.createElement("span");
const totalHits = searchResult["hits"]["total"].hasOwnProperty("value") const totalHits = searchResult["aggregations"]["total_count"]["value"];
? searchResult["hits"]["total"]["value"] : searchResult["hits"]["total"];
stat.appendChild(document.createTextNode(totalHits + " results in " + searchResult["took"] + "ms")); stat.appendChild(document.createTextNode(totalHits + " results in " + searchResult["took"] + "ms"));
statsCardBody.appendChild(stat); statsCardBody.appendChild(stat);

56
web/js/jquery-smartphoto.min.js vendored Normal file

File diff suppressed because one or more lines are too long

View File

@@ -6,7 +6,8 @@ let tagTree;
let searchBar = document.getElementById("searchBar"); let searchBar = document.getElementById("searchBar");
let pathBar = document.getElementById("pathBar"); let pathBar = document.getElementById("pathBar");
let scroll_id = null; let lastDoc = null;
let reachedEnd = false;
let docCount = 0; let docCount = 0;
let coolingDown = false; let coolingDown = false;
let searchBusy = true; let searchBusy = true;
@@ -82,7 +83,10 @@ $.jsonPost("i").then(resp => {
}); });
function getDocumentInfo(id) { function getDocumentInfo(id) {
return $.getJSON("d/" + id) return $.getJSON("d/" + id).fail(e => {
console.log(e);
showEsError();
})
} }
function handleTreeClick(tree) { function handleTreeClick(tree) {
@@ -148,8 +152,8 @@ $.jsonPost("es", {
target: '#mimeTree' target: '#mimeTree'
}); });
mimeTree.on("node.click", handleTreeClick(mimeTree)); mimeTree.on("node.click", handleTreeClick(mimeTree));
mimeTree.select(); mimeTree.deselect();
mimeTree.node("any").deselect(); mimeTree.node("any").select();
}); });
function leafTag(tag) { function leafTag(tag) {
@@ -256,41 +260,18 @@ function insertHits(resultContainer, hits) {
} }
window.addEventListener("scroll", function () { window.addEventListener("scroll", function () {
if (!coolingDown && !searchBusy) { if (!searchBusy) {
let threshold = 400; let threshold = 400;
if ((window.innerHeight + window.scrollY) >= document.body.offsetHeight - threshold) { if ((window.innerHeight + window.scrollY) >= document.body.offsetHeight - threshold) {
coolingDown = true; if (!reachedEnd) {
doScroll(); coolingDown = true;
search(lastDoc);
}
} }
} }
}); });
function doScroll() {
$.get("scroll", {scroll_id: scroll_id})
.then(searchResult => {
let searchResults = document.getElementById("searchResults");
let hits = searchResult["hits"]["hits"];
//Page indicator
let pageIndicator = makePageIndicator(searchResult);
searchResults.appendChild(pageIndicator);
//Result container
let resultContainer = makeResultContainer();
searchResults.appendChild(resultContainer);
insertHits(resultContainer, hits);
if (hits.length === SIZE) {
coolingDown = false;
}
})
.fail(() => {
window.location.reload();
})
}
function getSelectedNodes(tree) { function getSelectedNodes(tree) {
let selectedNodes = []; let selectedNodes = [];
@@ -311,21 +292,25 @@ function getSelectedNodes(tree) {
return selectedNodes return selectedNodes
} }
function search() { function search(after = null) {
lastDoc = null;
if (searchBusy) { if (searchBusy) {
return; return;
} }
searchBusy = true; searchBusy = true;
//Clear old search results
let searchResults = document.getElementById("searchResults"); let searchResults = document.getElementById("searchResults");
while (searchResults.firstChild) { //Clear old search results
searchResults.removeChild(searchResults.firstChild); let preload;
if (!after) {
while (searchResults.firstChild) {
searchResults.removeChild(searchResults.firstChild);
}
preload = makePreloader();
searchResults.appendChild(preload);
} }
const preload = makePreloader();
searchResults.appendChild(preload);
let query = searchBar.value; let query = searchBar.value;
let empty = query === ""; let empty = query === "";
let condition = empty ? "should" : "must"; let condition = empty ? "should" : "must";
@@ -359,25 +344,25 @@ function search() {
filters.push([{terms: {"tag": tags}}]); filters.push([{terms: {"tag": tags}}]);
} }
$.jsonPost("es?scroll=1", { let q = {
"_source": { "_source": {
excludes: ["content"] excludes: ["content", "_tie"]
}, },
query: { query: {
bool: { bool: {
[condition]: { [condition]: {
multi_match: { simple_query_string: {
query: query, query: query,
type: "most_fields",
fields: fields, fields: fields,
operator: "and" default_operator: "and"
} }
}, },
filter: filters filter: filters
} }
}, },
sort: [ "sort": [
"_score" {"_score": {"order": "desc"}},
{"_tie": {"order": "asc"}}
], ],
highlight: { highlight: {
pre_tags: ["<mark>"], pre_tags: ["<mark>"],
@@ -390,24 +375,45 @@ function search() {
font_name: {}, font_name: {},
} }
}, },
aggs: { aggs:
total_size: {"sum": {"field": "size"}} {
}, total_size: {"sum": {"field": "size"}},
total_count: {"value_count": {"field": "size"}}
},
size: SIZE, size: SIZE,
}).then(searchResult => { };
scroll_id = searchResult["_scroll_id"];
preload.remove(); if (after) {
//Search stats q.search_after = [after["_score"], after["_id"]];
searchResults.appendChild(makeStatsCard(searchResult)); }
$.jsonPost("es", q).then(searchResult => {
let hits = searchResult["hits"]["hits"];
if (hits) {
lastDoc = hits[hits.length - 1];
}
if (!after) {
preload.remove();
searchResults.appendChild(makeStatsCard(searchResult));
} else {
let pageIndicator = makePageIndicator(searchResult);
searchResults.appendChild(pageIndicator);
}
//Setup page //Setup page
let resultContainer = makeResultContainer(); let resultContainer = makeResultContainer();
searchResults.appendChild(resultContainer); searchResults.appendChild(resultContainer);
docCount = 0; window.setTimeout(() => {
insertHits(resultContainer, searchResult["hits"]["hits"]); $(".sp").SmartPhoto({animationSpeed: 0, swipeTopToClose: true, showAnimation: false, forceInterval: 50});
}, 100);
if (!after) {
docCount = 0;
}
reachedEnd = hits.length !== SIZE;
insertHits(resultContainer, hits);
searchBusy = false; searchBusy = false;
}); });
} }
@@ -468,6 +474,15 @@ function updateIndices() {
document.getElementById("indices").addEventListener("change", updateIndices); document.getElementById("indices").addEventListener("change", updateIndices);
updateIndices(); updateIndices();
window.onkeyup = function(e) {
if (e.key === "/" || e.key === "Escape") {
const bar = document.getElementById("searchBar");
bar.scrollIntoView();
bar.focus();
}
console.log(e)
};
//Suggest //Suggest
function getPathChoices() { function getPathChoices() {
return new Promise(getPaths => { return new Promise(getPaths => {

View File

@@ -11,7 +11,7 @@
<nav class="navbar navbar-expand-lg"> <nav class="navbar navbar-expand-lg">
<a class="navbar-brand" href="/">sist2</a> <a class="navbar-brand" href="/">sist2</a>
<span class="badge badge-pill version">v1.2.2</span> <span class="badge badge-pill version">v1.2.12</span>
<span class="tagline">Lightning-fast file system indexer and search tool </span> <span class="tagline">Lightning-fast file system indexer and search tool </span>
<a style="margin-left: auto" id="theme" class="btn" title="Toggle theme" href="/">Theme</a> <a style="margin-left: auto" id="theme" class="btn" title="Toggle theme" href="/">Theme</a>
</nav> </nav>
@@ -25,13 +25,20 @@
<div class="input-group"> <div class="input-group">
<div class="input-group-prepend"> <div class="input-group-prepend">
<div class="input-group-text"> <div class="input-group-text">
<span title="Toggle fuzzy searching" onclick="document.getElementById('fuzzyToggle').click()">Fuzzy&nbsp</span> <span title="Toggle fuzzy searching" onclick="document.getElementById('fuzzyToggle').click()">Fuzzy&nbsp</span>
<input title="Toggle fuzzy searching" type="checkbox" id="fuzzyToggle" <input title="Toggle fuzzy searching" type="checkbox" id="fuzzyToggle"
onclick="toggleFuzzy()" checked> onclick="toggleFuzzy()" checked>
</div> </div>
</div> </div>
<input id="searchBar" type="search" class="form-control" placeholder="Search"> <input id="searchBar" type="search" class="form-control" placeholder="Search">
<div class="input-group-append">
<button class="btn btn-outline-secondary small-btn" type="button" data-toggle="modal"
data-target="#help">?
</button>
<button class="btn btn-outline-secondary large-btn" type="button" data-toggle="modal"
data-target="#help">Help
</button>
</div>
</div> </div>
<input title="File size" id="sizeSlider" name="size"> <input title="File size" id="sizeSlider" name="size">
@@ -45,10 +52,12 @@
<div class="col" id="treeTabs"> <div class="col" id="treeTabs">
<ul class="nav nav-tabs" role="tablist"> <ul class="nav nav-tabs" role="tablist">
<li class="nav-item"> <li class="nav-item">
<a class="nav-link active" data-toggle="tab" href="#mime" role="tab" aria-controls="home" aria-selected="true">Mime Types</a> <a class="nav-link active" data-toggle="tab" href="#mime" role="tab" aria-controls="home"
aria-selected="true">Mime Types</a>
</li> </li>
<li class="nav-item"> <li class="nav-item">
<a class="nav-link" data-toggle="tab" href="#tag" role="tab" aria-controls="profile" aria-selected="false" title="User-defined tags">Tags</a> <a class="nav-link" data-toggle="tab" href="#tag" role="tab" aria-controls="profile"
aria-selected="false" title="User-defined tags">Tags</a>
</li> </li>
</ul> </ul>
<div class="tab-content" id="myTabContent"> <div class="tab-content" id="myTabContent">
@@ -79,6 +88,69 @@
</div> </div>
</div> </div>
<div class="modal" id="help" tabindex="-1" role="dialog" aria-labelledby="modal-title" aria-hidden="true">
<div class="modal-dialog modal-lg modal-dialog-centered" role="document">
<div class="modal-content">
<div class="modal-header">
<h5 class="modal-title">Search help</h5>
<button type="button" class="close" data-dismiss="modal" aria-label="Close">
<span aria-hidden="true">&times;</span>
</button>
</div>
<div class="modal-body">
<table class="table">
<tbody>
<tr>
<td><code>+</code></td>
<td>signifies AND operation</td>
</tr>
<tr>
<td><code>|</code></td>
<td>signifies OR operation</td>
</tr>
<tr>
<td><code>-</code></td>
<td>negates a single token</td>
</tr>
<tr>
<td><code>""</code></td>
<td>wraps a number of tokens to signify a phrase for searching</td>
</tr>
<tr>
<td><code>*</code></td>
<td>at the end of a term signifies a prefix query</td>
</tr>
<tr>
<td><code>(</code> and <code>)</code></td>
<td>signify precedence</td>
</tr>
<tr>
<td><code>~N</code></td>
<td>after a word signifies edit distance (fuzziness)</td>
</tr>
<tr>
<td><code>~N</code></td>
<td>after a phrase signifies slop amount</td>
</tr>
</tbody>
</table>
<p>For example: <code>"fried eggs" +(eggplant | potato) -frittata</code> will match the phrase
<i>fried eggs</i> and either <i>eggplant</i> or <i>potato</i>, but will ignore results
containing <i>frittata</i>.</p>
<p>When neither <code>+</code> or <code>|</code> is specified, the default operator is <code>+</code> (and).</p>
<p>When the <b>Fuzzy</b> option is checked, partial matches are also returned.</p>
<br>
<p>For more information, see <a target="_blank"
href="//www.elastic.co/guide/en/elasticsearch/reference/current/query-dsl-simple-query-string-query.html">Elasticsearch
documentation</a></p>
</div>
</div>
</div>
</div>
<div id="searchResults"></div> <div id="searchResults"></div>
</div> </div>