Compare commits
187 Commits
process-po
...
d44679131b
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
d44679131b | ||
|
|
4dd5e70406 | ||
|
|
5a82581992 | ||
|
|
0dc18a56c0 | ||
|
|
258b2e31e6 | ||
|
|
c726074029 | ||
|
|
7873ef003d | ||
|
|
d41266e136 | ||
|
|
0e946092eb | ||
|
|
95b19e2e67 | ||
|
|
bd98eb2522 | ||
|
|
3d99add79e | ||
|
|
2d6553d5d2 | ||
|
|
7d67354b96 | ||
|
|
1b77daef16 | ||
|
|
d7038be35b | ||
|
|
c1573a803e | ||
| 2436e52a62 | |||
|
|
c3a09d0683 | ||
| b9f82593ce | |||
| 59bc418a95 | |||
| fc06b3e378 | |||
| 89e1968994 | |||
| 7009c082e1 | |||
| 64d6bc04a7 | |||
| a2655edf2f | |||
| 86212ece64 | |||
| 61170ce503 | |||
| 7ae410dcc7 | |||
|
|
8714e7e41a | ||
|
|
4a804b7319 | ||
| 4f83a044c7 | |||
| 6e15201a05 | |||
| 6bb12a563a | |||
| 4567f52668 | |||
| 774efe062f | |||
| 7a7a0686c2 | |||
| 7bc2ef9e6c | |||
| f65cca5a02 | |||
| 6423643e24 | |||
| f99ea74e3f | |||
| 1f8f65044c | |||
| 0981a1f421 | |||
| ff066a3962 | |||
|
|
1e778b6f2a | ||
|
|
ff27a540eb | ||
|
|
83259eedee | ||
|
|
aff69fb3eb | ||
|
|
08b6323176 | ||
| 2307fc6e15 | |||
| d679e4c3ca | |||
| f423a17543 | |||
|
|
1bdf4d71dd | ||
|
|
f58e66352c | ||
|
|
a672822811 | ||
|
|
ae317e590d | ||
|
|
410283f14a | ||
|
|
2936240df8 | ||
| af5059f366 | |||
|
|
03983ce00a | ||
|
|
80528857e9 | ||
| ffa7f2ae84 | |||
| 6ade3395d5 | |||
| a2d5e774b3 | |||
|
|
19ea1169ff | ||
|
|
1225fd6bac | ||
| 687b645840 | |||
| d2c8f9209d | |||
|
|
3ea375b37d | ||
|
|
bff89d93e6 | ||
| f423863acb | |||
| 49a21a5a25 | |||
| 560aa82ce7 | |||
| b8c905bd64 | |||
| 8299237ea0 | |||
| 31646a2747 | |||
| d9d77de47f | |||
| 5f0957d029 | |||
| 1cc48f7f33 | |||
| e1e22fd79a | |||
| 786bbc3859 | |||
| 9698ea0c37 | |||
| f345fc1a9a | |||
| 660fbf75d8 | |||
| 33ae585879 | |||
| 5729cbd6b4 | |||
| a19ec3305a | |||
| 8fdb832c85 | |||
| b81ccebdb1 | |||
| b2d214a19a | |||
| 69438464bf | |||
|
|
aa60b526f4 | ||
| 2ea8b51b34 | |||
| 64b4b201d5 | |||
| 857f3315c2 | |||
| 5771693b1a | |||
| 27188b6fa0 | |||
| f56cfb0f2f | |||
| 70242846ae | |||
| b833acf522 | |||
| 5fa2da5eef | |||
| ec518887ee | |||
| 0b0b7fe951 | |||
| ba863e4e6c | |||
| cbab4c2841 | |||
| 930361e78c | |||
| 92478ec47c | |||
| 0d81d7c43b | |||
| 9f175cb0f0 | |||
| 6225cf81de | |||
| d7058ab645 | |||
| 84958502b1 | |||
| a0b6eed037 | |||
| 06d6910151 | |||
| b99e4ddf13 | |||
| d14139ba44 | |||
|
|
13960337aa | ||
| 2596361af5 | |||
| 5a1a04629f | |||
| 242dd67416 | |||
| 54d902146a | |||
| 3b0ab3679a | |||
|
|
58ce0ef414 | ||
| f984baf7fd | |||
| ce242d1053 | |||
| 71deab7fa2 | |||
|
|
b0462f9378 | ||
| ca845d80e8 | |||
| e2025df2c0 | |||
| 7eb064162e | |||
| 7bc4b73e43 | |||
| ca2e308d89 | |||
| c03c148273 | |||
| 5522bcfa9b | |||
| f0fd708082 | |||
| 6bf2b4c74d | |||
| d907576406 | |||
| 7659b481fa | |||
|
|
e81e5ee457 | ||
| d4820d2fad | |||
| b3b3005692 | |||
| 610882112d | |||
| e2e0cf260f | |||
| 3ffa30cc6f | |||
| 7920318406 | |||
| 41ef940623 | |||
| cdec1cebc6 | |||
| 0ce341d8e6 | |||
| 7d96d62983 | |||
| 72293d26f2 | |||
| 944c224904 | |||
| 63027dd5ca | |||
|
|
ac942947e4 | ||
| 1cfceba518 | |||
| 35cfd3b3b1 | |||
| b286c652ad | |||
| 2d8685f8f5 | |||
| c930ef7840 | |||
| d32bda0d68 | |||
| 499ed0be79 | |||
| dc39c0ec4b | |||
| b5cdd9a5df | |||
| a8b6886f7b | |||
| a7e9b6af96 | |||
| 0710dc6d3d | |||
| 75b66b5982 | |||
| 9813646c11 | |||
| ebc9468251 | |||
| 7baaca5078 | |||
| 6c4bdc87cf | |||
| 1ea78887c3 | |||
| 886fa720ec | |||
| d43aac735f | |||
| faf438a798 | |||
| 5b3b9911bd | |||
| 237d55ec9c | |||
|
|
ced4c7de88 | ||
| 90ee318981 | |||
| 785121e46c | |||
| 585c57a2ad | |||
| 42abbbce95 | |||
|
|
e8607df26f | ||
| f1726ca0a9 | |||
| 3ef675abcf | |||
|
|
81658efb19 | ||
| 60c77678b4 | |||
|
|
bf1d2f7d55 |
@@ -38,3 +38,6 @@ build/
|
|||||||
__pycache__/
|
__pycache__/
|
||||||
sist2-vue/dist
|
sist2-vue/dist
|
||||||
sist2-admin/frontend/dist
|
sist2-admin/frontend/dist
|
||||||
|
*.fts
|
||||||
|
.git
|
||||||
|
third-party/libscan/third-party/ext_*/*
|
||||||
88
.drone.yml
@@ -7,11 +7,36 @@ platform:
|
|||||||
arch: amd64
|
arch: amd64
|
||||||
|
|
||||||
steps:
|
steps:
|
||||||
|
- name: submodules
|
||||||
|
image: alpine/git
|
||||||
|
commands:
|
||||||
|
- git submodule update --init --recursive
|
||||||
|
- name: docker
|
||||||
|
image: plugins/docker
|
||||||
|
depends_on:
|
||||||
|
- submodules
|
||||||
|
settings:
|
||||||
|
username:
|
||||||
|
from_secret: DOCKER_USER
|
||||||
|
password:
|
||||||
|
from_secret: DOCKER_PASSWORD
|
||||||
|
repo: sist2app/sist2
|
||||||
|
context: ./
|
||||||
|
dockerfile: ./Dockerfile
|
||||||
|
auto_tag: true
|
||||||
|
auto_tag_suffix: x64-linux
|
||||||
|
when:
|
||||||
|
event:
|
||||||
|
- tag
|
||||||
- name: build
|
- name: build
|
||||||
image: simon987/sist2-build
|
image: sist2app/sist2-build
|
||||||
|
depends_on:
|
||||||
|
- submodules
|
||||||
commands:
|
commands:
|
||||||
- ./scripts/build.sh
|
- ./scripts/build.sh
|
||||||
- name: scp files
|
- name: scp files
|
||||||
|
depends_on:
|
||||||
|
- build
|
||||||
image: appleboy/drone-scp
|
image: appleboy/drone-scp
|
||||||
settings:
|
settings:
|
||||||
host:
|
host:
|
||||||
@@ -22,26 +47,11 @@ steps:
|
|||||||
from_secret: SSH_USER
|
from_secret: SSH_USER
|
||||||
key:
|
key:
|
||||||
from_secret: SSH_KEY
|
from_secret: SSH_KEY
|
||||||
target: /files/sist2/${DRONE_REPO_OWNER}_${DRONE_REPO_NAME}/${DRONE_BRANCH}_${DRONE_BUILD_NUMBER}_${DRONE_COMMIT}/
|
target: ~/files/sist2/${DRONE_REPO_OWNER}_${DRONE_REPO_NAME}/${DRONE_BRANCH}_${DRONE_BUILD_NUMBER}_${DRONE_COMMIT}/
|
||||||
source:
|
source:
|
||||||
- ./VERSION
|
- ./VERSION
|
||||||
- ./sist2-x64-linux
|
- ./sist2-x64-linux
|
||||||
- ./sist2-x64-linux-debug
|
- ./sist2-x64-linux-debug
|
||||||
- name: docker
|
|
||||||
image: plugins/docker
|
|
||||||
settings:
|
|
||||||
username:
|
|
||||||
from_secret: DOCKER_USER
|
|
||||||
password:
|
|
||||||
from_secret: DOCKER_PASSWORD
|
|
||||||
repo: simon987/sist2
|
|
||||||
context: ./
|
|
||||||
dockerfile: ./Dockerfile
|
|
||||||
auto_tag: true
|
|
||||||
auto_tag_suffix: x64-linux
|
|
||||||
when:
|
|
||||||
event:
|
|
||||||
- tag
|
|
||||||
|
|
||||||
---
|
---
|
||||||
kind: pipeline
|
kind: pipeline
|
||||||
@@ -52,11 +62,36 @@ platform:
|
|||||||
arch: arm64
|
arch: arm64
|
||||||
|
|
||||||
steps:
|
steps:
|
||||||
|
- name: submodules
|
||||||
|
image: alpine/git
|
||||||
|
commands:
|
||||||
|
- git submodule update --init --recursive
|
||||||
|
- name: docker
|
||||||
|
image: plugins/docker
|
||||||
|
depends_on:
|
||||||
|
- submodules
|
||||||
|
settings:
|
||||||
|
username:
|
||||||
|
from_secret: DOCKER_USER
|
||||||
|
password:
|
||||||
|
from_secret: DOCKER_PASSWORD
|
||||||
|
repo: sist2app/sist2
|
||||||
|
context: ./
|
||||||
|
dockerfile: ./Dockerfile.arm64
|
||||||
|
auto_tag: true
|
||||||
|
auto_tag_suffix: arm64-linux
|
||||||
|
when:
|
||||||
|
event:
|
||||||
|
- tag
|
||||||
- name: build
|
- name: build
|
||||||
image: simon987/sist2-build-arm64
|
image: sist2app/sist2-build-arm64
|
||||||
|
depends_on:
|
||||||
|
- submodules
|
||||||
commands:
|
commands:
|
||||||
- ./scripts/build_arm64.sh
|
- ./scripts/build_arm64.sh
|
||||||
- name: scp files
|
- name: scp files
|
||||||
|
depends_on:
|
||||||
|
- build
|
||||||
image: appleboy/drone-scp
|
image: appleboy/drone-scp
|
||||||
settings:
|
settings:
|
||||||
host:
|
host:
|
||||||
@@ -67,22 +102,7 @@ steps:
|
|||||||
from_secret: SSH_USER
|
from_secret: SSH_USER
|
||||||
key:
|
key:
|
||||||
from_secret: SSH_KEY
|
from_secret: SSH_KEY
|
||||||
target: /files/sist2/${DRONE_REPO_OWNER}_${DRONE_REPO_NAME}/arm_${DRONE_BRANCH}_${DRONE_BUILD_NUMBER}_${DRONE_COMMIT}/
|
target: ~/files/sist2/${DRONE_REPO_OWNER}_${DRONE_REPO_NAME}/arm_${DRONE_BRANCH}_${DRONE_BUILD_NUMBER}_${DRONE_COMMIT}/
|
||||||
source:
|
source:
|
||||||
- ./sist2-arm64-linux
|
- ./sist2-arm64-linux
|
||||||
- ./sist2-arm64-linux-debug
|
- ./sist2-arm64-linux-debug
|
||||||
- name: docker
|
|
||||||
image: plugins/docker
|
|
||||||
settings:
|
|
||||||
username:
|
|
||||||
from_secret: DOCKER_USER
|
|
||||||
password:
|
|
||||||
from_secret: DOCKER_PASSWORD
|
|
||||||
repo: simon987/sist2
|
|
||||||
context: ./
|
|
||||||
dockerfile: ./Dockerfile.arm64
|
|
||||||
auto_tag: true
|
|
||||||
auto_tag_suffix: arm64-linux
|
|
||||||
when:
|
|
||||||
event:
|
|
||||||
- tag
|
|
||||||
|
|||||||
3
.gitattributes
vendored
@@ -1,3 +0,0 @@
|
|||||||
CMakeModules/* linguist-vendored
|
|
||||||
**/*_generated.c linguist-vendored
|
|
||||||
**/*_generated.h linguist-vendored
|
|
||||||
3
.gitignore
vendored
@@ -3,6 +3,7 @@ thumbs
|
|||||||
*.cbp
|
*.cbp
|
||||||
CMakeCache.txt
|
CMakeCache.txt
|
||||||
CMakeFiles
|
CMakeFiles
|
||||||
|
cmake-build-default-event-trace
|
||||||
cmake-build-debug
|
cmake-build-debug
|
||||||
cmake_install.cmake
|
cmake_install.cmake
|
||||||
Makefile
|
Makefile
|
||||||
@@ -44,3 +45,5 @@ src/index/static_generated.c
|
|||||||
*.sist2
|
*.sist2
|
||||||
*-shm
|
*-shm
|
||||||
*-journal
|
*-journal
|
||||||
|
.vscode
|
||||||
|
*.fts
|
||||||
@@ -5,6 +5,7 @@ set(CMAKE_C_STANDARD 11)
|
|||||||
|
|
||||||
option(SIST_DEBUG "Build a debug executable" on)
|
option(SIST_DEBUG "Build a debug executable" on)
|
||||||
option(SIST_FAST "Enable more optimisation flags" off)
|
option(SIST_FAST "Enable more optimisation flags" off)
|
||||||
|
option(SIST_DEBUG_INFO "Turn on debug information in web interface" on)
|
||||||
|
|
||||||
add_compile_definitions(
|
add_compile_definitions(
|
||||||
"SIST_PLATFORM=${SIST_PLATFORM}"
|
"SIST_PLATFORM=${SIST_PLATFORM}"
|
||||||
@@ -14,13 +15,24 @@ if (SIST_DEBUG)
|
|||||||
add_compile_definitions(
|
add_compile_definitions(
|
||||||
"SIST_DEBUG=${SIST_DEBUG}"
|
"SIST_DEBUG=${SIST_DEBUG}"
|
||||||
)
|
)
|
||||||
endif()
|
set(VCPKG_BUILD_TYPE debug)
|
||||||
|
else ()
|
||||||
|
set(VCPKG_BUILD_TYPE release)
|
||||||
|
endif ()
|
||||||
|
|
||||||
|
if (SIST_DEBUG_INFO)
|
||||||
|
add_compile_definitions(
|
||||||
|
"SIST_DEBUG_INFO=${SIST_DEBUG_INFO}"
|
||||||
|
)
|
||||||
|
endif ()
|
||||||
|
|
||||||
|
|
||||||
add_subdirectory(third-party/libscan)
|
add_subdirectory(third-party/libscan)
|
||||||
set(ARGPARSE_SHARED off)
|
set(ARGPARSE_SHARED off)
|
||||||
add_subdirectory(third-party/argparse)
|
add_subdirectory(third-party/argparse)
|
||||||
|
|
||||||
add_executable(sist2
|
add_executable(
|
||||||
|
sist2
|
||||||
# argparse
|
# argparse
|
||||||
third-party/argparse/argparse.h third-party/argparse/argparse.c
|
third-party/argparse/argparse.h third-party/argparse/argparse.c
|
||||||
|
|
||||||
@@ -41,13 +53,16 @@ add_executable(sist2
|
|||||||
src/types.h
|
src/types.h
|
||||||
src/log.c src/log.h
|
src/log.c src/log.h
|
||||||
src/cli.c src/cli.h
|
src/cli.c src/cli.h
|
||||||
src/parsing/sidecar.c src/parsing/sidecar.h
|
|
||||||
src/database/database.c src/database/database.h
|
src/database/database.c src/database/database.h
|
||||||
src/parsing/fs_util.h
|
src/parsing/fs_util.h
|
||||||
|
|
||||||
src/auth0/auth0_c_api.h src/auth0/auth0_c_api.cpp
|
src/auth0/auth0_c_api.h src/auth0/auth0_c_api.cpp
|
||||||
|
|
||||||
src/database/database_stats.c src/database/database_stats.h src/database/database_schema.c)
|
src/database/database_stats.c
|
||||||
|
src/database/database_schema.c
|
||||||
|
src/database/database_fts.c
|
||||||
|
src/web/web_fts.c
|
||||||
|
src/database/database_embeddings.c)
|
||||||
set_target_properties(sist2 PROPERTIES LINKER_LANGUAGE C)
|
set_target_properties(sist2 PROPERTIES LINKER_LANGUAGE C)
|
||||||
|
|
||||||
target_link_directories(sist2 PRIVATE BEFORE ${_VCPKG_INSTALLED_DIR}/${VCPKG_TARGET_TRIPLET}/lib/)
|
target_link_directories(sist2 PRIVATE BEFORE ${_VCPKG_INSTALLED_DIR}/${VCPKG_TARGET_TRIPLET}/lib/)
|
||||||
@@ -60,6 +75,7 @@ find_package(unofficial-mongoose CONFIG REQUIRED)
|
|||||||
find_package(CURL CONFIG REQUIRED)
|
find_package(CURL CONFIG REQUIRED)
|
||||||
find_library(MAGIC_LIB NAMES libmagic.a REQUIRED)
|
find_library(MAGIC_LIB NAMES libmagic.a REQUIRED)
|
||||||
find_package(unofficial-sqlite3 CONFIG REQUIRED)
|
find_package(unofficial-sqlite3 CONFIG REQUIRED)
|
||||||
|
find_package(OpenBLAS CONFIG REQUIRED)
|
||||||
|
|
||||||
|
|
||||||
target_include_directories(
|
target_include_directories(
|
||||||
@@ -85,7 +101,7 @@ if (SIST_DEBUG)
|
|||||||
-fno-omit-frame-pointer
|
-fno-omit-frame-pointer
|
||||||
-fsanitize=address
|
-fsanitize=address
|
||||||
-fno-inline
|
-fno-inline
|
||||||
# -O2
|
# -O2
|
||||||
)
|
)
|
||||||
target_link_options(
|
target_link_options(
|
||||||
sist2
|
sist2
|
||||||
@@ -115,6 +131,7 @@ else ()
|
|||||||
PRIVATE
|
PRIVATE
|
||||||
|
|
||||||
-Ofast
|
-Ofast
|
||||||
|
# -g
|
||||||
-fno-stack-protector
|
-fno-stack-protector
|
||||||
-fomit-frame-pointer
|
-fomit-frame-pointer
|
||||||
-w
|
-w
|
||||||
@@ -130,6 +147,7 @@ add_dependencies(
|
|||||||
target_link_libraries(
|
target_link_libraries(
|
||||||
sist2
|
sist2
|
||||||
|
|
||||||
|
m
|
||||||
z
|
z
|
||||||
argparse
|
argparse
|
||||||
unofficial::mongoose::mongoose
|
unofficial::mongoose::mongoose
|
||||||
@@ -141,6 +159,7 @@ target_link_libraries(
|
|||||||
|
|
||||||
${MAGIC_LIB}
|
${MAGIC_LIB}
|
||||||
unofficial::sqlite3::sqlite3
|
unofficial::sqlite3::sqlite3
|
||||||
|
OpenBLAS::OpenBLAS
|
||||||
)
|
)
|
||||||
|
|
||||||
add_custom_target(
|
add_custom_target(
|
||||||
|
|||||||
41
Dockerfile
@@ -1,10 +1,4 @@
|
|||||||
FROM simon987/sist2-build as build
|
FROM sist2app/sist2-build as build
|
||||||
MAINTAINER simon987 <me@simon987.net>
|
|
||||||
|
|
||||||
ENV DEBIAN_FRONTEND=noninteractive
|
|
||||||
|
|
||||||
RUN curl -fsSL https://deb.nodesource.com/setup_16.x | bash
|
|
||||||
RUN apt update -y; apt install -y nodejs && rm -rf /var/lib/apt/lists/*
|
|
||||||
|
|
||||||
WORKDIR /build/
|
WORKDIR /build/
|
||||||
|
|
||||||
@@ -19,14 +13,12 @@ COPY sist2-admin sist2-admin
|
|||||||
RUN cd sist2-vue/ && npm install && npm run build
|
RUN cd sist2-vue/ && npm install && npm run build
|
||||||
RUN cd sist2-admin/frontend/ && npm install && npm run build
|
RUN cd sist2-admin/frontend/ && npm install && npm run build
|
||||||
|
|
||||||
RUN mkdir build && cd build && cmake -DSIST_PLATFORM=x64_linux -DSIST_DEBUG=off -DBUILD_TESTS=off -DCMAKE_TOOLCHAIN_FILE=/vcpkg/scripts/buildsystems/vcpkg.cmake ..
|
RUN mkdir build && cd build && cmake -DSIST_PLATFORM=x64_linux_docker -DSIST_DEBUG_INFO=on -DSIST_DEBUG=off -DBUILD_TESTS=off -DCMAKE_TOOLCHAIN_FILE=/vcpkg/scripts/buildsystems/vcpkg.cmake ..
|
||||||
RUN cd build && make -j$(nproc)
|
RUN cd build && make -j$(nproc)
|
||||||
RUN strip build/sist2 || mv build/sist2_debug build/sist2
|
RUN strip build/sist2 || mv build/sist2_debug build/sist2
|
||||||
|
|
||||||
FROM --platform="linux/amd64" ubuntu@sha256:965fbcae990b0467ed5657caceaec165018ef44a4d2d46c7cdea80a9dff0d1ea
|
FROM --platform="linux/amd64" ubuntu@sha256:965fbcae990b0467ed5657caceaec165018ef44a4d2d46c7cdea80a9dff0d1ea
|
||||||
|
|
||||||
WORKDIR /root
|
|
||||||
|
|
||||||
ENV LANG C.UTF-8
|
ENV LANG C.UTF-8
|
||||||
ENV LC_ALL C.UTF-8
|
ENV LC_ALL C.UTF-8
|
||||||
|
|
||||||
@@ -37,21 +29,24 @@ RUN apt update && DEBIAN_FRONTEND=noninteractive apt install -y curl libasan5 li
|
|||||||
|
|
||||||
RUN mkdir -p /usr/share/tessdata && \
|
RUN mkdir -p /usr/share/tessdata && \
|
||||||
cd /usr/share/tessdata/ && \
|
cd /usr/share/tessdata/ && \
|
||||||
curl -o /usr/share/tessdata/hin.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/hin.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/hin.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/hin.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/jpn.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/jpn.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/jpn.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/jpn.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/eng.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/eng.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/eng.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/eng.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/fra.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/fra.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/fra.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/fra.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/rus.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/rus.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/rus.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/rus.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/osd.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/osd.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/osd.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/osd.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/spa.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/spa.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/spa.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/spa.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/deu.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/deu.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/deu.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/deu.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/equ.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/equ.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/equ.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/equ.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/chi_sim.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/chi_sim.traineddata
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/pol.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/pol.traineddata &&\
|
||||||
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/chi_sim.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/chi_sim.traineddata
|
||||||
|
|
||||||
# sist2
|
# sist2
|
||||||
COPY --from=build /build/build/sist2 /root/sist2
|
COPY --from=build /build/build/sist2 /root/sist2
|
||||||
|
|
||||||
# sist2-admin
|
# sist2-admin
|
||||||
COPY sist2-admin/requirements.txt sist2-admin/
|
WORKDIR /root/sist2-admin
|
||||||
RUN python3 -m pip install --no-cache -r sist2-admin/requirements.txt
|
COPY sist2-admin/requirements.txt /root/sist2-admin/
|
||||||
COPY --from=build /build/sist2-admin/ sist2-admin/
|
RUN ln /usr/bin/python3 /usr/bin/python
|
||||||
|
RUN python -m pip install --no-cache -r /root/sist2-admin/requirements.txt
|
||||||
|
COPY --from=build /build/sist2-admin/ /root/sist2-admin/
|
||||||
|
|||||||
@@ -1,9 +1,21 @@
|
|||||||
FROM simon987/sist2-build-arm64 as build
|
FROM sist2app/sist2-build-arm64 as build
|
||||||
MAINTAINER simon987 <me@simon987.net>
|
|
||||||
|
WORKDIR /build/
|
||||||
|
|
||||||
|
COPY scripts scripts
|
||||||
|
COPY schema schema
|
||||||
|
COPY CMakeLists.txt .
|
||||||
|
COPY third-party third-party
|
||||||
|
COPY src src
|
||||||
|
COPY sist2-vue sist2-vue
|
||||||
|
COPY sist2-admin sist2-admin
|
||||||
|
|
||||||
|
RUN cd sist2-vue/ && npm install && npm run build
|
||||||
|
RUN cd sist2-admin/frontend/ && npm install && npm run build
|
||||||
|
|
||||||
WORKDIR /build/
|
WORKDIR /build/
|
||||||
ADD . /build/
|
ADD . /build/
|
||||||
RUN mkdir build && cd build && cmake -DSIST_PLATFORM=arm64_linux -DSIST_DEBUG=off -DBUILD_TESTS=off -DCMAKE_TOOLCHAIN_FILE=/vcpkg/scripts/buildsystems/vcpkg.cmake ..
|
RUN mkdir build && cd build && cmake -DSIST_PLATFORM=arm64_linux_docker -DSIST_DEBUG_INFO=on -DSIST_DEBUG=off -DBUILD_TESTS=off -DCMAKE_TOOLCHAIN_FILE=/vcpkg/scripts/buildsystems/vcpkg.cmake ..
|
||||||
RUN cd build && make -j$(nproc)
|
RUN cd build && make -j$(nproc)
|
||||||
RUN strip build/sist2 || mv build/sist2_debug build/sist2
|
RUN strip build/sist2 || mv build/sist2_debug build/sist2
|
||||||
|
|
||||||
@@ -20,16 +32,17 @@ RUN apt update && apt install -y curl libasan5 libmagic1 tesseract-ocr python3-p
|
|||||||
|
|
||||||
RUN mkdir -p /usr/share/tessdata && \
|
RUN mkdir -p /usr/share/tessdata && \
|
||||||
cd /usr/share/tessdata/ && \
|
cd /usr/share/tessdata/ && \
|
||||||
curl -o /usr/share/tessdata/hin.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/hin.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/hin.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/hin.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/jpn.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/jpn.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/jpn.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/jpn.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/eng.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/eng.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/eng.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/eng.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/fra.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/fra.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/fra.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/fra.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/rus.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/rus.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/rus.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/rus.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/osd.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/osd.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/osd.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/osd.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/spa.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/spa.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/spa.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/spa.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/deu.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/deu.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/deu.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/deu.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/equ.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/equ.traineddata &&\
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/equ.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/equ.traineddata &&\
|
||||||
curl -o /usr/share/tessdata/chi_sim.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/chi_sim.traineddata
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/pol.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/pol.traineddata &&\
|
||||||
|
curl -o /usr/share/tesseract-ocr/4.00/tessdata/chi_sim.traineddata https://raw.githubusercontent.com/tesseract-ocr/tessdata/master/chi_sim.traineddata
|
||||||
|
|
||||||
# sist2
|
# sist2
|
||||||
COPY --from=build /build/build/sist2 /root/sist2
|
COPY --from=build /build/build/sist2 /root/sist2
|
||||||
|
|||||||
147
README.md
@@ -4,19 +4,21 @@
|
|||||||
|
|
||||||
**Demo**: [sist2.simon987.net](https://sist2.simon987.net/)
|
**Demo**: [sist2.simon987.net](https://sist2.simon987.net/)
|
||||||
|
|
||||||
|
**Community URL:** [Discord](https://discord.gg/2PEjDy3Rfs)
|
||||||
|
|
||||||
# sist2
|
# sist2
|
||||||
|
|
||||||
sist2 (Simple incremental search tool)
|
sist2 (Simple incremental search tool)
|
||||||
|
|
||||||
*Warning: sist2 is in early development*
|
*Warning: sist2 is in early development*
|
||||||
|
|
||||||

|

|
||||||
|
|
||||||
## Features
|
## Features
|
||||||
|
|
||||||
* Fast, low memory usage, multi-threaded
|
* Fast, low memory usage, multi-threaded
|
||||||
|
* Manage & schedule scan jobs with simple web interface (Docker only)
|
||||||
* Mobile-friendly Web interface
|
* Mobile-friendly Web interface
|
||||||
* Portable (all its features are packaged in a single executable)
|
|
||||||
* Extracts text and metadata from common file types \*
|
* Extracts text and metadata from common file types \*
|
||||||
* Generates thumbnails \*
|
* Generates thumbnails \*
|
||||||
* Incremental scanning
|
* Incremental scanning
|
||||||
@@ -24,47 +26,74 @@ sist2 (Simple incremental search tool)
|
|||||||
* Recursive scan inside archive files \*\*
|
* Recursive scan inside archive files \*\*
|
||||||
* OCR support with tesseract \*\*\*
|
* OCR support with tesseract \*\*\*
|
||||||
* Stats page & disk utilisation visualization
|
* Stats page & disk utilisation visualization
|
||||||
|
* Named-entity recognition (client-side) \*\*\*\*
|
||||||
|
|
||||||
\* See [format support](#format-support)
|
\* See [format support](#format-support)
|
||||||
\*\* See [Archive files](#archive-files)
|
\*\* See [Archive files](#archive-files)
|
||||||
\*\*\* See [OCR](#ocr)
|
\*\*\* See [OCR](#ocr)
|
||||||
|
\*\*\*\* See [Named-Entity Recognition](#NER)
|
||||||

|
|
||||||
|
|
||||||
## Getting Started
|
## Getting Started
|
||||||
|
|
||||||
1. Have an Elasticsearch (>= 6.8.X, ideally >=7.14.0) instance running
|
### Using Docker Compose *(Windows/Linux/Mac)*
|
||||||
|
|
||||||
|
```yaml
|
||||||
|
version: "3"
|
||||||
|
|
||||||
|
services:
|
||||||
|
elasticsearch:
|
||||||
|
image: elasticsearch:7.17.9
|
||||||
|
restart: unless-stopped
|
||||||
|
volumes:
|
||||||
|
# This directory must have 1000:1000 permissions (or update PUID & PGID below)
|
||||||
|
- /data/sist2-es-data/:/usr/share/elasticsearch/data
|
||||||
|
environment:
|
||||||
|
- "discovery.type=single-node"
|
||||||
|
- "ES_JAVA_OPTS=-Xms2g -Xmx2g"
|
||||||
|
- "PUID=1000"
|
||||||
|
- "PGID=1000"
|
||||||
|
sist2-admin:
|
||||||
|
image: simon987/sist2:3.4.2-x64-linux
|
||||||
|
restart: unless-stopped
|
||||||
|
volumes:
|
||||||
|
- /data/sist2-admin-data/:/sist2-admin/
|
||||||
|
- /<path to index>/:/host
|
||||||
|
ports:
|
||||||
|
- 4090:4090
|
||||||
|
# NOTE: Don't expose this port publicly!
|
||||||
|
- 8080:8080
|
||||||
|
working_dir: /root/sist2-admin/
|
||||||
|
entrypoint: python3
|
||||||
|
command:
|
||||||
|
- /root/sist2-admin/sist2_admin/app.py
|
||||||
|
```
|
||||||
|
|
||||||
|
Navigate to http://localhost:8080/ to configure sist2-admin.
|
||||||
|
|
||||||
|
### Using the executable file *(Linux/WSL only)*
|
||||||
|
|
||||||
|
1. Choose search backend (See [comparison](#search-backends)):
|
||||||
|
* **Elasticsearch**: have an Elasticsearch (version >= 6.8.X, ideally >=7.14.0) instance running
|
||||||
1. Download [from official website](https://www.elastic.co/downloads/elasticsearch)
|
1. Download [from official website](https://www.elastic.co/downloads/elasticsearch)
|
||||||
1. *(or)* Run using docker:
|
2. *(or)* Run using docker:
|
||||||
```bash
|
```bash
|
||||||
docker run -d -p 9200:9200 -e "discovery.type=single-node" elasticsearch:7.17.9
|
docker run -d -p 9200:9200 -e "discovery.type=single-node" elasticsearch:7.17.9
|
||||||
```
|
```
|
||||||
1. *(or)* Run using docker-compose:
|
* **SQLite**: No installation required
|
||||||
```yaml
|
|
||||||
elasticsearch:
|
|
||||||
image: docker.elastic.co/elasticsearch/elasticsearch:7.17.9
|
|
||||||
environment:
|
|
||||||
- discovery.type=single-node
|
|
||||||
- "ES_JAVA_OPTS=-Xms1G -Xmx2G"
|
|
||||||
```
|
|
||||||
1. Download sist2 executable
|
|
||||||
1. Download the [latest sist2 release](https://github.com/simon987/sist2/releases).
|
|
||||||
Select the file corresponding to your CPU architecture and mark the binary as executable with `chmod +x` *
|
|
||||||
2. *(or)* Download a [development snapshot](https://files.simon987.net/.gate/sist2/simon987_sist2/) *(Not
|
|
||||||
recommended!)*
|
|
||||||
3. *(or)* `docker pull simon987/sist2:2.12.1-x64-linux`
|
|
||||||
|
|
||||||
1. See [Usage guide](docs/USAGE.md)
|
2. Download the [latest sist2 release](https://github.com/simon987/sist2/releases).
|
||||||
|
Select the file corresponding to your CPU architecture and mark the binary as executable with `chmod +x`.
|
||||||
|
3. See [usage guide](docs/USAGE.md) for command line usage.
|
||||||
|
|
||||||
\* *Windows users*: **sist2** runs under [WSL](https://en.wikipedia.org/wiki/Windows_Subsystem_for_Linux)
|
Example usage:
|
||||||
|
|
||||||
## Example usage
|
1. Scan a directory: `sist2 scan ~/Documents --output ./documents.sist2`
|
||||||
|
2. Prepare search index:
|
||||||
See [Usage guide](docs/USAGE.md) for more details
|
* **Elasticsearch**: `sist2 index --es-url http://localhost:9200 ./documents.sist2`
|
||||||
|
* **SQLite**: `sist2 sqlite-index --search-index ./search.sist2 ./documents.sist2`
|
||||||
1. Scan a directory: `sist2 scan ~/Documents -o ./docs_idx`
|
3. Start web interface:
|
||||||
1. Push index to Elasticsearch: `sist2 index ./docs_idx`
|
* **Elasticsearch**: `sist2 web ./documents.sist2`
|
||||||
1. Start web interface: `sist2 web ./docs_idx`
|
* **SQLite**: `sist2 web --search-index ./search.sist2 ./documents.sist2`
|
||||||
|
|
||||||
## Format support
|
## Format support
|
||||||
|
|
||||||
@@ -82,7 +111,7 @@ See [Usage guide](docs/USAGE.md) for more details
|
|||||||
| tar, zip, rar, 7z, ar ... | Libarchive | yes\* | - | no |
|
| tar, zip, rar, 7z, ar ... | Libarchive | yes\* | - | no |
|
||||||
| docx, xlsx, pptx | [libscan](https://github.com/simon987/sist2/tree/master/third-party/libscan) | yes | if embedded | creator, modified_by, title |
|
| docx, xlsx, pptx | [libscan](https://github.com/simon987/sist2/tree/master/third-party/libscan) | yes | if embedded | creator, modified_by, title |
|
||||||
| doc (MS Word 97-2003) | antiword | yes | no | author, title |
|
| doc (MS Word 97-2003) | antiword | yes | no | author, title |
|
||||||
| mobi, azw, azw3 | libmobi | yes | no | author, title |
|
| mobi, azw, azw3 | libmobi | yes | yes | author, title |
|
||||||
| wpd (WordPerfect) | libwpd | yes | no | *planned* |
|
| wpd (WordPerfect) | libwpd | yes | no | *planned* |
|
||||||
| json, jsonl, ndjson | [libscan](https://github.com/simon987/sist2/tree/master/third-party/libscan) | yes | - | - |
|
| json, jsonl, ndjson | [libscan](https://github.com/simon987/sist2/tree/master/third-party/libscan) | yes | - | - |
|
||||||
|
|
||||||
@@ -109,7 +138,7 @@ Download the language data files with your package manager (`apt install tessera
|
|||||||
directly [from Github](https://github.com/tesseract-ocr/tesseract/wiki/Data-Files).
|
directly [from Github](https://github.com/tesseract-ocr/tesseract/wiki/Data-Files).
|
||||||
|
|
||||||
The `simon987/sist2` image comes with common languages
|
The `simon987/sist2` image comes with common languages
|
||||||
(hin, jpn, eng, fra, rus, spa, chi_sim, deu) pre-installed.
|
(hin, jpn, eng, fra, rus, spa, chi_sim, deu, pol) pre-installed.
|
||||||
|
|
||||||
You can use the `+` separator to specify multiple languages. The language
|
You can use the `+` separator to specify multiple languages. The language
|
||||||
name must be identical to the `*.traineddata` file installed on your system
|
name must be identical to the `*.traineddata` file installed on your system
|
||||||
@@ -123,20 +152,61 @@ sist2 scan --ocr-images --ocr-lang eng ~/Images/Screenshots/
|
|||||||
sist2 scan --ocr-ebooks --ocr-images --ocr-lang eng+chi_sim ~/Chinese-Bilingual/
|
sist2 scan --ocr-ebooks --ocr-images --ocr-lang eng+chi_sim ~/Chinese-Bilingual/
|
||||||
```
|
```
|
||||||
|
|
||||||
|
### Search backends
|
||||||
|
|
||||||
|
sist2 v3.0.7+ supports SQLite search backend. The SQLite search backend has
|
||||||
|
fewer features and generally comparable query performance for medium-size
|
||||||
|
indices, but it uses much less memory and is easier to set up.
|
||||||
|
|
||||||
|
| | SQLite | Elasticsearch |
|
||||||
|
|----------------------------------------------|:---------------------------------------------------:|:-------------------------------------------------------------------------------------------------------------------------------------:|
|
||||||
|
| Requires separate search engine installation | | ✓ |
|
||||||
|
| Memory footprint | ~20MB | >500MB |
|
||||||
|
| Query syntax | [fts5](https://www.sqlite.org/fts5.html) | [query_string](https://www.elastic.co/guide/en/elasticsearch/reference/current/query-dsl-query-string-query.html#query-string-syntax) |
|
||||||
|
| Fuzzy search | | ✓ |
|
||||||
|
| Media Types tree real-time updating | | ✓ |
|
||||||
|
| Manual tagging | ✓ | ✓ |
|
||||||
|
| User scripts | ✓ | ✓ |
|
||||||
|
| Media Type breakdown for search results | | ✓ |
|
||||||
|
| Embeddings search | ✓ *O(n)* | ✓ *O(logn)* |
|
||||||
|
|
||||||
|
### NER
|
||||||
|
|
||||||
|
sist2 v3.0.4+ supports named-entity recognition (NER). Simply add a supported repository URL to
|
||||||
|
**Configuration** > **Machine learning options** > **Model repositories**
|
||||||
|
to enable it.
|
||||||
|
|
||||||
|
The text processing is done in your browser, no data is sent to any third-party services.
|
||||||
|
See [simon987/sist2-ner-models](https://github.com/simon987/sist2-ner-models) for more details.
|
||||||
|
|
||||||
|
#### List of available repositories:
|
||||||
|
|
||||||
|
| URL | Maintainer | Purpose |
|
||||||
|
|---------------------------------------------------------------------------------------------------------|-----------------------------------------|---------|
|
||||||
|
| [simon987/sist2-ner-models](https://raw.githubusercontent.com/simon987/sist2-ner-models/main/repo.json) | [simon987](https://github.com/simon987) | General |
|
||||||
|
|
||||||
|
<details>
|
||||||
|
<summary>Screenshot</summary>
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
</details>
|
||||||
|
|
||||||
## Build from source
|
## Build from source
|
||||||
|
|
||||||
You can compile **sist2** by yourself if you don't want to use the pre-compiled binaries
|
You can compile **sist2** by yourself if you don't want to use the pre-compiled binaries
|
||||||
|
|
||||||
### With docker (recommended)
|
### Using docker
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
git clone --recursive https://github.com/simon987/sist2/
|
git clone --recursive https://github.com/simon987/sist2/
|
||||||
cd sist2
|
cd sist2
|
||||||
docker build . -f ./Dockerfile -t my-sist2-image
|
docker build . -t my-sist2-image
|
||||||
|
# Copy sist2 executable from docker image
|
||||||
docker run --rm --entrypoint cat my-sist2-image /root/sist2 > sist2-x64-linux
|
docker run --rm --entrypoint cat my-sist2-image /root/sist2 > sist2-x64-linux
|
||||||
```
|
```
|
||||||
|
|
||||||
### On a linux computer
|
### Using a linux computer
|
||||||
|
|
||||||
1. Install compile-time dependencies
|
1. Install compile-time dependencies
|
||||||
|
|
||||||
@@ -144,15 +214,14 @@ docker run --rm --entrypoint cat my-sist2-image /root/sist2 > sist2-x64-linux
|
|||||||
apt install gcc g++ python3 yasm ragel automake autotools-dev wget libtool libssl-dev curl zip unzip tar xorg-dev libglu1-mesa-dev libxcursor-dev libxml2-dev libxinerama-dev gettext nasm git nodejs
|
apt install gcc g++ python3 yasm ragel automake autotools-dev wget libtool libssl-dev curl zip unzip tar xorg-dev libglu1-mesa-dev libxcursor-dev libxml2-dev libxinerama-dev gettext nasm git nodejs
|
||||||
```
|
```
|
||||||
|
|
||||||
1. Apply vcpkg patches, as per [sist2-build](https://github.com/simon987/sist2-build) Dockerfile
|
2. Install vcpkg using my fork: https://github.com/simon987/vcpkg
|
||||||
|
3. Install vcpkg dependencies
|
||||||
1. Install vcpkg dependencies
|
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
vcpkg install curl[core,openssl] sqlite3 cpp-jwt pcre cjson brotli libarchive[core,bzip2,libxml2,lz4,lzma,lzo] pthread tesseract libxml2 libmupdf gtest mongoose libmagic libraw gumbo ffmpeg[core,avcodec,avformat,swscale,swresample]
|
vcpkg install openblas curl[core,openssl] sqlite3[core,fts5,json1] cpp-jwt pcre cjson brotli libarchive[core,bzip2,libxml2,lz4,lzma,lzo] pthread tesseract libxml2 libmupdf[ocr] gtest mongoose libmagic libraw gumbo ffmpeg[core,avcodec,avformat,swscale,swresample,webp,opus,mp3lame,vpx,zlib]
|
||||||
```
|
```
|
||||||
|
|
||||||
1. Build
|
4. Build
|
||||||
```bash
|
```bash
|
||||||
git clone --recursive https://github.com/simon987/sist2/
|
git clone --recursive https://github.com/simon987/sist2/
|
||||||
(cd sist2-vue; npm install; npm run build)
|
(cd sist2-vue; npm install; npm run build)
|
||||||
|
|||||||
@@ -4,16 +4,21 @@ services:
|
|||||||
elasticsearch:
|
elasticsearch:
|
||||||
image: elasticsearch:7.17.9
|
image: elasticsearch:7.17.9
|
||||||
container_name: sist2-es
|
container_name: sist2-es
|
||||||
|
volumes:
|
||||||
|
# This directory must have 1000:1000 permissions (or update PUID & PGID below)
|
||||||
|
- /data/sist2-es-data/:/usr/share/elasticsearch/data
|
||||||
environment:
|
environment:
|
||||||
- "discovery.type=single-node"
|
- "discovery.type=single-node"
|
||||||
- "ES_JAVA_OPTS=-Xms2g -Xmx2g"
|
- "ES_JAVA_OPTS=-Xms2g -Xmx2g"
|
||||||
|
- "PUID=1000"
|
||||||
|
- "PGID=1000"
|
||||||
sist2-admin:
|
sist2-admin:
|
||||||
build:
|
build:
|
||||||
context: .
|
context: .
|
||||||
container_name: sist2-admin
|
container_name: sist2-admin
|
||||||
volumes:
|
volumes:
|
||||||
- /mnt/array/sist2-admin-data/:/sist2-admin/
|
- /data/sist2-admin-data/:/sist2-admin/
|
||||||
- /:/host
|
- /<path to index>/:/host
|
||||||
ports:
|
ports:
|
||||||
- 4090:4090
|
- 4090:4090
|
||||||
# NOTE: Don't export this port publicly!
|
# NOTE: Don't export this port publicly!
|
||||||
|
|||||||
361
docs/USAGE.md
@@ -1,78 +1,68 @@
|
|||||||
# Usage
|
# Usage
|
||||||
|
|
||||||
*More examples (specifically with docker/compose) are in progress*
|
|
||||||
|
|
||||||
* [scan](#scan)
|
|
||||||
* [options](#scan-options)
|
|
||||||
* [examples](#scan-examples)
|
|
||||||
* [index format](#index-format)
|
|
||||||
* [index](#index)
|
|
||||||
* [options](#index-options)
|
|
||||||
* [examples](#index-examples)
|
|
||||||
* [web](#web)
|
|
||||||
* [options](#web-options)
|
|
||||||
* [examples](#web-examples)
|
|
||||||
* [rewrite_url](#rewrite_url)
|
|
||||||
* [elasticsearch](#elasticsearch)
|
|
||||||
* [exec-script](#exec-script)
|
|
||||||
* [tagging](#tagging)
|
|
||||||
* [sidecar files](#sidecar-files)
|
|
||||||
|
|
||||||
```
|
```
|
||||||
Usage: sist2 scan [OPTION]... PATH
|
Usage: sist2 scan [OPTION]... PATH
|
||||||
or: sist2 index [OPTION]... INDEX
|
or: sist2 index [OPTION]... INDEX
|
||||||
|
or: sist2 sqlite-index [OPTION]... INDEX
|
||||||
or: sist2 web [OPTION]... INDEX...
|
or: sist2 web [OPTION]... INDEX...
|
||||||
or: sist2 exec-script [OPTION]... INDEX
|
|
||||||
Lightning-fast file system indexer and search tool.
|
Lightning-fast file system indexer and search tool.
|
||||||
|
|
||||||
-h, --help show this help message and exit
|
-h, --help show this help message and exit
|
||||||
-v, --version Show version and exit
|
-v, --version Print version and exit.
|
||||||
--verbose Turn on logging
|
--verbose Turn on logging.
|
||||||
--very-verbose Turn on debug messages
|
--very-verbose Turn on debug messages.
|
||||||
|
--json-logs Output logs in JSON format.
|
||||||
|
|
||||||
Scan options
|
Scan options
|
||||||
-t, --threads=<int> Number of threads. DEFAULT=1
|
-t, --threads=<int> Number of threads. DEFAULT: 1
|
||||||
--mem-throttle=<int> Total memory threshold in MiB for scan throttling. DEFAULT=0
|
-q, --thumbnail_count-quality=<int> Thumbnail quality, on a scale of 0 to 100, 100 being the best. DEFAULT: 50
|
||||||
-q, --thumbnail-quality=<int> Thumbnail quality, on a scale of 2 to 31, 2 being the best. DEFAULT=2
|
--thumbnail_count-size=<int> Thumbnail size, in pixels. DEFAULT: 552
|
||||||
--thumbnail-size=<int> Thumbnail size, in pixels. DEFAULT=500
|
--thumbnail_count-count=<int> Number of thumbnails to generate. Set a value > 1 to create video previews, set to 0 to disable thumbnails. DEFAULT: 1
|
||||||
--thumbnail-count=<int> Number of thumbnails to generate. Set a value > 1 to create video previews, set to 0 to disable thumbnails. DEFAULT=1
|
--content-size=<int> Number of bytes to be extracted from text documents. Set to 0 to disable. DEFAULT: 32768
|
||||||
--content-size=<int> Number of bytes to be extracted from text documents. Set to 0 to disable. DEFAULT=32768
|
-o, --output=<str> Output index file path. DEFAULT: index.sist2
|
||||||
--incremental=<str> Reuse an existing index and only scan modified files.
|
--incremental If the output file path exists, only scan new or modified files.
|
||||||
-o, --output=<str> Output directory. DEFAULT=index.sist2/
|
--optimize-index Defragment index file after scan to reduce its file size.
|
||||||
--rewrite-url=<str> Serve files from this url instead of from disk.
|
--rewrite-url=<str> Serve files from this url instead of from disk.
|
||||||
--name=<str> Index display name. DEFAULT: (name of the directory)
|
--name=<str> Index display name. DEFAULT: index
|
||||||
--depth=<int> Scan up to DEPTH subdirectories deep. Use 0 to only scan files in PATH. DEFAULT: -1
|
--depth=<int> Scan up to DEPTH subdirectories deep. Use 0 to only scan files in PATH. DEFAULT: -1
|
||||||
--archive=<str> Archive file mode (skip|list|shallow|recurse). skip: Don't parse, list: only get file names as text, shallow: Don't parse archives inside archives. DEFAULT: recurse
|
--archive=<str> Archive file mode (skip|list|shallow|recurse). skip: don't scan, list: only save file names as text, shallow: don't scan archives inside archives. DEFAULT: recurse
|
||||||
--archive-passphrase=<str> Passphrase for encrypted archive files
|
--archive-passphrase=<str> Passphrase for encrypted archive files
|
||||||
--ocr-lang=<str> Tesseract language (use 'tesseract --list-langs' to see which are installed on your machine)
|
--ocr-lang=<str> Tesseract language (use 'tesseract --list-langs' to see which are installed on your machine)
|
||||||
--ocr-images Enable OCR'ing of image files.
|
--ocr-images Enable OCR'ing of image files.
|
||||||
--ocr-ebooks Enable OCR'ing of ebook files.
|
--ocr-ebooks Enable OCR'ing of ebook files.
|
||||||
-e, --exclude=<str> Files that match this regex will not be scanned
|
-e, --exclude=<str> Files that match this regex will not be scanned.
|
||||||
--fast Only index file names & mime type
|
--fast Only index file names & mime type.
|
||||||
--treemap-threshold=<str> Relative size threshold for treemap (see USAGE.md). DEFAULT: 0.0005
|
--treemap-threshold=<str> Relative size threshold for treemap (see USAGE.md). DEFAULT: 0.0005
|
||||||
--mem-buffer=<int> Maximum memory buffer size per thread in MiB for files inside archives (see USAGE.md). DEFAULT: 2000
|
--mem-buffer=<int> Maximum memory buffer size per thread in MiB for files inside archives (see USAGE.md). DEFAULT: 2000
|
||||||
--read-subtitles Read subtitles from media files.
|
--read-subtitles Read subtitles from media files.
|
||||||
--fast-epub Faster but less accurate EPUB parsing (no thumbnails, metadata)
|
--fast-epub Faster but less accurate EPUB parsing (no thumbnails, metadata).
|
||||||
--checksums Calculate file checksums when scanning.
|
--checksums Calculate file checksums when scanning.
|
||||||
--list-file=<str> Specify a list of newline-delimited paths to be scanned instead of normal directory traversal. Use '-' to read from stdin.
|
--list-file=<str> Specify a list of newline-delimited paths to be scanned instead of normal directory traversal. Use '-' to read from stdin.
|
||||||
|
|
||||||
Index options
|
Index options
|
||||||
-t, --threads=<int> Number of threads. DEFAULT=1
|
-t, --threads=<int> Number of threads. DEFAULT: 1
|
||||||
--es-url=<str> Elasticsearch url with port. DEFAULT=http://localhost:9200
|
--es-url=<str> Elasticsearch url with port. DEFAULT: http://localhost:9200
|
||||||
--es-index=<str> Elasticsearch index name. DEFAULT=sist2
|
--es-insecure-ssl Do not verify SSL connections to Elasticsearch.
|
||||||
-p, --print Just print JSON documents to stdout.
|
--es-index=<str> Elasticsearch index name. DEFAULT: sist2
|
||||||
--incremental-index Conduct incremental indexing, assumes that the old index is already digested by Elasticsearch.
|
-p, --print Print JSON documents to stdout instead of indexing to elasticsearch.
|
||||||
|
--incremental-index Conduct incremental indexing. Assumes that the old index is already ingested in Elasticsearch.
|
||||||
--script-file=<str> Path to user script.
|
--script-file=<str> Path to user script.
|
||||||
--mappings-file=<str> Path to Elasticsearch mappings.
|
--mappings-file=<str> Path to Elasticsearch mappings.
|
||||||
--settings-file=<str> Path to Elasticsearch settings.
|
--settings-file=<str> Path to Elasticsearch settings.
|
||||||
--async-script Execute user script asynchronously.
|
--async-script Execute user script asynchronously.
|
||||||
--batch-size=<int> Index batch size. DEFAULT: 100
|
--batch-size=<int> Index batch size. DEFAULT: 70
|
||||||
-f, --force-reset Reset Elasticsearch mappings and settings. (You must use this option the first time you use the index command)
|
-f, --force-reset Reset Elasticsearch mappings and settings.
|
||||||
|
|
||||||
|
sqlite-index options
|
||||||
|
--search-index=<str> Path to search index. Will be created if it does not exist yet.
|
||||||
|
|
||||||
Web options
|
Web options
|
||||||
--es-url=<str> Elasticsearch url. DEFAULT=http://localhost:9200
|
--es-url=<str> Elasticsearch url. DEFAULT: http://localhost:9200
|
||||||
--es-index=<str> Elasticsearch index name. DEFAULT=sist2
|
--es-insecure-ssl Do not verify SSL connections to Elasticsearch.
|
||||||
--bind=<str> Listen on this address. DEFAULT=localhost:4090
|
--search-index=<str> Path to SQLite search index.
|
||||||
|
--es-index=<str> Elasticsearch index name. DEFAULT: sist2
|
||||||
|
--bind=<str> Listen for connections on this address. DEFAULT: localhost:4090
|
||||||
--auth=<str> Basic auth in user:password format
|
--auth=<str> Basic auth in user:password format
|
||||||
--auth0-audience=<str> API audience/identifier
|
--auth0-audience=<str> API audience/identifier
|
||||||
--auth0-domain=<str> Application domain
|
--auth0-domain=<str> Application domain
|
||||||
@@ -83,89 +73,18 @@ Web options
|
|||||||
--dev Serve html & js files from disk (for development)
|
--dev Serve html & js files from disk (for development)
|
||||||
--lang=<str> Default UI language. Can be changed by the user
|
--lang=<str> Default UI language. Can be changed by the user
|
||||||
|
|
||||||
Exec-script options
|
|
||||||
--es-url=<str> Elasticsearch url. DEFAULT=http://localhost:9200
|
|
||||||
--es-index=<str> Elasticsearch index name. DEFAULT=sist2
|
|
||||||
--script-file=<str> Path to user script.
|
|
||||||
--async-script Execute user script asynchronously.
|
|
||||||
Made by simon987 <me@simon987.net>. Released under GPL-3.0
|
Made by simon987 <me@simon987.net>. Released under GPL-3.0
|
||||||
```
|
```
|
||||||
|
|
||||||
## Scan
|
|
||||||
|
|
||||||
### Scan options
|
|
||||||
|
|
||||||
* `-t, --threads`
|
|
||||||
Number of threads for file parsing. **Do not set a number higher than `$(nproc)` or `$(Get-CimInstance Win32_ComputerSystem).NumberOfLogicalProcessors` in Windows!**
|
|
||||||
* `--mem-throttle`
|
|
||||||
Total memory threshold in MiB for scan throttling. Worker threads will not start a new parse job
|
|
||||||
until the total memory usage of sist2 is below this threshold. Set to 0 to disable. DEFAULT=0
|
|
||||||
* `-q, --thumbnail-quality`
|
|
||||||
Thumbnail quality, on a scale of 2 to 32, 2 being the best. See section below for a rough estimate of thumbnail database size
|
|
||||||
* `--thumbnail-size`
|
|
||||||
Thumbnail size in pixels.
|
|
||||||
* `--thumbnail-count`
|
|
||||||
Maximum number of thumbnails to generate. When set to a value >= 2, thumbnails for video previews
|
|
||||||
will be generated. The actual number of thumbnails generated depends on the length of the video (maximum 1 image
|
|
||||||
every ~7s). Set to 0 to completely disable thumbnails.
|
|
||||||
* `--content-size`
|
|
||||||
Number of bytes of text to be extracted from the content of files (plain text, PDFs etc.).
|
|
||||||
Repeated whitespace and special characters do not count toward this limit.
|
|
||||||
Set to 0 to completely disable content parsing.
|
|
||||||
* `--incremental`
|
|
||||||
Specify an existing index. Information about files in this index that were not modified (based on *mtime* attribute)
|
|
||||||
will be copied to the new index and will not be parsed again.
|
|
||||||
* `-o, --output` Output directory.
|
|
||||||
* `--rewrite-url` Set the `rewrite_url` option for the web module (See [rewrite_url](#rewrite_url))
|
|
||||||
* `--name` Set the `name` option for the web module
|
|
||||||
* `--depth` Maximum scan dept. Set to 0 only scan files directly in the root directory, set to -1 for infinite depth
|
|
||||||
* `--archive` Archive file mode.
|
|
||||||
* skip: Don't parse
|
|
||||||
* list: Only get file names as text
|
|
||||||
* shallow: Don't parse archives inside archives.
|
|
||||||
* recurse: Scan archives recursively (default)
|
|
||||||
* `--ocr-lang`, `--ocr-ebooks`, `--ocr-images` See [OCR](../README.md#OCR)
|
|
||||||
* `-e, --exclude` Regex pattern to exclude files. A file is excluded if the pattern matches any
|
|
||||||
part of the full absolute path.
|
|
||||||
|
|
||||||
Examples:
|
|
||||||
* `-e ".*\.ttf"`: Ignore ttf files
|
|
||||||
* `-e ".*\.(ttf|rar)"`: Ignore ttf and rar files
|
|
||||||
* `-e "^/mnt/backups/"`: Ignore all files in the `/mnt/backups/` directory
|
|
||||||
* `-e "^/mnt/Data[12]/"`: Ignore all files in the `/mnt/Data1/` and `/mnt/Data2/` directory
|
|
||||||
* `-e "(^/usr/)|(^/var/)|(^/media/DRIVE-A/tmp/)|(^/media/DRIVE-B/Trash/)"` Exclude the
|
|
||||||
`/usr`, `/var`, `/media/DRIVE-A/tmp`, `/media/DRIVE-B/Trash` directories
|
|
||||||
* `--fast` Only index file names and mime type
|
|
||||||
* `--treemap-threshold` Directories smaller than (`treemap-threshold` * `<total size of the index>`)
|
|
||||||
will not be considered for the disk utilisation visualization; their size will be added to
|
|
||||||
the parent directory. If the parent directory is still smaller than the threshold, it will also be "merged upwards"
|
|
||||||
and so on.
|
|
||||||
|
|
||||||
In effect, smaller `treemap-threshold` values will yield a more detailed
|
|
||||||
(but also a more cluttered and harder to read) visualization.
|
|
||||||
|
|
||||||
* `--mem-buffer` Maximum memory buffer size in MiB (per thread) for files inside archives. Media files
|
|
||||||
larger than this number will be read sequentially and no *seek* operations will be supported.
|
|
||||||
|
|
||||||
To check if a media file can be parsed without *seek*, execute `cat file.mp4 | ffprobe -`
|
|
||||||
* `--read-subtitles` When enabled, will attempt to read the subtitles stream from media files.
|
|
||||||
* `--fast-epub` Much faster but less accurate EPUB parsing. When enabled, sist2 will use a simple HTML parser to read epub files instead of the MuPDF library. No thumbnails are generated and author/title metadata are not parsed.
|
|
||||||
* `--checksums` Calculate file checksums (SHA1) when scanning files. This option does not cause any additional read
|
|
||||||
operations. Checksums are not calculated for all file types, unless the file is inside an archive. When enabled, duplicate
|
|
||||||
files are hidden in the web UI (this behaviour can be toggled in the Configuration page).
|
|
||||||
|
|
||||||
|
|
||||||
#### Thumbnail database size estimation
|
#### Thumbnail database size estimation
|
||||||
|
|
||||||
See chart below for rough estimate of thumbnail size vs. thumbnail size & quality arguments:
|
See chart below for rough estimate of thumbnail_count size vs. thumbnail_count size & quality arguments:
|
||||||
|
|
||||||
For example, `--thumbnail-size=500`, `--thumbnail-quality=2` for a directory with 8 million images will create a thumbnail database
|
For example, `--thumbnail_count-size=500`, `--thumbnail_count-quality=50` for a directory with 8 million images will create a thumbnail_count database
|
||||||
that is about `8000000 * 36kB = 288GB`.
|
that is about `8000000 * 11.8kB = 94.4GB`.
|
||||||
|
|
||||||

|

|
||||||
|
|
||||||
// TODO: add note about LMDB page size 4096
|
|
||||||
|
|
||||||
### Scan examples
|
### Scan examples
|
||||||
|
|
||||||
Simple scan
|
Simple scan
|
||||||
@@ -173,132 +92,70 @@ Simple scan
|
|||||||
sist2 scan ~/Documents
|
sist2 scan ~/Documents
|
||||||
|
|
||||||
sist2 scan \
|
sist2 scan \
|
||||||
--threads 4 --content-size 16000000 --thumbnail-quality 2 --archive shallow \
|
--threads 4 --content-size 16000000 --thumbnail_count-quality 2 --archive shallow \
|
||||||
--name "My Documents" --rewrite-url "http://nas.domain.local/My Documents/" \
|
--name "My Documents" --rewrite-url "http://nas.domain.local/My Documents/" \
|
||||||
~/Documents -o ./documents.idx/
|
~/Documents -o ./documents.sist2
|
||||||
```
|
```
|
||||||
|
|
||||||
Incremental scan
|
Incremental scan
|
||||||
```
|
|
||||||
sist2 scan --incremental ./orig_idx/ -o ./updated_idx/ ~/Documents
|
|
||||||
```
|
|
||||||
|
|
||||||
### Index format
|
If the index file does not exist, `--incremental` has no effect.
|
||||||
|
|
||||||
A typical `ndjson` type index structure looks like this:
|
|
||||||
```
|
|
||||||
documents.idx/
|
|
||||||
├── descriptor.json
|
|
||||||
├── _index_main.ndjson.zst
|
|
||||||
├── treemap.csv
|
|
||||||
├── agg_mime.csv
|
|
||||||
├── agg_date.csv
|
|
||||||
├── add_size.csv
|
|
||||||
├── thumbs/
|
|
||||||
| ├── data.mdb
|
|
||||||
| └── lock.mdb
|
|
||||||
├── tags/
|
|
||||||
| ├── data.mdb
|
|
||||||
| └── lock.mdb
|
|
||||||
└── meta/
|
|
||||||
├── data.mdb
|
|
||||||
└── lock.mdb
|
|
||||||
```
|
|
||||||
|
|
||||||
The `_index_*.ndjson.zst` files contain the document data in JSON format, in a compressed newline-delemited file.
|
|
||||||
|
|
||||||
The `thumbs/` folder is a [LMDB](https://en.wikipedia.org/wiki/Lightning_Memory-Mapped_Database)
|
|
||||||
database containing the thumbnails.
|
|
||||||
|
|
||||||
The `descriptor.json` file contains general information about the index. The
|
|
||||||
following fields are safe to modify manually: `root`, `name`, [rewrite_url](#rewrite_url) and `timestamp`.
|
|
||||||
|
|
||||||
The `.csv` are pre-computed aggregations necessary for the stats page.
|
|
||||||
|
|
||||||
*thumbs/*:
|
|
||||||
|
|
||||||
LMDB key-value store. Keys are **binary** 16-byte md5 hash* (`_id` field)
|
|
||||||
and values are raw image bytes.
|
|
||||||
|
|
||||||
*\* Hash is calculated from the full path of the file, including the extension, relative to the index root*
|
|
||||||
|
|
||||||
|
|
||||||
## Index
|
|
||||||
### Index options
|
|
||||||
* `--es-url`
|
|
||||||
Elasticsearch url and port. If you are using docker, make sure that both containers are on the
|
|
||||||
same network.
|
|
||||||
* `--es-index`
|
|
||||||
Elasticsearch index name. DEFAULT=sist2
|
|
||||||
* `-p, --print`
|
|
||||||
Print index in JSON format to stdout.
|
|
||||||
* `--incremental-index`
|
|
||||||
Conduct incremental indexing. Assumes that the old index is already ingested in Elasticsearch.
|
|
||||||
Only the new changes since the last scan will be sent.
|
|
||||||
* `--script-file`
|
|
||||||
Path to user script. See [Scripting](scripting.md).
|
|
||||||
* `--mappings-file`
|
|
||||||
Path to custom Elasticsearch mappings. If none is specified, [the bundled mappings](https://github.com/simon987/sist2/tree/master/schema) will be used.
|
|
||||||
* `--settings-file`
|
|
||||||
Path to custom Elasticsearch settings. *(See above)*
|
|
||||||
* `--async-script`
|
|
||||||
Use `wait_for_completion=false` elasticsearch option while executing user script.
|
|
||||||
(See [Elasticsearch documentation](https://www.elastic.co/guide/en/elasticsearch/reference/current/tasks.html))
|
|
||||||
* `--batch-size=<int>`
|
|
||||||
Index batch size. Indexing is generally faster with larger batches, but payloads that
|
|
||||||
are too large will fail and additional overhead for retrying with smaller sizes may slow
|
|
||||||
down the process.
|
|
||||||
* `-f, --force-reset`
|
|
||||||
Reset Elasticsearch mappings and settings.
|
|
||||||
* `-t, --threads` Number of threads to use. Ideally, choose a number equal to the number of logical cores of the machine hosting Elasticsearch.
|
|
||||||
|
|
||||||
### Index examples
|
|
||||||
|
|
||||||
**Push to elasticsearch**
|
|
||||||
```bash
|
```bash
|
||||||
sist2 index --force-reset --batch-size 1000 --es-url http://localhost:9200 ./my_index/
|
sist scan ~/Documents -o ./documents.sist2
|
||||||
sist2 index ./my_index/
|
sist scan ~/Documents -o ./documents.sist2 --incremental
|
||||||
|
# or
|
||||||
|
sist scan ~/Documents -o ./documents.sist2 --incremental
|
||||||
|
sist scan ~/Documents -o ./documents.sist2 --incremental
|
||||||
|
```
|
||||||
|
|
||||||
|
### Index documents to Elasticsearch search backend
|
||||||
|
|
||||||
|
```bash
|
||||||
|
sist2 index --force-reset --batch-size 1000 --es-url http://localhost:9200 ./my_index.sist2
|
||||||
|
sist2 index ./my_index.sist2
|
||||||
|
```
|
||||||
|
|
||||||
|
#### Index documents to SQLite search backend
|
||||||
|
```bash
|
||||||
|
# The search index will be created if it does not exist already
|
||||||
|
sist2 sqlite-index ./index1.sist2 --search-index search.sist2
|
||||||
|
sist2 sqlite-index ./index2.sist2 --search-index search.sist2
|
||||||
```
|
```
|
||||||
|
|
||||||
**Save index in JSON format**
|
**Save index in JSON format**
|
||||||
```bash
|
```bash
|
||||||
sist2 index --print ./my_index/ > my_index.ndjson
|
sist2 index --print ./my_index.sist2 > my_index.ndjson
|
||||||
```
|
```
|
||||||
|
|
||||||
**Inspect contents of an index**
|
**Inspect contents of an index**
|
||||||
```bash
|
```bash
|
||||||
sist2 index --print ./my_index/ | jq | less
|
sist2 index --print ./my_index.sist2 | jq | less
|
||||||
```
|
```
|
||||||
|
|
||||||
## Web
|
## Web
|
||||||
|
|
||||||
### Web options
|
|
||||||
* `--es-url=<str>` Elasticsearch url.
|
|
||||||
* `--es-index`
|
|
||||||
Elasticsearch index name. DEFAULT=sist2
|
|
||||||
* `--bind=<str>` Listen on this address.
|
|
||||||
* `--auth=<str>` Basic auth in user:password format
|
|
||||||
* `--tag-auth=<str>` Basic auth in user:password format. Works the same way as the
|
|
||||||
`--auth` argument, but authentication is only applied the `/tag/` endpoint.
|
|
||||||
* `--tagline=<str>` When specified, will replace the default tagline in the navbar.
|
|
||||||
* `--dev` Serve html & js files from disk (for development, used to modify frontend files without having to recompile)
|
|
||||||
* `--lang=<str>` Set the default web UI language (See #180 for a list of supported languages, default
|
|
||||||
is `en`). The user can change the language in the configuration page
|
|
||||||
* `--auth0-audience`, `--auth0-domain`, `--auth0-client-id`, `--auth0-public-key-file` See [Authentication with Auth0](auth0.md)
|
|
||||||
|
|
||||||
### Web examples
|
### Web examples
|
||||||
|
|
||||||
**Single index**
|
**Single index (Elasticsearch backend)**
|
||||||
```bash
|
```bash
|
||||||
sist2 web --auth admin:hunter2 --bind 0.0.0.0:8888 my_index
|
sist2 web --auth admin:hunter2 --bind 0.0.0.0:8888 my_index.sist2
|
||||||
```
|
```
|
||||||
|
|
||||||
**Multiple indices**
|
**Multiple indices (Elasticsearch backend)**
|
||||||
```bash
|
```bash
|
||||||
# Indices will be displayed in this order in the web interface
|
# Indices will be displayed in this order in the web interface
|
||||||
sist2 web index1 index2 index3 index4
|
sist2 web index1.sist2 index2.sist2 index3.sist2 index4.sist2
|
||||||
```
|
```
|
||||||
|
|
||||||
|
**SQLite search backend**
|
||||||
|
```bash
|
||||||
|
sist2 web --search-index search.sist2 index1.sist2
|
||||||
|
```
|
||||||
|
|
||||||
|
#### Auth0 authentication
|
||||||
|
|
||||||
|
See [auth0.md](auth0.md)
|
||||||
|
|
||||||
### rewrite_url
|
### rewrite_url
|
||||||
|
|
||||||
When the `rewrite_url` field is not empty, the web module ignores the `root`
|
When the `rewrite_url` field is not empty, the web module ignores the `root`
|
||||||
@@ -315,12 +172,37 @@ Using a version >=7.14.0 is recommended to enable the following features:
|
|||||||
|
|
||||||
- Bug fix for large documents (See #198)
|
- Bug fix for large documents (See #198)
|
||||||
|
|
||||||
|
Using a version >=8.0.0 is recommended to enable the following features:
|
||||||
|
|
||||||
|
- Approximate KNN search for Embeddings search (faster queries).
|
||||||
|
|
||||||
When using a legacy version of ES, a notice will be displayed next to the sist2 version in the web UI.
|
When using a legacy version of ES, a notice will be displayed next to the sist2 version in the web UI.
|
||||||
If you don't care about the features above, you can ignore it or disable it in the configuration page.
|
If you don't care about the features above, you can ignore it or disable it in the configuration page.
|
||||||
|
|
||||||
## exec-script
|
# Embeddings search
|
||||||
|
|
||||||
The `exec-script` command is used to execute a user script for an index that has already been imported to Elasticsearch with the `index` command. Note that the documents will not be reset to their default state before each execution as the `index` command does: if you make undesired changes to the documents by accident, you will need to run `index` again to revert to the original state.
|
Since v3.2.0, User scripts can be used to generate _embeddings_ (vector of float32 numbers) which are stored in the .sist2 index file
|
||||||
|
(see [scripting](scripting.md)). Embeddings can be used for:
|
||||||
|
|
||||||
|
* Nearest-neighbor queries (e.g. "return the documents most similar to this one")
|
||||||
|
* Semantic searches (e.g. "return the documents that are most closely related to the given topic")
|
||||||
|
|
||||||
|
In theory, embeddings can be created for any type of documents (image, text, audio etc.).
|
||||||
|
|
||||||
|
For example, the [clip](https://github.com/sist2app/sist2-script-clip) User Script, generates 512-d embeddings of images
|
||||||
|
(videos are also supported using the thumbnails generated by sist2). When the user enters a query in the "Embeddings Search"
|
||||||
|
textbox, the query's embedding is generated in their browser, leveraging the ONNX web runtime.
|
||||||
|
|
||||||
|
<details>
|
||||||
|
<summary>Screenshots</summary>
|
||||||
|
|
||||||
|

|
||||||
|

|
||||||
|
|
||||||
|
1. Embeddings search bar. You can select the model using the dropdown on the left.
|
||||||
|
2. This icon appears for indices with embeddings search enabled.
|
||||||
|
3. Documents with this icon have embeddings. Click on the icon to perform KNN search.
|
||||||
|
</details>
|
||||||
|
|
||||||
|
|
||||||
# Tagging
|
# Tagging
|
||||||
@@ -348,42 +230,3 @@ See [Automatic tagging](#automatic-tagging) for information about tag
|
|||||||
### Automatic tagging
|
### Automatic tagging
|
||||||
|
|
||||||
See [scripting](scripting.md) documentation.
|
See [scripting](scripting.md) documentation.
|
||||||
|
|
||||||
# Sidecar files
|
|
||||||
|
|
||||||
When scanning, sist2 will read metadata from `.s2meta` JSON files and overwrite the
|
|
||||||
original document's indexed metadata (does not modify the actual file). Sidecar metadata files will also work inside archives.
|
|
||||||
Sidecar files themselves are not saved in the index.
|
|
||||||
|
|
||||||
This feature is useful to leverage third-party applications such as speech-to-text or
|
|
||||||
OCR to add additional metadata to a file.
|
|
||||||
|
|
||||||
**Example**
|
|
||||||
|
|
||||||
```
|
|
||||||
~/Documents/
|
|
||||||
├── Video.mp4
|
|
||||||
└── Video.mp4.s2meta
|
|
||||||
```
|
|
||||||
|
|
||||||
The sidecar file must have exactly the same file path and the `.s2meta` suffix.
|
|
||||||
|
|
||||||
`Video.mp4.s2meta`:
|
|
||||||
```json
|
|
||||||
{
|
|
||||||
"content": "This sidecar file will overwrite some metadata fields of Video.mp4",
|
|
||||||
"author": "Some author",
|
|
||||||
"duration": 12345,
|
|
||||||
"bitrate": 67890,
|
|
||||||
"some_arbitrary_field": [1,2,3]
|
|
||||||
}
|
|
||||||
```
|
|
||||||
|
|
||||||
```
|
|
||||||
sist2 scan ~/Documents -o ./docs.idx
|
|
||||||
sist2 index ./docs.idx
|
|
||||||
```
|
|
||||||
|
|
||||||
*NOTE*: It is technically possible to overwrite the `tag` value using sidecar files, however,
|
|
||||||
it is not currently possible to restore both manual tags and sidecar tags without user scripts
|
|
||||||
while reindexing.
|
|
||||||
|
|||||||
BIN
docs/embeddings-1.png
Normal file
|
After Width: | Height: | Size: 90 KiB |
BIN
docs/embeddings-2.png
Normal file
|
After Width: | Height: | Size: 996 KiB |
BIN
docs/ner.png
Normal file
|
After Width: | Height: | Size: 448 KiB |
@@ -1,6 +1,34 @@
|
|||||||
## User scripts
|
## User scripts
|
||||||
|
|
||||||
*This document is under construction, more in-depth guide coming soon*
|
User scripts are used to augment your sist2 index with additional metadata, neural network embeddings, tags etc.
|
||||||
|
|
||||||
|
|
||||||
|
Since version 3.2.0, user scripts are written in Python, and are ran against the sist2 index file. User scripts do not
|
||||||
|
need a connection to the search backend.
|
||||||
|
|
||||||
|
You can create a user script based on a template from the sist2-admin interface:
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
User scripts leverage the [sist2-python](https://github.com/simon987/sist2-python) library to interface with the
|
||||||
|
index file*. You can find sist2-python documentation and examples
|
||||||
|
here: [sist2-python.readthedocs.io](https://sist2-python.readthedocs.io/).
|
||||||
|
|
||||||
|
If you are not using the sist2-admin interface, you can run user scripts manually from the command line:
|
||||||
|
|
||||||
|
```
|
||||||
|
pip install git+https://github.com/simon987/sist2-python.git
|
||||||
|
|
||||||
|
python my_script.py /path/to/my_index.sist2
|
||||||
|
```
|
||||||
|
|
||||||
|
\* It is possible to manually update the index using raw SQL queries, but the database schema is not stable and
|
||||||
|
can change at any time; it is recommended to use the more stable sist2-python wrapper instead.
|
||||||
|
|
||||||
|
<hr>
|
||||||
|
|
||||||
|
<details>
|
||||||
|
<summary>Legacy user scripts (sist2 version < 3.2.0)</summary>
|
||||||
|
|
||||||
During the `index` step, you can use the `--script-file <script>` option to
|
During the `index` step, you can use the `--script-file <script>` option to
|
||||||
modify documents or add user tags. This option is mainly used to
|
modify documents or add user tags. This option is mainly used to
|
||||||
@@ -13,6 +41,7 @@ without programming experience at all if you're somewhat familiar with
|
|||||||
regex.
|
regex.
|
||||||
|
|
||||||
This is the base structure of the documents we're working with:
|
This is the base structure of the documents we're working with:
|
||||||
|
|
||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
"_id": "e171405c-fdb5-4feb-bb32-82637bc32084",
|
"_id": "e171405c-fdb5-4feb-bb32-82637bc32084",
|
||||||
@@ -35,6 +64,7 @@ This is the base structure of the documents we're working with:
|
|||||||
|
|
||||||
This script checks if the `genre` attribute exists, if it does
|
This script checks if the `genre` attribute exists, if it does
|
||||||
it adds the `genre.<genre>` tag.
|
it adds the `genre.<genre>` tag.
|
||||||
|
|
||||||
```Java
|
```Java
|
||||||
ArrayList tags = ctx._source.tag = new ArrayList();
|
ArrayList tags = ctx._source.tag = new ArrayList();
|
||||||
|
|
||||||
@@ -47,11 +77,12 @@ You can use `.` to create a hierarchical tag tree:
|
|||||||
|
|
||||||

|

|
||||||
|
|
||||||
|
|
||||||
To use regular expressions, you need to add this line in `/etc/elasticsearch/elasticsearch.yml`
|
To use regular expressions, you need to add this line in `/etc/elasticsearch/elasticsearch.yml`
|
||||||
|
|
||||||
```yaml
|
```yaml
|
||||||
script.painless.regex.enabled: true
|
script.painless.regex.enabled: true
|
||||||
```
|
```
|
||||||
|
|
||||||
Or, if you're using docker add `-e "script.painless.regex.enabled=true"`
|
Or, if you're using docker add `-e "script.painless.regex.enabled=true"`
|
||||||
|
|
||||||
**Tag color**
|
**Tag color**
|
||||||
@@ -62,6 +93,7 @@ hexadecimal color code (`#RRGGBBAA`) to the tag name.
|
|||||||
### Examples
|
### Examples
|
||||||
|
|
||||||
If `(20XX)` is in the file name, add the `year.<year>` tag:
|
If `(20XX)` is in the file name, add the `year.<year>` tag:
|
||||||
|
|
||||||
```Java
|
```Java
|
||||||
ArrayList tags = ctx._source.tag = new ArrayList();
|
ArrayList tags = ctx._source.tag = new ArrayList();
|
||||||
|
|
||||||
@@ -72,6 +104,7 @@ if (m.find()) {
|
|||||||
```
|
```
|
||||||
|
|
||||||
Use default *Calibre* folder structure to infer author.
|
Use default *Calibre* folder structure to infer author.
|
||||||
|
|
||||||
```Java
|
```Java
|
||||||
ArrayList tags = ctx._source.tag = new ArrayList();
|
ArrayList tags = ctx._source.tag = new ArrayList();
|
||||||
|
|
||||||
@@ -86,6 +119,7 @@ if (ctx._source.name.contains("-") && ctx._source.extension == "pdf") {
|
|||||||
|
|
||||||
If the file matches a specific pattern `AAAA-000 fName1 lName1, <fName2 lName2>...`, add the `actress.<actress>` and
|
If the file matches a specific pattern `AAAA-000 fName1 lName1, <fName2 lName2>...`, add the `actress.<actress>` and
|
||||||
`studio.<studio>` tag:
|
`studio.<studio>` tag:
|
||||||
|
|
||||||
```Java
|
```Java
|
||||||
ArrayList tags = ctx._source.tag = new ArrayList();
|
ArrayList tags = ctx._source.tag = new ArrayList();
|
||||||
|
|
||||||
@@ -102,16 +136,18 @@ if (m.find()) {
|
|||||||
```
|
```
|
||||||
|
|
||||||
Set the name of the last folder (`/path/to/<studio>/file.mp4`) to `studio.<studio>` tag
|
Set the name of the last folder (`/path/to/<studio>/file.mp4`) to `studio.<studio>` tag
|
||||||
|
|
||||||
```Java
|
```Java
|
||||||
ArrayList tags = ctx._source.tag = new ArrayList();
|
ArrayList tags = ctx._source.tag = new ArrayList();
|
||||||
|
|
||||||
if (ctx._source.path != "") {
|
if (ctx._source.path != "") {
|
||||||
String[] names = ctx._source.path.splitOnToken('/');
|
String[] names = ctx._source.path.splitOnToken('/');
|
||||||
tags.add("studio." + names[names.length-1]);
|
tags.add("studio." + names[names.length-1]);
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
Parse `EXIF:F Number` tag
|
Parse `EXIF:F Number` tag
|
||||||
|
|
||||||
```Java
|
```Java
|
||||||
if (ctx._source?.exif_fnumber != null) {
|
if (ctx._source?.exif_fnumber != null) {
|
||||||
String[] values = ctx._source.exif_fnumber.splitOnToken(' ');
|
String[] values = ctx._source.exif_fnumber.splitOnToken(' ');
|
||||||
@@ -124,6 +160,7 @@ if (ctx._source?.exif_fnumber != null) {
|
|||||||
```
|
```
|
||||||
|
|
||||||
Display year and months from `EXIF:DateTime` tag
|
Display year and months from `EXIF:DateTime` tag
|
||||||
|
|
||||||
```Java
|
```Java
|
||||||
if (ctx._source?.exif_datetime != null) {
|
if (ctx._source?.exif_datetime != null) {
|
||||||
SimpleDateFormat parser = new SimpleDateFormat("yyyy:MM:dd HH:mm:ss");
|
SimpleDateFormat parser = new SimpleDateFormat("yyyy:MM:dd HH:mm:ss");
|
||||||
@@ -140,3 +177,6 @@ if (ctx._source?.exif_datetime != null) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
```
|
```
|
||||||
|
|
||||||
|
</details>
|
||||||
|
|
||||||
|
|||||||
BIN
docs/sist2-admin-scripts.png
Normal file
|
After Width: | Height: | Size: 78 KiB |
BIN
docs/sist2.gif
Normal file
|
After Width: | Height: | Size: 3.7 MiB |
BIN
docs/sist2.png
|
Before Width: | Height: | Size: 1011 KiB |
|
Before Width: | Height: | Size: 180 KiB After Width: | Height: | Size: 169 KiB |
@@ -68,7 +68,7 @@
|
|||||||
},
|
},
|
||||||
"mtime": {
|
"mtime": {
|
||||||
"type": "date",
|
"type": "date",
|
||||||
"format": "epoch_millis"
|
"format": "epoch_second"
|
||||||
},
|
},
|
||||||
"size": {
|
"size": {
|
||||||
"type": "long"
|
"type": "long"
|
||||||
@@ -202,6 +202,46 @@
|
|||||||
},
|
},
|
||||||
"modified_by": {
|
"modified_by": {
|
||||||
"type": "text"
|
"type": "text"
|
||||||
|
},
|
||||||
|
"emb.384.*": {
|
||||||
|
"type": "dense_vector",
|
||||||
|
"dims": 384
|
||||||
|
},
|
||||||
|
"emb.idx_384.*": {
|
||||||
|
"type": "dense_vector",
|
||||||
|
"dims": 384,
|
||||||
|
"index": true,
|
||||||
|
"similarity": "cosine"
|
||||||
|
},
|
||||||
|
"emb.idx_512.clip": {
|
||||||
|
"type": "dense_vector",
|
||||||
|
"dims": 512,
|
||||||
|
"index": true,
|
||||||
|
"similarity": "cosine"
|
||||||
|
},
|
||||||
|
"emb.512.*": {
|
||||||
|
"type": "dense_vector",
|
||||||
|
"dims": 512
|
||||||
|
},
|
||||||
|
"emb.idx_768.*": {
|
||||||
|
"type": "dense_vector",
|
||||||
|
"dims": 768,
|
||||||
|
"index": true,
|
||||||
|
"similarity": "cosine"
|
||||||
|
},
|
||||||
|
"emb.768.*": {
|
||||||
|
"type": "dense_vector",
|
||||||
|
"dims": 768
|
||||||
|
},
|
||||||
|
"emb.idx_1024.*": {
|
||||||
|
"type": "dense_vector",
|
||||||
|
"dims": 1024,
|
||||||
|
"index": true,
|
||||||
|
"similarity": "cosine"
|
||||||
|
},
|
||||||
|
"emb.1024.*": {
|
||||||
|
"type": "dense_vector",
|
||||||
|
"dims": 1024
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,12 +2,24 @@
|
|||||||
|
|
||||||
VCPKG_ROOT="/vcpkg"
|
VCPKG_ROOT="/vcpkg"
|
||||||
|
|
||||||
git submodule update --init --recursive
|
(
|
||||||
|
cd sist2-vue/
|
||||||
|
npm install
|
||||||
|
npm run build
|
||||||
|
) &
|
||||||
|
|
||||||
|
(
|
||||||
|
cd sist2-admin/frontend/
|
||||||
|
npm install
|
||||||
|
npm run build
|
||||||
|
) &
|
||||||
|
|
||||||
|
wait
|
||||||
|
|
||||||
mkdir build
|
mkdir build
|
||||||
(
|
(
|
||||||
cd build
|
cd build
|
||||||
cmake -DSIST_PLATFORM=x64_linux -DSIST_DEBUG=off -DBUILD_TESTS=off -DCMAKE_TOOLCHAIN_FILE="${VCPKG_ROOT}/scripts/buildsystems/vcpkg.cmake" ..
|
cmake -DSIST_PLATFORM=x64_linux -DSIST_DEBUG_INFO=on -DSIST_DEBUG=off -DBUILD_TESTS=off -DCMAKE_TOOLCHAIN_FILE="${VCPKG_ROOT}/scripts/buildsystems/vcpkg.cmake" ..
|
||||||
make -j $(nproc)
|
make -j $(nproc)
|
||||||
strip sist2
|
strip sist2
|
||||||
./sist2 -v > VERSION
|
./sist2 -v > VERSION
|
||||||
@@ -17,7 +29,7 @@ mv build/sist2 sist2-x64-linux
|
|||||||
(
|
(
|
||||||
cd build
|
cd build
|
||||||
rm -rf CMakeFiles CMakeCache.txt
|
rm -rf CMakeFiles CMakeCache.txt
|
||||||
cmake -DSIST_PLATFORM=x64_linux -DSIST_DEBUG=on -DBUILD_TESTS=off -DCMAKE_TOOLCHAIN_FILE="${VCPKG_ROOT}/scripts/buildsystems/vcpkg.cmake" ..
|
cmake -DSIST_PLATFORM=x64_linux -DSIST_DEBUG_INFO=on -DSIST_DEBUG=on -DBUILD_TESTS=off -DCMAKE_TOOLCHAIN_FILE="${VCPKG_ROOT}/scripts/buildsystems/vcpkg.cmake" ..
|
||||||
make -j $(nproc)
|
make -j $(nproc)
|
||||||
)
|
)
|
||||||
mv build/sist2_debug sist2-x64-linux-debug
|
mv build/sist2_debug sist2-x64-linux-debug
|
||||||
@@ -4,10 +4,24 @@ VCPKG_ROOT="/vcpkg"
|
|||||||
|
|
||||||
git submodule update --init --recursive
|
git submodule update --init --recursive
|
||||||
|
|
||||||
|
(
|
||||||
|
cd sist2-vue/
|
||||||
|
npm install
|
||||||
|
npm run build
|
||||||
|
) &
|
||||||
|
|
||||||
|
(
|
||||||
|
cd sist2-admin/frontend/
|
||||||
|
npm install
|
||||||
|
npm run build
|
||||||
|
) &
|
||||||
|
|
||||||
|
wait
|
||||||
|
|
||||||
mkdir build
|
mkdir build
|
||||||
(
|
(
|
||||||
cd build
|
cd build
|
||||||
cmake -DSIST_PLATFORM=arm64_linux -DSIST_DEBUG=off -DBUILD_TESTS=off -DCMAKE_TOOLCHAIN_FILE="${VCPKG_ROOT}/scripts/buildsystems/vcpkg.cmake" .
|
cmake -DSIST_PLATFORM=arm64_linux -DSIST_DEBUG_INFO=on -DSIST_DEBUG=off -DBUILD_TESTS=off -DCMAKE_TOOLCHAIN_FILE="${VCPKG_ROOT}/scripts/buildsystems/vcpkg.cmake" ..
|
||||||
make -j $(nproc)
|
make -j $(nproc)
|
||||||
strip sist2
|
strip sist2
|
||||||
)
|
)
|
||||||
@@ -16,7 +30,7 @@ mv build/sist2 sist2-arm64-linux
|
|||||||
rm -rf CMakeFiles CMakeCache.txt
|
rm -rf CMakeFiles CMakeCache.txt
|
||||||
(
|
(
|
||||||
cd build
|
cd build
|
||||||
cmake -DSIST_PLATFORM=arm64_linux -DSIST_DEBUG=on -DBUILD_TESTS=off -DCMAKE_TOOLCHAIN_FILE="${VCPKG_ROOT}/scripts/buildsystems/vcpkg.cmake" .
|
cmake -DSIST_PLATFORM=arm64_linux -DSIST_DEBUG_INFO=on -DSIST_DEBUG=on -DBUILD_TESTS=off -DCMAKE_TOOLCHAIN_FILE="${VCPKG_ROOT}/scripts/buildsystems/vcpkg.cmake" ..
|
||||||
make -j $(nproc)
|
make -j $(nproc)
|
||||||
)
|
)
|
||||||
mv build/sist2_debug sist2-arm64-linux-debug
|
mv build/sist2_debug sist2-arm64-linux-debug
|
||||||
@@ -1,8 +1,16 @@
|
|||||||
|
|
||||||
try:
|
MAGIC_PATHS = [
|
||||||
with open("/usr/lib/file/magic.mgc", "rb") as f:
|
"/vcpkg/installed/x64-linux/share/libmagic/misc/magic.mgc",
|
||||||
|
"/work/vcpkg/installed/x64-linux/share/libmagic/misc/magic.mgc",
|
||||||
|
"/usr/lib/file/magic.mgc"
|
||||||
|
]
|
||||||
|
|
||||||
|
for path in MAGIC_PATHS:
|
||||||
|
try:
|
||||||
|
with open(path, "rb") as f:
|
||||||
data = f.read()
|
data = f.read()
|
||||||
except:
|
break
|
||||||
data = bytes([])
|
except:
|
||||||
|
continue
|
||||||
|
|
||||||
print("char magic_database_buffer[%d] = {%s};" % (len(data), ",".join(str(int(b)) for b in data)))
|
print("char magic_database_buffer[%d] = {%s};" % (len(data), ",".join(str(int(b)) for b in data)))
|
||||||
|
|||||||
@@ -450,4 +450,3 @@ image/x-sony-arw, arw
|
|||||||
image/x-sony-sr2, sr2
|
image/x-sony-sr2, sr2
|
||||||
image/x-sony-srf, srf
|
image/x-sony-srf, srf
|
||||||
image/x-epson-erf, erf
|
image/x-epson-erf, erf
|
||||||
sist2/sidecar, s2meta
|
|
||||||
|
@@ -3,6 +3,7 @@ import zlib
|
|||||||
mimes = {}
|
mimes = {}
|
||||||
noparse = set()
|
noparse = set()
|
||||||
ext_in_hash = set()
|
ext_in_hash = set()
|
||||||
|
mime_ids = {}
|
||||||
|
|
||||||
major_mime = {
|
major_mime = {
|
||||||
"sist2": 0,
|
"sist2": 0,
|
||||||
@@ -102,6 +103,9 @@ cnt = 1
|
|||||||
|
|
||||||
|
|
||||||
def mime_id(mime):
|
def mime_id(mime):
|
||||||
|
if mime in mime_ids:
|
||||||
|
return mime_ids[mime]
|
||||||
|
|
||||||
global cnt
|
global cnt
|
||||||
major = mime.split("/")[0]
|
major = mime.split("/")[0]
|
||||||
mime_id = str((major_mime[major] << 16) + cnt)
|
mime_id = str((major_mime[major] << 16) + cnt)
|
||||||
@@ -127,9 +131,7 @@ def mime_id(mime):
|
|||||||
elif mime == "application/x-empty":
|
elif mime == "application/x-empty":
|
||||||
cnt -= 1
|
cnt -= 1
|
||||||
return "1"
|
return "1"
|
||||||
elif mime == "sist2/sidecar":
|
mime_ids[mime] = mime_id
|
||||||
cnt -= 1
|
|
||||||
return "2"
|
|
||||||
return mime_id
|
return mime_id
|
||||||
|
|
||||||
|
|
||||||
@@ -197,4 +199,12 @@ with open("scripts/mime.csv") as f:
|
|||||||
print(f"case {crc(mime)}: return {clean(mime)};")
|
print(f"case {crc(mime)}: return {clean(mime)};")
|
||||||
|
|
||||||
print("default: return 0;}}")
|
print("default: return 0;}}")
|
||||||
|
|
||||||
|
# mime list
|
||||||
|
|
||||||
|
mime_list = ",".join(mime_id(x) for x in mimes.keys()) + ",0"
|
||||||
|
|
||||||
|
print(f"unsigned int mime_ids[] = {{{mime_list}}};")
|
||||||
|
print("unsigned int* get_mime_ids() { return mime_ids; }")
|
||||||
|
|
||||||
print("#endif")
|
print("#endif")
|
||||||
|
|||||||
84
scripts/sqlite_extension.c
Normal file
@@ -0,0 +1,84 @@
|
|||||||
|
#include <sqlite3ext.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
|
||||||
|
SQLITE_EXTENSION_INIT1
|
||||||
|
|
||||||
|
static int sep_rfind(const char *str) {
|
||||||
|
for (int i = (int) strlen(str); i >= 0; i--) {
|
||||||
|
if (str[i] == '/') {
|
||||||
|
return i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
void path_parent_func(sqlite3_context *ctx, int argc, sqlite3_value **argv) {
|
||||||
|
if (argc != 1 || sqlite3_value_type(argv[0]) != SQLITE_TEXT) {
|
||||||
|
sqlite3_result_error(ctx, "Invalid parameters", -1);
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *value = (const char *) sqlite3_value_text(argv[0]);
|
||||||
|
|
||||||
|
int stop = sep_rfind(value);
|
||||||
|
if (stop == -1) {
|
||||||
|
sqlite3_result_null(ctx);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
char parent[4096 * 3];
|
||||||
|
strncpy(parent, value, stop);
|
||||||
|
|
||||||
|
sqlite3_result_text(ctx, parent, stop, SQLITE_TRANSIENT);
|
||||||
|
}
|
||||||
|
|
||||||
|
void random_func(sqlite3_context *ctx, int argc, sqlite3_value **argv) {
|
||||||
|
if (argc != 1 || sqlite3_value_type(argv[0]) != SQLITE_INTEGER) {
|
||||||
|
sqlite3_result_error(ctx, "Invalid parameters", -1);
|
||||||
|
}
|
||||||
|
|
||||||
|
char state_buf[32] = {0,};
|
||||||
|
struct random_data buf;
|
||||||
|
int result;
|
||||||
|
|
||||||
|
long seed = sqlite3_value_int64(argv[0]);
|
||||||
|
|
||||||
|
initstate_r((int) seed, state_buf, sizeof(state_buf), &buf);
|
||||||
|
|
||||||
|
random_r(&buf, &result);
|
||||||
|
|
||||||
|
sqlite3_result_int(ctx, result);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int sqlite3_extension_init(
|
||||||
|
sqlite3 *db,
|
||||||
|
char **pzErrMsg,
|
||||||
|
const sqlite3_api_routines *pApi
|
||||||
|
) {
|
||||||
|
SQLITE_EXTENSION_INIT2(pApi);
|
||||||
|
|
||||||
|
|
||||||
|
sqlite3_create_function(
|
||||||
|
db,
|
||||||
|
"path_parent",
|
||||||
|
1,
|
||||||
|
SQLITE_UTF8,
|
||||||
|
NULL,
|
||||||
|
path_parent_func,
|
||||||
|
NULL,
|
||||||
|
NULL
|
||||||
|
);
|
||||||
|
|
||||||
|
sqlite3_create_function(
|
||||||
|
db,
|
||||||
|
"random_seeded",
|
||||||
|
1,
|
||||||
|
SQLITE_UTF8,
|
||||||
|
NULL,
|
||||||
|
random_func,
|
||||||
|
NULL,
|
||||||
|
NULL
|
||||||
|
);
|
||||||
|
|
||||||
|
return SQLITE_OK;
|
||||||
|
}
|
||||||
1
scripts/sqlite_extension_compile.sh
Executable file
@@ -0,0 +1 @@
|
|||||||
|
gcc -I/mnt/work/vcpkg/installed/x64-linux/include -g -fPIC -shared sqlite_extension.c -o sist2funcs.so
|
||||||
@@ -1,3 +1,3 @@
|
|||||||
docker run --rm -it --name "sist2-dev-es"\
|
docker run --rm -it --name "sist2-dev-es3"\
|
||||||
-p 9200:9200 -e "discovery.type=single-node" \
|
-p 9200:9200 -e "discovery.type=single-node" \
|
||||||
-e "ES_JAVA_OPTS=-Xms8g -Xmx8g" elasticsearch:7.17.9
|
-e "ES_JAVA_OPTS=-Xms8g -Xmx8g" elasticsearch:7.17.9
|
||||||
|
|||||||
@@ -1,3 +1,3 @@
|
|||||||
docker run --rm -it --name "sist2-dev-es"\
|
docker run --rm -it --name "sist2-dev-es3"\
|
||||||
-p 9200:9200 -p 9300:9300 -e "discovery.type=single-node" \
|
-p 9200:9200 -p 9300:9300 -e "discovery.type=single-node" \
|
||||||
-e "ES_JAVA_OPTS=-Xms8g -Xmx8g" elasticsearch:8.1.2
|
-e "ES_JAVA_OPTS=-Xms8g -Xmx8g" elasticsearch:8.7.0
|
||||||
|
|||||||
7
scripts/test_in_docker.sh
Normal file
@@ -0,0 +1,7 @@
|
|||||||
|
docker build . -t tmp
|
||||||
|
|
||||||
|
docker run --rm -it\
|
||||||
|
-v $(pwd):/host \
|
||||||
|
tmp \
|
||||||
|
scan --ocr-lang eng --ocr-ebooks -t6 --incremental --very-verbose \
|
||||||
|
-o /host/docker.sist2 /host/third-party/libscan/libscan-test-files/test_files/
|
||||||
2716
sist2-admin/frontend/package-lock.json
generated
@@ -8,7 +8,7 @@
|
|||||||
"watch": "vue-cli-service build --watch"
|
"watch": "vue-cli-service build --watch"
|
||||||
},
|
},
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"axios": "^0.27.2",
|
"axios": "^1.6.0",
|
||||||
"bootstrap-vue": "^2.21.2",
|
"bootstrap-vue": "^2.21.2",
|
||||||
"core-js": "^3.6.5",
|
"core-js": "^3.6.5",
|
||||||
"moment": "^2.29.3",
|
"moment": "^2.29.3",
|
||||||
|
|||||||
@@ -4,9 +4,10 @@
|
|||||||
<b-container class="pt-4">
|
<b-container class="pt-4">
|
||||||
<b-alert show dismissible variant="info">
|
<b-alert show dismissible variant="info">
|
||||||
This is a beta version of sist2-admin. Please submit bug reports, usability issues and feature requests
|
This is a beta version of sist2-admin. Please submit bug reports, usability issues and feature requests
|
||||||
to the <a href="https://github.com/simon987/sist2/issues/new/choose" target="_blank">issue tracker on Github</a>. Thank you!
|
to the <a href="https://github.com/sist2app/sist2/issues/new/choose" target="_blank">issue tracker on
|
||||||
|
Github</a>. Thank you!
|
||||||
</b-alert>
|
</b-alert>
|
||||||
<router-view/>
|
<router-view v-if="$store.state.sist2AdminInfo"/>
|
||||||
</b-container>
|
</b-container>
|
||||||
</div>
|
</div>
|
||||||
</template>
|
</template>
|
||||||
@@ -31,7 +32,11 @@ export default {
|
|||||||
},
|
},
|
||||||
methods: {
|
methods: {
|
||||||
connectNotifications() {
|
connectNotifications() {
|
||||||
|
if (window.location.protocol === "https:") {
|
||||||
|
this.socket = new WebSocket(`wss://${window.location.host}/notifications`);
|
||||||
|
} else {
|
||||||
this.socket = new WebSocket(`ws://${window.location.host}/notifications`);
|
this.socket = new WebSocket(`ws://${window.location.host}/notifications`);
|
||||||
|
}
|
||||||
this.socket.onopen = () => {
|
this.socket.onopen = () => {
|
||||||
this.socket.send("Hello from client");
|
this.socket.send("Hello from client");
|
||||||
}
|
}
|
||||||
@@ -66,10 +71,12 @@ html, body {
|
|||||||
|
|
||||||
.info-icon {
|
.info-icon {
|
||||||
width: 1rem;
|
width: 1rem;
|
||||||
|
min-width: 1rem;
|
||||||
margin-right: 0.2rem;
|
margin-right: 0.2rem;
|
||||||
cursor: pointer;
|
cursor: pointer;
|
||||||
line-height: 1rem;
|
line-height: 1rem;
|
||||||
height: 1rem;
|
height: 1rem;
|
||||||
|
min-height: 1rem;
|
||||||
background-image: url(data:image/svg+xml;base64,PHN2ZyB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciIHhtbG5zOnhsaW5rPSJodHRwOi8vd3d3LnczLm9yZy8xOTk5L3hsaW5rIiB4PSIwcHgiIHk9IjBweCIKICAgICB2aWV3Qm94PSIwIDAgNDI2LjY2NyA0MjYuNjY3IiBzdHlsZT0iZW5hYmxlLWJhY2tncm91bmQ6bmV3IDAgMCA0MjYuNjY3IDQyNi42Njc7IiBmaWxsPSIjZmZmIj4KPGc+CiAgICA8Zz4KICAgICAgICA8Zz4KICAgICAgICAgICAgPHJlY3QgeD0iMTkyIiB5PSIxOTIiIHdpZHRoPSI0Mi42NjciIGhlaWdodD0iMTI4Ii8+CiAgICAgICAgICAgIDxwYXRoIGQ9Ik0yMTMuMzMzLDBDOTUuNDY3LDAsMCw5NS40NjcsMCwyMTMuMzMzczk1LjQ2NywyMTMuMzMzLDIxMy4zMzMsMjEzLjMzM1M0MjYuNjY3LDMzMS4yLDQyNi42NjcsMjEzLjMzMwogICAgICAgICAgICAgICAgUzMzMS4yLDAsMjEzLjMzMywweiBNMjEzLjMzMywzODRjLTk0LjA4LDAtMTcwLjY2Ny03Ni41ODctMTcwLjY2Ny0xNzAuNjY3UzExOS4yNTMsNDIuNjY3LDIxMy4zMzMsNDIuNjY3CiAgICAgICAgICAgICAgICBTMzg0LDExOS4yNTMsMzg0LDIxMy4zMzNTMzA3LjQxMywzODQsMjEzLjMzMywzODR6Ii8+CiAgICAgICAgICAgIDxyZWN0IHg9IjE5MiIgeT0iMTA2LjY2NyIgd2lkdGg9IjQyLjY2NyIgaGVpZ2h0PSI0Mi42NjciLz4KICAgICAgICA8L2c+CiAgICA8L2c+CjwvZz4KPC9zdmc+Cg==);
|
background-image: url(data:image/svg+xml;base64,PHN2ZyB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciIHhtbG5zOnhsaW5rPSJodHRwOi8vd3d3LnczLm9yZy8xOTk5L3hsaW5rIiB4PSIwcHgiIHk9IjBweCIKICAgICB2aWV3Qm94PSIwIDAgNDI2LjY2NyA0MjYuNjY3IiBzdHlsZT0iZW5hYmxlLWJhY2tncm91bmQ6bmV3IDAgMCA0MjYuNjY3IDQyNi42Njc7IiBmaWxsPSIjZmZmIj4KPGc+CiAgICA8Zz4KICAgICAgICA8Zz4KICAgICAgICAgICAgPHJlY3QgeD0iMTkyIiB5PSIxOTIiIHdpZHRoPSI0Mi42NjciIGhlaWdodD0iMTI4Ii8+CiAgICAgICAgICAgIDxwYXRoIGQ9Ik0yMTMuMzMzLDBDOTUuNDY3LDAsMCw5NS40NjcsMCwyMTMuMzMzczk1LjQ2NywyMTMuMzMzLDIxMy4zMzMsMjEzLjMzM1M0MjYuNjY3LDMzMS4yLDQyNi42NjcsMjEzLjMzMwogICAgICAgICAgICAgICAgUzMzMS4yLDAsMjEzLjMzMywweiBNMjEzLjMzMywzODRjLTk0LjA4LDAtMTcwLjY2Ny03Ni41ODctMTcwLjY2Ny0xNzAuNjY3UzExOS4yNTMsNDIuNjY3LDIxMy4zMzMsNDIuNjY3CiAgICAgICAgICAgICAgICBTMzg0LDExOS4yNTMsMzg0LDIxMy4zMzNTMzA3LjQxMywzODQsMjEzLjMzMywzODR6Ii8+CiAgICAgICAgICAgIDxyZWN0IHg9IjE5MiIgeT0iMTA2LjY2NyIgd2lkdGg9IjQyLjY2NyIgaGVpZ2h0PSI0Mi42NjciLz4KICAgICAgICA8L2c+CiAgICA8L2c+CjwvZz4KPC9zdmc+Cg==);
|
||||||
filter: brightness(45%);
|
filter: brightness(45%);
|
||||||
display: block;
|
display: block;
|
||||||
|
|||||||
@@ -7,15 +7,15 @@ class Sist2AdminApi {
|
|||||||
}
|
}
|
||||||
|
|
||||||
getJobs() {
|
getJobs() {
|
||||||
return axios.get(`${this.baseUrl}/api/job/`);
|
return axios.get(`${this.baseUrl}/api/job`);
|
||||||
}
|
}
|
||||||
|
|
||||||
getFrontends() {
|
getFrontends() {
|
||||||
return axios.get(`${this.baseUrl}/api/frontend/`);
|
return axios.get(`${this.baseUrl}/api/frontend`);
|
||||||
}
|
}
|
||||||
|
|
||||||
getTasks() {
|
getTasks() {
|
||||||
return axios.get(`${this.baseUrl}/api/task/`);
|
return axios.get(`${this.baseUrl}/api/task`);
|
||||||
}
|
}
|
||||||
|
|
||||||
killTask(taskId) {
|
killTask(taskId) {
|
||||||
@@ -33,9 +33,26 @@ class Sist2AdminApi {
|
|||||||
return axios.get(`${this.baseUrl}/api/job/${name}`);
|
return axios.get(`${this.baseUrl}/api/job/${name}`);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
getSearchBackend(name) {
|
||||||
* @param {string} name
|
return axios.get(`${this.baseUrl}/api/search_backend/${name}`);
|
||||||
*/
|
}
|
||||||
|
|
||||||
|
updateSearchBackend(name, data) {
|
||||||
|
return axios.put(`${this.baseUrl}/api/search_backend/${name}`, data);
|
||||||
|
}
|
||||||
|
|
||||||
|
getSearchBackends() {
|
||||||
|
return axios.get(`${this.baseUrl}/api/search_backend`);
|
||||||
|
}
|
||||||
|
|
||||||
|
deleteBackend(name) {
|
||||||
|
return axios.delete(`${this.baseUrl}/api/search_backend/${name}`)
|
||||||
|
}
|
||||||
|
|
||||||
|
createBackend(name) {
|
||||||
|
return axios.post(`${this.baseUrl}/api/search_backend/${name}`);
|
||||||
|
}
|
||||||
|
|
||||||
getFrontend(name) {
|
getFrontend(name) {
|
||||||
return axios.get(`${this.baseUrl}/api/frontend/${name}`);
|
return axios.get(`${this.baseUrl}/api/frontend/${name}`);
|
||||||
}
|
}
|
||||||
@@ -72,9 +89,12 @@ class Sist2AdminApi {
|
|||||||
|
|
||||||
/**
|
/**
|
||||||
* @param {string} name
|
* @param {string} name
|
||||||
|
* @param {bool} full
|
||||||
*/
|
*/
|
||||||
runJob(name) {
|
runJob(name, full) {
|
||||||
return axios.get(`${this.baseUrl}/api/job/${name}/run`);
|
return axios.get(`${this.baseUrl}/api/job/${name}/run`, {
|
||||||
|
params: {full}
|
||||||
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -110,7 +130,49 @@ class Sist2AdminApi {
|
|||||||
}
|
}
|
||||||
|
|
||||||
getSist2AdminInfo() {
|
getSist2AdminInfo() {
|
||||||
return axios.get(`${this.baseUrl}/api/`);
|
return axios.get(`${this.baseUrl}/api`);
|
||||||
|
}
|
||||||
|
|
||||||
|
getLogsToDelete(jobName, n) {
|
||||||
|
return axios.get(`${this.baseUrl}/api/job/${jobName}/logs_to_delete`, {
|
||||||
|
params: {n: n}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
deleteTaskLogs(taskId) {
|
||||||
|
return axios.post(`${this.baseUrl}/api/task/${taskId}/delete_logs`);
|
||||||
|
}
|
||||||
|
|
||||||
|
getUserScripts() {
|
||||||
|
return axios.get(`${this.baseUrl}/api/user_script`);
|
||||||
|
}
|
||||||
|
|
||||||
|
getUserScript(name) {
|
||||||
|
return axios.get(`${this.baseUrl}/api/user_script/${name}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
createUserScript(name, template) {
|
||||||
|
return axios.post(`${this.baseUrl}/api/user_script/${name}`, null, {
|
||||||
|
params: {
|
||||||
|
template: template
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
updateUserScript(name, data) {
|
||||||
|
return axios.put(`${this.baseUrl}/api/user_script/${name}`, data);
|
||||||
|
}
|
||||||
|
|
||||||
|
deleteUserScript(name) {
|
||||||
|
return axios.delete(`${this.baseUrl}/api/user_script/${name}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
testUserScript(name, job) {
|
||||||
|
return axios.get(`${this.baseUrl}/api/user_script/${name}/run`, {
|
||||||
|
params: {
|
||||||
|
job: job
|
||||||
|
}
|
||||||
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,64 +0,0 @@
|
|||||||
<template>
|
|
||||||
<div>
|
|
||||||
<label>{{ $t("indexOptions.threads") }}</label>
|
|
||||||
<b-form-input v-model="options.threads" type="number" min="1" @change="update()"></b-form-input>
|
|
||||||
|
|
||||||
<label>{{ $t("webOptions.esUrl") }}</label>
|
|
||||||
<b-alert :variant="esTestOk ? 'success' : 'danger'" :show="showEsTestAlert" class="mt-1">
|
|
||||||
{{ esTestMessage }}
|
|
||||||
</b-alert>
|
|
||||||
<b-input-group>
|
|
||||||
<b-form-input v-model="options.es_url" @change="update()"></b-form-input>
|
|
||||||
<b-input-group-append>
|
|
||||||
<b-button variant="outline-primary" @click="testEs()">{{ $t("test") }}</b-button>
|
|
||||||
</b-input-group-append>
|
|
||||||
</b-input-group>
|
|
||||||
|
|
||||||
<label>{{ $t("indexOptions.esIndex") }}</label>
|
|
||||||
<b-form-input v-model="options.es_index" @change="update()"></b-form-input>
|
|
||||||
|
|
||||||
<br>
|
|
||||||
<b-form-checkbox v-model="options.es_insecure_ssl" :disabled="!options.es_url.startsWith('https')" @change="update()">
|
|
||||||
{{ $t("webOptions.esInsecure") }}
|
|
||||||
</b-form-checkbox>
|
|
||||||
|
|
||||||
<label>{{ $t("indexOptions.batchSize") }}</label>
|
|
||||||
<b-form-input v-model="options.batch_size" type="number" min="1" @change="update()"></b-form-input>
|
|
||||||
|
|
||||||
<label>{{ $t("indexOptions.script") }}</label>
|
|
||||||
<b-form-textarea v-model="options.script" rows="6" @change="update()"></b-form-textarea>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
</template>
|
|
||||||
|
|
||||||
<script>
|
|
||||||
import sist2AdminApi from "@/Sist2AdminApi";
|
|
||||||
|
|
||||||
export default {
|
|
||||||
name: "IndexOptions",
|
|
||||||
props: ["options"],
|
|
||||||
data() {
|
|
||||||
return {
|
|
||||||
showEsTestAlert: false,
|
|
||||||
esTestOk: false,
|
|
||||||
esTestMessage: "",
|
|
||||||
}
|
|
||||||
},
|
|
||||||
methods: {
|
|
||||||
update() {
|
|
||||||
this.$emit("change", this.options);
|
|
||||||
},
|
|
||||||
testEs() {
|
|
||||||
sist2AdminApi.pingEs(this.options.es_url, this.options.es_insecure_ssl).then((resp) => {
|
|
||||||
this.showEsTestAlert = true;
|
|
||||||
this.esTestOk = resp.data.ok;
|
|
||||||
this.esTestMessage = resp.data.message;
|
|
||||||
});
|
|
||||||
}
|
|
||||||
},
|
|
||||||
}
|
|
||||||
</script>
|
|
||||||
|
|
||||||
<style scoped>
|
|
||||||
|
|
||||||
</style>
|
|
||||||
@@ -9,13 +9,18 @@
|
|||||||
@input="frontend.jobs = $event; $emit('input')"
|
@input="frontend.jobs = $event; $emit('input')"
|
||||||
>
|
>
|
||||||
<div v-for="job in jobs" :key="job.name">
|
<div v-for="job in jobs" :key="job.name">
|
||||||
<b-form-checkbox :disabled="job.status !== 'indexed'" :value="job.name">[{{ job.name }}]</b-form-checkbox>
|
<b-form-checkbox :disabled="job.status !== 'indexed'"
|
||||||
|
:value="job.name">
|
||||||
|
<template #default><span
|
||||||
|
:title="job.status !== 'indexed' ? $t('jobOptions.notIndexed') : ''"
|
||||||
|
>[{{ job.name }}]</span></template>
|
||||||
|
</b-form-checkbox>
|
||||||
<br/>
|
<br/>
|
||||||
</div>
|
</div>
|
||||||
</b-form-checkbox-group>
|
</b-form-checkbox-group>
|
||||||
<div v-else>
|
<div v-else>
|
||||||
<span class="text-muted">{{ $t('jobOptions.noJobAvailable') }}</span>
|
<span class="text-muted">{{ $t('jobOptions.noJobAvailable') }}</span>
|
||||||
<router-link to="/">{{$t("create")}}</router-link>
|
<router-link to="/">{{ $t("create") }}</router-link>
|
||||||
</div>
|
</div>
|
||||||
</b-form-group>
|
</b-form-group>
|
||||||
</div>
|
</div>
|
||||||
@@ -29,13 +34,20 @@ export default {
|
|||||||
props: ["frontend"],
|
props: ["frontend"],
|
||||||
mounted() {
|
mounted() {
|
||||||
Sist2AdminApi.getJobs().then(resp => {
|
Sist2AdminApi.getJobs().then(resp => {
|
||||||
this.jobs = resp.data;
|
this._jobs = resp.data;
|
||||||
this.loading = false;
|
this.loading = false;
|
||||||
});
|
});
|
||||||
},
|
},
|
||||||
|
computed: {
|
||||||
|
jobs() {
|
||||||
|
return this._jobs
|
||||||
|
.filter(job => job.index_options.search_backend === this.frontend.web_options.search_backend)
|
||||||
|
}
|
||||||
|
},
|
||||||
data() {
|
data() {
|
||||||
return {
|
return {
|
||||||
loading: true,
|
loading: true,
|
||||||
|
_jobs: null
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -44,8 +44,7 @@ export default {
|
|||||||
return "";
|
return "";
|
||||||
}
|
}
|
||||||
|
|
||||||
const date = Date.parse(dateString);
|
return moment.utc(dateString).local().fromNow();
|
||||||
return moment(date).fromNow();
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -9,18 +9,31 @@
|
|||||||
</b-form-checkbox>
|
</b-form-checkbox>
|
||||||
|
|
||||||
<label>{{ $t("jobOptions.cron") }}</label>
|
<label>{{ $t("jobOptions.cron") }}</label>
|
||||||
<b-form-input class="text-monospace" :state="cronValid" v-model="job.cron_expression" :disabled="!job.schedule_enabled" @change="update()"></b-form-input>
|
<b-form-input class="text-monospace" :state="cronValid" v-model="job.cron_expression"
|
||||||
|
:disabled="!job.schedule_enabled" @change="update()"></b-form-input>
|
||||||
|
|
||||||
|
<label>{{ $t("jobOptions.keepNLogs") }}</label>
|
||||||
|
<b-input-group>
|
||||||
|
<b-form-input type="number" v-model="job.keep_last_n_logs" @change="update()"></b-form-input>
|
||||||
|
<b-input-group-append>
|
||||||
|
<b-button variant="danger" @click="onDeleteNowClick()">{{ $t("jobOptions.deleteNow") }}</b-button>
|
||||||
|
</b-input-group-append>
|
||||||
|
</b-input-group>
|
||||||
|
|
||||||
</div>
|
</div>
|
||||||
</template>
|
</template>
|
||||||
|
|
||||||
<script>
|
<script>
|
||||||
|
|
||||||
|
import Sist2AdminApi from "@/Sist2AdminApi";
|
||||||
|
|
||||||
export default {
|
export default {
|
||||||
name: "JobOptions",
|
name: "JobOptions",
|
||||||
props: ["job"],
|
props: ["job"],
|
||||||
data() {
|
data() {
|
||||||
return {
|
return {
|
||||||
cronValid: undefined
|
cronValid: undefined,
|
||||||
|
logsToDelete: null
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
computed: {
|
computed: {
|
||||||
@@ -52,6 +65,30 @@ export default {
|
|||||||
this.$emit("change", this.job);
|
this.$emit("change", this.job);
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
onDeleteNowClick() {
|
||||||
|
Sist2AdminApi.getLogsToDelete(this.job.name, this.job.keep_last_n_logs).then(resp => {
|
||||||
|
const toDelete = resp.data;
|
||||||
|
const message = `Delete ${toDelete.length} log files?`;
|
||||||
|
|
||||||
|
this.$bvModal.msgBoxConfirm(message, {
|
||||||
|
title: this.$t("confirmation"),
|
||||||
|
size: "sm",
|
||||||
|
buttonSize: "sm",
|
||||||
|
okVariant: "danger",
|
||||||
|
okTitle: this.$t("delete"),
|
||||||
|
cancelTitle: this.$t("cancel"),
|
||||||
|
footerClass: "p-2",
|
||||||
|
hideHeaderClose: false,
|
||||||
|
centered: true
|
||||||
|
}).then(value => {
|
||||||
|
if (value) {
|
||||||
|
toDelete.forEach(row => {
|
||||||
|
Sist2AdminApi.deleteTaskLogs(row["id"]);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
});
|
||||||
|
})
|
||||||
|
}
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
</script>
|
</script>
|
||||||
34
sist2-admin/frontend/src/components/JobSelect.vue
Normal file
@@ -0,0 +1,34 @@
|
|||||||
|
<template>
|
||||||
|
<b-progress v-if="loading" striped animated value="100"></b-progress>
|
||||||
|
<span v-else-if="jobs.length === 0"></span>
|
||||||
|
<b-form-select v-else :options="jobs" text-field="name" value-field="name"
|
||||||
|
@change="$emit('change', $event)" :value="$t('selectJob')"></b-form-select>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
import Sist2AdminApi from "@/Sist2AdminApi";
|
||||||
|
|
||||||
|
export default {
|
||||||
|
name: "JobSelect",
|
||||||
|
mounted() {
|
||||||
|
Sist2AdminApi.getJobs().then(resp => {
|
||||||
|
this._jobs = resp.data;
|
||||||
|
this.loading = false;
|
||||||
|
});
|
||||||
|
},
|
||||||
|
computed: {
|
||||||
|
jobs() {
|
||||||
|
return [
|
||||||
|
{name: this.$t("selectJob"), disabled: true},
|
||||||
|
...this._jobs.filter(job => job.index_path)
|
||||||
|
]
|
||||||
|
}
|
||||||
|
},
|
||||||
|
data() {
|
||||||
|
return {
|
||||||
|
loading: true,
|
||||||
|
_jobs: null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</script>
|
||||||
@@ -7,7 +7,7 @@
|
|||||||
<b-form-input type="number" min="1" v-model="options.threads" @change="update()"></b-form-input>
|
<b-form-input type="number" min="1" v-model="options.threads" @change="update()"></b-form-input>
|
||||||
|
|
||||||
<label>{{ $t("scanOptions.thumbnailQuality") }}</label>
|
<label>{{ $t("scanOptions.thumbnailQuality") }}</label>
|
||||||
<b-form-input type="number" min="1" max="31" v-model="options.thumbnail_quality" @change="update()"></b-form-input>
|
<b-form-input type="number" min="0" max="100" v-model="options.thumbnail_quality" @change="update()"></b-form-input>
|
||||||
|
|
||||||
<label>{{ $t("scanOptions.thumbnailCount") }}</label>
|
<label>{{ $t("scanOptions.thumbnailCount") }}</label>
|
||||||
<b-form-input type="number" min="0" max="1000" v-model="options.thumbnail_count" @change="update()"></b-form-input>
|
<b-form-input type="number" min="0" max="1000" v-model="options.thumbnail_count" @change="update()"></b-form-input>
|
||||||
@@ -95,6 +95,7 @@ export default {
|
|||||||
methods: {
|
methods: {
|
||||||
onOcrLangChange() {
|
onOcrLangChange() {
|
||||||
this.options.ocr_lang = this.selectedOcrLangs.join("+");
|
this.options.ocr_lang = this.selectedOcrLangs.join("+");
|
||||||
|
this.update();
|
||||||
},
|
},
|
||||||
update() {
|
update() {
|
||||||
this.disableOcrLang = this.options.ocr_images === false && this.options.ocr_ebooks === false;
|
this.disableOcrLang = this.options.ocr_images === false && this.options.ocr_ebooks === false;
|
||||||
|
|||||||
@@ -0,0 +1,24 @@
|
|||||||
|
<template>
|
||||||
|
<b-list-group-item action :to="`/searchBackend/${backend.name}`">
|
||||||
|
|
||||||
|
<div class="d-flex w-100 justify-content-between">
|
||||||
|
<h5 class="mb-1">
|
||||||
|
{{ backend.name }}
|
||||||
|
</h5>
|
||||||
|
|
||||||
|
<div>
|
||||||
|
<b-badge v-if="backend.backend_type === 'sqlite'" variant="info">SQLite</b-badge>
|
||||||
|
<b-badge v-else variant="info">Elasticsearch</b-badge>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
</b-list-group-item>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
|
||||||
|
export default {
|
||||||
|
name: "SearchBackendListItem",
|
||||||
|
props: ["backend"],
|
||||||
|
}
|
||||||
|
</script>
|
||||||
37
sist2-admin/frontend/src/components/SearchBackendSelect.vue
Normal file
@@ -0,0 +1,37 @@
|
|||||||
|
<template>
|
||||||
|
<b-progress v-if="loading" striped animated value="100"></b-progress>
|
||||||
|
<div v-else>
|
||||||
|
<label>{{$t("backendOptions.searchBackend")}}</label>
|
||||||
|
<b-select :options="options" :value="value" @change="$emit('change', $event)"></b-select>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
import Sist2AdminApi from "@/Sist2AdminApi";
|
||||||
|
|
||||||
|
export default {
|
||||||
|
name: "SearchBackendSelect",
|
||||||
|
props: ["value"],
|
||||||
|
data() {
|
||||||
|
return {
|
||||||
|
loading: true,
|
||||||
|
backends: null,
|
||||||
|
}
|
||||||
|
},
|
||||||
|
computed: {
|
||||||
|
options() {
|
||||||
|
return this.backends.map(backend => backend.name)
|
||||||
|
}
|
||||||
|
},
|
||||||
|
mounted() {
|
||||||
|
Sist2AdminApi.getSearchBackends().then(resp => {
|
||||||
|
this.loading = false;
|
||||||
|
this.backends = resp.data
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<style scoped>
|
||||||
|
|
||||||
|
</style>
|
||||||
18
sist2-admin/frontend/src/components/UserScriptListItem.vue
Normal file
@@ -0,0 +1,18 @@
|
|||||||
|
<template>
|
||||||
|
<b-list-group-item action :to="`/userScript/${script.name}`">
|
||||||
|
|
||||||
|
<div class="d-flex w-100 justify-content-between">
|
||||||
|
<h5 class="mb-1">
|
||||||
|
{{ script.name }}
|
||||||
|
</h5>
|
||||||
|
</div>
|
||||||
|
</b-list-group-item>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
|
||||||
|
export default {
|
||||||
|
name: "UserScriptListItem",
|
||||||
|
props: ["script"],
|
||||||
|
}
|
||||||
|
</script>
|
||||||
88
sist2-admin/frontend/src/components/UserScriptPicker.vue
Normal file
@@ -0,0 +1,88 @@
|
|||||||
|
<template>
|
||||||
|
<b-progress v-if="loading" striped animated value="100"></b-progress>
|
||||||
|
|
||||||
|
<b-row v-else>
|
||||||
|
<b-col cols="6">
|
||||||
|
<h5>Selected scripts</h5>
|
||||||
|
<b-list-group>
|
||||||
|
<b-list-group-item v-for="script in selectedScripts" :key="script"
|
||||||
|
button
|
||||||
|
@click="onRemoveScript(script)"
|
||||||
|
class="d-flex justify-content-between align-items-center">
|
||||||
|
{{ script }}
|
||||||
|
<b-button-group>
|
||||||
|
<b-button variant="light" @click.stop="moveUpScript(script)">↑</b-button>
|
||||||
|
<b-button variant="light" @click.stop="moveDownScript(script)">↓</b-button>
|
||||||
|
</b-button-group>
|
||||||
|
</b-list-group-item>
|
||||||
|
</b-list-group>
|
||||||
|
</b-col>
|
||||||
|
<b-col cols="6">
|
||||||
|
<h5>Available scripts</h5>
|
||||||
|
<b-list-group>
|
||||||
|
<b-list-group-item v-for="script in availableScripts" :key="script" button
|
||||||
|
@click="onSelectScript(script)">
|
||||||
|
{{ script }}
|
||||||
|
</b-list-group-item>
|
||||||
|
</b-list-group>
|
||||||
|
</b-col>
|
||||||
|
</b-row>
|
||||||
|
|
||||||
|
<!-- <b-checkbox-group v-else :options="scripts" stacked :checked="selectedScripts"-->
|
||||||
|
<!-- @input="$emit('change', $event)"></b-checkbox-group>-->
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
import Sist2AdminApi from "@/Sist2AdminApi";
|
||||||
|
|
||||||
|
export default {
|
||||||
|
name: "UserScriptPicker",
|
||||||
|
props: ["selectedScripts"],
|
||||||
|
data() {
|
||||||
|
return {
|
||||||
|
loading: true,
|
||||||
|
scripts: []
|
||||||
|
}
|
||||||
|
},
|
||||||
|
computed: {
|
||||||
|
availableScripts() {
|
||||||
|
return this.scripts.filter(script => !this.selectedScripts.includes(script))
|
||||||
|
}
|
||||||
|
},
|
||||||
|
mounted() {
|
||||||
|
Sist2AdminApi.getUserScripts().then(resp => {
|
||||||
|
this.scripts = resp.data.map(script => script.name);
|
||||||
|
this.loading = false;
|
||||||
|
});
|
||||||
|
},
|
||||||
|
methods: {
|
||||||
|
onSelectScript(name) {
|
||||||
|
this.selectedScripts.push(name);
|
||||||
|
this.$emit("change", this.selectedScripts)
|
||||||
|
},
|
||||||
|
onRemoveScript(name) {
|
||||||
|
this.selectedScripts.splice(this.selectedScripts.indexOf(name), 1);
|
||||||
|
this.$emit("change", this.selectedScripts);
|
||||||
|
},
|
||||||
|
moveUpScript(name) {
|
||||||
|
const index = this.selectedScripts.indexOf(name);
|
||||||
|
if (index > 0) {
|
||||||
|
this.selectedScripts.splice(index, 1);
|
||||||
|
this.selectedScripts.splice(index - 1, 0, name);
|
||||||
|
}
|
||||||
|
this.$emit("change", this.selectedScripts);
|
||||||
|
},
|
||||||
|
moveDownScript(name) {
|
||||||
|
const index = this.selectedScripts.indexOf(name);
|
||||||
|
if (index < this.selectedScripts.length - 1) {
|
||||||
|
this.selectedScripts.splice(index, 1);
|
||||||
|
this.selectedScripts.splice(index + 1, 0, name);
|
||||||
|
}
|
||||||
|
this.$emit("change", this.selectedScripts);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<style scoped>
|
||||||
|
</style>
|
||||||
@@ -1,26 +1,10 @@
|
|||||||
<template>
|
<template>
|
||||||
<div>
|
<div>
|
||||||
<label>{{ $t("webOptions.esUrl") }}</label>
|
<h4>{{ $t("webOptions.title") }}</h4>
|
||||||
<b-alert :variant="esTestOk ? 'success' : 'danger'" :show="showEsTestAlert" class="mt-1">
|
<b-card>
|
||||||
{{ esTestMessage }}
|
|
||||||
</b-alert>
|
|
||||||
|
|
||||||
<b-input-group>
|
|
||||||
<b-form-input v-model="options.es_url" @change="update()"></b-form-input>
|
|
||||||
<b-input-group-append>
|
|
||||||
<b-button variant="outline-primary" @click="testEs()">{{ $t("test") }}</b-button>
|
|
||||||
</b-input-group-append>
|
|
||||||
</b-input-group>
|
|
||||||
|
|
||||||
<b-form-checkbox v-model="options.es_insecure_ssl" :disabled="!this.options.es_url.startsWith('https')" @change="update()">
|
|
||||||
{{ $t("webOptions.esInsecure") }}
|
|
||||||
</b-form-checkbox>
|
|
||||||
|
|
||||||
<label>{{ $t("webOptions.esIndex") }}</label>
|
|
||||||
<b-form-input v-model="options.es_index" @change="update()"></b-form-input>
|
|
||||||
|
|
||||||
<label>{{ $t("webOptions.lang") }}</label>
|
<label>{{ $t("webOptions.lang") }}</label>
|
||||||
<b-form-select v-model="options.lang" :options="['en', 'fr', 'zh-CN']" @change="update()"></b-form-select>
|
<b-form-select v-model="options.lang" :options="['en', 'fr', 'zh-CN', 'pl', 'de']"
|
||||||
|
@change="update()"></b-form-select>
|
||||||
|
|
||||||
<label>{{ $t("webOptions.bind") }}</label>
|
<label>{{ $t("webOptions.bind") }}</label>
|
||||||
<b-form-input v-model="options.bind" @change="update()"></b-form-input>
|
<b-form-input v-model="options.bind" @change="update()"></b-form-input>
|
||||||
@@ -32,10 +16,16 @@
|
|||||||
<b-form-input v-model="options.auth" @change="update()"></b-form-input>
|
<b-form-input v-model="options.auth" @change="update()"></b-form-input>
|
||||||
|
|
||||||
<label>{{ $t("webOptions.tagAuth") }}</label>
|
<label>{{ $t("webOptions.tagAuth") }}</label>
|
||||||
<b-form-input v-model="options.tag_auth" @change="update()"></b-form-input>
|
<b-form-input v-model="options.tag_auth" @change="update()" :disabled="Boolean(options.auth)"></b-form-input>
|
||||||
|
|
||||||
|
<b-form-checkbox v-model="options.verbose" @change="update()">
|
||||||
|
{{$t("webOptions.verbose")}}
|
||||||
|
</b-form-checkbox>
|
||||||
|
</b-card>
|
||||||
|
|
||||||
<br>
|
<br>
|
||||||
<h5>Auth0 options</h5>
|
<h4>Auth0 options</h4>
|
||||||
|
<b-card>
|
||||||
<label>{{ $t("webOptions.auth0Audience") }}</label>
|
<label>{{ $t("webOptions.auth0Audience") }}</label>
|
||||||
<b-form-input v-model="options.auth0_audience" @change="update()"></b-form-input>
|
<b-form-input v-model="options.auth0_audience" @change="update()"></b-form-input>
|
||||||
|
|
||||||
@@ -47,16 +37,12 @@
|
|||||||
|
|
||||||
<label>{{ $t("webOptions.auth0PublicKey") }}</label>
|
<label>{{ $t("webOptions.auth0PublicKey") }}</label>
|
||||||
<b-textarea rows="10" v-model="options.auth0_public_key" @change="update()"></b-textarea>
|
<b-textarea rows="10" v-model="options.auth0_public_key" @change="update()"></b-textarea>
|
||||||
|
</b-card>
|
||||||
|
|
||||||
|
|
||||||
</div>
|
</div>
|
||||||
</template>
|
</template>
|
||||||
|
|
||||||
<script>
|
<script>
|
||||||
|
|
||||||
import sist2AdminApi from "@/Sist2AdminApi";
|
|
||||||
|
|
||||||
export default {
|
export default {
|
||||||
name: "WebOptions",
|
name: "WebOptions",
|
||||||
props: ["options", "frontendName"],
|
props: ["options", "frontendName"],
|
||||||
@@ -64,24 +50,20 @@ export default {
|
|||||||
return {
|
return {
|
||||||
showEsTestAlert: false,
|
showEsTestAlert: false,
|
||||||
esTestOk: false,
|
esTestOk: false,
|
||||||
esTestMessage: "",
|
esTestMessage: ""
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
methods: {
|
methods: {
|
||||||
update() {
|
update() {
|
||||||
if (!this.options.es_url.startsWith("https")) {
|
|
||||||
this.options.es_insecure_ssl = false;
|
console.log(this.options)
|
||||||
|
if (this.options.auth && this.options.tag_auth) {
|
||||||
|
// If both are set, remove tagAuth
|
||||||
|
this.options.tag_auth = "";
|
||||||
}
|
}
|
||||||
|
|
||||||
this.$emit("change", this.options);
|
this.$emit("change", this.options);
|
||||||
},
|
},
|
||||||
testEs() {
|
|
||||||
sist2AdminApi.pingEs(this.options.es_url, this.options.es_insecure_ssl).then((resp) => {
|
|
||||||
this.showEsTestAlert = true;
|
|
||||||
this.esTestOk = resp.data.ok;
|
|
||||||
this.esTestMessage = resp.data.message;
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
</script>
|
</script>
|
||||||
|
|||||||
@@ -5,10 +5,14 @@ export default {
|
|||||||
go: "Go",
|
go: "Go",
|
||||||
online: "online",
|
online: "online",
|
||||||
offline: "offline",
|
offline: "offline",
|
||||||
|
view: "View",
|
||||||
delete: "Delete",
|
delete: "Delete",
|
||||||
runNow: "Index now",
|
runNow: "Index now",
|
||||||
|
runNowFull: "Full re-index",
|
||||||
create: "Create",
|
create: "Create",
|
||||||
|
cancel: "Cancel",
|
||||||
test: "Test",
|
test: "Test",
|
||||||
|
confirmation: "Confirmation",
|
||||||
|
|
||||||
jobTitle: "job configuration",
|
jobTitle: "job configuration",
|
||||||
tasks: "Tasks",
|
tasks: "Tasks",
|
||||||
@@ -45,12 +49,29 @@ export default {
|
|||||||
extraQueryArgs: "Extra query arguments when launching from sist2-admin",
|
extraQueryArgs: "Extra query arguments when launching from sist2-admin",
|
||||||
customUrl: "Custom URL when launching from sist2-admin",
|
customUrl: "Custom URL when launching from sist2-admin",
|
||||||
|
|
||||||
selectJobs: "Select jobs",
|
searchBackends: "Search backends",
|
||||||
|
searchBackendTitle: "search backend configuration",
|
||||||
|
newBackendName: "New search backend name",
|
||||||
|
|
||||||
|
frontendTab: "Frontend",
|
||||||
|
backendTab: "Backend",
|
||||||
|
scripts: "User Scripts",
|
||||||
|
script: "User Script",
|
||||||
|
testScript: "Test/debug User Script",
|
||||||
|
newScriptName: "New script name",
|
||||||
|
scriptType: "Script type",
|
||||||
|
scriptCode: "Script code (Python)",
|
||||||
|
scriptOptions: "User scripts",
|
||||||
|
gitRepository: "Git repository URL",
|
||||||
|
extraArgs: "Extra command line arguments",
|
||||||
|
|
||||||
|
couldNotStartFrontend: "Could not start frontend",
|
||||||
|
couldNotStartFrontendBody: "Unable to start the frontend, check server logs for more details.",
|
||||||
|
|
||||||
|
selectJobs: "Available jobs",
|
||||||
|
selectJob: "Select a job",
|
||||||
webOptions: {
|
webOptions: {
|
||||||
title: "Web options",
|
title: "Web options",
|
||||||
esUrl: "Elasticsearch URL",
|
|
||||||
esIndex: "Elasticsearch index name",
|
|
||||||
esInsecure: "Do not verify SSL connections to Elasticsearch.",
|
|
||||||
lang: "UI Language",
|
lang: "UI Language",
|
||||||
bind: "Listen address",
|
bind: "Listen address",
|
||||||
tagline: "Tagline in navbar",
|
tagline: "Tagline in navbar",
|
||||||
@@ -60,13 +81,26 @@ export default {
|
|||||||
auth0Domain: "Auth0 domain",
|
auth0Domain: "Auth0 domain",
|
||||||
auth0ClientId: "Auth0 client ID",
|
auth0ClientId: "Auth0 client ID",
|
||||||
auth0PublicKey: "Auth0 public key",
|
auth0PublicKey: "Auth0 public key",
|
||||||
|
verbose: "Verbose logs"
|
||||||
|
},
|
||||||
|
backendOptions: {
|
||||||
|
title: "Search backend options",
|
||||||
|
searchBackend: "Search backend",
|
||||||
|
type: "Search backend type",
|
||||||
|
esUrl: "Elasticsearch URL",
|
||||||
|
esIndex: "Elasticsearch index name",
|
||||||
|
esInsecure: "Do not verify SSL connections to Elasticsearch.",
|
||||||
|
threads: "Number of threads",
|
||||||
|
batchSize: "Index batch size",
|
||||||
|
script: "User script",
|
||||||
|
searchIndex: "Search index file location"
|
||||||
},
|
},
|
||||||
scanOptions: {
|
scanOptions: {
|
||||||
title: "Scanning options",
|
title: "Scanning options",
|
||||||
path: "Path",
|
path: "Path",
|
||||||
threads: "Number of threads",
|
threads: "Number of threads",
|
||||||
memThrottle: "Total memory threshold in MiB for scan throttling",
|
memThrottle: "Total memory threshold in MiB for scan throttling",
|
||||||
thumbnailQuality: "Thumbnail quality, on a scale of 2 to 32, 2 being the best",
|
thumbnailQuality: "Thumbnail quality, on a scale of 0 to 100, 100 being the best",
|
||||||
thumbnailCount: "Number of thumbnails to generate. Set a value > 1 to create video previews, set to 0 to disable thumbnails.",
|
thumbnailCount: "Number of thumbnails to generate. Set a value > 1 to create video previews, set to 0 to disable thumbnails.",
|
||||||
thumbnailSize: "Thumbnail size, in pixels",
|
thumbnailSize: "Thumbnail size, in pixels",
|
||||||
contentSize: "Number of bytes to be extracted from text documents. Set to 0 to disable",
|
contentSize: "Number of bytes to be extracted from text documents. Set to 0 to disable",
|
||||||
@@ -87,24 +121,19 @@ export default {
|
|||||||
treemapThreshold: "Relative size threshold for treemap",
|
treemapThreshold: "Relative size threshold for treemap",
|
||||||
optimizeIndex: "Defragment index file after scan to reduce its file size."
|
optimizeIndex: "Defragment index file after scan to reduce its file size."
|
||||||
},
|
},
|
||||||
indexOptions: {
|
|
||||||
title: "Indexing options",
|
|
||||||
threads: "Number of threads",
|
|
||||||
esUrl: "Elasticsearch URL",
|
|
||||||
esIndex: "Elasticsearch index name",
|
|
||||||
esInsecure: "Do not verify SSL connections to Elasticsearch.",
|
|
||||||
batchSize: "Index batch size",
|
|
||||||
script: "User script"
|
|
||||||
},
|
|
||||||
jobOptions: {
|
jobOptions: {
|
||||||
title: "Job options",
|
title: "Job options",
|
||||||
cron: "Job schedule",
|
cron: "Job schedule",
|
||||||
|
keepNLogs: "Keep last N log files. Set to -1 to keep all logs.",
|
||||||
|
deleteNow: "Delete now",
|
||||||
scheduleEnabled: "Enable scheduled re-scan",
|
scheduleEnabled: "Enable scheduled re-scan",
|
||||||
noJobAvailable: "No jobs available.",
|
noJobAvailable: "No jobs available for this search backend.",
|
||||||
|
notIndexed: "Has not been indexed yet",
|
||||||
|
noBackendError: "You must select a search backend to run this job",
|
||||||
desktopNotifications: "Desktop notifications"
|
desktopNotifications: "Desktop notifications"
|
||||||
},
|
},
|
||||||
frontendOptions: {
|
frontendOptions: {
|
||||||
title: "Frontend options",
|
title: "Advanced options",
|
||||||
noJobSelectedWarning: "You must select at least one job to start this frontend"
|
noJobSelectedWarning: "You must select at least one job to start this frontend"
|
||||||
},
|
},
|
||||||
notifications: {
|
notifications: {
|
||||||
|
|||||||
@@ -5,12 +5,19 @@ import Job from "@/views/Job";
|
|||||||
import Tasks from "@/views/Tasks";
|
import Tasks from "@/views/Tasks";
|
||||||
import Frontend from "@/views/Frontend";
|
import Frontend from "@/views/Frontend";
|
||||||
import Tail from "@/views/Tail";
|
import Tail from "@/views/Tail";
|
||||||
|
import SearchBackend from "@/views/SearchBackend.vue";
|
||||||
|
import UserScript from "@/views/UserScript.vue";
|
||||||
|
|
||||||
Vue.use(VueRouter);
|
Vue.use(VueRouter);
|
||||||
|
|
||||||
const routes = [
|
const routes = [
|
||||||
{
|
{
|
||||||
path: "/",
|
path: "/task",
|
||||||
|
name: "Tasks",
|
||||||
|
component: Tasks
|
||||||
|
},
|
||||||
|
{
|
||||||
|
path: "/:tab?",
|
||||||
name: "Home",
|
name: "Home",
|
||||||
component: Home
|
component: Home
|
||||||
},
|
},
|
||||||
@@ -19,16 +26,21 @@ const routes = [
|
|||||||
name: "Job",
|
name: "Job",
|
||||||
component: Job
|
component: Job
|
||||||
},
|
},
|
||||||
{
|
|
||||||
path: "/task/",
|
|
||||||
name: "Tasks",
|
|
||||||
component: Tasks
|
|
||||||
},
|
|
||||||
{
|
{
|
||||||
path: "/frontend/:name",
|
path: "/frontend/:name",
|
||||||
name: "Frontend",
|
name: "Frontend",
|
||||||
component: Frontend
|
component: Frontend
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
path: "/searchBackend/:name",
|
||||||
|
name: "SearchBackend",
|
||||||
|
component: SearchBackend
|
||||||
|
},
|
||||||
|
{
|
||||||
|
path: "/userScript/:name",
|
||||||
|
name: "UserScript",
|
||||||
|
component: UserScript
|
||||||
|
},
|
||||||
{
|
{
|
||||||
path: "/log/:taskId",
|
path: "/log/:taskId",
|
||||||
name: "Tail",
|
name: "Tail",
|
||||||
|
|||||||
@@ -8,6 +8,7 @@
|
|||||||
</small>
|
</small>
|
||||||
</b-card-title>
|
</b-card-title>
|
||||||
|
|
||||||
|
<!-- Action buttons-->
|
||||||
<div class="mb-3" v-if="!loading">
|
<div class="mb-3" v-if="!loading">
|
||||||
<b-button class="mr-1" :disabled="frontend.running || !valid" variant="success" @click="start()">{{
|
<b-button class="mr-1" :disabled="frontend.running || !valid" variant="success" @click="start()">{{
|
||||||
$t("start")
|
$t("start")
|
||||||
@@ -23,10 +24,26 @@
|
|||||||
<b-button variant="danger" @click="deleteFrontend()">{{ $t("delete") }}</b-button>
|
<b-button variant="danger" @click="deleteFrontend()">{{ $t("delete") }}</b-button>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
|
||||||
<b-progress v-if="loading" striped animated value="100"></b-progress>
|
<b-progress v-if="loading" striped animated value="100"></b-progress>
|
||||||
<b-card-body v-else>
|
<b-card-body v-else>
|
||||||
|
|
||||||
|
<h4>{{ $t("backendOptions.title") }}</h4>
|
||||||
|
<b-card>
|
||||||
|
<b-alert v-if="!valid" variant="warning" show>{{ $t("frontendOptions.noJobSelectedWarning") }}</b-alert>
|
||||||
|
|
||||||
|
<SearchBackendSelect :value="frontend.web_options.search_backend"
|
||||||
|
@change="onBackendSelect($event)"></SearchBackendSelect>
|
||||||
|
|
||||||
|
<br>
|
||||||
|
<JobCheckboxGroup :frontend="frontend" @input="update()"></JobCheckboxGroup>
|
||||||
|
</b-card>
|
||||||
|
|
||||||
|
<br/>
|
||||||
|
|
||||||
|
<WebOptions :options="frontend.web_options" :frontend-name="$route.params.name"
|
||||||
|
@change="update()"></WebOptions>
|
||||||
|
<br/>
|
||||||
|
|
||||||
<h4>{{ $t("frontendOptions.title") }}</h4>
|
<h4>{{ $t("frontendOptions.title") }}</h4>
|
||||||
<b-card>
|
<b-card>
|
||||||
<b-form-checkbox v-model="frontend.auto_start" @change="update()">
|
<b-form-checkbox v-model="frontend.auto_start" @change="update()">
|
||||||
@@ -38,22 +55,8 @@
|
|||||||
|
|
||||||
<label>{{ $t("customUrl") }}</label>
|
<label>{{ $t("customUrl") }}</label>
|
||||||
<b-form-input v-model="frontend.custom_url" @change="update()" placeholder="http://"></b-form-input>
|
<b-form-input v-model="frontend.custom_url" @change="update()" placeholder="http://"></b-form-input>
|
||||||
|
|
||||||
<br/>
|
|
||||||
|
|
||||||
<b-alert v-if="!valid" variant="warning" show>{{ $t("frontendOptions.noJobSelectedWarning") }}</b-alert>
|
|
||||||
|
|
||||||
<JobCheckboxGroup :frontend="frontend" @input="update()"></JobCheckboxGroup>
|
|
||||||
</b-card>
|
|
||||||
|
|
||||||
<br/>
|
|
||||||
|
|
||||||
<h4>{{ $t("webOptions.title") }}</h4>
|
|
||||||
<b-card>
|
|
||||||
<WebOptions :options="frontend.web_options" :frontend-name="$route.params.name" @change="update()"></WebOptions>
|
|
||||||
</b-card>
|
</b-card>
|
||||||
</b-card-body>
|
</b-card-body>
|
||||||
|
|
||||||
</b-card>
|
</b-card>
|
||||||
</template>
|
</template>
|
||||||
|
|
||||||
@@ -62,10 +65,11 @@
|
|||||||
import Sist2AdminApi from "@/Sist2AdminApi";
|
import Sist2AdminApi from "@/Sist2AdminApi";
|
||||||
import JobCheckboxGroup from "@/components/JobCheckboxGroup";
|
import JobCheckboxGroup from "@/components/JobCheckboxGroup";
|
||||||
import WebOptions from "@/components/WebOptions";
|
import WebOptions from "@/components/WebOptions";
|
||||||
|
import SearchBackendSelect from "@/components/SearchBackendSelect.vue";
|
||||||
|
|
||||||
export default {
|
export default {
|
||||||
name: 'Frontend',
|
name: 'Frontend',
|
||||||
components: {JobCheckboxGroup, WebOptions},
|
components: {SearchBackendSelect, JobCheckboxGroup, WebOptions},
|
||||||
data() {
|
data() {
|
||||||
return {
|
return {
|
||||||
loading: true,
|
loading: true,
|
||||||
@@ -109,8 +113,15 @@ export default {
|
|||||||
},
|
},
|
||||||
methods: {
|
methods: {
|
||||||
start() {
|
start() {
|
||||||
|
Sist2AdminApi.startFrontend(this.name).then(() => {
|
||||||
this.frontend.running = true;
|
this.frontend.running = true;
|
||||||
Sist2AdminApi.startFrontend(this.name)
|
}).catch(() => {
|
||||||
|
this.$bvToast.toast(this.$t("couldNotStartFrontendBody"), {
|
||||||
|
title: this.$t("couldNotStartFrontend"),
|
||||||
|
variant: "danger",
|
||||||
|
toaster: "b-toaster-bottom-right"
|
||||||
|
});
|
||||||
|
});
|
||||||
},
|
},
|
||||||
stop() {
|
stop() {
|
||||||
this.frontend.running = false;
|
this.frontend.running = false;
|
||||||
@@ -118,12 +129,17 @@ export default {
|
|||||||
},
|
},
|
||||||
deleteFrontend() {
|
deleteFrontend() {
|
||||||
Sist2AdminApi.deleteFrontend(this.name).then(() => {
|
Sist2AdminApi.deleteFrontend(this.name).then(() => {
|
||||||
this.$router.push("/frontends");
|
this.$router.push("/");
|
||||||
});
|
});
|
||||||
},
|
},
|
||||||
update() {
|
update() {
|
||||||
Sist2AdminApi.updateFrontend(this.name, this.frontend);
|
Sist2AdminApi.updateFrontend(this.name, this.frontend);
|
||||||
},
|
},
|
||||||
|
onBackendSelect(backend) {
|
||||||
|
this.frontend.web_options.search_backend = backend;
|
||||||
|
this.frontend.jobs = [];
|
||||||
|
this.update();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
</script>
|
</script>
|
||||||
@@ -1,5 +1,34 @@
|
|||||||
<template>
|
<template>
|
||||||
<div>
|
<div>
|
||||||
|
<b-tabs content-class="mt-3" v-model="tab" @input="onTabChange($event)">
|
||||||
|
<b-tab :title="$t('backendTab')">
|
||||||
|
|
||||||
|
<b-card>
|
||||||
|
<b-card-title>{{ $t("searchBackends") }}</b-card-title>
|
||||||
|
|
||||||
|
<b-row>
|
||||||
|
<b-col>
|
||||||
|
<b-input v-model="newBackendName" :placeholder="$t('newBackendName')"></b-input>
|
||||||
|
</b-col>
|
||||||
|
<b-col>
|
||||||
|
<b-button variant="primary" @click="createBackend()"
|
||||||
|
:disabled="!backendNameValid(newBackendName)">
|
||||||
|
{{ $t("create") }}
|
||||||
|
</b-button>
|
||||||
|
</b-col>
|
||||||
|
</b-row>
|
||||||
|
|
||||||
|
<hr/>
|
||||||
|
|
||||||
|
<b-progress v-if="backendsLoading" striped animated value="100"></b-progress>
|
||||||
|
<b-list-group v-else>
|
||||||
|
<SearchBackendListItem v-for="backend in backends"
|
||||||
|
:key="backend.name" :backend="backend"></SearchBackendListItem>
|
||||||
|
</b-list-group>
|
||||||
|
</b-card>
|
||||||
|
|
||||||
|
<br/>
|
||||||
|
|
||||||
<b-card>
|
<b-card>
|
||||||
<b-card-title>{{ $t("jobs") }}</b-card-title>
|
<b-card-title>{{ $t("jobs") }}</b-card-title>
|
||||||
<b-row>
|
<b-row>
|
||||||
@@ -15,7 +44,8 @@
|
|||||||
></b-popover>
|
></b-popover>
|
||||||
</b-col>
|
</b-col>
|
||||||
<b-col>
|
<b-col>
|
||||||
<b-button variant="primary" @click="createJob()" :disabled="!jobNameValid(newJobName)">{{ $t("create") }}
|
<b-button variant="primary" @click="createJob()" :disabled="!jobNameValid(newJobName)">
|
||||||
|
{{ $t("create") }}
|
||||||
</b-button>
|
</b-button>
|
||||||
</b-col>
|
</b-col>
|
||||||
</b-row>
|
</b-row>
|
||||||
@@ -27,9 +57,39 @@
|
|||||||
<JobListItem v-for="job in jobs" :key="job.name" :job="job"></JobListItem>
|
<JobListItem v-for="job in jobs" :key="job.name" :job="job"></JobListItem>
|
||||||
</b-list-group>
|
</b-list-group>
|
||||||
</b-card>
|
</b-card>
|
||||||
|
</b-tab>
|
||||||
|
<b-tab :title="$t('scripts')">
|
||||||
|
|
||||||
<br/>
|
<b-progress v-if="scriptsLoading" striped animated value="100"></b-progress>
|
||||||
|
<b-card v-else>
|
||||||
|
<b-card-title>{{ $t("scripts") }}</b-card-title>
|
||||||
|
|
||||||
|
<label>Select template</label>
|
||||||
|
<b-form-radio-group stacked :options="scriptTemplates" v-model="scriptTemplate"></b-form-radio-group>
|
||||||
|
<br>
|
||||||
|
|
||||||
|
<b-row>
|
||||||
|
<b-col>
|
||||||
|
<b-form-input v-model="newScriptName" :disabled="!scriptTemplate" :placeholder="$t('newScriptName')"></b-form-input>
|
||||||
|
</b-col>
|
||||||
|
<b-col>
|
||||||
|
<b-button variant="primary" @click="createScript()"
|
||||||
|
:disabled="!scriptNameValid(newScriptName)">
|
||||||
|
{{ $t("create") }}
|
||||||
|
</b-button>
|
||||||
|
</b-col>
|
||||||
|
</b-row>
|
||||||
|
|
||||||
|
<hr/>
|
||||||
|
|
||||||
|
<b-list-group>
|
||||||
|
<UserScriptListItem v-for="script in scripts"
|
||||||
|
:key="script.name" :script="script"></UserScriptListItem>
|
||||||
|
</b-list-group>
|
||||||
|
|
||||||
|
</b-card>
|
||||||
|
</b-tab>
|
||||||
|
<b-tab :title="$t('frontendTab')">
|
||||||
<b-card>
|
<b-card>
|
||||||
|
|
||||||
<b-card-title>{{ $t("frontends") }}</b-card-title>
|
<b-card-title>{{ $t("frontends") }}</b-card-title>
|
||||||
@@ -39,7 +99,8 @@
|
|||||||
<b-input v-model="newFrontendName" :placeholder="$t('newFrontendName')"></b-input>
|
<b-input v-model="newFrontendName" :placeholder="$t('newFrontendName')"></b-input>
|
||||||
</b-col>
|
</b-col>
|
||||||
<b-col>
|
<b-col>
|
||||||
<b-button variant="primary" @click="createFrontend()" :disabled="!frontendNameValid(newFrontendName)">
|
<b-button variant="primary" @click="createFrontend()"
|
||||||
|
:disabled="!frontendNameValid(newFrontendName)">
|
||||||
{{ $t("create") }}
|
{{ $t("create") }}
|
||||||
</b-button>
|
</b-button>
|
||||||
</b-col>
|
</b-col>
|
||||||
@@ -54,6 +115,8 @@
|
|||||||
</b-list-group>
|
</b-list-group>
|
||||||
|
|
||||||
</b-card>
|
</b-card>
|
||||||
|
</b-tab>
|
||||||
|
</b-tabs>
|
||||||
</div>
|
</div>
|
||||||
</template>
|
</template>
|
||||||
|
|
||||||
@@ -62,10 +125,12 @@ import JobListItem from "@/components/JobListItem";
|
|||||||
import {formatBindAddress} from "@/util";
|
import {formatBindAddress} from "@/util";
|
||||||
import Sist2AdminApi from "@/Sist2AdminApi";
|
import Sist2AdminApi from "@/Sist2AdminApi";
|
||||||
import FrontendListItem from "@/components/FrontendListItem";
|
import FrontendListItem from "@/components/FrontendListItem";
|
||||||
|
import SearchBackendListItem from "@/components/SearchBackendListItem.vue";
|
||||||
|
import UserScriptListItem from "@/components/UserScriptListItem.vue";
|
||||||
|
|
||||||
export default {
|
export default {
|
||||||
name: "Jobs",
|
name: "Jobs",
|
||||||
components: {JobListItem, FrontendListItem},
|
components: {UserScriptListItem, SearchBackendListItem, JobListItem, FrontendListItem},
|
||||||
data() {
|
data() {
|
||||||
return {
|
return {
|
||||||
jobsLoading: true,
|
jobsLoading: true,
|
||||||
@@ -77,11 +142,28 @@ export default {
|
|||||||
formatBindAddress,
|
formatBindAddress,
|
||||||
newFrontendName: "",
|
newFrontendName: "",
|
||||||
|
|
||||||
showHelp: false
|
backends: [],
|
||||||
|
backendsLoading: true,
|
||||||
|
newBackendName: "",
|
||||||
|
|
||||||
|
scripts: [],
|
||||||
|
scriptTemplates: [],
|
||||||
|
newScriptName: "",
|
||||||
|
scriptTemplate: null,
|
||||||
|
scriptsLoading: true,
|
||||||
|
|
||||||
|
showHelp: false,
|
||||||
|
tab: 0
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
mounted() {
|
mounted() {
|
||||||
this.loading = true;
|
this.loading = true;
|
||||||
|
if (this.$route.params.tab) {
|
||||||
|
console.log("mounted " + this.$route.params.tab)
|
||||||
|
window.setTimeout(() => {
|
||||||
|
this.tab = Math.round(Number(this.$route.params.tab));
|
||||||
|
}, 1)
|
||||||
|
}
|
||||||
this.reload();
|
this.reload();
|
||||||
},
|
},
|
||||||
methods: {
|
methods: {
|
||||||
@@ -93,7 +175,24 @@ export default {
|
|||||||
return /^[a-zA-Z0-9-_,.; ]+$/.test(name);
|
return /^[a-zA-Z0-9-_,.; ]+$/.test(name);
|
||||||
},
|
},
|
||||||
frontendNameValid(name) {
|
frontendNameValid(name) {
|
||||||
if (this.frontends.some(job => job.name === name)) {
|
if (this.frontends.some(frontend => frontend.name === name)) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return /^[a-zA-Z0-9-_,.; ]+$/.test(name);
|
||||||
|
},
|
||||||
|
backendNameValid(name) {
|
||||||
|
if (this.backends.some(backend => backend.name === name)) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return /^[a-zA-Z0-9-_,.; ]+$/.test(name);
|
||||||
|
},
|
||||||
|
scriptNameValid(name) {
|
||||||
|
if (this.scripts.some(script => script.name === name)) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (name.length > 16) {
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -103,19 +202,38 @@ export default {
|
|||||||
Sist2AdminApi.getJobs().then(resp => {
|
Sist2AdminApi.getJobs().then(resp => {
|
||||||
this.jobs = resp.data;
|
this.jobs = resp.data;
|
||||||
this.jobsLoading = false;
|
this.jobsLoading = false;
|
||||||
|
|
||||||
this.showHelp = this.jobs.length === 0;
|
this.showHelp = this.jobs.length === 0;
|
||||||
});
|
});
|
||||||
Sist2AdminApi.getFrontends().then(resp => {
|
Sist2AdminApi.getFrontends().then(resp => {
|
||||||
this.frontends = resp.data;
|
this.frontends = resp.data;
|
||||||
this.frontendsLoading = false;
|
this.frontendsLoading = false;
|
||||||
});
|
});
|
||||||
|
Sist2AdminApi.getSearchBackends().then(resp => {
|
||||||
|
this.backends = resp.data;
|
||||||
|
this.backendsLoading = false;
|
||||||
|
})
|
||||||
|
Sist2AdminApi.getUserScripts().then(resp => {
|
||||||
|
this.scripts = resp.data;
|
||||||
|
this.scriptTemplates = this.$store.state.sist2AdminInfo.user_script_templates;
|
||||||
|
this.scriptsLoading = false;
|
||||||
|
})
|
||||||
},
|
},
|
||||||
createJob() {
|
createJob() {
|
||||||
Sist2AdminApi.createJob(this.newJobName).then(this.reload);
|
Sist2AdminApi.createJob(this.newJobName).then(this.reload);
|
||||||
},
|
},
|
||||||
createFrontend() {
|
createFrontend() {
|
||||||
Sist2AdminApi.createFrontend(this.newFrontendName).then(this.reload)
|
Sist2AdminApi.createFrontend(this.newFrontendName).then(this.reload)
|
||||||
|
},
|
||||||
|
createBackend() {
|
||||||
|
Sist2AdminApi.createBackend(this.newBackendName).then(this.reload);
|
||||||
|
},
|
||||||
|
createScript() {
|
||||||
|
Sist2AdminApi.createUserScript(this.newScriptName, this.scriptTemplate).then(this.reload)
|
||||||
|
},
|
||||||
|
onTabChange(tab) {
|
||||||
|
if (this.$route.params.tab != tab) {
|
||||||
|
this.$router.push({params: {tab: tab}})
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -6,7 +6,19 @@
|
|||||||
</b-card-title>
|
</b-card-title>
|
||||||
|
|
||||||
<div class="mb-3">
|
<div class="mb-3">
|
||||||
<b-button class="mr-1" variant="primary" @click="runJob()">{{ $t("runNow") }}</b-button>
|
|
||||||
|
<b-dropdown
|
||||||
|
split
|
||||||
|
split-variant="primary"
|
||||||
|
variant="primary"
|
||||||
|
:text="$t('runNow')"
|
||||||
|
class="mr-1"
|
||||||
|
:disabled="!valid"
|
||||||
|
@click="runJob()"
|
||||||
|
>
|
||||||
|
<b-dropdown-item href="#" @click="runJob(true)">{{ $t("runNowFull") }}</b-dropdown-item>
|
||||||
|
</b-dropdown>
|
||||||
|
|
||||||
<b-button variant="danger" @click="deleteJob()">{{ $t("delete") }}</b-button>
|
<b-button variant="danger" @click="deleteJob()">{{ $t("delete") }}</b-button>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
@@ -24,16 +36,25 @@
|
|||||||
|
|
||||||
<br/>
|
<br/>
|
||||||
|
|
||||||
<h4>{{ $t("scanOptions.title") }}</h4>
|
<h4>{{ $t("backendOptions.title") }}</h4>
|
||||||
<b-card>
|
<b-card>
|
||||||
<ScanOptions :options="job.scan_options" @change="update()"></ScanOptions>
|
<b-alert v-if="!valid" variant="warning" show>{{ $t("jobOptions.noBackendError") }}</b-alert>
|
||||||
|
<SearchBackendSelect :value="job.index_options.search_backend"
|
||||||
|
@change="onBackendSelect($event)"></SearchBackendSelect>
|
||||||
|
</b-card>
|
||||||
|
<br/>
|
||||||
|
|
||||||
|
<h4>{{ $t("scriptOptions") }}</h4>
|
||||||
|
<b-card>
|
||||||
|
<UserScriptPicker :selected-scripts="job.user_scripts"
|
||||||
|
@change="onScriptChange($event)"></UserScriptPicker>
|
||||||
</b-card>
|
</b-card>
|
||||||
|
|
||||||
<br/>
|
<br/>
|
||||||
|
|
||||||
<h4>{{ $t("indexOptions.title") }}</h4>
|
<h4>{{ $t("scanOptions.title") }}</h4>
|
||||||
<b-card>
|
<b-card>
|
||||||
<IndexOptions :options="job.index_options" @change="update()"></IndexOptions>
|
<ScanOptions :options="job.scan_options" @change="update()"></ScanOptions>
|
||||||
</b-card>
|
</b-card>
|
||||||
|
|
||||||
</b-card-body>
|
</b-card-body>
|
||||||
@@ -44,20 +65,23 @@
|
|||||||
<script>
|
<script>
|
||||||
import ScanOptions from "@/components/ScanOptions";
|
import ScanOptions from "@/components/ScanOptions";
|
||||||
import Sist2AdminApi from "@/Sist2AdminApi";
|
import Sist2AdminApi from "@/Sist2AdminApi";
|
||||||
import IndexOptions from "@/components/IndexOptions";
|
|
||||||
import JobOptions from "@/components/JobOptions";
|
import JobOptions from "@/components/JobOptions";
|
||||||
|
import SearchBackendSelect from "@/components/SearchBackendSelect.vue";
|
||||||
|
import UserScriptPicker from "@/components/UserScriptPicker.vue";
|
||||||
|
|
||||||
export default {
|
export default {
|
||||||
name: "Job",
|
name: "Job",
|
||||||
components: {
|
components: {
|
||||||
IndexOptions,
|
UserScriptPicker,
|
||||||
|
SearchBackendSelect,
|
||||||
ScanOptions,
|
ScanOptions,
|
||||||
JobOptions
|
JobOptions
|
||||||
},
|
},
|
||||||
data() {
|
data() {
|
||||||
return {
|
return {
|
||||||
loading: true,
|
loading: true,
|
||||||
job: null
|
job: null,
|
||||||
|
console: console
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
methods: {
|
methods: {
|
||||||
@@ -67,8 +91,8 @@ export default {
|
|||||||
update() {
|
update() {
|
||||||
Sist2AdminApi.updateJob(this.getName(), this.job);
|
Sist2AdminApi.updateJob(this.getName(), this.job);
|
||||||
},
|
},
|
||||||
runJob() {
|
runJob(full = false) {
|
||||||
Sist2AdminApi.runJob(this.getName()).then(() => {
|
Sist2AdminApi.runJob(this.getName(), full).then(() => {
|
||||||
this.$bvToast.toast(this.$t("runJobConfirmation"), {
|
this.$bvToast.toast(this.$t("runJobConfirmation"), {
|
||||||
title: this.$t("runJobConfirmationTitle"),
|
title: this.$t("runJobConfirmationTitle"),
|
||||||
variant: "success",
|
variant: "success",
|
||||||
@@ -77,9 +101,26 @@ export default {
|
|||||||
});
|
});
|
||||||
},
|
},
|
||||||
deleteJob() {
|
deleteJob() {
|
||||||
Sist2AdminApi.deleteJob(this.getName()).then(() => {
|
Sist2AdminApi.deleteJob(this.getName())
|
||||||
|
.then(() => {
|
||||||
this.$router.push("/");
|
this.$router.push("/");
|
||||||
})
|
})
|
||||||
|
.catch(err => {
|
||||||
|
this.$bvToast.toast("Cannot delete job " +
|
||||||
|
"because it is referenced by a frontend", {
|
||||||
|
title: "Error",
|
||||||
|
variant: "danger",
|
||||||
|
toaster: "b-toaster-bottom-right"
|
||||||
|
});
|
||||||
|
})
|
||||||
|
},
|
||||||
|
onBackendSelect(backend) {
|
||||||
|
this.job.index_options.search_backend = backend;
|
||||||
|
this.update();
|
||||||
|
},
|
||||||
|
onScriptChange(scripts) {
|
||||||
|
this.job.user_scripts = scripts;
|
||||||
|
this.update();
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
mounted() {
|
mounted() {
|
||||||
@@ -87,6 +128,11 @@ export default {
|
|||||||
this.loading = false;
|
this.loading = false;
|
||||||
this.job = resp.data;
|
this.job = resp.data;
|
||||||
})
|
})
|
||||||
|
},
|
||||||
|
computed: {
|
||||||
|
valid() {
|
||||||
|
return this.job?.index_options.search_backend != null;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
</script>
|
</script>
|
||||||
123
sist2-admin/frontend/src/views/SearchBackend.vue
Normal file
@@ -0,0 +1,123 @@
|
|||||||
|
<template>
|
||||||
|
|
||||||
|
<b-card>
|
||||||
|
<b-card-title>
|
||||||
|
<span class="text-monospace">{{ getName() }}</span>
|
||||||
|
{{ $t("searchBackendTitle") }}
|
||||||
|
</b-card-title>
|
||||||
|
|
||||||
|
<div class="mb-3">
|
||||||
|
<b-button variant="danger" @click="deleteBackend()">{{ $t("delete") }}</b-button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<b-progress v-if="loading" striped animated value="100"></b-progress>
|
||||||
|
<b-card-body v-else>
|
||||||
|
|
||||||
|
<label>{{ $t("backendOptions.type") }}</label>
|
||||||
|
<b-select :options="backendTypeOptions" v-model="backend.backend_type" @change="update()"></b-select>
|
||||||
|
|
||||||
|
<hr/>
|
||||||
|
|
||||||
|
<template v-if="backend.backend_type === 'elasticsearch'">
|
||||||
|
<b-alert :variant="esTestOk ? 'success' : 'danger'" :show="showEsTestAlert" class="mt-1">
|
||||||
|
{{ esTestMessage }}
|
||||||
|
</b-alert>
|
||||||
|
|
||||||
|
<label>{{ $t("backendOptions.esUrl") }}</label>
|
||||||
|
<b-input-group>
|
||||||
|
<b-form-input v-model="backend.es_url" @change="update()"></b-form-input>
|
||||||
|
<b-input-group-append>
|
||||||
|
<b-button variant="outline-primary" @click="testEs()">{{ $t("test") }}</b-button>
|
||||||
|
</b-input-group-append>
|
||||||
|
</b-input-group>
|
||||||
|
|
||||||
|
<b-form-checkbox v-model="backend.es_insecure_ssl" :disabled="!this.backend.es_url.startsWith('https')"
|
||||||
|
@change="update()">
|
||||||
|
{{ $t("backendOptions.esInsecure") }}
|
||||||
|
</b-form-checkbox>
|
||||||
|
|
||||||
|
<label>{{ $t("backendOptions.esIndex") }}</label>
|
||||||
|
<b-form-input v-model="backend.es_index" @change="update()"></b-form-input>
|
||||||
|
|
||||||
|
<label>{{ $t("backendOptions.threads") }}</label>
|
||||||
|
<b-form-input v-model="backend.threads" type="number" min="1" @change="update()"></b-form-input>
|
||||||
|
|
||||||
|
<label>{{ $t("backendOptions.batchSize") }}</label>
|
||||||
|
<b-form-input v-model="backend.batch_size" type="number" min="1" @change="update()"></b-form-input>
|
||||||
|
</template>
|
||||||
|
<template v-else>
|
||||||
|
<label>{{ $t("backendOptions.searchIndex") }}</label>
|
||||||
|
<b-form-input v-model="backend.search_index" disabled></b-form-input>
|
||||||
|
</template>
|
||||||
|
</b-card-body>
|
||||||
|
|
||||||
|
</b-card>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
import sist2AdminApi from "@/Sist2AdminApi";
|
||||||
|
import Sist2AdminApi from "@/Sist2AdminApi";
|
||||||
|
|
||||||
|
export default {
|
||||||
|
name: "SearchBackend",
|
||||||
|
data() {
|
||||||
|
return {
|
||||||
|
showEsTestAlert: false,
|
||||||
|
esTestOk: false,
|
||||||
|
esTestMessage: "",
|
||||||
|
loading: true,
|
||||||
|
backend: null,
|
||||||
|
backendTypeOptions: [
|
||||||
|
{
|
||||||
|
text: "Elasticsearch",
|
||||||
|
value: "elasticsearch"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
text: "SQLite",
|
||||||
|
value: "sqlite"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
},
|
||||||
|
mounted() {
|
||||||
|
Sist2AdminApi.getSearchBackend(this.getName()).then(resp => {
|
||||||
|
this.backend = resp.data;
|
||||||
|
this.loading = false;
|
||||||
|
});
|
||||||
|
},
|
||||||
|
methods: {
|
||||||
|
getName() {
|
||||||
|
return this.$route.params.name;
|
||||||
|
},
|
||||||
|
testEs() {
|
||||||
|
sist2AdminApi.pingEs(this.backend.es_url, this.backend.es_insecure_ssl)
|
||||||
|
.then((resp) => {
|
||||||
|
this.showEsTestAlert = true;
|
||||||
|
this.esTestOk = resp.data.ok;
|
||||||
|
this.esTestMessage = resp.data.message;
|
||||||
|
});
|
||||||
|
},
|
||||||
|
update() {
|
||||||
|
Sist2AdminApi.updateSearchBackend(this.getName(), this.backend);
|
||||||
|
},
|
||||||
|
deleteBackend() {
|
||||||
|
Sist2AdminApi.deleteBackend(this.getName())
|
||||||
|
.then(() => {
|
||||||
|
this.$router.push("/");
|
||||||
|
})
|
||||||
|
.catch(err => {
|
||||||
|
this.$bvToast.toast("Cannot delete search backend " +
|
||||||
|
"because it is referenced by a job or frontend", {
|
||||||
|
title: "Error",
|
||||||
|
variant: "danger",
|
||||||
|
toaster: "b-toaster-bottom-right"
|
||||||
|
});
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<style scoped>
|
||||||
|
|
||||||
|
</style>
|
||||||
@@ -65,7 +65,11 @@ export default {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const n = this.mode === "follow" ? 32 : 9999999999;
|
const n = this.mode === "follow" ? 32 : 9999999999;
|
||||||
|
if (window.location.protocol === "https:") {
|
||||||
|
this.socket = new WebSocket(`wss://${window.location.host}/log/${this.taskId}?n=${n}`);
|
||||||
|
} else {
|
||||||
this.socket = new WebSocket(`ws://${window.location.host}/log/${this.taskId}?n=${n}`);
|
this.socket = new WebSocket(`ws://${window.location.host}/log/${this.taskId}?n=${n}`);
|
||||||
|
}
|
||||||
this.socket.onopen = () => {
|
this.socket.onopen = () => {
|
||||||
this.socket.send("Hello from client");
|
this.socket.send("Hello from client");
|
||||||
}
|
}
|
||||||
@@ -88,6 +92,9 @@ export default {
|
|||||||
if ("stderr" in message) {
|
if ("stderr" in message) {
|
||||||
message.level = "ERROR";
|
message.level = "ERROR";
|
||||||
message.message = message["stderr"];
|
message.message = message["stderr"];
|
||||||
|
} else if ("stdout" in message) {
|
||||||
|
message.level = "INFO";
|
||||||
|
message.message = message["stdout"];
|
||||||
} else {
|
} else {
|
||||||
message.level = "ADMIN";
|
message.level = "ADMIN";
|
||||||
message.message = message["sist2-admin"];
|
message.message = message["sist2-admin"];
|
||||||
@@ -163,6 +170,6 @@ span.ADMIN {
|
|||||||
margin: 3px;
|
margin: 3px;
|
||||||
white-space: pre;
|
white-space: pre;
|
||||||
color: #000;
|
color: #000;
|
||||||
overflow: hidden;
|
overflow-y: hidden;
|
||||||
}
|
}
|
||||||
</style>
|
</style>
|
||||||
@@ -23,7 +23,18 @@
|
|||||||
:per-page="10"
|
:per-page="10"
|
||||||
>
|
>
|
||||||
<template #cell(logs)="data">
|
<template #cell(logs)="data">
|
||||||
<router-link :to="`/log/${data.item.logs}`">{{ $t("logs") }}</router-link>
|
<template v-if="data.item._row.has_logs">
|
||||||
|
<b-button variant="link" size="sm" :to="`/log/${data.item.id}`">
|
||||||
|
{{ $t("view") }}
|
||||||
|
</b-button>
|
||||||
|
/
|
||||||
|
<b-button variant="link" size="sm" @click="deleteLogs(data.item.id)">
|
||||||
|
{{ $t("delete") }}
|
||||||
|
</b-button>
|
||||||
|
</template>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<template #cell(delete)="data">
|
||||||
</template>
|
</template>
|
||||||
|
|
||||||
</b-table>
|
</b-table>
|
||||||
@@ -70,7 +81,7 @@ function humanDuration(sec_num) {
|
|||||||
return `${seconds}s`;
|
return `${seconds}s`;
|
||||||
}
|
}
|
||||||
|
|
||||||
return "<0s";
|
return "<1s";
|
||||||
}
|
}
|
||||||
|
|
||||||
export default {
|
export default {
|
||||||
@@ -121,9 +132,10 @@ export default {
|
|||||||
id: row.id,
|
id: row.id,
|
||||||
name: row.name,
|
name: row.name,
|
||||||
duration: this.taskDuration(row),
|
duration: this.taskDuration(row),
|
||||||
time: moment(row.started).format("dd, MMM Do YYYY, HH:mm:ss"),
|
time: moment.utc(row.started).local().format("dd, MMM Do YYYY, HH:mm:ss"),
|
||||||
logs: row.id,
|
logs: null,
|
||||||
status: row.return_code === 0 ? "ok" : "failed"
|
status: row.return_code === 0 ? "ok" : "failed",
|
||||||
|
_row: row
|
||||||
}));
|
}));
|
||||||
});
|
});
|
||||||
},
|
},
|
||||||
@@ -137,6 +149,11 @@ export default {
|
|||||||
const end = moment.utc(task.ended);
|
const end = moment.utc(task.ended);
|
||||||
|
|
||||||
return humanDuration(end.diff(start))
|
return humanDuration(end.diff(start))
|
||||||
|
},
|
||||||
|
deleteLogs(taskId) {
|
||||||
|
Sist2AdminApi.deleteTaskLogs(taskId).then(() => {
|
||||||
|
this.updateHistory();
|
||||||
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -147,4 +164,8 @@ export default {
|
|||||||
font-family: monospace;
|
font-family: monospace;
|
||||||
font-size: 12px;
|
font-size: 12px;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.btn-link {
|
||||||
|
padding: 0;
|
||||||
|
}
|
||||||
</style>
|
</style>
|
||||||
|
|||||||
117
sist2-admin/frontend/src/views/UserScript.vue
Normal file
@@ -0,0 +1,117 @@
|
|||||||
|
<template>
|
||||||
|
<b-progress v-if="loading" striped animated value="100"></b-progress>
|
||||||
|
<b-card v-else>
|
||||||
|
<b-card-title>
|
||||||
|
{{ $route.params.name }}
|
||||||
|
{{ $t("script") }}
|
||||||
|
</b-card-title>
|
||||||
|
|
||||||
|
<div class="mb-3">
|
||||||
|
<b-button variant="danger" @click="deleteScript()">{{ $t("delete") }}</b-button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<b-card>
|
||||||
|
<h5>{{ $t("testScript") }}</h5>
|
||||||
|
|
||||||
|
<b-row>
|
||||||
|
<b-col cols="11">
|
||||||
|
<JobSelect @change="onJobSelect($event)"></JobSelect>
|
||||||
|
</b-col>
|
||||||
|
<b-col cols="1">
|
||||||
|
<b-button :disabled="!selectedTestJob" variant="primary" @click="testScript()">{{ $t("test") }}
|
||||||
|
</b-button>
|
||||||
|
</b-col>
|
||||||
|
</b-row>
|
||||||
|
|
||||||
|
</b-card>
|
||||||
|
<br/>
|
||||||
|
|
||||||
|
<label>{{ $t("scriptType") }}</label>
|
||||||
|
<b-form-select :options="['git', 'simple']" v-model="script.type" @change="update()"></b-form-select>
|
||||||
|
|
||||||
|
<template v-if="script.type === 'git'">
|
||||||
|
<label>{{ $t("gitRepository") }}</label>
|
||||||
|
<b-form-input v-model="script.git_repository" placeholder="https://github.com/example/example.git"
|
||||||
|
@change="update()"></b-form-input>
|
||||||
|
|
||||||
|
<label>{{ $t("extraArgs") }}</label>
|
||||||
|
<b-form-input v-model="script.extra_args" @change="update()" class="text-monospace"></b-form-input>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<template v-if="script.type === 'simple'">
|
||||||
|
|
||||||
|
<label>{{ $t("scriptCode") }}</label>
|
||||||
|
<p>Find sist2-python documentation <a href="https://sist2-python.readthedocs.io/" target="_blank">here</a></p>
|
||||||
|
<b-textarea rows="15" class="text-monospace" v-model="script.script" @change="update()" spellcheck="false"></b-textarea>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<template v-if="script.type === 'local'">
|
||||||
|
<!-- TODO-->
|
||||||
|
</template>
|
||||||
|
|
||||||
|
|
||||||
|
</b-card>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
|
||||||
|
import Sist2AdminApi from "@/Sist2AdminApi";
|
||||||
|
import JobOptions from "@/components/JobOptions.vue";
|
||||||
|
import JobCheckboxGroup from "@/components/JobCheckboxGroup.vue";
|
||||||
|
import JobSelect from "@/components/JobSelect.vue";
|
||||||
|
|
||||||
|
export default {
|
||||||
|
name: "UserScript",
|
||||||
|
components: {JobSelect, JobCheckboxGroup, JobOptions},
|
||||||
|
data() {
|
||||||
|
return {
|
||||||
|
loading: true,
|
||||||
|
script: null,
|
||||||
|
selectedTestJob: null
|
||||||
|
}
|
||||||
|
},
|
||||||
|
methods: {
|
||||||
|
update() {
|
||||||
|
Sist2AdminApi.updateUserScript(this.name, this.script);
|
||||||
|
},
|
||||||
|
onJobSelect(job) {
|
||||||
|
this.selectedTestJob = job;
|
||||||
|
},
|
||||||
|
deleteScript() {
|
||||||
|
Sist2AdminApi.deleteUserScript(this.name)
|
||||||
|
.then(() => {
|
||||||
|
this.$router.push("/");
|
||||||
|
})
|
||||||
|
.catch(err => {
|
||||||
|
this.$bvToast.toast("Cannot delete user script " +
|
||||||
|
"because it is referenced by a job", {
|
||||||
|
title: "Error",
|
||||||
|
variant: "danger",
|
||||||
|
toaster: "b-toaster-bottom-right"
|
||||||
|
});
|
||||||
|
})
|
||||||
|
},
|
||||||
|
testScript() {
|
||||||
|
Sist2AdminApi.testUserScript(this.name, this.selectedTestJob)
|
||||||
|
.then(() => {
|
||||||
|
this.$bvToast.toast(this.$t("runJobConfirmation"), {
|
||||||
|
title: this.$t("runJobConfirmationTitle"),
|
||||||
|
variant: "success",
|
||||||
|
toaster: "b-toaster-bottom-right"
|
||||||
|
});
|
||||||
|
})
|
||||||
|
}
|
||||||
|
},
|
||||||
|
mounted() {
|
||||||
|
Sist2AdminApi.getUserScript(this.name).then(resp => {
|
||||||
|
this.script = resp.data;
|
||||||
|
this.loading = false;
|
||||||
|
});
|
||||||
|
},
|
||||||
|
computed: {
|
||||||
|
name() {
|
||||||
|
return this.$route.params.name;
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
</script>
|
||||||
@@ -3,3 +3,5 @@ git+https://github.com/simon987/hexlib.git
|
|||||||
uvicorn
|
uvicorn
|
||||||
websockets
|
websockets
|
||||||
pycron
|
pycron
|
||||||
|
GitPython
|
||||||
|
git+https://github.com/sist2app/sist2-python.git
|
||||||
@@ -2,6 +2,7 @@ import asyncio
|
|||||||
import os
|
import os
|
||||||
import signal
|
import signal
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
from time import sleep
|
||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
import requests
|
import requests
|
||||||
@@ -18,11 +19,14 @@ from websockets.exceptions import ConnectionClosed
|
|||||||
|
|
||||||
import cron
|
import cron
|
||||||
from config import LOG_FOLDER, logger, WEBSERVER_PORT, DATA_FOLDER, SIST2_BINARY
|
from config import LOG_FOLDER, logger, WEBSERVER_PORT, DATA_FOLDER, SIST2_BINARY
|
||||||
from jobs import Sist2Job, Sist2ScanTask, TaskQueue, Sist2IndexTask, JobStatus
|
from jobs import Sist2Job, Sist2ScanTask, TaskQueue, Sist2IndexTask, JobStatus, Sist2UserScriptTask
|
||||||
from notifications import Subscribe, Notifications
|
from notifications import Subscribe, Notifications
|
||||||
from sist2 import Sist2
|
from sist2 import Sist2, Sist2SearchBackend
|
||||||
from state import migrate_v1_to_v2, RUNNING_FRONTENDS, TESSERACT_LANGS, DB_SCHEMA_VERSION
|
from state import migrate_v1_to_v2, RUNNING_FRONTENDS, TESSERACT_LANGS, DB_SCHEMA_VERSION, migrate_v3_to_v4, \
|
||||||
|
get_log_files_to_remove, delete_log_file, create_default_search_backends
|
||||||
from web import Sist2Frontend
|
from web import Sist2Frontend
|
||||||
|
from script import UserScript, SCRIPT_TEMPLATES
|
||||||
|
from util import tail_sync, pid_is_running
|
||||||
|
|
||||||
sist2 = Sist2(SIST2_BINARY, DATA_FOLDER)
|
sist2 = Sist2(SIST2_BINARY, DATA_FOLDER)
|
||||||
db = PersistentState(dbfile=os.path.join(DATA_FOLDER, "state.db"))
|
db = PersistentState(dbfile=os.path.join(DATA_FOLDER, "state.db"))
|
||||||
@@ -51,7 +55,8 @@ async def home():
|
|||||||
async def api():
|
async def api():
|
||||||
return {
|
return {
|
||||||
"tesseract_langs": TESSERACT_LANGS,
|
"tesseract_langs": TESSERACT_LANGS,
|
||||||
"logs_folder": LOG_FOLDER
|
"logs_folder": LOG_FOLDER,
|
||||||
|
"user_script_templates": list(SCRIPT_TEMPLATES.keys())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@@ -73,16 +78,14 @@ async def get_frontend(name: str):
|
|||||||
raise HTTPException(status_code=404)
|
raise HTTPException(status_code=404)
|
||||||
|
|
||||||
|
|
||||||
@app.get("/api/job/")
|
@app.get("/api/job")
|
||||||
async def get_jobs():
|
async def get_jobs():
|
||||||
return list(db["jobs"])
|
return list(db["jobs"])
|
||||||
|
|
||||||
|
|
||||||
@app.put("/api/job/{name:str}")
|
@app.put("/api/job/{name:str}")
|
||||||
async def update_job(name: str, new_job: Sist2Job):
|
async def update_job(name: str, new_job: Sist2Job):
|
||||||
# TODO: Check etag
|
new_job.last_modified = datetime.utcnow()
|
||||||
|
|
||||||
new_job.last_modified = datetime.now()
|
|
||||||
job = db["jobs"][name]
|
job = db["jobs"][name]
|
||||||
if not job:
|
if not job:
|
||||||
raise HTTPException(status_code=404)
|
raise HTTPException(status_code=404)
|
||||||
@@ -114,12 +117,10 @@ async def update_job(name: str, new_job: Sist2Job):
|
|||||||
async def update_frontend(name: str, frontend: Sist2Frontend):
|
async def update_frontend(name: str, frontend: Sist2Frontend):
|
||||||
db["frontends"][name] = frontend
|
db["frontends"][name] = frontend
|
||||||
|
|
||||||
# TODO: Check etag
|
|
||||||
|
|
||||||
return "ok"
|
return "ok"
|
||||||
|
|
||||||
|
|
||||||
@app.get("/api/task/")
|
@app.get("/api/task")
|
||||||
async def get_tasks():
|
async def get_tasks():
|
||||||
return list(map(lambda t: t.json(), task_queue.tasks()))
|
return list(map(lambda t: t.json(), task_queue.tasks()))
|
||||||
|
|
||||||
@@ -134,43 +135,102 @@ async def kill_job(task_id: str):
|
|||||||
return task_queue.kill_task(task_id)
|
return task_queue.kill_task(task_id)
|
||||||
|
|
||||||
|
|
||||||
|
@app.post("/api/task/{task_id:str}/delete_logs")
|
||||||
|
async def delete_task_logs(task_id: str):
|
||||||
|
if not db["task_done"][task_id]:
|
||||||
|
raise HTTPException(status_code=404)
|
||||||
|
|
||||||
|
delete_log_file(db, task_id)
|
||||||
|
|
||||||
|
return "ok"
|
||||||
|
|
||||||
|
|
||||||
def _run_job(job: Sist2Job):
|
def _run_job(job: Sist2Job):
|
||||||
job.last_modified = datetime.now()
|
job.last_modified = datetime.utcnow()
|
||||||
if job.status == JobStatus("created"):
|
if job.status == JobStatus("created"):
|
||||||
job.status = JobStatus("started")
|
job.status = JobStatus("started")
|
||||||
db["jobs"][job.name] = job
|
db["jobs"][job.name] = job
|
||||||
|
|
||||||
scan_task = Sist2ScanTask(job, f"Scan [{job.name}]")
|
scan_task = Sist2ScanTask(job, f"Scan [{job.name}]")
|
||||||
index_task = Sist2IndexTask(job, f"Index [{job.name}]", depends_on=scan_task)
|
|
||||||
|
index_depends_on = scan_task
|
||||||
|
script_tasks = []
|
||||||
|
for script_name in job.user_scripts:
|
||||||
|
script = db["user_scripts"][script_name]
|
||||||
|
|
||||||
|
task = Sist2UserScriptTask(script, job, f"Script <{script_name}> [{job.name}]", depends_on=scan_task)
|
||||||
|
script_tasks.append(task)
|
||||||
|
index_depends_on = task
|
||||||
|
|
||||||
|
index_task = Sist2IndexTask(job, f"Index [{job.name}]", depends_on=index_depends_on)
|
||||||
|
|
||||||
task_queue.submit(scan_task)
|
task_queue.submit(scan_task)
|
||||||
|
for task in script_tasks:
|
||||||
|
task_queue.submit(task)
|
||||||
task_queue.submit(index_task)
|
task_queue.submit(index_task)
|
||||||
|
|
||||||
|
|
||||||
@app.get("/api/job/{name:str}/run")
|
@app.get("/api/job/{name:str}/run")
|
||||||
async def run_job(name: str):
|
async def run_job(name: str, full: bool = False):
|
||||||
job = db["jobs"][name]
|
job: Sist2Job = db["jobs"][name]
|
||||||
if not job:
|
if not job:
|
||||||
raise HTTPException(status_code=404)
|
raise HTTPException(status_code=404)
|
||||||
|
|
||||||
|
if full:
|
||||||
|
job.do_full_scan = True
|
||||||
|
|
||||||
_run_job(job)
|
_run_job(job)
|
||||||
|
|
||||||
return "ok"
|
return "ok"
|
||||||
|
|
||||||
|
|
||||||
|
@app.get("/api/user_script/{name:str}/run")
|
||||||
|
def run_user_script(name: str, job: str):
|
||||||
|
script = db["user_scripts"][name]
|
||||||
|
if not script:
|
||||||
|
raise HTTPException(status_code=404)
|
||||||
|
job = db["jobs"][job]
|
||||||
|
if not job:
|
||||||
|
raise HTTPException(status_code=404)
|
||||||
|
|
||||||
|
script_task = Sist2UserScriptTask(script, job, f"Script <{name}> [{job.name}]")
|
||||||
|
|
||||||
|
task_queue.submit(script_task)
|
||||||
|
|
||||||
|
return "ok"
|
||||||
|
|
||||||
|
|
||||||
|
@app.get("/api/job/{name:str}/logs_to_delete")
|
||||||
|
async def task_history(n: int, name: str):
|
||||||
|
return get_log_files_to_remove(db, name, n)
|
||||||
|
|
||||||
|
|
||||||
@app.delete("/api/job/{name:str}")
|
@app.delete("/api/job/{name:str}")
|
||||||
async def delete_job(name: str):
|
async def delete_job(name: str):
|
||||||
job = db["jobs"][name]
|
job: Sist2Job = db["jobs"][name]
|
||||||
if job:
|
if not job:
|
||||||
del db["jobs"][name]
|
|
||||||
else:
|
|
||||||
raise HTTPException(status_code=404)
|
raise HTTPException(status_code=404)
|
||||||
|
|
||||||
|
if any(name in frontend.jobs for frontend in db["frontends"]):
|
||||||
|
raise HTTPException(status_code=400, detail="in use (frontend)")
|
||||||
|
|
||||||
|
try:
|
||||||
|
os.remove(job.previous_index)
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
|
||||||
|
del db["jobs"][name]
|
||||||
|
|
||||||
|
return "ok"
|
||||||
|
|
||||||
|
|
||||||
@app.delete("/api/frontend/{name:str}")
|
@app.delete("/api/frontend/{name:str}")
|
||||||
async def delete_frontend(name: str):
|
async def delete_frontend(name: str):
|
||||||
if name in RUNNING_FRONTENDS:
|
if name in RUNNING_FRONTENDS:
|
||||||
|
try:
|
||||||
os.kill(RUNNING_FRONTENDS[name], signal.SIGTERM)
|
os.kill(RUNNING_FRONTENDS[name], signal.SIGTERM)
|
||||||
|
except ProcessLookupError:
|
||||||
|
pass
|
||||||
del RUNNING_FRONTENDS[name]
|
del RUNNING_FRONTENDS[name]
|
||||||
|
|
||||||
frontend = db["frontends"][name]
|
frontend = db["frontends"][name]
|
||||||
@@ -215,7 +275,7 @@ def check_es_version(es_url: str, insecure: bool):
|
|||||||
es_url = f"{url.scheme}://{url.hostname}:{url.port}"
|
es_url = f"{url.scheme}://{url.hostname}:{url.port}"
|
||||||
else:
|
else:
|
||||||
auth = None
|
auth = None
|
||||||
r = requests.get(es_url, verify=insecure, auth=auth)
|
r = requests.get(es_url, verify=not insecure, auth=auth)
|
||||||
except SSLError:
|
except SSLError:
|
||||||
return {
|
return {
|
||||||
"ok": False,
|
"ok": False,
|
||||||
@@ -251,10 +311,33 @@ def check_es_version(es_url: str, insecure: bool):
|
|||||||
|
|
||||||
|
|
||||||
def start_frontend_(frontend: Sist2Frontend):
|
def start_frontend_(frontend: Sist2Frontend):
|
||||||
frontend.web_options.indices = list(map(lambda j: db["jobs"][j].index_path, frontend.jobs))
|
frontend.web_options.indices = [
|
||||||
|
os.path.join(DATA_FOLDER, db["jobs"][j].index_path)
|
||||||
|
for j in frontend.jobs
|
||||||
|
]
|
||||||
|
|
||||||
|
backend_name = frontend.web_options.search_backend
|
||||||
|
search_backend = db["search_backends"][backend_name]
|
||||||
|
if search_backend is None:
|
||||||
|
logger.error(
|
||||||
|
f"Error while running task: search backend not found: {backend_name}")
|
||||||
|
return -1
|
||||||
|
|
||||||
|
logger.debug(f"Fetched search backend options for {backend_name}")
|
||||||
|
|
||||||
|
pid = sist2.web(frontend.web_options, search_backend, frontend.name)
|
||||||
|
|
||||||
|
sleep(0.2)
|
||||||
|
if not pid_is_running(pid):
|
||||||
|
frontend_log = frontend.get_log_path(LOG_FOLDER)
|
||||||
|
logger.error(f"Frontend exited too quickly, check {frontend_log} for more details:")
|
||||||
|
for line in tail_sync(frontend.get_log_path(LOG_FOLDER), 3):
|
||||||
|
logger.error(line.strip())
|
||||||
|
|
||||||
|
return False
|
||||||
|
|
||||||
pid = sist2.web(frontend.web_options, frontend.name)
|
|
||||||
RUNNING_FRONTENDS[frontend.name] = pid
|
RUNNING_FRONTENDS[frontend.name] = pid
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
@app.post("/api/frontend/{name:str}/start")
|
@app.post("/api/frontend/{name:str}/start")
|
||||||
@@ -263,7 +346,12 @@ async def start_frontend(name: str):
|
|||||||
if not frontend:
|
if not frontend:
|
||||||
raise HTTPException(status_code=404)
|
raise HTTPException(status_code=404)
|
||||||
|
|
||||||
start_frontend_(frontend)
|
ok = start_frontend_(frontend)
|
||||||
|
|
||||||
|
if not ok:
|
||||||
|
raise HTTPException(status_code=500)
|
||||||
|
|
||||||
|
return "ok"
|
||||||
|
|
||||||
|
|
||||||
@app.post("/api/frontend/{name:str}/stop")
|
@app.post("/api/frontend/{name:str}/stop")
|
||||||
@@ -273,7 +361,7 @@ async def stop_frontend(name: str):
|
|||||||
del RUNNING_FRONTENDS[name]
|
del RUNNING_FRONTENDS[name]
|
||||||
|
|
||||||
|
|
||||||
@app.get("/api/frontend/")
|
@app.get("/api/frontend")
|
||||||
async def get_frontends():
|
async def get_frontends():
|
||||||
res = []
|
res = []
|
||||||
for frontend in db["frontends"]:
|
for frontend in db["frontends"]:
|
||||||
@@ -283,6 +371,115 @@ async def get_frontends():
|
|||||||
return res
|
return res
|
||||||
|
|
||||||
|
|
||||||
|
@app.get("/api/search_backend")
|
||||||
|
async def get_search_backends():
|
||||||
|
return list(db["search_backends"])
|
||||||
|
|
||||||
|
|
||||||
|
@app.put("/api/search_backend/{name:str}")
|
||||||
|
async def update_search_backend(name: str, backend: Sist2SearchBackend):
|
||||||
|
if not db["search_backends"][name]:
|
||||||
|
raise HTTPException(status_code=404)
|
||||||
|
|
||||||
|
db["search_backends"][name] = backend
|
||||||
|
return "ok"
|
||||||
|
|
||||||
|
|
||||||
|
@app.get("/api/search_backend/{name:str}")
|
||||||
|
def get_search_backend(name: str):
|
||||||
|
backend = db["search_backends"][name]
|
||||||
|
if not backend:
|
||||||
|
raise HTTPException(status_code=404)
|
||||||
|
|
||||||
|
return backend
|
||||||
|
|
||||||
|
|
||||||
|
@app.delete("/api/search_backend/{name:str}")
|
||||||
|
def delete_search_backend(name: str):
|
||||||
|
backend: Sist2SearchBackend = db["search_backends"][name]
|
||||||
|
if not backend:
|
||||||
|
raise HTTPException(status_code=404)
|
||||||
|
|
||||||
|
if any(frontend.web_options.search_backend == name for frontend in db["frontends"]):
|
||||||
|
raise HTTPException(status_code=400, detail="in use (frontend)")
|
||||||
|
|
||||||
|
if any(job.index_options.search_backend == name for job in db["jobs"]):
|
||||||
|
raise HTTPException(status_code=400, detail="in use (job)")
|
||||||
|
|
||||||
|
del db["search_backends"][name]
|
||||||
|
|
||||||
|
try:
|
||||||
|
os.remove(os.path.join(DATA_FOLDER, backend.search_index))
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
|
||||||
|
return "ok"
|
||||||
|
|
||||||
|
|
||||||
|
@app.post("/api/search_backend/{name:str}")
|
||||||
|
def create_search_backend(name: str):
|
||||||
|
if db["search_backends"][name] is not None:
|
||||||
|
return HTTPException(status_code=400, detail="already exists")
|
||||||
|
|
||||||
|
backend = Sist2SearchBackend.create_default(name)
|
||||||
|
db["search_backends"][name] = backend
|
||||||
|
|
||||||
|
return backend
|
||||||
|
|
||||||
|
|
||||||
|
@app.delete("/api/user_script/{name:str}")
|
||||||
|
def delete_user_script(name: str):
|
||||||
|
if db["user_scripts"][name] is None:
|
||||||
|
return HTTPException(status_code=404)
|
||||||
|
|
||||||
|
if any(name in job.user_scripts for job in db["jobs"]):
|
||||||
|
raise HTTPException(status_code=400, detail="in use (job)")
|
||||||
|
|
||||||
|
script: UserScript = db["user_scripts"][name]
|
||||||
|
script.delete_dir()
|
||||||
|
|
||||||
|
del db["user_scripts"][name]
|
||||||
|
|
||||||
|
return "ok"
|
||||||
|
|
||||||
|
|
||||||
|
@app.post("/api/user_script/{name:str}")
|
||||||
|
def create_user_script(name: str, template: str):
|
||||||
|
if db["user_scripts"][name] is not None:
|
||||||
|
return HTTPException(status_code=400, detail="already exists")
|
||||||
|
|
||||||
|
script = SCRIPT_TEMPLATES[template](name)
|
||||||
|
db["user_scripts"][name] = script
|
||||||
|
|
||||||
|
return script
|
||||||
|
|
||||||
|
|
||||||
|
@app.get("/api/user_script")
|
||||||
|
async def get_user_scripts():
|
||||||
|
return list(db["user_scripts"])
|
||||||
|
|
||||||
|
|
||||||
|
@app.get("/api/user_script/{name:str}")
|
||||||
|
async def get_user_script(name: str):
|
||||||
|
backend = db["user_scripts"][name]
|
||||||
|
if not backend:
|
||||||
|
raise HTTPException(status_code=404)
|
||||||
|
|
||||||
|
return backend
|
||||||
|
|
||||||
|
|
||||||
|
@app.put("/api/user_script/{name:str}")
|
||||||
|
async def update_user_script(name: str, script: UserScript):
|
||||||
|
previous_version: UserScript = db["user_scripts"][name]
|
||||||
|
|
||||||
|
if previous_version and previous_version.git_repository != script.git_repository:
|
||||||
|
script.force_clone = True
|
||||||
|
|
||||||
|
db["user_scripts"][name] = script
|
||||||
|
|
||||||
|
return "ok"
|
||||||
|
|
||||||
|
|
||||||
def tail(filepath: str, n: int):
|
def tail(filepath: str, n: int):
|
||||||
with open(filepath) as file:
|
with open(filepath) as file:
|
||||||
|
|
||||||
@@ -321,7 +518,6 @@ async def ws_tail_log(websocket: WebSocket):
|
|||||||
async with Subscribe(notifications) as ob:
|
async with Subscribe(notifications) as ob:
|
||||||
async for notification in ob.notifications():
|
async for notification in ob.notifications():
|
||||||
await websocket.send_json(notification)
|
await websocket.send_json(notification)
|
||||||
print(notification)
|
|
||||||
|
|
||||||
except ConnectionClosed:
|
except ConnectionClosed:
|
||||||
return
|
return
|
||||||
@@ -352,7 +548,7 @@ async def ws_tail_log(websocket: WebSocket, task_id: str, n: int):
|
|||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
uvicorn.run(app, port=WEBSERVER_PORT, host="0.0.0.0")
|
uvicorn.run(app, port=WEBSERVER_PORT, host="0.0.0.0", timeout_graceful_shutdown=0)
|
||||||
|
|
||||||
|
|
||||||
def initialize_db():
|
def initialize_db():
|
||||||
@@ -361,6 +557,8 @@ def initialize_db():
|
|||||||
frontend = Sist2Frontend.create_default("default")
|
frontend = Sist2Frontend.create_default("default")
|
||||||
db["frontends"]["default"] = frontend
|
db["frontends"]["default"] = frontend
|
||||||
|
|
||||||
|
create_default_search_backends(db)
|
||||||
|
|
||||||
logger.info("Initialized database.")
|
logger.info("Initialized database.")
|
||||||
|
|
||||||
|
|
||||||
@@ -381,6 +579,13 @@ if __name__ == '__main__':
|
|||||||
if db["sist2_admin"]["info"]["version"] == "2":
|
if db["sist2_admin"]["info"]["version"] == "2":
|
||||||
logger.error("Cannot migrate database from v2 to v3. Delete state.db to proceed.")
|
logger.error("Cannot migrate database from v2 to v3. Delete state.db to proceed.")
|
||||||
exit(-1)
|
exit(-1)
|
||||||
|
if db["sist2_admin"]["info"]["version"] == "3":
|
||||||
|
logger.info("Migrating to v4 database schema")
|
||||||
|
migrate_v3_to_v4(db)
|
||||||
|
|
||||||
|
if db["sist2_admin"]["info"]["version"] != DB_SCHEMA_VERSION:
|
||||||
|
raise Exception(f"Incompatible database {db.dbfile}. "
|
||||||
|
f"Automatic migration is not available, please delete the database file to continue.")
|
||||||
|
|
||||||
start_frontends()
|
start_frontends()
|
||||||
cron.initialize(db, _run_job)
|
cron.initialize(db, _run_job)
|
||||||
|
|||||||
@@ -9,9 +9,11 @@ MAX_LOG_SIZE = 1 * 1024 * 1024
|
|||||||
SIST2_BINARY = os.environ.get("SIST2_BINARY", "/root/sist2")
|
SIST2_BINARY = os.environ.get("SIST2_BINARY", "/root/sist2")
|
||||||
DATA_FOLDER = os.environ.get("DATA_FOLDER", "/sist2-admin/")
|
DATA_FOLDER = os.environ.get("DATA_FOLDER", "/sist2-admin/")
|
||||||
LOG_FOLDER = os.path.join(DATA_FOLDER, "logs")
|
LOG_FOLDER = os.path.join(DATA_FOLDER, "logs")
|
||||||
|
SCRIPT_FOLDER = os.path.join(DATA_FOLDER, "scripts")
|
||||||
WEBSERVER_PORT = 8080
|
WEBSERVER_PORT = 8080
|
||||||
|
|
||||||
os.makedirs(LOG_FOLDER, exist_ok=True)
|
os.makedirs(LOG_FOLDER, exist_ok=True)
|
||||||
|
os.makedirs(SCRIPT_FOLDER, exist_ok=True)
|
||||||
os.makedirs(DATA_FOLDER, exist_ok=True)
|
os.makedirs(DATA_FOLDER, exist_ok=True)
|
||||||
|
|
||||||
logger = logging.Logger("sist2-admin")
|
logger = logging.Logger("sist2-admin")
|
||||||
|
|||||||
@@ -10,7 +10,9 @@ from jobs import Sist2Job
|
|||||||
|
|
||||||
|
|
||||||
def _check_schedule(db: PersistentState, run_job):
|
def _check_schedule(db: PersistentState, run_job):
|
||||||
for job in db["jobs"]:
|
jobs = list(db["jobs"])
|
||||||
|
|
||||||
|
for job in jobs:
|
||||||
job: Sist2Job
|
job: Sist2Job
|
||||||
|
|
||||||
if job.schedule_enabled:
|
if job.schedule_enabled:
|
||||||
|
|||||||
@@ -1,23 +1,29 @@
|
|||||||
import json
|
import json
|
||||||
import logging
|
import logging
|
||||||
import os.path
|
import os.path
|
||||||
|
import shlex
|
||||||
import signal
|
import signal
|
||||||
import uuid
|
import uuid
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from enum import Enum
|
from enum import Enum
|
||||||
|
from io import TextIOWrapper
|
||||||
from logging import FileHandler
|
from logging import FileHandler
|
||||||
|
from subprocess import Popen
|
||||||
|
import subprocess
|
||||||
from threading import Lock, Thread
|
from threading import Lock, Thread
|
||||||
from time import sleep
|
from time import sleep
|
||||||
|
from typing import List
|
||||||
from uuid import uuid4, UUID
|
from uuid import uuid4, UUID
|
||||||
|
|
||||||
from hexlib.db import PersistentState
|
from hexlib.db import PersistentState
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
|
|
||||||
from config import logger, LOG_FOLDER
|
from config import logger, LOG_FOLDER, DATA_FOLDER
|
||||||
from notifications import Notifications
|
from notifications import Notifications
|
||||||
from sist2 import ScanOptions, IndexOptions, Sist2
|
from sist2 import ScanOptions, IndexOptions, Sist2
|
||||||
from state import RUNNING_FRONTENDS
|
from state import RUNNING_FRONTENDS, get_log_files_to_remove, delete_log_file
|
||||||
from web import Sist2Frontend
|
from web import Sist2Frontend
|
||||||
|
from script import UserScript
|
||||||
|
|
||||||
|
|
||||||
class JobStatus(Enum):
|
class JobStatus(Enum):
|
||||||
@@ -32,9 +38,13 @@ class Sist2Job(BaseModel):
|
|||||||
scan_options: ScanOptions
|
scan_options: ScanOptions
|
||||||
index_options: IndexOptions
|
index_options: IndexOptions
|
||||||
|
|
||||||
|
user_scripts: List[str] = []
|
||||||
|
|
||||||
cron_expression: str
|
cron_expression: str
|
||||||
schedule_enabled: bool = False
|
schedule_enabled: bool = False
|
||||||
|
|
||||||
|
keep_last_n_logs: int = -1
|
||||||
|
|
||||||
previous_index: str = None
|
previous_index: str = None
|
||||||
index_path: str = None
|
index_path: str = None
|
||||||
previous_index_path: str = None
|
previous_index_path: str = None
|
||||||
@@ -53,15 +63,10 @@ class Sist2Job(BaseModel):
|
|||||||
name=name,
|
name=name,
|
||||||
scan_options=ScanOptions(path="/"),
|
scan_options=ScanOptions(path="/"),
|
||||||
index_options=IndexOptions(),
|
index_options=IndexOptions(),
|
||||||
last_modified=datetime.now(),
|
last_modified=datetime.utcnow(),
|
||||||
cron_expression="0 0 * * *"
|
cron_expression="0 0 * * *"
|
||||||
)
|
)
|
||||||
|
|
||||||
# @validator("etag", always=True)
|
|
||||||
# def validate_etag(cls, value, values):
|
|
||||||
# s = values["name"] + values["scan_options"].json() + values["index_options"].json() + values["cron_expression"]
|
|
||||||
# return md5(s.encode()).hexdigest()
|
|
||||||
|
|
||||||
|
|
||||||
class Sist2TaskProgress:
|
class Sist2TaskProgress:
|
||||||
|
|
||||||
@@ -111,10 +116,14 @@ class Sist2Task:
|
|||||||
self._logger.info(json.dumps(log_json))
|
self._logger.info(json.dumps(log_json))
|
||||||
|
|
||||||
def run(self, sist2: Sist2, db: PersistentState):
|
def run(self, sist2: Sist2, db: PersistentState):
|
||||||
self.started = datetime.now()
|
self.started = datetime.utcnow()
|
||||||
|
|
||||||
logger.info(f"Started task {self.display_name}")
|
logger.info(f"Started task {self.display_name}")
|
||||||
|
|
||||||
|
def set_pid(self, pid):
|
||||||
|
self.pid = pid
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class Sist2ScanTask(Sist2Task):
|
class Sist2ScanTask(Sist2Task):
|
||||||
|
|
||||||
@@ -128,18 +137,17 @@ class Sist2ScanTask(Sist2Task):
|
|||||||
else:
|
else:
|
||||||
self.job.scan_options.output = None
|
self.job.scan_options.output = None
|
||||||
|
|
||||||
def set_pid(pid):
|
return_code = sist2.scan(self.job.scan_options, logs_cb=self.log_callback, set_pid_cb=self.set_pid)
|
||||||
self.pid = pid
|
self.ended = datetime.utcnow()
|
||||||
|
|
||||||
return_code = sist2.scan(self.job.scan_options, logs_cb=self.log_callback, set_pid_cb=set_pid)
|
is_ok = (return_code in (0, 1)) if "debug" in sist2.bin_path else (return_code == 0)
|
||||||
self.ended = datetime.now()
|
|
||||||
|
|
||||||
if return_code != 0:
|
if not is_ok:
|
||||||
self._logger.error(json.dumps({"sist2-admin": f"Process returned non-zero exit code ({return_code})"}))
|
self._logger.error(json.dumps({"sist2-admin": f"Process returned non-zero exit code ({return_code})"}))
|
||||||
logger.info(f"Task {self.display_name} failed ({return_code})")
|
logger.info(f"Task {self.display_name} failed ({return_code})")
|
||||||
else:
|
else:
|
||||||
self.job.index_path = self.job.scan_options.output
|
self.job.index_path = self.job.scan_options.output
|
||||||
self.job.last_index_date = datetime.now()
|
self.job.last_index_date = datetime.utcnow()
|
||||||
self.job.do_full_scan = False
|
self.job.do_full_scan = False
|
||||||
db["jobs"][self.job.name] = self.job
|
db["jobs"][self.job.name] = self.job
|
||||||
self._logger.info(json.dumps({"sist2-admin": f"Save last_index_date={self.job.last_index_date}"}))
|
self._logger.info(json.dumps({"sist2-admin": f"Save last_index_date={self.job.last_index_date}"}))
|
||||||
@@ -147,7 +155,7 @@ class Sist2ScanTask(Sist2Task):
|
|||||||
logger.info(f"Completed {self.display_name} ({return_code=})")
|
logger.info(f"Completed {self.display_name} ({return_code=})")
|
||||||
|
|
||||||
# Remove old index
|
# Remove old index
|
||||||
if return_code == 0:
|
if is_ok:
|
||||||
if self.job.previous_index_path is not None and self.job.previous_index_path != self.job.index_path:
|
if self.job.previous_index_path is not None and self.job.previous_index_path != self.job.index_path:
|
||||||
self._logger.info(json.dumps({"sist2-admin": f"Remove {self.job.previous_index_path=}"}))
|
self._logger.info(json.dumps({"sist2-admin": f"Remove {self.job.previous_index_path=}"}))
|
||||||
try:
|
try:
|
||||||
@@ -158,6 +166,9 @@ class Sist2ScanTask(Sist2Task):
|
|||||||
self.job.previous_index_path = self.job.index_path
|
self.job.previous_index_path = self.job.index_path
|
||||||
db["jobs"][self.job.name] = self.job
|
db["jobs"][self.job.name] = self.job
|
||||||
|
|
||||||
|
if is_ok:
|
||||||
|
return 0
|
||||||
|
|
||||||
return return_code
|
return return_code
|
||||||
|
|
||||||
|
|
||||||
@@ -171,12 +182,19 @@ class Sist2IndexTask(Sist2Task):
|
|||||||
|
|
||||||
self.job.index_options.path = self.job.scan_options.output
|
self.job.index_options.path = self.job.scan_options.output
|
||||||
|
|
||||||
return_code = sist2.index(self.job.index_options, logs_cb=self.log_callback)
|
search_backend = db["search_backends"][self.job.index_options.search_backend]
|
||||||
self.ended = datetime.now()
|
if search_backend is None:
|
||||||
|
logger.error(f"Error while running task: search backend not found: {self.job.index_options.search_backend}")
|
||||||
|
return -1
|
||||||
|
|
||||||
|
logger.debug(f"Fetched search backend options for {self.job.index_options.search_backend}")
|
||||||
|
|
||||||
|
return_code = sist2.index(self.job.index_options, search_backend, logs_cb=self.log_callback, set_pid_cb=self.set_pid)
|
||||||
|
self.ended = datetime.utcnow()
|
||||||
|
|
||||||
duration = self.ended - self.started
|
duration = self.ended - self.started
|
||||||
|
|
||||||
ok = return_code == 0
|
ok = return_code in (0, 1)
|
||||||
|
|
||||||
if ok:
|
if ok:
|
||||||
self.restart_running_frontends(db, sist2)
|
self.restart_running_frontends(db, sist2)
|
||||||
@@ -186,7 +204,7 @@ class Sist2IndexTask(Sist2Task):
|
|||||||
self.job.previous_index_path = self.job.index_path
|
self.job.previous_index_path = self.job.index_path
|
||||||
db["jobs"][self.job.name] = self.job
|
db["jobs"][self.job.name] = self.job
|
||||||
|
|
||||||
self._logger.info(json.dumps({"sist2-admin": f"Sist2Scan task finished {return_code=}, {duration=}"}))
|
self._logger.info(json.dumps({"sist2-admin": f"Sist2Scan task finished {return_code=}, {duration=}, {ok=}"}))
|
||||||
|
|
||||||
logger.info(f"Completed {self.display_name} ({return_code=})")
|
logger.info(f"Completed {self.display_name} ({return_code=})")
|
||||||
|
|
||||||
@@ -206,14 +224,84 @@ class Sist2IndexTask(Sist2Task):
|
|||||||
except ChildProcessError:
|
except ChildProcessError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
frontend.web_options.indices = map(lambda j: db["jobs"][j].index_path, frontend.jobs)
|
backend_name = frontend.web_options.search_backend
|
||||||
|
search_backend = db["search_backends"][backend_name]
|
||||||
|
if search_backend is None:
|
||||||
|
logger.error(f"Error while running task: search backend not found: {backend_name}")
|
||||||
|
return -1
|
||||||
|
|
||||||
pid = sist2.web(frontend.web_options, frontend.name)
|
logger.debug(f"Fetched search backend options for {backend_name}")
|
||||||
|
|
||||||
|
frontend.web_options.indices = [
|
||||||
|
os.path.join(DATA_FOLDER, db["jobs"][j].index_path)
|
||||||
|
for j in frontend.jobs
|
||||||
|
]
|
||||||
|
|
||||||
|
pid = sist2.web(frontend.web_options, search_backend, frontend.name)
|
||||||
RUNNING_FRONTENDS[frontend_name] = pid
|
RUNNING_FRONTENDS[frontend_name] = pid
|
||||||
|
|
||||||
self._logger.info(json.dumps({"sist2-admin": f"Restart frontend {pid=} {frontend_name=}"}))
|
self._logger.info(json.dumps({"sist2-admin": f"Restart frontend {pid=} {frontend_name=}"}))
|
||||||
|
|
||||||
|
|
||||||
|
class Sist2UserScriptTask(Sist2Task):
|
||||||
|
|
||||||
|
def __init__(self, user_script: UserScript, job: Sist2Job, display_name: str, depends_on: Sist2Task = None):
|
||||||
|
super().__init__(job, display_name, depends_on=depends_on.id if depends_on else None)
|
||||||
|
self.user_script = user_script
|
||||||
|
|
||||||
|
def run(self, sist2: Sist2, db: PersistentState):
|
||||||
|
super().run(sist2, db)
|
||||||
|
|
||||||
|
try:
|
||||||
|
self.user_script.setup(self.log_callback, self.set_pid)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Setup for {self.user_script.name} failed: ")
|
||||||
|
logger.exception(e)
|
||||||
|
self.log_callback({"sist2-admin": f"Setup for {self.user_script.name} failed: {e}"})
|
||||||
|
return -1
|
||||||
|
|
||||||
|
executable = self.user_script.get_executable()
|
||||||
|
index_path = os.path.join(DATA_FOLDER, self.job.index_path)
|
||||||
|
extra_args = self.user_script.extra_args
|
||||||
|
|
||||||
|
args = [
|
||||||
|
executable,
|
||||||
|
index_path,
|
||||||
|
*shlex.split(extra_args)
|
||||||
|
]
|
||||||
|
|
||||||
|
self.log_callback({"sist2-admin": f"Starting user script with {executable=}, {index_path=}, {extra_args=}"})
|
||||||
|
|
||||||
|
proc = Popen(args, stdout=subprocess.PIPE, stderr=subprocess.PIPE, cwd=self.user_script.script_dir())
|
||||||
|
self.set_pid(proc.pid)
|
||||||
|
|
||||||
|
t_stderr = Thread(target=self._consume_logs, args=(self.log_callback, proc, "stderr", False))
|
||||||
|
t_stderr.start()
|
||||||
|
|
||||||
|
self._consume_logs(self.log_callback, proc, "stdout", True)
|
||||||
|
|
||||||
|
self.ended = datetime.utcnow()
|
||||||
|
|
||||||
|
return 0
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def _consume_logs(logs_cb, proc, stream, wait):
|
||||||
|
pipe_wrapper = TextIOWrapper(getattr(proc, stream), encoding="utf8", errors="ignore")
|
||||||
|
try:
|
||||||
|
for line in pipe_wrapper:
|
||||||
|
if line.strip() == "":
|
||||||
|
continue
|
||||||
|
if line.startswith("$PROGRESS"):
|
||||||
|
progress = json.loads(line[len("$PROGRESS "):])
|
||||||
|
logs_cb({"progress": progress})
|
||||||
|
continue
|
||||||
|
logs_cb({stream: line})
|
||||||
|
finally:
|
||||||
|
if wait:
|
||||||
|
proc.wait()
|
||||||
|
pipe_wrapper.close()
|
||||||
|
|
||||||
|
|
||||||
class TaskQueue:
|
class TaskQueue:
|
||||||
def __init__(self, sist2: Sist2, db: PersistentState, notifications: Notifications):
|
def __init__(self, sist2: Sist2, db: PersistentState, notifications: Notifications):
|
||||||
self._lock = Lock()
|
self._lock = Lock()
|
||||||
@@ -301,8 +389,14 @@ class TaskQueue:
|
|||||||
"ended": task.ended,
|
"ended": task.ended,
|
||||||
"started": task.started,
|
"started": task.started,
|
||||||
"name": task.display_name,
|
"name": task.display_name,
|
||||||
"return_code": task_result
|
"return_code": task_result,
|
||||||
|
"has_logs": 1
|
||||||
}
|
}
|
||||||
|
|
||||||
|
logs_to_delete = get_log_files_to_remove(self._db, task.job.name, task.job.keep_last_n_logs)
|
||||||
|
for row in logs_to_delete:
|
||||||
|
delete_log_file(self._db, row["id"])
|
||||||
|
|
||||||
if isinstance(task, Sist2IndexTask):
|
if isinstance(task, Sist2IndexTask):
|
||||||
self._notifications.notify({
|
self._notifications.notify({
|
||||||
"message": "notifications.indexCompleted",
|
"message": "notifications.indexCompleted",
|
||||||
|
|||||||
130
sist2-admin/sist2_admin/script.py
Normal file
@@ -0,0 +1,130 @@
|
|||||||
|
import os
|
||||||
|
import shutil
|
||||||
|
import stat
|
||||||
|
import subprocess
|
||||||
|
from enum import Enum
|
||||||
|
|
||||||
|
from git import Repo
|
||||||
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
from config import SCRIPT_FOLDER
|
||||||
|
|
||||||
|
|
||||||
|
class ScriptType(Enum):
|
||||||
|
LOCAL = "local"
|
||||||
|
SIMPLE = "simple"
|
||||||
|
GIT = "git"
|
||||||
|
|
||||||
|
|
||||||
|
def set_executable(file):
|
||||||
|
os.chmod(file, os.stat(file).st_mode | stat.S_IEXEC)
|
||||||
|
|
||||||
|
|
||||||
|
def _initialize_git_repository(url, path, log_cb, force_clone, set_pid_cb):
|
||||||
|
log_cb({"sist2-admin": f"Cloning {url}"})
|
||||||
|
|
||||||
|
if force_clone or not os.path.exists(os.path.join(path, ".git")):
|
||||||
|
if force_clone:
|
||||||
|
shutil.rmtree(path, ignore_errors=True)
|
||||||
|
Repo.clone_from(url, path)
|
||||||
|
else:
|
||||||
|
repo = Repo(path)
|
||||||
|
repo.remote("origin").pull()
|
||||||
|
|
||||||
|
setup_script = os.path.join(path, "setup.sh")
|
||||||
|
if setup_script:
|
||||||
|
log_cb({"sist2-admin": f"Executing setup script {setup_script}"})
|
||||||
|
|
||||||
|
set_executable(setup_script)
|
||||||
|
proc = subprocess.Popen([setup_script], cwd=path, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
|
||||||
|
set_pid_cb(proc.pid)
|
||||||
|
proc.wait()
|
||||||
|
stdout = proc.stdout.read()
|
||||||
|
|
||||||
|
for line in stdout.split(b"\n"):
|
||||||
|
if line:
|
||||||
|
log_cb({"stdout": line.decode()})
|
||||||
|
|
||||||
|
log_cb({"stdout": f"Executed setup script {setup_script}, return code = {proc.returncode}"})
|
||||||
|
|
||||||
|
if proc.returncode != 0:
|
||||||
|
raise Exception("Error when running setup script!")
|
||||||
|
|
||||||
|
log_cb({"sist2-admin": f"Initialized git repository in {path}"})
|
||||||
|
|
||||||
|
|
||||||
|
class UserScript(BaseModel):
|
||||||
|
name: str
|
||||||
|
type: ScriptType
|
||||||
|
git_repository: str = None
|
||||||
|
force_clone: bool = False
|
||||||
|
script: str = None
|
||||||
|
extra_args: str = ""
|
||||||
|
|
||||||
|
def script_dir(self):
|
||||||
|
return os.path.join(SCRIPT_FOLDER, self.name)
|
||||||
|
|
||||||
|
def setup(self, log_cb, set_pid_cb):
|
||||||
|
os.makedirs(self.script_dir(), exist_ok=True)
|
||||||
|
|
||||||
|
if self.type == ScriptType.GIT:
|
||||||
|
_initialize_git_repository(self.git_repository, self.script_dir(), log_cb, self.force_clone, set_pid_cb)
|
||||||
|
self.force_clone = False
|
||||||
|
elif self.type == ScriptType.SIMPLE:
|
||||||
|
self._setup_simple()
|
||||||
|
|
||||||
|
set_executable(self.get_executable())
|
||||||
|
|
||||||
|
def _setup_simple(self):
|
||||||
|
with open(self.get_executable(), "w") as f:
|
||||||
|
f.write(
|
||||||
|
"#!/bin/bash\n"
|
||||||
|
"python run.py \"$@\""
|
||||||
|
)
|
||||||
|
|
||||||
|
with open(os.path.join(self.script_dir(), "run.py"), "w") as f:
|
||||||
|
f.write(self.script)
|
||||||
|
|
||||||
|
def get_executable(self):
|
||||||
|
return os.path.join(self.script_dir(), "run.sh")
|
||||||
|
|
||||||
|
def delete_dir(self):
|
||||||
|
shutil.rmtree(self.script_dir(), ignore_errors=True)
|
||||||
|
|
||||||
|
|
||||||
|
SCRIPT_TEMPLATES = {
|
||||||
|
"CLIP - Generate embeddings to predict the most relevant image based on the text prompt": lambda name: UserScript(
|
||||||
|
name=name,
|
||||||
|
type=ScriptType.GIT,
|
||||||
|
git_repository="https://github.com/sist2app/sist2-script-clip",
|
||||||
|
extra_args="--num-tags=1 --tags-file=general.txt --color=#dcd7ff"
|
||||||
|
),
|
||||||
|
"Whisper - Speech to text with OpenAI Whisper": lambda name: UserScript(
|
||||||
|
name=name,
|
||||||
|
type=ScriptType.GIT,
|
||||||
|
git_repository="https://github.com/simon987/sist2-script-whisper",
|
||||||
|
extra_args="--model=base --num-threads=4 --color=#51da4c --tag"
|
||||||
|
),
|
||||||
|
"Hamburger - Simple script example": lambda name: UserScript(
|
||||||
|
name=name,
|
||||||
|
type=ScriptType.SIMPLE,
|
||||||
|
script=
|
||||||
|
'from sist2 import Sist2Index\n'
|
||||||
|
'import sys\n'
|
||||||
|
'\n'
|
||||||
|
'index = Sist2Index(sys.argv[1])\n'
|
||||||
|
'for doc in index.document_iter():\n'
|
||||||
|
' doc.json_data["tag"] = ["hamburger.#00FF00"]\n'
|
||||||
|
' index.update_document(doc)\n'
|
||||||
|
'\n'
|
||||||
|
'index.sync_tag_table()\n'
|
||||||
|
'index.commit()\n'
|
||||||
|
'\n'
|
||||||
|
'print("Done!")\n'
|
||||||
|
),
|
||||||
|
"(Blank)": lambda name: UserScript(
|
||||||
|
name=name,
|
||||||
|
type=ScriptType.SIMPLE,
|
||||||
|
script=""
|
||||||
|
)
|
||||||
|
}
|
||||||
@@ -2,9 +2,11 @@ import datetime
|
|||||||
import json
|
import json
|
||||||
import logging
|
import logging
|
||||||
import os.path
|
import os.path
|
||||||
|
import sys
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
from enum import Enum
|
||||||
from io import TextIOWrapper
|
from io import TextIOWrapper
|
||||||
from logging import FileHandler
|
from logging import FileHandler, StreamHandler
|
||||||
from subprocess import Popen, PIPE
|
from subprocess import Popen, PIPE
|
||||||
from tempfile import NamedTemporaryFile
|
from tempfile import NamedTemporaryFile
|
||||||
from threading import Thread
|
from threading import Thread
|
||||||
@@ -12,7 +14,7 @@ from typing import List
|
|||||||
|
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
|
|
||||||
from config import logger, LOG_FOLDER
|
from config import logger, LOG_FOLDER, DATA_FOLDER
|
||||||
|
|
||||||
|
|
||||||
class Sist2Version:
|
class Sist2Version:
|
||||||
@@ -25,74 +27,53 @@ class Sist2Version:
|
|||||||
return f"{self.major}.{self.minor}.{self.patch}"
|
return f"{self.major}.{self.minor}.{self.patch}"
|
||||||
|
|
||||||
|
|
||||||
class WebOptions(BaseModel):
|
class SearchBackendType(Enum):
|
||||||
indices: List[str] = []
|
SQLITE = "sqlite"
|
||||||
|
ELASTICSEARCH = "elasticsearch"
|
||||||
|
|
||||||
|
|
||||||
|
class Sist2SearchBackend(BaseModel):
|
||||||
|
backend_type: SearchBackendType = SearchBackendType("elasticsearch")
|
||||||
|
name: str
|
||||||
|
|
||||||
|
search_index: str = ""
|
||||||
|
|
||||||
es_url: str = "http://elasticsearch:9200"
|
es_url: str = "http://elasticsearch:9200"
|
||||||
es_insecure_ssl: bool = False
|
es_insecure_ssl: bool = False
|
||||||
es_index: str = "sist2"
|
es_index: str = "sist2"
|
||||||
bind: str = "0.0.0.0:4090"
|
threads: int = 1
|
||||||
auth: str = None
|
batch_size: int = 70
|
||||||
tag_auth: str = None
|
|
||||||
tagline: str = "Lightning-fast file system indexer and search tool"
|
|
||||||
dev: bool = False
|
|
||||||
lang: str = "en"
|
|
||||||
auth0_audience: str = None
|
|
||||||
auth0_domain: str = None
|
|
||||||
auth0_client_id: str = None
|
|
||||||
auth0_public_key: str = None
|
|
||||||
auth0_public_key_file: str = None
|
|
||||||
|
|
||||||
def __init__(self, **kwargs):
|
@staticmethod
|
||||||
super().__init__(**kwargs)
|
def create_default(name: str, backend_type: SearchBackendType = SearchBackendType("elasticsearch")):
|
||||||
|
return Sist2SearchBackend(
|
||||||
def args(self):
|
name=name,
|
||||||
args = ["web", f"--es-url={self.es_url}", f"--bind={self.bind}",
|
search_index=f"search-index-{name.replace('/', '_')}.sist2",
|
||||||
f"--tagline={self.tagline}", f"--lang={self.lang}"]
|
backend_type=backend_type
|
||||||
|
)
|
||||||
if self.auth0_audience:
|
|
||||||
args.append(f"--auth0-audience={self.auth0_audience}")
|
|
||||||
if self.auth0_domain:
|
|
||||||
args.append(f"--auth0-domain={self.auth0_domain}")
|
|
||||||
if self.auth0_client_id:
|
|
||||||
args.append(f"--auth0-client-id={self.auth0_client_id}")
|
|
||||||
if self.auth0_public_key_file:
|
|
||||||
args.append(f"--auth0-public-key-file={self.auth0_public_key_file}")
|
|
||||||
if self.es_insecure_ssl:
|
|
||||||
args.append(f"--es-insecure-ssl")
|
|
||||||
if self.auth:
|
|
||||||
args.append(f"--auth={self.auth}")
|
|
||||||
if self.tag_auth:
|
|
||||||
args.append(f"--tag-auth={self.tag_auth}")
|
|
||||||
if self.dev:
|
|
||||||
args.append(f"--dev")
|
|
||||||
|
|
||||||
args.extend(self.indices)
|
|
||||||
|
|
||||||
return args
|
|
||||||
|
|
||||||
|
|
||||||
class IndexOptions(BaseModel):
|
class IndexOptions(BaseModel):
|
||||||
path: str = None
|
path: str = None
|
||||||
threads: int = 1
|
|
||||||
es_url: str = "http://elasticsearch:9200"
|
|
||||||
es_insecure_ssl: bool = False
|
|
||||||
es_index: str = "sist2"
|
|
||||||
incremental_index: bool = True
|
incremental_index: bool = True
|
||||||
script: str = ""
|
search_backend: str = None
|
||||||
script_file: str = None
|
|
||||||
batch_size: int = 70
|
|
||||||
|
|
||||||
def __init__(self, **kwargs):
|
def __init__(self, **kwargs):
|
||||||
super().__init__(**kwargs)
|
super().__init__(**kwargs)
|
||||||
|
|
||||||
def args(self):
|
def args(self, search_backend):
|
||||||
|
absolute_path = os.path.join(DATA_FOLDER, self.path)
|
||||||
|
|
||||||
args = ["index", self.path, f"--threads={self.threads}", f"--es-url={self.es_url}",
|
if search_backend.backend_type == SearchBackendType("sqlite"):
|
||||||
f"--es-index={self.es_index}", f"--batch-size={self.batch_size}"]
|
search_index_absolute = os.path.join(DATA_FOLDER, search_backend.search_index)
|
||||||
|
args = ["sqlite-index", absolute_path, "--search-index", search_index_absolute]
|
||||||
|
else:
|
||||||
|
args = ["index", absolute_path, f"--threads={search_backend.threads}",
|
||||||
|
f"--es-url={search_backend.es_url}",
|
||||||
|
f"--es-index={search_backend.es_index}",
|
||||||
|
f"--batch-size={search_backend.batch_size}"]
|
||||||
|
|
||||||
if self.script_file:
|
if search_backend.es_insecure_ssl:
|
||||||
args.append(f"--script-file={self.script_file}")
|
|
||||||
if self.es_insecure_ssl:
|
|
||||||
args.append(f"--es-insecure-ssl")
|
args.append(f"--es-insecure-ssl")
|
||||||
if self.incremental_index:
|
if self.incremental_index:
|
||||||
args.append(f"--incremental-index")
|
args.append(f"--incremental-index")
|
||||||
@@ -109,7 +90,7 @@ ARCHIVE_RECURSE = "recurse"
|
|||||||
class ScanOptions(BaseModel):
|
class ScanOptions(BaseModel):
|
||||||
path: str
|
path: str
|
||||||
threads: int = 1
|
threads: int = 1
|
||||||
thumbnail_quality: int = 2
|
thumbnail_quality: int = 50
|
||||||
thumbnail_size: int = 552
|
thumbnail_size: int = 552
|
||||||
thumbnail_count: int = 1
|
thumbnail_count: int = 1
|
||||||
content_size: int = 32768
|
content_size: int = 32768
|
||||||
@@ -137,9 +118,12 @@ class ScanOptions(BaseModel):
|
|||||||
super().__init__(**kwargs)
|
super().__init__(**kwargs)
|
||||||
|
|
||||||
def args(self):
|
def args(self):
|
||||||
|
|
||||||
|
output_path = os.path.join(DATA_FOLDER, self.output)
|
||||||
|
|
||||||
args = ["scan", self.path, f"--threads={self.threads}", f"--thumbnail-quality={self.thumbnail_quality}",
|
args = ["scan", self.path, f"--threads={self.threads}", f"--thumbnail-quality={self.thumbnail_quality}",
|
||||||
f"--thumbnail-count={self.thumbnail_count}", f"--thumbnail-size={self.thumbnail_size}",
|
f"--thumbnail-count={self.thumbnail_count}", f"--thumbnail-size={self.thumbnail_size}",
|
||||||
f"--content-size={self.content_size}", f"--output={self.output}", f"--depth={self.depth}",
|
f"--content-size={self.content_size}", f"--output={output_path}", f"--depth={self.depth}",
|
||||||
f"--archive={self.archive}", f"--mem-buffer={self.mem_buffer}"]
|
f"--archive={self.archive}", f"--mem-buffer={self.mem_buffer}"]
|
||||||
|
|
||||||
if self.incremental:
|
if self.incremental:
|
||||||
@@ -201,30 +185,83 @@ class Sist2Index:
|
|||||||
return self._descriptor["name"]
|
return self._descriptor["name"]
|
||||||
|
|
||||||
|
|
||||||
|
class WebOptions(BaseModel):
|
||||||
|
indices: List[str] = []
|
||||||
|
|
||||||
|
search_backend: str = "elasticsearch"
|
||||||
|
|
||||||
|
bind: str = "0.0.0.0:4090"
|
||||||
|
auth: str = None
|
||||||
|
tag_auth: str = None
|
||||||
|
tagline: str = "Lightning-fast file system indexer and search tool"
|
||||||
|
dev: bool = False
|
||||||
|
lang: str = "en"
|
||||||
|
auth0_audience: str = None
|
||||||
|
auth0_domain: str = None
|
||||||
|
auth0_client_id: str = None
|
||||||
|
auth0_public_key: str = None
|
||||||
|
auth0_public_key_file: str = None
|
||||||
|
verbose: bool = False
|
||||||
|
|
||||||
|
def __init__(self, **kwargs):
|
||||||
|
super().__init__(**kwargs)
|
||||||
|
|
||||||
|
def args(self, search_backend: Sist2SearchBackend):
|
||||||
|
args = ["web", f"--bind={self.bind}", f"--tagline={self.tagline}",
|
||||||
|
f"--lang={self.lang}"]
|
||||||
|
|
||||||
|
if search_backend.backend_type == SearchBackendType("sqlite"):
|
||||||
|
search_index_absolute = os.path.join(DATA_FOLDER, search_backend.search_index)
|
||||||
|
args.append(f"--search-index={search_index_absolute}")
|
||||||
|
else:
|
||||||
|
args.append(f"--es-url={search_backend.es_url}")
|
||||||
|
args.append(f"--es-index={search_backend.es_index}")
|
||||||
|
if search_backend.es_insecure_ssl:
|
||||||
|
args.append(f"--es-insecure-ssl")
|
||||||
|
|
||||||
|
if self.auth0_audience:
|
||||||
|
args.append(f"--auth0-audience={self.auth0_audience}")
|
||||||
|
if self.auth0_domain:
|
||||||
|
args.append(f"--auth0-domain={self.auth0_domain}")
|
||||||
|
if self.auth0_client_id:
|
||||||
|
args.append(f"--auth0-client-id={self.auth0_client_id}")
|
||||||
|
if self.auth0_public_key_file:
|
||||||
|
args.append(f"--auth0-public-key-file={self.auth0_public_key_file}")
|
||||||
|
if self.auth:
|
||||||
|
args.append(f"--auth={self.auth}")
|
||||||
|
if self.tag_auth:
|
||||||
|
args.append(f"--tag-auth={self.tag_auth}")
|
||||||
|
if self.dev:
|
||||||
|
args.append(f"--dev")
|
||||||
|
if self.verbose:
|
||||||
|
args.append(f"--very-verbose")
|
||||||
|
|
||||||
|
args.extend(self.indices)
|
||||||
|
|
||||||
|
return args
|
||||||
|
|
||||||
|
|
||||||
class Sist2:
|
class Sist2:
|
||||||
|
|
||||||
def __init__(self, bin_path: str, data_directory: str):
|
def __init__(self, bin_path: str, data_directory: str):
|
||||||
self._bin_path = bin_path
|
self.bin_path = bin_path
|
||||||
self._data_dir = data_directory
|
self._data_dir = data_directory
|
||||||
|
|
||||||
def index(self, options: IndexOptions, logs_cb):
|
def index(self, options: IndexOptions, search_backend: Sist2SearchBackend, logs_cb, set_pid_cb):
|
||||||
|
|
||||||
if options.script:
|
|
||||||
with NamedTemporaryFile("w", prefix="sist2-admin", suffix=".painless", delete=False) as f:
|
|
||||||
f.write(options.script)
|
|
||||||
options.script_file = f.name
|
|
||||||
else:
|
|
||||||
options.script_file = None
|
|
||||||
|
|
||||||
args = [
|
args = [
|
||||||
self._bin_path,
|
self.bin_path,
|
||||||
*options.args(),
|
*options.args(search_backend),
|
||||||
"--json-logs",
|
"--json-logs",
|
||||||
"--very-verbose"
|
"--very-verbose"
|
||||||
]
|
]
|
||||||
|
|
||||||
|
logs_cb({"sist2-admin": f"Starting sist2 command with args {args}"})
|
||||||
proc = Popen(args, stdout=PIPE, stderr=PIPE)
|
proc = Popen(args, stdout=PIPE, stderr=PIPE)
|
||||||
|
|
||||||
t_stderr = Thread(target=self._consume_logs_stderr, args=(logs_cb, proc))
|
set_pid_cb(proc.pid)
|
||||||
|
|
||||||
|
t_stderr = Thread(target=self._consume_logs_stderr, args=(logs_cb, None, proc))
|
||||||
t_stderr.start()
|
t_stderr.start()
|
||||||
|
|
||||||
self._consume_logs_stdout(logs_cb, proc)
|
self._consume_logs_stdout(logs_cb, proc)
|
||||||
@@ -236,13 +273,10 @@ class Sist2:
|
|||||||
def scan(self, options: ScanOptions, logs_cb, set_pid_cb):
|
def scan(self, options: ScanOptions, logs_cb, set_pid_cb):
|
||||||
|
|
||||||
if options.output is None:
|
if options.output is None:
|
||||||
options.output = os.path.join(
|
options.output = f"scan-{options.name.replace('/', '_')}-{datetime.utcnow()}.sist2"
|
||||||
self._data_dir,
|
|
||||||
f"scan-{options.name.replace('/', '_')}-{datetime.now()}.sist2"
|
|
||||||
)
|
|
||||||
|
|
||||||
args = [
|
args = [
|
||||||
self._bin_path,
|
self.bin_path,
|
||||||
*options.args(),
|
*options.args(),
|
||||||
"--json-logs",
|
"--json-logs",
|
||||||
"--very-verbose"
|
"--very-verbose"
|
||||||
@@ -254,7 +288,7 @@ class Sist2:
|
|||||||
|
|
||||||
set_pid_cb(proc.pid)
|
set_pid_cb(proc.pid)
|
||||||
|
|
||||||
t_stderr = Thread(target=self._consume_logs_stderr, args=(logs_cb, proc))
|
t_stderr = Thread(target=self._consume_logs_stderr, args=(logs_cb, None, proc))
|
||||||
t_stderr.start()
|
t_stderr.start()
|
||||||
|
|
||||||
self._consume_logs_stdout(logs_cb, proc)
|
self._consume_logs_stdout(logs_cb, proc)
|
||||||
@@ -264,7 +298,7 @@ class Sist2:
|
|||||||
return proc.returncode
|
return proc.returncode
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def _consume_logs_stderr(logs_cb, proc):
|
def _consume_logs_stderr(logs_cb, exit_cb, proc):
|
||||||
pipe_wrapper = TextIOWrapper(proc.stderr, encoding="utf8", errors="ignore")
|
pipe_wrapper = TextIOWrapper(proc.stderr, encoding="utf8", errors="ignore")
|
||||||
try:
|
try:
|
||||||
for line in pipe_wrapper:
|
for line in pipe_wrapper:
|
||||||
@@ -272,7 +306,9 @@ class Sist2:
|
|||||||
continue
|
continue
|
||||||
logs_cb({"stderr": line})
|
logs_cb({"stderr": line})
|
||||||
finally:
|
finally:
|
||||||
proc.wait()
|
return_code = proc.wait()
|
||||||
|
if exit_cb:
|
||||||
|
exit_cb(return_code)
|
||||||
pipe_wrapper.close()
|
pipe_wrapper.close()
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
@@ -290,7 +326,7 @@ class Sist2:
|
|||||||
except NameError:
|
except NameError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
def web(self, options: WebOptions, name: str):
|
def web(self, options: WebOptions, search_backend: Sist2SearchBackend, name: str):
|
||||||
|
|
||||||
if options.auth0_public_key:
|
if options.auth0_public_key:
|
||||||
with NamedTemporaryFile("w", prefix="sist2-admin", suffix=".txt", delete=False) as f:
|
with NamedTemporaryFile("w", prefix="sist2-admin", suffix=".txt", delete=False) as f:
|
||||||
@@ -300,21 +336,25 @@ class Sist2:
|
|||||||
options.auth0_public_key_file = None
|
options.auth0_public_key_file = None
|
||||||
|
|
||||||
args = [
|
args = [
|
||||||
self._bin_path,
|
self.bin_path,
|
||||||
*options.args()
|
*options.args(search_backend)
|
||||||
]
|
]
|
||||||
|
|
||||||
web_logger = logging.Logger(name=f"sist2-frontend-{name}")
|
web_logger = logging.Logger(name=f"sist2-frontend-{name}")
|
||||||
web_logger.addHandler(FileHandler(os.path.join(LOG_FOLDER, f"frontend-{name}.log")))
|
web_logger.addHandler(FileHandler(os.path.join(LOG_FOLDER, f"frontend-{name}.log")))
|
||||||
|
web_logger.addHandler(StreamHandler())
|
||||||
|
|
||||||
def logs_cb(message):
|
def logs_cb(message):
|
||||||
web_logger.info(json.dumps(message))
|
web_logger.info(json.dumps(message))
|
||||||
|
|
||||||
|
def exit_cb(return_code):
|
||||||
|
logger.info(f"Web frontend exited with return code {return_code}")
|
||||||
|
|
||||||
logger.info(f"Starting frontend {' '.join(args)}")
|
logger.info(f"Starting frontend {' '.join(args)}")
|
||||||
|
|
||||||
proc = Popen(args, stdout=PIPE, stderr=PIPE)
|
proc = Popen(args, stdout=PIPE, stderr=PIPE)
|
||||||
|
|
||||||
t_stderr = Thread(target=self._consume_logs_stderr, args=(logs_cb, proc))
|
t_stderr = Thread(target=self._consume_logs_stderr, args=(logs_cb, exit_cb, proc))
|
||||||
t_stderr.start()
|
t_stderr.start()
|
||||||
|
|
||||||
t_stdout = Thread(target=self._consume_logs_stdout, args=(logs_cb, proc))
|
t_stdout = Thread(target=self._consume_logs_stdout, args=(logs_cb, proc))
|
||||||
|
|||||||
@@ -1,16 +1,20 @@
|
|||||||
from typing import Dict
|
from typing import Dict
|
||||||
|
import os
|
||||||
import shutil
|
import shutil
|
||||||
|
|
||||||
from hexlib.db import Table, PersistentState
|
from hexlib.db import Table, PersistentState
|
||||||
import pickle
|
import pickle
|
||||||
|
|
||||||
from tesseract import get_tesseract_langs
|
from tesseract import get_tesseract_langs
|
||||||
|
import sqlite3
|
||||||
|
from config import LOG_FOLDER, logger
|
||||||
|
from sist2 import SearchBackendType, Sist2SearchBackend
|
||||||
|
|
||||||
RUNNING_FRONTENDS: Dict[str, int] = {}
|
RUNNING_FRONTENDS: Dict[str, int] = {}
|
||||||
|
|
||||||
TESSERACT_LANGS = get_tesseract_langs()
|
TESSERACT_LANGS = get_tesseract_langs()
|
||||||
|
|
||||||
DB_SCHEMA_VERSION = "3"
|
DB_SCHEMA_VERSION = "5"
|
||||||
|
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
|
|
||||||
@@ -50,8 +54,35 @@ class PickleTable(Table):
|
|||||||
yield dict((k, _deserialize(v)) for k, v in row.items())
|
yield dict((k, _deserialize(v)) for k, v in row.items())
|
||||||
|
|
||||||
|
|
||||||
def migrate_v1_to_v2(db: PersistentState):
|
def get_log_files_to_remove(db: PersistentState, job_name: str, n: int):
|
||||||
|
if n < 0:
|
||||||
|
return []
|
||||||
|
|
||||||
|
counter = 0
|
||||||
|
to_remove = []
|
||||||
|
|
||||||
|
for row in db["task_done"].sql("WHERE has_logs=1 ORDER BY started DESC"):
|
||||||
|
if row["name"].endswith(f"[{job_name}]"):
|
||||||
|
counter += 1
|
||||||
|
|
||||||
|
if counter > n:
|
||||||
|
to_remove.append(row)
|
||||||
|
|
||||||
|
return to_remove
|
||||||
|
|
||||||
|
|
||||||
|
def delete_log_file(db: PersistentState, task_id: str):
|
||||||
|
db["task_done"][task_id] = {
|
||||||
|
"has_logs": 0
|
||||||
|
}
|
||||||
|
|
||||||
|
try:
|
||||||
|
os.remove(os.path.join(LOG_FOLDER, f"sist2-{task_id}.log"))
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
def migrate_v1_to_v2(db: PersistentState):
|
||||||
shutil.copy(db.dbfile, db.dbfile + "-before-migrate-v2.bak")
|
shutil.copy(db.dbfile, db.dbfile + "-before-migrate-v2.bak")
|
||||||
|
|
||||||
# Frontends
|
# Frontends
|
||||||
@@ -77,3 +108,29 @@ def migrate_v1_to_v2(db: PersistentState):
|
|||||||
db["sist2_admin"]["info"] = {
|
db["sist2_admin"]["info"] = {
|
||||||
"version": "2"
|
"version": "2"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def create_default_search_backends(db: PersistentState):
|
||||||
|
es_backend = Sist2SearchBackend.create_default(name="elasticsearch",
|
||||||
|
backend_type=SearchBackendType("elasticsearch"))
|
||||||
|
db["search_backends"]["elasticsearch"] = es_backend
|
||||||
|
sqlite_backend = Sist2SearchBackend.create_default(name="sqlite", backend_type=SearchBackendType("sqlite"))
|
||||||
|
db["search_backends"]["sqlite"] = sqlite_backend
|
||||||
|
|
||||||
|
|
||||||
|
def migrate_v3_to_v4(db: PersistentState):
|
||||||
|
shutil.copy(db.dbfile, db.dbfile + "-before-migrate-v4.bak")
|
||||||
|
|
||||||
|
create_default_search_backends(db)
|
||||||
|
|
||||||
|
try:
|
||||||
|
conn = sqlite3.connect(db.dbfile)
|
||||||
|
conn.execute("ALTER TABLE task_done ADD COLUMN has_logs INTEGER DEFAULT 1")
|
||||||
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
|
except Exception as e:
|
||||||
|
logger.exception(e)
|
||||||
|
|
||||||
|
db["sist2_admin"]["info"] = {
|
||||||
|
"version": "4"
|
||||||
|
}
|
||||||
|
|||||||
41
sist2-admin/sist2_admin/util.py
Normal file
@@ -0,0 +1,41 @@
|
|||||||
|
from glob import glob
|
||||||
|
import os
|
||||||
|
from config import DATA_FOLDER
|
||||||
|
|
||||||
|
|
||||||
|
def get_old_index_files(name):
|
||||||
|
files = glob(os.path.join(DATA_FOLDER, f"scan-{name.replace('/', '_')}-*.sist2"))
|
||||||
|
files = list(sorted(files, key=lambda f: os.stat(f).st_mtime))
|
||||||
|
files = files[-1:]
|
||||||
|
|
||||||
|
return files
|
||||||
|
|
||||||
|
|
||||||
|
def tail_sync(filename, lines=1, _buffer=4098):
|
||||||
|
with open(filename) as f:
|
||||||
|
lines_found = []
|
||||||
|
|
||||||
|
block_counter = -1
|
||||||
|
|
||||||
|
while len(lines_found) < lines:
|
||||||
|
try:
|
||||||
|
f.seek(block_counter * _buffer, os.SEEK_END)
|
||||||
|
except IOError:
|
||||||
|
f.seek(0)
|
||||||
|
lines_found = f.readlines()
|
||||||
|
break
|
||||||
|
|
||||||
|
lines_found = f.readlines()
|
||||||
|
|
||||||
|
block_counter -= 1
|
||||||
|
|
||||||
|
return lines_found[-lines:]
|
||||||
|
|
||||||
|
|
||||||
|
def pid_is_running(pid):
|
||||||
|
try:
|
||||||
|
os.kill(pid, 0)
|
||||||
|
except OSError:
|
||||||
|
return False
|
||||||
|
|
||||||
|
return True
|
||||||
15457
sist2-vue/package-lock.json
generated
@@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "sist2",
|
"name": "sist2",
|
||||||
"version": "2.11.0",
|
"version": "1.0.0",
|
||||||
"private": true,
|
"private": true,
|
||||||
"scripts": {
|
"scripts": {
|
||||||
"serve": "vue-cli-service serve",
|
"serve": "vue-cli-service serve",
|
||||||
@@ -9,7 +9,7 @@
|
|||||||
"dependencies": {
|
"dependencies": {
|
||||||
"@auth0/auth0-spa-js": "^2.0.2",
|
"@auth0/auth0-spa-js": "^2.0.2",
|
||||||
"@egjs/vue-infinitegrid": "3.3.0",
|
"@egjs/vue-infinitegrid": "3.3.0",
|
||||||
"axios": "^0.25.0",
|
"axios": "^1.6.0",
|
||||||
"bootstrap-vue": "^2.21.2",
|
"bootstrap-vue": "^2.21.2",
|
||||||
"core-js": "^3.6.5",
|
"core-js": "^3.6.5",
|
||||||
"d3": "^5.6.1",
|
"d3": "^5.6.1",
|
||||||
@@ -17,6 +17,7 @@
|
|||||||
"dom-to-image": "^2.6.0",
|
"dom-to-image": "^2.6.0",
|
||||||
"fslightbox-vue": "fslightbox-vue.tgz",
|
"fslightbox-vue": "fslightbox-vue.tgz",
|
||||||
"nouislider": "^15.2.0",
|
"nouislider": "^15.2.0",
|
||||||
|
"onnxruntime-web": "1.15.1",
|
||||||
"underscore": "^1.13.1",
|
"underscore": "^1.13.1",
|
||||||
"vue": "^2.6.12",
|
"vue": "^2.6.12",
|
||||||
"vue-color": "^2.8.1",
|
"vue-color": "^2.8.1",
|
||||||
@@ -28,9 +29,9 @@
|
|||||||
},
|
},
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
"@babel/polyfill": "^7.12.1",
|
"@babel/polyfill": "^7.12.1",
|
||||||
|
"@types/underscore": "^1.11.6",
|
||||||
"@vue/cli-plugin-babel": "~5.0.8",
|
"@vue/cli-plugin-babel": "~5.0.8",
|
||||||
"@vue/cli-plugin-router": "~5.0.8",
|
"@vue/cli-plugin-router": "~5.0.8",
|
||||||
"@vue/cli-plugin-typescript": "^5.0.8",
|
|
||||||
"@vue/cli-plugin-vuex": "~5.0.8",
|
"@vue/cli-plugin-vuex": "~5.0.8",
|
||||||
"@vue/cli-service": "^5.0.8",
|
"@vue/cli-service": "^5.0.8",
|
||||||
"@vue/test-utils": "^1.0.3",
|
"@vue/test-utils": "^1.0.3",
|
||||||
@@ -42,8 +43,7 @@
|
|||||||
"portal-vue": "^2.1.7",
|
"portal-vue": "^2.1.7",
|
||||||
"sass": "^1.26.11",
|
"sass": "^1.26.11",
|
||||||
"sass-loader": "^10.0.2",
|
"sass-loader": "^10.0.2",
|
||||||
"typescript": "~4.1.5",
|
"vue-cli-plugin-bootstrap-vue": "~0.8.2",
|
||||||
"vue-cli-plugin-bootstrap-vue": "~0.7.0",
|
|
||||||
"vue-template-compiler": "^2.6.11"
|
"vue-template-compiler": "^2.6.11"
|
||||||
},
|
},
|
||||||
"browserslist": [
|
"browserslist": [
|
||||||
|
|||||||
@@ -19,12 +19,6 @@
|
|||||||
We're sorry but <%= htmlWebpackPlugin.options.title %> doesn't work properly without JavaScript enabled.
|
We're sorry but <%= htmlWebpackPlugin.options.title %> doesn't work properly without JavaScript enabled.
|
||||||
Please enable it to continue.
|
Please enable it to continue.
|
||||||
</strong>
|
</strong>
|
||||||
<br/>
|
|
||||||
<strong>
|
|
||||||
Nous sommes désolés mais <%= htmlWebpackPlugin.options.title %> ne fonctionne pas correctement
|
|
||||||
si JavaScript est activé.
|
|
||||||
Veuillez l'activer pour continuer.
|
|
||||||
</strong>
|
|
||||||
</div>
|
</div>
|
||||||
</noscript>
|
</noscript>
|
||||||
<div id="app"></div>
|
<div id="app"></div>
|
||||||
|
|||||||
@@ -10,7 +10,7 @@
|
|||||||
<b-spinner type="grow" variant="primary"></b-spinner>
|
<b-spinner type="grow" variant="primary"></b-spinner>
|
||||||
</div>
|
</div>
|
||||||
<div class="loading-text">
|
<div class="loading-text">
|
||||||
Loading • Chargement • 装载 • Wird geladen
|
Loading • Chargement • 装载 • Wird geladen • Ładowanie
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
</template>
|
</template>
|
||||||
@@ -19,7 +19,10 @@
|
|||||||
import NavBar from "@/components/NavBar";
|
import NavBar from "@/components/NavBar";
|
||||||
import {mapActions, mapGetters, mapMutations} from "vuex";
|
import {mapActions, mapGetters, mapMutations} from "vuex";
|
||||||
import Sist2Api from "@/Sist2Api";
|
import Sist2Api from "@/Sist2Api";
|
||||||
|
import ModelsRepo from "@/ml/modelsRepo";
|
||||||
import {setupAuth0} from "@/main";
|
import {setupAuth0} from "@/main";
|
||||||
|
import Sist2ElasticsearchQuery from "@/Sist2ElasticsearchQuery";
|
||||||
|
import Sist2SqliteQuery from "@/Sist2SqliteQuery";
|
||||||
|
|
||||||
export default {
|
export default {
|
||||||
components: {NavBar},
|
components: {NavBar},
|
||||||
@@ -36,6 +39,17 @@ export default {
|
|||||||
mounted() {
|
mounted() {
|
||||||
this.$store.dispatch("loadConfiguration").then(() => {
|
this.$store.dispatch("loadConfiguration").then(() => {
|
||||||
this.$root.$i18n.locale = this.$store.state.optLang;
|
this.$root.$i18n.locale = this.$store.state.optLang;
|
||||||
|
ModelsRepo.init(this.$store.getters.mlRepositoryList).catch(err => {
|
||||||
|
this.$bvToast.toast(
|
||||||
|
this.$t("ml.repoFetchError"),
|
||||||
|
{
|
||||||
|
title: this.$t("ml.repoFetchErrorTitle"),
|
||||||
|
noAutoHide: true,
|
||||||
|
toaster: "b-toaster-bottom-right",
|
||||||
|
headerClass: "toast-header-warning",
|
||||||
|
bodyClass: "toast-body-warning",
|
||||||
|
});
|
||||||
|
});
|
||||||
});
|
});
|
||||||
|
|
||||||
this.$store.subscribe((mutation) => {
|
this.$store.subscribe((mutation) => {
|
||||||
@@ -76,6 +90,13 @@ export default {
|
|||||||
|
|
||||||
this.setSist2Info(data);
|
this.setSist2Info(data);
|
||||||
this.setIndices(data.indices)
|
this.setIndices(data.indices)
|
||||||
|
|
||||||
|
if (Sist2Api.backend() === "sqlite") {
|
||||||
|
Sist2Api.init(Sist2SqliteQuery.searchQuery);
|
||||||
|
this.$store.commit("setUiSqliteMode", true);
|
||||||
|
} else {
|
||||||
|
Sist2Api.init(Sist2ElasticsearchQuery.searchQuery);
|
||||||
|
}
|
||||||
});
|
});
|
||||||
},
|
},
|
||||||
methods: {
|
methods: {
|
||||||
@@ -287,15 +308,21 @@ html, body {
|
|||||||
|
|
||||||
.info-icon {
|
.info-icon {
|
||||||
width: 1rem;
|
width: 1rem;
|
||||||
|
min-width: 1rem;
|
||||||
margin-right: 0.2rem;
|
margin-right: 0.2rem;
|
||||||
cursor: pointer;
|
cursor: pointer;
|
||||||
line-height: 1rem;
|
line-height: 1rem;
|
||||||
height: 1rem;
|
height: 1rem;
|
||||||
|
min-height: 1rem;
|
||||||
background-image: url(data:image/svg+xml;base64,PHN2ZyB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciIHhtbG5zOnhsaW5rPSJodHRwOi8vd3d3LnczLm9yZy8xOTk5L3hsaW5rIiB4PSIwcHgiIHk9IjBweCIKICAgICB2aWV3Qm94PSIwIDAgNDI2LjY2NyA0MjYuNjY3IiBzdHlsZT0iZW5hYmxlLWJhY2tncm91bmQ6bmV3IDAgMCA0MjYuNjY3IDQyNi42Njc7IiBmaWxsPSIjZmZmIj4KPGc+CiAgICA8Zz4KICAgICAgICA8Zz4KICAgICAgICAgICAgPHJlY3QgeD0iMTkyIiB5PSIxOTIiIHdpZHRoPSI0Mi42NjciIGhlaWdodD0iMTI4Ii8+CiAgICAgICAgICAgIDxwYXRoIGQ9Ik0yMTMuMzMzLDBDOTUuNDY3LDAsMCw5NS40NjcsMCwyMTMuMzMzczk1LjQ2NywyMTMuMzMzLDIxMy4zMzMsMjEzLjMzM1M0MjYuNjY3LDMzMS4yLDQyNi42NjcsMjEzLjMzMwogICAgICAgICAgICAgICAgUzMzMS4yLDAsMjEzLjMzMywweiBNMjEzLjMzMywzODRjLTk0LjA4LDAtMTcwLjY2Ny03Ni41ODctMTcwLjY2Ny0xNzAuNjY3UzExOS4yNTMsNDIuNjY3LDIxMy4zMzMsNDIuNjY3CiAgICAgICAgICAgICAgICBTMzg0LDExOS4yNTMsMzg0LDIxMy4zMzNTMzA3LjQxMywzODQsMjEzLjMzMywzODR6Ii8+CiAgICAgICAgICAgIDxyZWN0IHg9IjE5MiIgeT0iMTA2LjY2NyIgd2lkdGg9IjQyLjY2NyIgaGVpZ2h0PSI0Mi42NjciLz4KICAgICAgICA8L2c+CiAgICA8L2c+CjwvZz4KPC9zdmc+Cg==);
|
background-image: url(data:image/svg+xml;base64,PHN2ZyB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciIHhtbG5zOnhsaW5rPSJodHRwOi8vd3d3LnczLm9yZy8xOTk5L3hsaW5rIiB4PSIwcHgiIHk9IjBweCIKICAgICB2aWV3Qm94PSIwIDAgNDI2LjY2NyA0MjYuNjY3IiBzdHlsZT0iZW5hYmxlLWJhY2tncm91bmQ6bmV3IDAgMCA0MjYuNjY3IDQyNi42Njc7IiBmaWxsPSIjZmZmIj4KPGc+CiAgICA8Zz4KICAgICAgICA8Zz4KICAgICAgICAgICAgPHJlY3QgeD0iMTkyIiB5PSIxOTIiIHdpZHRoPSI0Mi42NjciIGhlaWdodD0iMTI4Ii8+CiAgICAgICAgICAgIDxwYXRoIGQ9Ik0yMTMuMzMzLDBDOTUuNDY3LDAsMCw5NS40NjcsMCwyMTMuMzMzczk1LjQ2NywyMTMuMzMzLDIxMy4zMzMsMjEzLjMzM1M0MjYuNjY3LDMzMS4yLDQyNi42NjcsMjEzLjMzMwogICAgICAgICAgICAgICAgUzMzMS4yLDAsMjEzLjMzMywweiBNMjEzLjMzMywzODRjLTk0LjA4LDAtMTcwLjY2Ny03Ni41ODctMTcwLjY2Ny0xNzAuNjY3UzExOS4yNTMsNDIuNjY3LDIxMy4zMzMsNDIuNjY3CiAgICAgICAgICAgICAgICBTMzg0LDExOS4yNTMsMzg0LDIxMy4zMzNTMzA3LjQxMywzODQsMjEzLjMzMywzODR6Ii8+CiAgICAgICAgICAgIDxyZWN0IHg9IjE5MiIgeT0iMTA2LjY2NyIgd2lkdGg9IjQyLjY2NyIgaGVpZ2h0PSI0Mi42NjciLz4KICAgICAgICA8L2c+CiAgICA8L2c+CjwvZz4KPC9zdmc+Cg==);
|
||||||
filter: brightness(45%);
|
filter: brightness(45%);
|
||||||
display: block;
|
display: block;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.theme-black .info-icon {
|
||||||
|
filter: brightness(80%);
|
||||||
|
}
|
||||||
|
|
||||||
.tabs {
|
.tabs {
|
||||||
margin-top: 10px;
|
margin-top: 10px;
|
||||||
}
|
}
|
||||||
|
|||||||
628
sist2-vue/src/Sist2Api.js
Normal file
@@ -0,0 +1,628 @@
|
|||||||
|
import axios from "axios";
|
||||||
|
import {strUnescape, lum, sid} from "./util";
|
||||||
|
import Sist2Query from "@/Sist2ElasticsearchQuery";
|
||||||
|
import store from "@/store";
|
||||||
|
|
||||||
|
|
||||||
|
class Sist2Api {
|
||||||
|
|
||||||
|
baseUrl;
|
||||||
|
sist2Info;
|
||||||
|
queryfunc;
|
||||||
|
|
||||||
|
constructor(baseUrl) {
|
||||||
|
this.baseUrl = baseUrl;
|
||||||
|
}
|
||||||
|
|
||||||
|
init(queryFunc) {
|
||||||
|
this.queryfunc = queryFunc;
|
||||||
|
}
|
||||||
|
|
||||||
|
backend() {
|
||||||
|
return this.sist2Info.searchBackend;
|
||||||
|
}
|
||||||
|
|
||||||
|
models() {
|
||||||
|
const allModels = this.sist2Info.indices
|
||||||
|
.map(idx => idx.models)
|
||||||
|
.flat();
|
||||||
|
|
||||||
|
return allModels
|
||||||
|
.filter((v, i, a) => a.findIndex(v2 => (v2.id === v.id)) === i)
|
||||||
|
}
|
||||||
|
|
||||||
|
getSist2Info() {
|
||||||
|
return axios.get(`${this.baseUrl}i`).then(resp => {
|
||||||
|
this.sist2Info = resp.data;
|
||||||
|
|
||||||
|
return resp.data;
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
setHitProps(hit) {
|
||||||
|
hit["_props"] = {};
|
||||||
|
|
||||||
|
const mimeCategory = hit._source.mime == null ? null : hit._source.mime.split("/")[0];
|
||||||
|
|
||||||
|
if ("parent" in hit._source) {
|
||||||
|
hit._props.isSubDocument = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
if ("thumbnail" in hit._source && hit._source.thumbnail > 0) {
|
||||||
|
hit._props.hasThumbnail = true;
|
||||||
|
|
||||||
|
if (Number.isNaN(Number(hit._source.thumbnail))) {
|
||||||
|
// Backwards compatibility
|
||||||
|
hit._props.tnNum = 1;
|
||||||
|
hit._props.hasVidPreview = false;
|
||||||
|
} else {
|
||||||
|
hit._props.tnNum = Number(hit._source.thumbnail);
|
||||||
|
hit._props.hasVidPreview = hit._props.tnNum > 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
switch (mimeCategory) {
|
||||||
|
case "image":
|
||||||
|
if (hit._source.videoc === "gif") {
|
||||||
|
hit._props.isGif = true;
|
||||||
|
} else {
|
||||||
|
hit._props.isImage = true;
|
||||||
|
}
|
||||||
|
if ("width" in hit._source && !hit._props.isSubDocument && hit._source.videoc !== "tiff"
|
||||||
|
&& hit._source.videoc !== "raw" && hit._source.videoc !== "ppm") {
|
||||||
|
hit._props.isPlayableImage = true;
|
||||||
|
}
|
||||||
|
if ("width" in hit._source && "height" in hit._source) {
|
||||||
|
hit._props.imageAspectRatio = hit._source.width / hit._source.height;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case "video":
|
||||||
|
if ("videoc" in hit._source) {
|
||||||
|
hit._props.isVideo = true;
|
||||||
|
}
|
||||||
|
if (hit._props.isVideo) {
|
||||||
|
const videoc = hit._source.videoc;
|
||||||
|
const mime = hit._source.mime;
|
||||||
|
|
||||||
|
hit._props.isPlayableVideo = mime != null &&
|
||||||
|
mime.startsWith("video/") &&
|
||||||
|
!hit._props.isSubDocument &&
|
||||||
|
hit._source.extension !== "mkv" &&
|
||||||
|
hit._source.extension !== "avi" &&
|
||||||
|
hit._source.extension !== "mov" &&
|
||||||
|
videoc !== "hevc" &&
|
||||||
|
videoc !== "mpeg1video" &&
|
||||||
|
videoc !== "mpeg2video" &&
|
||||||
|
videoc !== "wmv3";
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case "audio":
|
||||||
|
if ("audioc" in hit._source && !hit._props.isSubDocument) {
|
||||||
|
hit._props.isAudio = true;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
setHitTags(hit) {
|
||||||
|
const tags = [];
|
||||||
|
|
||||||
|
// User tags
|
||||||
|
if ("tag" in hit._source) {
|
||||||
|
hit._source.tag.forEach(tag => {
|
||||||
|
tags.push(this.createUserTag(tag));
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
hit._tags = tags;
|
||||||
|
}
|
||||||
|
|
||||||
|
createUserTag(tag) {
|
||||||
|
const tokens = tag.split(".");
|
||||||
|
|
||||||
|
const colorToken = tokens.pop();
|
||||||
|
|
||||||
|
const bg = colorToken;
|
||||||
|
const fg = lum(colorToken) > 50 ? "#000" : "#fff";
|
||||||
|
|
||||||
|
return {
|
||||||
|
style: "user",
|
||||||
|
fg: fg,
|
||||||
|
bg: bg,
|
||||||
|
text: tokens.join("."),
|
||||||
|
rawText: tag,
|
||||||
|
userTag: true,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
search() {
|
||||||
|
if (this.backend() === "sqlite") {
|
||||||
|
return this.ftsQuery(this.queryfunc())
|
||||||
|
} else {
|
||||||
|
return this.esQuery(this.queryfunc());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
_getIndexRoot(indexId) {
|
||||||
|
return this.sist2Info.indices.find(idx => idx.id === indexId).root;
|
||||||
|
}
|
||||||
|
|
||||||
|
esQuery(query) {
|
||||||
|
return axios.post(`${this.baseUrl}es`, query).then(resp => {
|
||||||
|
const res = resp.data;
|
||||||
|
|
||||||
|
if (res.hits?.hits) {
|
||||||
|
res.hits.hits.forEach((hit) => {
|
||||||
|
hit["_source"]["name"] = strUnescape(hit["_source"]["name"]);
|
||||||
|
hit["_source"]["path"] = strUnescape(hit["_source"]["path"]);
|
||||||
|
hit["_source"]["indexRoot"] = this._getIndexRoot(hit["_source"]["index"]);
|
||||||
|
|
||||||
|
this.setHitProps(hit);
|
||||||
|
this.setHitTags(hit);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
return res;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
ftsQuery(query) {
|
||||||
|
return axios.post(`${this.baseUrl}fts/search`, query).then(resp => {
|
||||||
|
const res = resp.data;
|
||||||
|
|
||||||
|
if (res.hits.hits) {
|
||||||
|
res.hits.hits.forEach(hit => {
|
||||||
|
hit["_source"]["name"] = strUnescape(hit["_source"]["name"]);
|
||||||
|
hit["_source"]["path"] = strUnescape(hit["_source"]["path"]);
|
||||||
|
|
||||||
|
this.setHitProps(hit);
|
||||||
|
this.setHitTags(hit);
|
||||||
|
|
||||||
|
if ("highlight" in hit) {
|
||||||
|
hit["highlight"]["name"] = [hit["highlight"]["name"]];
|
||||||
|
hit["highlight"]["content"] = [hit["highlight"]["content"]];
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
return res;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
getMimeTypesEs(query) {
|
||||||
|
const AGGS = {
|
||||||
|
mimeTypes: {
|
||||||
|
terms: {
|
||||||
|
field: "mime",
|
||||||
|
size: 10000
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
if (!query) {
|
||||||
|
query = {
|
||||||
|
aggs: AGGS,
|
||||||
|
size: 0,
|
||||||
|
};
|
||||||
|
} else {
|
||||||
|
query.size = 0;
|
||||||
|
query.aggs = AGGS;
|
||||||
|
}
|
||||||
|
|
||||||
|
return this.esQuery(query).then(resp => {
|
||||||
|
return resp["aggregations"]["mimeTypes"]["buckets"].map(bucket => ({
|
||||||
|
mime: bucket.key,
|
||||||
|
count: bucket.doc_count
|
||||||
|
}));
|
||||||
|
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
getMimeTypesSqlite() {
|
||||||
|
return axios.get(`${this.baseUrl}fts/mimetypes`)
|
||||||
|
.then(resp => {
|
||||||
|
return resp.data;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
async getMimeTypes(query = undefined) {
|
||||||
|
let buckets;
|
||||||
|
|
||||||
|
if (this.backend() === "sqlite") {
|
||||||
|
buckets = await this.getMimeTypesSqlite();
|
||||||
|
} else {
|
||||||
|
buckets = await this.getMimeTypesEs(query);
|
||||||
|
}
|
||||||
|
|
||||||
|
const mimeMap = [];
|
||||||
|
|
||||||
|
buckets.sort((a, b) => a.mime > b.mime).forEach((bucket) => {
|
||||||
|
const tmp = bucket.mime.split("/");
|
||||||
|
const category = tmp[0];
|
||||||
|
const mime = tmp[1];
|
||||||
|
|
||||||
|
let category_exists = false;
|
||||||
|
|
||||||
|
const child = {
|
||||||
|
"id": bucket.mime,
|
||||||
|
"text": `${mime} (${bucket.count})`
|
||||||
|
};
|
||||||
|
|
||||||
|
mimeMap.forEach(node => {
|
||||||
|
if (node.text === category) {
|
||||||
|
node.children.push(child);
|
||||||
|
category_exists = true;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!category_exists) {
|
||||||
|
mimeMap.push({text: category, children: [child], id: category});
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
mimeMap.forEach(node => {
|
||||||
|
if (node.children) {
|
||||||
|
node.children.sort((a, b) => a.id.localeCompare(b.id));
|
||||||
|
}
|
||||||
|
})
|
||||||
|
mimeMap.sort((a, b) => a.id.localeCompare(b.id))
|
||||||
|
|
||||||
|
return {buckets, mimeMap};
|
||||||
|
}
|
||||||
|
|
||||||
|
_createEsTag(tag, count) {
|
||||||
|
const tokens = tag.split(".");
|
||||||
|
|
||||||
|
if (/.*\.#[0-9a-fA-F]{6}/.test(tag)) {
|
||||||
|
return {
|
||||||
|
id: tokens.slice(0, -1).join("."),
|
||||||
|
color: tokens.pop(),
|
||||||
|
isLeaf: true,
|
||||||
|
count: count
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
return {
|
||||||
|
id: tag,
|
||||||
|
count: count,
|
||||||
|
isLeaf: false,
|
||||||
|
color: undefined
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
getTagsEs() {
|
||||||
|
return this.esQuery({
|
||||||
|
aggs: {
|
||||||
|
tags: {
|
||||||
|
terms: {
|
||||||
|
field: "tag",
|
||||||
|
size: 65535
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
size: 0,
|
||||||
|
}).then(resp => {
|
||||||
|
return resp["aggregations"]["tags"]["buckets"]
|
||||||
|
.sort((a, b) => a["key"].localeCompare(b["key"]))
|
||||||
|
.map((bucket) => this._createEsTag(bucket["key"], bucket["doc_count"]));
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
getTagsSqlite() {
|
||||||
|
return axios.get(`${this.baseUrl}/fts/tags`)
|
||||||
|
.then(resp => {
|
||||||
|
return resp.data.map(tag => this._createEsTag(tag.tag, tag.count))
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
async getTags() {
|
||||||
|
let tags;
|
||||||
|
if (this.backend() === "sqlite") {
|
||||||
|
tags = await this.getTagsSqlite();
|
||||||
|
} else {
|
||||||
|
tags = await this.getTagsEs();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Remove duplicates (same tag with different color)
|
||||||
|
const seen = new Set();
|
||||||
|
|
||||||
|
return tags.filter((t) => {
|
||||||
|
if (seen.has(t.id)) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
seen.add(t.id);
|
||||||
|
return true;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
saveTag(tag, hit) {
|
||||||
|
return axios.post(`${this.baseUrl}tag/${sid(hit)}`, {
|
||||||
|
delete: false,
|
||||||
|
name: tag,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
deleteTag(tag, hit) {
|
||||||
|
return axios.post(`${this.baseUrl}tag/${sid(hit)}`, {
|
||||||
|
delete: true,
|
||||||
|
name: tag,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
searchPaths(indexId, minDepth, maxDepth, prefix = null) {
|
||||||
|
if (this.backend() === "sqlite") {
|
||||||
|
return this.searchPathsSqlite(indexId, minDepth, minDepth, prefix);
|
||||||
|
} else {
|
||||||
|
return this.searchPathsEs(indexId, minDepth, maxDepth, prefix);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
searchPathsSqlite(indexId, minDepth, maxDepth, prefix) {
|
||||||
|
return axios.post(`${this.baseUrl}fts/paths`, {
|
||||||
|
indexId, minDepth, maxDepth, prefix
|
||||||
|
}).then(resp => {
|
||||||
|
return resp.data;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
searchPathsEs(indexId, minDepth, maxDepth, prefix) {
|
||||||
|
|
||||||
|
const query = {
|
||||||
|
query: {
|
||||||
|
bool: {
|
||||||
|
filter: [
|
||||||
|
{term: {index: indexId}},
|
||||||
|
{range: {_depth: {gte: minDepth, lte: maxDepth}}},
|
||||||
|
]
|
||||||
|
}
|
||||||
|
},
|
||||||
|
aggs: {
|
||||||
|
paths: {
|
||||||
|
terms: {
|
||||||
|
field: "path",
|
||||||
|
size: 10000
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
size: 0
|
||||||
|
};
|
||||||
|
|
||||||
|
if (prefix != null) {
|
||||||
|
query["query"]["bool"]["must"] = {
|
||||||
|
prefix: {
|
||||||
|
path: prefix,
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
return this.esQuery(query).then(resp => {
|
||||||
|
const buckets = resp["aggregations"]["paths"]["buckets"];
|
||||||
|
|
||||||
|
if (!buckets) {
|
||||||
|
return [];
|
||||||
|
}
|
||||||
|
|
||||||
|
return buckets
|
||||||
|
.map(bucket => ({
|
||||||
|
path: bucket.key,
|
||||||
|
count: bucket.doc_count
|
||||||
|
}));
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
getDateRangeSqlite() {
|
||||||
|
return axios.get(`${this.baseUrl}fts/dateRange`)
|
||||||
|
.then(resp => ({
|
||||||
|
min: resp.data.dateMin,
|
||||||
|
max: (resp.data.dateMax === resp.data.dateMin)
|
||||||
|
? resp.data.dateMax + 1
|
||||||
|
: resp.data.dateMax,
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
getDateRange() {
|
||||||
|
if (this.backend() === "sqlite") {
|
||||||
|
return this.getDateRangeSqlite();
|
||||||
|
} else {
|
||||||
|
return this.getDateRangeEs();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
getDateRangeEs() {
|
||||||
|
return this.esQuery({
|
||||||
|
// TODO: filter current selected indices
|
||||||
|
aggs: {
|
||||||
|
dateMin: {min: {field: "mtime"}},
|
||||||
|
dateMax: {max: {field: "mtime"}},
|
||||||
|
},
|
||||||
|
size: 0
|
||||||
|
}).then(res => {
|
||||||
|
const range = {
|
||||||
|
min: res.aggregations.dateMin.value / 1000,
|
||||||
|
max: res.aggregations.dateMax.value / 1000,
|
||||||
|
}
|
||||||
|
|
||||||
|
if (range.min == null) {
|
||||||
|
range.min = 0;
|
||||||
|
range.max = 1;
|
||||||
|
} else if (range.min === range.max) {
|
||||||
|
range.max += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
return range;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
getPathSuggestionsSqlite(text) {
|
||||||
|
return axios.post(`${this.baseUrl}fts/paths`, {
|
||||||
|
prefix: text,
|
||||||
|
minDepth: 1,
|
||||||
|
maxDepth: 10000
|
||||||
|
}).then(resp => {
|
||||||
|
return resp.data.map(bucket => bucket.path);
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
getPathSuggestionsEs(text) {
|
||||||
|
return this.esQuery({
|
||||||
|
suggest: {
|
||||||
|
path: {
|
||||||
|
prefix: text,
|
||||||
|
completion: {
|
||||||
|
field: "suggest-path",
|
||||||
|
skip_duplicates: true,
|
||||||
|
size: 10000
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}).then(resp => {
|
||||||
|
return resp["suggest"]["path"][0]["options"]
|
||||||
|
.map(opt => opt["_source"]["path"]);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
getPathSuggestions(text) {
|
||||||
|
if (this.backend() === "sqlite") {
|
||||||
|
return this.getPathSuggestionsSqlite(text);
|
||||||
|
} else {
|
||||||
|
return this.getPathSuggestionsEs(text)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
getTreemapStat(indexId) {
|
||||||
|
return `${this.baseUrl}s/${indexId}/TMAP`;
|
||||||
|
}
|
||||||
|
|
||||||
|
getMimeStat(indexId) {
|
||||||
|
return `${this.baseUrl}s/${indexId}/MAGG`;
|
||||||
|
}
|
||||||
|
|
||||||
|
getSizeStat(indexId) {
|
||||||
|
return `${this.baseUrl}s/${indexId}/SAGG`;
|
||||||
|
}
|
||||||
|
|
||||||
|
getDateStat(indexId) {
|
||||||
|
return `${this.baseUrl}s/${indexId}/DAGG`;
|
||||||
|
}
|
||||||
|
|
||||||
|
getDocumentEs(sid, highlight, fuzzy) {
|
||||||
|
const query = Sist2Query.searchQuery();
|
||||||
|
|
||||||
|
if (highlight) {
|
||||||
|
const fields = fuzzy
|
||||||
|
? {"content.nGram": {}}
|
||||||
|
: {content: {}};
|
||||||
|
|
||||||
|
query.highlight = {
|
||||||
|
pre_tags: ["<mark>"],
|
||||||
|
post_tags: ["</mark>"],
|
||||||
|
number_of_fragments: 0,
|
||||||
|
fields,
|
||||||
|
};
|
||||||
|
|
||||||
|
if (!store.state.sist2Info.esVersionLegacy) {
|
||||||
|
query.highlight.max_analyzed_offset = 999_999;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if ("knn" in query) {
|
||||||
|
query.query = {
|
||||||
|
bool: {
|
||||||
|
must: []
|
||||||
|
}
|
||||||
|
};
|
||||||
|
delete query.knn;
|
||||||
|
}
|
||||||
|
|
||||||
|
if ("function_score" in query.query) {
|
||||||
|
query.query = query.query.function_score.query;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!("must" in query.query.bool)) {
|
||||||
|
query.query.bool.must = [];
|
||||||
|
} else if (!Array.isArray(query.query.bool.must)) {
|
||||||
|
query.query.bool.must = [query.query.bool.must];
|
||||||
|
}
|
||||||
|
|
||||||
|
query.query.bool.must.push({match: {_id: sid}});
|
||||||
|
|
||||||
|
delete query["sort"];
|
||||||
|
delete query["aggs"];
|
||||||
|
delete query["search_after"];
|
||||||
|
delete query.query["function_score"];
|
||||||
|
|
||||||
|
query._source = {
|
||||||
|
includes: ["content", "name", "path", "extension"]
|
||||||
|
}
|
||||||
|
|
||||||
|
query.size = 1;
|
||||||
|
|
||||||
|
return this.esQuery(query).then(resp => {
|
||||||
|
if (resp.hits.hits.length === 1) {
|
||||||
|
return resp.hits.hits[0];
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
getDocumentSqlite(sid) {
|
||||||
|
return axios.get(`${this.baseUrl}/fts/d/${sid}`)
|
||||||
|
.then(resp => ({
|
||||||
|
_source: resp.data
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
getDocument(sid, highlight, fuzzy) {
|
||||||
|
if (this.backend() === "sqlite") {
|
||||||
|
return this.getDocumentSqlite(sid);
|
||||||
|
} else {
|
||||||
|
return this.getDocumentEs(sid, highlight, fuzzy);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
getTagSuggestions(prefix) {
|
||||||
|
if (this.backend() === "sqlite") {
|
||||||
|
return this.getTagSuggestionsSqlite(prefix);
|
||||||
|
} else {
|
||||||
|
return this.getTagSuggestionsEs(prefix);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
getTagSuggestionsSqlite(prefix) {
|
||||||
|
return axios.post(`${this.baseUrl}/fts/suggestTags`, prefix)
|
||||||
|
.then(resp => (resp.data));
|
||||||
|
}
|
||||||
|
|
||||||
|
getTagSuggestionsEs(prefix) {
|
||||||
|
return this.esQuery({
|
||||||
|
suggest: {
|
||||||
|
tag: {
|
||||||
|
prefix: prefix,
|
||||||
|
completion: {
|
||||||
|
field: "suggest-tag",
|
||||||
|
skip_duplicates: true,
|
||||||
|
size: 10000
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}).then(resp => {
|
||||||
|
const result = [];
|
||||||
|
resp["suggest"]["tag"][0]["options"].map(opt => opt["_source"]["tag"]).forEach(tags => {
|
||||||
|
tags.forEach(tag => {
|
||||||
|
const t = tag.slice(0, -8);
|
||||||
|
if (!result.find(x => x.slice(0, -8) === t)) {
|
||||||
|
result.push(tag);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
return result;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
getEmbeddings(sid, modelId) {
|
||||||
|
return axios.post(`${this.baseUrl}/e/${sid}/${modelId.toString().padStart(3, '0')}`)
|
||||||
|
.then(resp => (resp.data));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export default new Sist2Api("");
|
||||||
@@ -1,381 +0,0 @@
|
|||||||
import axios from "axios";
|
|
||||||
import {ext, strUnescape, lum} from "./util";
|
|
||||||
|
|
||||||
export interface EsTag {
|
|
||||||
id: string
|
|
||||||
count: number
|
|
||||||
color: string | undefined
|
|
||||||
isLeaf: boolean
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface Tag {
|
|
||||||
style: string
|
|
||||||
text: string
|
|
||||||
rawText: string
|
|
||||||
fg: string
|
|
||||||
bg: string
|
|
||||||
userTag: boolean
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface Index {
|
|
||||||
name: string
|
|
||||||
version: string
|
|
||||||
id: string
|
|
||||||
idPrefix: string
|
|
||||||
timestamp: number
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface EsHit {
|
|
||||||
_index: string
|
|
||||||
_id: string
|
|
||||||
_score: number
|
|
||||||
_type: string
|
|
||||||
_tags: Tag[]
|
|
||||||
_seq: number
|
|
||||||
_source: {
|
|
||||||
path: string
|
|
||||||
size: number
|
|
||||||
mime: string
|
|
||||||
name: string
|
|
||||||
extension: string
|
|
||||||
index: string
|
|
||||||
_depth: number
|
|
||||||
mtime: number
|
|
||||||
videoc: string
|
|
||||||
audioc: string
|
|
||||||
parent: string
|
|
||||||
width: number
|
|
||||||
height: number
|
|
||||||
duration: number
|
|
||||||
tag: string[]
|
|
||||||
checksum: string
|
|
||||||
thumbnail: string
|
|
||||||
}
|
|
||||||
_props: {
|
|
||||||
isSubDocument: boolean
|
|
||||||
isImage: boolean
|
|
||||||
isGif: boolean
|
|
||||||
isVideo: boolean
|
|
||||||
isPlayableVideo: boolean
|
|
||||||
isPlayableImage: boolean
|
|
||||||
isAudio: boolean
|
|
||||||
hasThumbnail: boolean
|
|
||||||
hasVidPreview: boolean
|
|
||||||
imageAspectRatio: number
|
|
||||||
/** Number of thumbnails available */
|
|
||||||
tnNum: number
|
|
||||||
}
|
|
||||||
highlight: {
|
|
||||||
name: string[] | undefined,
|
|
||||||
content: string[] | undefined,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function getIdPrefix(indices: Index[], id: string): string {
|
|
||||||
for (let i = 4; i < 32; i++) {
|
|
||||||
const prefix = id.slice(0, i);
|
|
||||||
|
|
||||||
if (indices.filter(idx => idx.id.slice(0, i) == prefix).length == 1) {
|
|
||||||
return prefix;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return id;
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface EsResult {
|
|
||||||
took: number
|
|
||||||
|
|
||||||
hits: {
|
|
||||||
// TODO: ES 6.X ?
|
|
||||||
total: {
|
|
||||||
value: number
|
|
||||||
}
|
|
||||||
hits: EsHit[]
|
|
||||||
}
|
|
||||||
|
|
||||||
aggregations: any
|
|
||||||
}
|
|
||||||
|
|
||||||
class Sist2Api {
|
|
||||||
|
|
||||||
private baseUrl: string
|
|
||||||
|
|
||||||
constructor(baseUrl: string) {
|
|
||||||
this.baseUrl = baseUrl;
|
|
||||||
}
|
|
||||||
|
|
||||||
getSist2Info(): Promise<any> {
|
|
||||||
return axios.get(`${this.baseUrl}i`).then(resp => {
|
|
||||||
const indices = resp.data.indices as Index[];
|
|
||||||
|
|
||||||
resp.data.indices = indices.map(idx => {
|
|
||||||
return {
|
|
||||||
id: idx.id,
|
|
||||||
name: idx.name,
|
|
||||||
timestamp: idx.timestamp,
|
|
||||||
version: idx.version,
|
|
||||||
idPrefix: getIdPrefix(indices, idx.id)
|
|
||||||
} as Index;
|
|
||||||
});
|
|
||||||
|
|
||||||
return resp.data;
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
setHitProps(hit: EsHit): void {
|
|
||||||
hit["_props"] = {} as any;
|
|
||||||
|
|
||||||
const mimeCategory = hit._source.mime == null ? null : hit._source.mime.split("/")[0];
|
|
||||||
|
|
||||||
if ("parent" in hit._source) {
|
|
||||||
hit._props.isSubDocument = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
if ("thumbnail" in hit._source) {
|
|
||||||
hit._props.hasThumbnail = true;
|
|
||||||
|
|
||||||
if (Number.isNaN(Number(hit._source.thumbnail))) {
|
|
||||||
// Backwards compatibility
|
|
||||||
hit._props.tnNum = 1;
|
|
||||||
hit._props.hasVidPreview = false;
|
|
||||||
} else {
|
|
||||||
hit._props.tnNum = Number(hit._source.thumbnail);
|
|
||||||
hit._props.hasVidPreview = hit._props.tnNum > 1;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
switch (mimeCategory) {
|
|
||||||
case "image":
|
|
||||||
if (hit._source.videoc === "gif") {
|
|
||||||
hit._props.isGif = true;
|
|
||||||
} else {
|
|
||||||
hit._props.isImage = true;
|
|
||||||
}
|
|
||||||
if ("width" in hit._source && !hit._props.isSubDocument && hit._source.videoc !== "tiff"
|
|
||||||
&& hit._source.videoc !== "raw" && hit._source.videoc !== "ppm") {
|
|
||||||
hit._props.isPlayableImage = true;
|
|
||||||
}
|
|
||||||
if ("width" in hit._source && "height" in hit._source) {
|
|
||||||
hit._props.imageAspectRatio = hit._source.width / hit._source.height;
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case "video":
|
|
||||||
if ("videoc" in hit._source) {
|
|
||||||
hit._props.isVideo = true;
|
|
||||||
}
|
|
||||||
if (hit._props.isVideo) {
|
|
||||||
const videoc = hit._source.videoc;
|
|
||||||
const mime = hit._source.mime;
|
|
||||||
|
|
||||||
hit._props.isPlayableVideo = mime != null &&
|
|
||||||
mime.startsWith("video/") &&
|
|
||||||
!hit._props.isSubDocument &&
|
|
||||||
hit._source.extension !== "mkv" &&
|
|
||||||
hit._source.extension !== "avi" &&
|
|
||||||
hit._source.extension !== "mov" &&
|
|
||||||
videoc !== "hevc" &&
|
|
||||||
videoc !== "mpeg1video" &&
|
|
||||||
videoc !== "mpeg2video" &&
|
|
||||||
videoc !== "wmv3";
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case "audio":
|
|
||||||
if ("audioc" in hit._source && !hit._props.isSubDocument) {
|
|
||||||
hit._props.isAudio = true;
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
setHitTags(hit: EsHit): void {
|
|
||||||
const tags = [] as Tag[];
|
|
||||||
|
|
||||||
// User tags
|
|
||||||
if ("tag" in hit._source) {
|
|
||||||
hit._source.tag.forEach(tag => {
|
|
||||||
tags.push(this.createUserTag(tag));
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
hit._tags = tags;
|
|
||||||
}
|
|
||||||
|
|
||||||
createUserTag(tag: string): Tag {
|
|
||||||
const tokens = tag.split(".");
|
|
||||||
|
|
||||||
const colorToken = tokens.pop() as string;
|
|
||||||
|
|
||||||
const bg = colorToken;
|
|
||||||
const fg = lum(colorToken) > 50 ? "#000" : "#fff";
|
|
||||||
|
|
||||||
return {
|
|
||||||
style: "user",
|
|
||||||
fg: fg,
|
|
||||||
bg: bg,
|
|
||||||
text: tokens.join("."),
|
|
||||||
rawText: tag,
|
|
||||||
userTag: true,
|
|
||||||
} as Tag;
|
|
||||||
}
|
|
||||||
|
|
||||||
esQuery(query: any): Promise<EsResult> {
|
|
||||||
return axios.post(`${this.baseUrl}es`, query).then(resp => {
|
|
||||||
const res = resp.data as EsResult;
|
|
||||||
|
|
||||||
if (res.hits?.hits) {
|
|
||||||
res.hits.hits.forEach((hit: EsHit) => {
|
|
||||||
hit["_source"]["name"] = strUnescape(hit["_source"]["name"]);
|
|
||||||
hit["_source"]["path"] = strUnescape(hit["_source"]["path"]);
|
|
||||||
|
|
||||||
this.setHitProps(hit);
|
|
||||||
this.setHitTags(hit);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
return res;
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
getMimeTypes(query = undefined) {
|
|
||||||
const AGGS = {
|
|
||||||
mimeTypes: {
|
|
||||||
terms: {
|
|
||||||
field: "mime",
|
|
||||||
size: 10000
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
if (!query) {
|
|
||||||
query = {
|
|
||||||
aggs: AGGS,
|
|
||||||
size: 0,
|
|
||||||
};
|
|
||||||
} else {
|
|
||||||
query.size = 0;
|
|
||||||
query.aggs = AGGS;
|
|
||||||
}
|
|
||||||
|
|
||||||
return this.esQuery(query).then(resp => {
|
|
||||||
const mimeMap: any[] = [];
|
|
||||||
const buckets = resp["aggregations"]["mimeTypes"]["buckets"];
|
|
||||||
|
|
||||||
buckets.sort((a: any, b: any) => a.key > b.key).forEach((bucket: any) => {
|
|
||||||
const tmp = bucket["key"].split("/");
|
|
||||||
const category = tmp[0];
|
|
||||||
const mime = tmp[1];
|
|
||||||
|
|
||||||
let category_exists = false;
|
|
||||||
|
|
||||||
const child = {
|
|
||||||
"id": bucket["key"],
|
|
||||||
"text": `${mime} (${bucket["doc_count"]})`
|
|
||||||
};
|
|
||||||
|
|
||||||
mimeMap.forEach(node => {
|
|
||||||
if (node.text === category) {
|
|
||||||
node.children.push(child);
|
|
||||||
category_exists = true;
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!category_exists) {
|
|
||||||
mimeMap.push({text: category, children: [child], id: category});
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
mimeMap.forEach(node => {
|
|
||||||
if (node.children) {
|
|
||||||
node.children.sort((a, b) => a.id.localeCompare(b.id));
|
|
||||||
}
|
|
||||||
})
|
|
||||||
mimeMap.sort((a, b) => a.id.localeCompare(b.id))
|
|
||||||
|
|
||||||
return {buckets, mimeMap};
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
_createEsTag(tag: string, count: number): EsTag {
|
|
||||||
const tokens = tag.split(".");
|
|
||||||
|
|
||||||
if (/.*\.#[0-9a-f]{6}/.test(tag)) {
|
|
||||||
return {
|
|
||||||
id: tokens.slice(0, -1).join("."),
|
|
||||||
color: tokens.pop(),
|
|
||||||
isLeaf: true,
|
|
||||||
count: count
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
return {
|
|
||||||
id: tag,
|
|
||||||
count: count,
|
|
||||||
isLeaf: false,
|
|
||||||
color: undefined
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
getTags() {
|
|
||||||
return this.esQuery({
|
|
||||||
aggs: {
|
|
||||||
tags: {
|
|
||||||
terms: {
|
|
||||||
field: "tag",
|
|
||||||
size: 10000
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
size: 0,
|
|
||||||
}).then(resp => {
|
|
||||||
const seen = new Set();
|
|
||||||
|
|
||||||
const tags = resp["aggregations"]["tags"]["buckets"]
|
|
||||||
.sort((a: any, b: any) => a["key"].localeCompare(b["key"]))
|
|
||||||
.map((bucket: any) => this._createEsTag(bucket["key"], bucket["doc_count"]));
|
|
||||||
|
|
||||||
// Remove duplicates (same tag with different color)
|
|
||||||
return tags.filter((t: EsTag) => {
|
|
||||||
if (seen.has(t.id)) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
seen.add(t.id);
|
|
||||||
return true;
|
|
||||||
});
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
saveTag(tag: string, hit: EsHit) {
|
|
||||||
return axios.post(`${this.baseUrl}tag/` + hit["_source"]["index"], {
|
|
||||||
delete: false,
|
|
||||||
name: tag,
|
|
||||||
doc_id: hit["_id"]
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
deleteTag(tag: string, hit: EsHit) {
|
|
||||||
return axios.post(`${this.baseUrl}tag/` + hit["_source"]["index"], {
|
|
||||||
delete: true,
|
|
||||||
name: tag,
|
|
||||||
doc_id: hit["_id"]
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
getTreemapCsvUrl(indexId: string) {
|
|
||||||
return `${this.baseUrl}s/${indexId}/1`;
|
|
||||||
}
|
|
||||||
|
|
||||||
getMimeCsvUrl(indexId: string) {
|
|
||||||
return `${this.baseUrl}s/${indexId}/2`;
|
|
||||||
}
|
|
||||||
|
|
||||||
getSizeCsv(indexId: string) {
|
|
||||||
return `${this.baseUrl}s/${indexId}/3`;
|
|
||||||
}
|
|
||||||
|
|
||||||
getDateCsv(indexId: string) {
|
|
||||||
return `${this.baseUrl}s/${indexId}/4`;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
export default new Sist2Api("");
|
|
||||||
@@ -1,5 +1,5 @@
|
|||||||
import store from "./store";
|
import store from "@/store";
|
||||||
import {EsHit, Index} from "@/Sist2Api";
|
import Sist2Api from "@/Sist2Api";
|
||||||
|
|
||||||
const SORT_MODES = {
|
const SORT_MODES = {
|
||||||
score: {
|
score: {
|
||||||
@@ -7,69 +7,62 @@ const SORT_MODES = {
|
|||||||
{_score: {order: "desc"}},
|
{_score: {order: "desc"}},
|
||||||
{_tie: {order: "asc"}}
|
{_tie: {order: "asc"}}
|
||||||
],
|
],
|
||||||
key: (hit: EsHit) => hit._score
|
key: (hit) => hit._score
|
||||||
},
|
},
|
||||||
random: {
|
random: {
|
||||||
mode: [
|
mode: [
|
||||||
{_score: {order: "desc"}},
|
{_score: {order: "desc"}},
|
||||||
{_tie: {order: "asc"}}
|
{_tie: {order: "asc"}}
|
||||||
],
|
],
|
||||||
key: (hit: EsHit) => hit._score
|
key: (hit) => hit._score
|
||||||
},
|
},
|
||||||
dateAsc: {
|
dateAsc: {
|
||||||
mode: [
|
mode: [
|
||||||
{mtime: {order: "asc"}},
|
{mtime: {order: "asc"}},
|
||||||
{_tie: {order: "asc"}}
|
{_tie: {order: "asc"}}
|
||||||
],
|
],
|
||||||
key: (hit: EsHit) => hit._source.mtime
|
key: (hit) => hit._source.mtime
|
||||||
},
|
},
|
||||||
dateDesc: {
|
dateDesc: {
|
||||||
mode: [
|
mode: [
|
||||||
{mtime: {order: "desc"}},
|
{mtime: {order: "desc"}},
|
||||||
{_tie: {order: "asc"}}
|
{_tie: {order: "asc"}}
|
||||||
],
|
],
|
||||||
key: (hit: EsHit) => hit._source.mtime
|
key: (hit) => hit._source.mtime
|
||||||
},
|
},
|
||||||
sizeAsc: {
|
sizeAsc: {
|
||||||
mode: [
|
mode: [
|
||||||
{size: {order: "asc"}},
|
{size: {order: "asc"}},
|
||||||
{_tie: {order: "asc"}}
|
{_tie: {order: "asc"}}
|
||||||
],
|
],
|
||||||
key: (hit: EsHit) => hit._source.size
|
key: (hit) => hit._source.size
|
||||||
},
|
},
|
||||||
sizeDesc: {
|
sizeDesc: {
|
||||||
mode: [
|
mode: [
|
||||||
{size: {order: "desc"}},
|
{size: {order: "desc"}},
|
||||||
{_tie: {order: "asc"}}
|
{_tie: {order: "asc"}}
|
||||||
],
|
],
|
||||||
key: (hit: EsHit) => hit._source.size
|
key: (hit) => hit._source.size
|
||||||
},
|
},
|
||||||
nameAsc: {
|
nameAsc: {
|
||||||
mode: [
|
mode: [
|
||||||
{name: {order: "asc"}},
|
{name: {order: "asc"}},
|
||||||
{_tie: {order: "asc"}}
|
{_tie: {order: "asc"}}
|
||||||
],
|
],
|
||||||
key: (hit: EsHit) => hit._source.name
|
key: (hit) => hit._source.name
|
||||||
},
|
},
|
||||||
nameDesc: {
|
nameDesc: {
|
||||||
mode: [
|
mode: [
|
||||||
{name: {order: "desc"}},
|
{name: {order: "desc"}},
|
||||||
{_tie: {order: "asc"}}
|
{_tie: {order: "asc"}}
|
||||||
],
|
],
|
||||||
key: (hit: EsHit) => hit._source.name
|
key: (hit) => hit._source.name
|
||||||
}
|
}
|
||||||
} as any;
|
};
|
||||||
|
|
||||||
interface SortMode {
|
class Sist2ElasticsearchQuery {
|
||||||
text: string
|
|
||||||
mode: any[]
|
|
||||||
key: (hit: EsHit) => any
|
|
||||||
}
|
|
||||||
|
|
||||||
|
searchQuery(blankSearch = false) {
|
||||||
class Sist2Query {
|
|
||||||
|
|
||||||
searchQuery(blankSearch: boolean = false): any {
|
|
||||||
|
|
||||||
const getters = store.getters;
|
const getters = store.getters;
|
||||||
|
|
||||||
@@ -83,15 +76,17 @@ class Sist2Query {
|
|||||||
const fuzzy = getters.fuzzy;
|
const fuzzy = getters.fuzzy;
|
||||||
const size = getters.size;
|
const size = getters.size;
|
||||||
const after = getters.lastDoc;
|
const after = getters.lastDoc;
|
||||||
const selectedIndexIds = getters.selectedIndices.map((idx: Index) => idx.id)
|
const selectedIndexIds = getters.selectedIndices.map((idx) => idx.id)
|
||||||
const selectedMimeTypes = getters.selectedMimeTypes;
|
const selectedMimeTypes = getters.selectedMimeTypes;
|
||||||
const selectedTags = getters.selectedTags;
|
const selectedTags = getters.selectedTags;
|
||||||
|
const sortMode = getters.embedding ? "score" : getters.sortMode;
|
||||||
|
|
||||||
const legacyES = store.state.sist2Info.esVersionLegacy;
|
const legacyES = store.state.sist2Info.esVersionLegacy;
|
||||||
|
const hasKnn = store.state.sist2Info.esVersionHasKnn;
|
||||||
|
|
||||||
const filters = [
|
const filters = [
|
||||||
{terms: {index: selectedIndexIds}}
|
{terms: {index: selectedIndexIds}}
|
||||||
] as any[];
|
];
|
||||||
|
|
||||||
const fields = [
|
const fields = [
|
||||||
"name^8",
|
"name^8",
|
||||||
@@ -143,7 +138,7 @@ class Sist2Query {
|
|||||||
if (getters.optTagOrOperator) {
|
if (getters.optTagOrOperator) {
|
||||||
filters.push({terms: {"tag": selectedTags}});
|
filters.push({terms: {"tag": selectedTags}});
|
||||||
} else {
|
} else {
|
||||||
selectedTags.forEach((tag: string) => filters.push({term: {"tag": tag}}));
|
selectedTags.forEach((tag) => filters.push({term: {"tag": tag}}));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -169,31 +164,76 @@ class Sist2Query {
|
|||||||
|
|
||||||
const q = {
|
const q = {
|
||||||
_source: {
|
_source: {
|
||||||
excludes: ["content", "_tie"]
|
excludes: ["content", "_tie", "emb.*"]
|
||||||
},
|
},
|
||||||
query: {
|
query: {
|
||||||
bool: {
|
bool: {
|
||||||
filter: filters,
|
filter: filters,
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
sort: SORT_MODES[getters.sortMode].mode,
|
sort: SORT_MODES[sortMode].mode,
|
||||||
aggs:
|
size: size,
|
||||||
{
|
};
|
||||||
|
|
||||||
|
if (!after) {
|
||||||
|
q.aggs = {
|
||||||
total_size: {"sum": {"field": "size"}},
|
total_size: {"sum": {"field": "size"}},
|
||||||
total_count: {"value_count": {"field": "size"}}
|
total_count: {"value_count": {"field": "size"}}
|
||||||
},
|
};
|
||||||
size: size,
|
}
|
||||||
} as any;
|
|
||||||
|
|
||||||
if (!empty && !blankSearch) {
|
if (!empty && !blankSearch) {
|
||||||
|
if (getters.embedding) {
|
||||||
|
filters.push(query)
|
||||||
|
} else {
|
||||||
q.query.bool.must = query;
|
q.query.bool.must = query;
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (getters.embedding) {
|
||||||
|
delete q.query;
|
||||||
|
|
||||||
|
const field = "emb." + Sist2Api.models().find(m => m.id === getters.embeddingsModel).path;
|
||||||
|
|
||||||
|
if (hasKnn) {
|
||||||
|
// Use knn (8.8+)
|
||||||
|
q.knn = {
|
||||||
|
field: field,
|
||||||
|
query_vector: getters.embedding,
|
||||||
|
|
||||||
|
k: 600,
|
||||||
|
num_candidates: 600,
|
||||||
|
|
||||||
|
filter: filters
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// Use brute-force as a fallback
|
||||||
|
|
||||||
|
filters.push({exists: {field: field}});
|
||||||
|
|
||||||
|
q.query = {
|
||||||
|
function_score: {
|
||||||
|
query: {
|
||||||
|
bool: {
|
||||||
|
must: filters,
|
||||||
|
}
|
||||||
|
},
|
||||||
|
script_score: {
|
||||||
|
script: {
|
||||||
|
source: `cosineSimilarity(params.query_vector, "${field}") + 1.0`,
|
||||||
|
params: {query_vector: getters.embedding}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if (after) {
|
if (after) {
|
||||||
q.search_after = [SORT_MODES[getters.sortMode].key(after), after["_id"]];
|
q.search_after = [SORT_MODES[sortMode].key(after), after["_id"]];
|
||||||
}
|
}
|
||||||
|
|
||||||
if (getters.optHighlight) {
|
if (getters.optHighlight && !getters.embedding) {
|
||||||
q.highlight = {
|
q.highlight = {
|
||||||
pre_tags: ["<mark>"],
|
pre_tags: ["<mark>"],
|
||||||
post_tags: ["</mark>"],
|
post_tags: ["</mark>"],
|
||||||
@@ -219,7 +259,7 @@ class Sist2Query {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (getters.sortMode === "random") {
|
if (sortMode === "random") {
|
||||||
q.query = {
|
q.query = {
|
||||||
function_score: {
|
function_score: {
|
||||||
query: {
|
query: {
|
||||||
@@ -249,4 +289,5 @@ class Sist2Query {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
export default new Sist2Query();
|
|
||||||
|
export default new Sist2ElasticsearchQuery();
|
||||||
116
sist2-vue/src/Sist2SqliteQuery.js
Normal file
@@ -0,0 +1,116 @@
|
|||||||
|
import store from "@/store";
|
||||||
|
|
||||||
|
const SORT_MODES = {
|
||||||
|
score: {
|
||||||
|
"sort": "score",
|
||||||
|
},
|
||||||
|
random: {
|
||||||
|
"sort": "random"
|
||||||
|
},
|
||||||
|
dateAsc: {
|
||||||
|
"sort": "mtime"
|
||||||
|
},
|
||||||
|
dateDesc: {
|
||||||
|
"sort": "mtime",
|
||||||
|
"sortAsc": false
|
||||||
|
},
|
||||||
|
sizeAsc: {
|
||||||
|
"sort": "size",
|
||||||
|
},
|
||||||
|
sizeDesc: {
|
||||||
|
"sort": "size",
|
||||||
|
"sortAsc": false
|
||||||
|
},
|
||||||
|
nameAsc: {
|
||||||
|
"sort": "name",
|
||||||
|
},
|
||||||
|
nameDesc: {
|
||||||
|
"sort": "name",
|
||||||
|
"sortAsc": false
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
class Sist2ElasticsearchQuery {
|
||||||
|
|
||||||
|
searchQuery() {
|
||||||
|
|
||||||
|
const getters = store.getters;
|
||||||
|
|
||||||
|
const searchText = getters.searchText;
|
||||||
|
const pathText = getters.pathText;
|
||||||
|
const sizeMin = getters.sizeMin;
|
||||||
|
const sizeMax = getters.sizeMax;
|
||||||
|
const dateMin = getters.dateMin;
|
||||||
|
const dateMax = getters.dateMax;
|
||||||
|
const size = getters.size;
|
||||||
|
const after = getters.lastDoc;
|
||||||
|
const selectedIndexIds = getters.selectedIndices.map((idx) => idx.id)
|
||||||
|
const selectedMimeTypes = getters.selectedMimeTypes;
|
||||||
|
const selectedTags = getters.selectedTags;
|
||||||
|
|
||||||
|
const q = {
|
||||||
|
"pageSize": size
|
||||||
|
}
|
||||||
|
|
||||||
|
Object.assign(q, SORT_MODES[getters.sortMode]);
|
||||||
|
|
||||||
|
if (!after) {
|
||||||
|
q["fetchAggregations"] = true;
|
||||||
|
}
|
||||||
|
if (searchText) {
|
||||||
|
q["query"] = searchText;
|
||||||
|
}
|
||||||
|
if (pathText) {
|
||||||
|
q["path"] = pathText.endsWith("/") ? pathText.slice(0, -1) : pathText;
|
||||||
|
}
|
||||||
|
if (sizeMin) {
|
||||||
|
q["sizeMin"] = sizeMin;
|
||||||
|
}
|
||||||
|
if (sizeMax) {
|
||||||
|
q["sizeMax"] = sizeMax;
|
||||||
|
}
|
||||||
|
if (dateMin) {
|
||||||
|
q["dateMin"] = dateMin;
|
||||||
|
}
|
||||||
|
if (dateMax) {
|
||||||
|
q["dateMax"] = dateMax;
|
||||||
|
}
|
||||||
|
if (after) {
|
||||||
|
q["after"] = after.sort;
|
||||||
|
}
|
||||||
|
if (selectedIndexIds.length > 0) {
|
||||||
|
q["indexIds"] = selectedIndexIds;
|
||||||
|
}
|
||||||
|
if (selectedMimeTypes.length > 0) {
|
||||||
|
q["mimeTypes"] = selectedMimeTypes;
|
||||||
|
}
|
||||||
|
if (selectedTags.length > 0) {
|
||||||
|
q["tags"] = selectedTags
|
||||||
|
}
|
||||||
|
if (getters.sortMode === "random") {
|
||||||
|
q["seed"] = getters.seed;
|
||||||
|
}
|
||||||
|
if (getters.optHighlight) {
|
||||||
|
q["highlight"] = true;
|
||||||
|
q["highlightContextSize"] = Number(getters.optFragmentSize);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (getters.embedding) {
|
||||||
|
q["model"] = getters.embeddingsModel;
|
||||||
|
q["embedding"] = getters.embedding;
|
||||||
|
q["sort"] = "embedding";
|
||||||
|
q["sortAsc"] = false;
|
||||||
|
} else if (getters.sortMode === "embedding") {
|
||||||
|
q["sort"] = "sort"
|
||||||
|
q["sortAsc"] = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
q["searchInPath"] = getters.optSearchInPath;
|
||||||
|
|
||||||
|
return q;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
export default new Sist2ElasticsearchQuery();
|
||||||
21
sist2-vue/src/components/AnalyzedContentSpan.vue
Normal file
@@ -0,0 +1,21 @@
|
|||||||
|
<template>
|
||||||
|
<span :style="getStyle()">{{span.text}}</span>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
|
||||||
|
|
||||||
|
import ModelsRepo from "@/ml/modelsRepo";
|
||||||
|
|
||||||
|
export default {
|
||||||
|
name: "AnalyzedContentSpan",
|
||||||
|
props: ["span", "text"],
|
||||||
|
methods: {
|
||||||
|
getStyle() {
|
||||||
|
return ModelsRepo.data[this.$store.getters.nerModel.name].labelStyles[this.span.label];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<style scoped></style>
|
||||||
75
sist2-vue/src/components/AnalyzedContentSpanContainer.vue
Normal file
@@ -0,0 +1,75 @@
|
|||||||
|
<template>
|
||||||
|
<div>
|
||||||
|
<b-card class="mb-2">
|
||||||
|
<AnalyzedContentSpan v-for="span of legend" :key="span.id" :span="span"
|
||||||
|
class="mr-2"></AnalyzedContentSpan>
|
||||||
|
</b-card>
|
||||||
|
<div class="content-div">
|
||||||
|
<AnalyzedContentSpan v-for="span of mergedSpans" :key="span.id" :span="span"></AnalyzedContentSpan>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
|
||||||
|
|
||||||
|
import AnalyzedContentSpan from "@/components/AnalyzedContentSpan.vue";
|
||||||
|
import ModelsRepo from "@/ml/modelsRepo";
|
||||||
|
|
||||||
|
export default {
|
||||||
|
name: "AnalyzedContentSpanContainer",
|
||||||
|
components: {AnalyzedContentSpan},
|
||||||
|
props: ["spans", "text"],
|
||||||
|
computed: {
|
||||||
|
legend() {
|
||||||
|
return Object.entries(ModelsRepo.data[this.$store.state.nerModel.name].legend)
|
||||||
|
.map(([label, name]) => ({
|
||||||
|
text: name,
|
||||||
|
id: label,
|
||||||
|
label: label
|
||||||
|
}));
|
||||||
|
},
|
||||||
|
mergedSpans() {
|
||||||
|
const spans = this.spans;
|
||||||
|
|
||||||
|
const merged = [];
|
||||||
|
|
||||||
|
let lastLabel = null;
|
||||||
|
let fixSpace = false;
|
||||||
|
for (let i = 0; i < spans.length; i++) {
|
||||||
|
|
||||||
|
if (spans[i].label !== lastLabel) {
|
||||||
|
let start = spans[i].wordIndex;
|
||||||
|
const nextSpan = spans.slice(i + 1).find(s => s.label !== spans[i].label)
|
||||||
|
let end = nextSpan ? nextSpan.wordIndex : undefined;
|
||||||
|
|
||||||
|
if (end !== undefined && this.text[end - 1] === " ") {
|
||||||
|
end -= 1;
|
||||||
|
fixSpace = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
merged.push({
|
||||||
|
text: this.text.slice(start, end),
|
||||||
|
label: spans[i].label,
|
||||||
|
id: spans[i].wordIndex
|
||||||
|
});
|
||||||
|
|
||||||
|
if (fixSpace) {
|
||||||
|
merged.push({
|
||||||
|
text: " ",
|
||||||
|
label: "O",
|
||||||
|
id: end
|
||||||
|
});
|
||||||
|
fixSpace = false;
|
||||||
|
}
|
||||||
|
lastLabel = spans[i].label;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return merged;
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<style scoped></style>
|
||||||
@@ -120,7 +120,7 @@ export default {
|
|||||||
update(indexId) {
|
update(indexId) {
|
||||||
const svg = d3.select("#date-histogram");
|
const svg = d3.select("#date-histogram");
|
||||||
|
|
||||||
d3.csv(Sist2Api.getDateCsv(indexId)).then(tabularData => {
|
d3.json(Sist2Api.getDateStat(indexId)).then(tabularData => {
|
||||||
dateHistogram(tabularData.slice(), svg, this.$t("d3.dateHistogram"));
|
dateHistogram(tabularData.slice(), svg, this.$t("d3.dateHistogram"));
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -91,7 +91,7 @@ export default {
|
|||||||
const mimeSvgCount = d3.select("#agg-mime-count");
|
const mimeSvgCount = d3.select("#agg-mime-count");
|
||||||
const fillOpacity = this.$store.state.optTheme === "black" ? 0.9 : 0.6;
|
const fillOpacity = this.$store.state.optTheme === "black" ? 0.9 : 0.6;
|
||||||
|
|
||||||
d3.csv(Sist2Api.getMimeCsvUrl(indexId)).then(tabularData => {
|
d3.json(Sist2Api.getMimeStat(indexId)).then(tabularData => {
|
||||||
mimeBarCount(tabularData.slice(), mimeSvgCount, fillOpacity, this.$t("d3.mimeCount"));
|
mimeBarCount(tabularData.slice(), mimeSvgCount, fillOpacity, this.$t("d3.mimeCount"));
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -90,7 +90,7 @@ export default {
|
|||||||
const mimeSvgSize = d3.select("#agg-mime-size");
|
const mimeSvgSize = d3.select("#agg-mime-size");
|
||||||
const fillOpacity = this.$store.state.optTheme === "black" ? 0.9 : 0.6;
|
const fillOpacity = this.$store.state.optTheme === "black" ? 0.9 : 0.6;
|
||||||
|
|
||||||
d3.csv(Sist2Api.getMimeCsvUrl(indexId)).then(tabularData => {
|
d3.json(Sist2Api.getMimeStat(indexId)).then(tabularData => {
|
||||||
mimeBarSize(tabularData.slice(), mimeSvgSize, fillOpacity, this.$t("d3.mimeSize"));
|
mimeBarSize(tabularData.slice(), mimeSvgSize, fillOpacity, this.$t("d3.mimeSize"));
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -117,7 +117,7 @@ export default {
|
|||||||
update(indexId) {
|
update(indexId) {
|
||||||
const svg = d3.select("#size-histogram");
|
const svg = d3.select("#size-histogram");
|
||||||
|
|
||||||
d3.csv(Sist2Api.getSizeCsv(indexId)).then(tabularData => {
|
d3.json(Sist2Api.getSizeStat(indexId)).then(tabularData => {
|
||||||
sizeHistogram(tabularData.slice(), svg, this.$t("d3.sizeHistogram"));
|
sizeHistogram(tabularData.slice(), svg, this.$t("d3.sizeHistogram"));
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -9,8 +9,7 @@
|
|||||||
|
|
||||||
<script>
|
<script>
|
||||||
import * as d3 from "d3";
|
import * as d3 from "d3";
|
||||||
import {burrow} from "@/util-js"
|
import {humanFileSize, burrow} from "@/util";
|
||||||
import {humanFileSize} from "@/util";
|
|
||||||
import Sist2Api from "@/Sist2Api";
|
import Sist2Api from "@/Sist2Api";
|
||||||
import domtoimage from "dom-to-image";
|
import domtoimage from "dom-to-image";
|
||||||
|
|
||||||
@@ -240,7 +239,7 @@ export default {
|
|||||||
.style("overflow", "visible")
|
.style("overflow", "visible")
|
||||||
.style("font", "10px sans-serif");
|
.style("font", "10px sans-serif");
|
||||||
|
|
||||||
d3.csv(Sist2Api.getTreemapCsvUrl(indexId)).then(tabularData => {
|
d3.json(Sist2Api.getTreemapStat(indexId)).then(tabularData => {
|
||||||
tabularData.forEach(row => {
|
tabularData.forEach(row => {
|
||||||
row.taxonomy = row.path.split("/");
|
row.taxonomy = row.path.split("/");
|
||||||
row.size = Number(row.size);
|
row.size = Number(row.size);
|
||||||
|
|||||||
@@ -31,8 +31,7 @@
|
|||||||
<script>
|
<script>
|
||||||
import noUiSlider from 'nouislider';
|
import noUiSlider from 'nouislider';
|
||||||
import 'nouislider/dist/nouislider.css';
|
import 'nouislider/dist/nouislider.css';
|
||||||
import {humanDate} from "@/util";
|
import {humanDate, mergeTooltips} from "@/util";
|
||||||
import {mergeTooltips} from "@/util-js";
|
|
||||||
|
|
||||||
export default {
|
export default {
|
||||||
name: "DateSlider",
|
name: "DateSlider",
|
||||||
|
|||||||
@@ -1,14 +1,18 @@
|
|||||||
<template>
|
<template>
|
||||||
<b-card class="mb-4 mt-4">
|
<b-card v-if="$store.state.sist2Info.showDebugInfo" class="mb-4 mt-4">
|
||||||
<b-card-title><DebugIcon class="mr-1"></DebugIcon>{{ $t("debug") }}</b-card-title>
|
<b-card-title>
|
||||||
|
<DebugIcon class="mr-1"></DebugIcon>
|
||||||
|
{{ $t("debug") }}
|
||||||
|
</b-card-title>
|
||||||
<p v-html="$t('debugDescription')"></p>
|
<p v-html="$t('debugDescription')"></p>
|
||||||
|
|
||||||
<b-card-body>
|
<b-card-body>
|
||||||
|
|
||||||
<b-table :items="tableItems" small borderless responsive="md" thead-class="hidden" class="mb-0"></b-table>
|
<b-table :items="tableItems" small borderless responsive="md" thead-class="hidden" class="mb-0"></b-table>
|
||||||
|
|
||||||
<hr />
|
<hr/>
|
||||||
<IndexDebugInfo v-for="idx of $store.state.sist2Info.indices" :key="idx.id" :index="idx" class="mt-2"></IndexDebugInfo>
|
<IndexDebugInfo v-for="idx of $store.state.sist2Info.indices" :key="idx.id" :index="idx"
|
||||||
|
class="mt-2"></IndexDebugInfo>
|
||||||
</b-card-body>
|
</b-card-body>
|
||||||
</b-card>
|
</b-card>
|
||||||
</template>
|
</template>
|
||||||
@@ -16,13 +20,17 @@
|
|||||||
<script>
|
<script>
|
||||||
import IndexDebugInfo from "@/components/IndexDebugInfo";
|
import IndexDebugInfo from "@/components/IndexDebugInfo";
|
||||||
import DebugIcon from "@/components/icons/DebugIcon";
|
import DebugIcon from "@/components/icons/DebugIcon";
|
||||||
|
import {mapGetters} from "vuex";
|
||||||
|
|
||||||
export default {
|
export default {
|
||||||
name: "DebugInfo.vue",
|
name: "DebugInfo.vue",
|
||||||
components: {DebugIcon, IndexDebugInfo},
|
components: {DebugIcon, IndexDebugInfo},
|
||||||
computed: {
|
computed: {
|
||||||
|
...mapGetters([
|
||||||
|
"uiSqliteMode",
|
||||||
|
]),
|
||||||
tableItems() {
|
tableItems() {
|
||||||
return [
|
const items = [
|
||||||
{key: "version", value: this.$store.state.sist2Info.version},
|
{key: "version", value: this.$store.state.sist2Info.version},
|
||||||
{key: "platform", value: this.$store.state.sist2Info.platform},
|
{key: "platform", value: this.$store.state.sist2Info.platform},
|
||||||
{key: "debugBinary", value: this.$store.state.sist2Info.debug},
|
{key: "debugBinary", value: this.$store.state.sist2Info.debug},
|
||||||
@@ -31,10 +39,18 @@ export default {
|
|||||||
{key: "tagline", value: this.$store.state.sist2Info.tagline},
|
{key: "tagline", value: this.$store.state.sist2Info.tagline},
|
||||||
{key: "dev", value: this.$store.state.sist2Info.dev},
|
{key: "dev", value: this.$store.state.sist2Info.dev},
|
||||||
{key: "mongooseVersion", value: this.$store.state.sist2Info.mongooseVersion},
|
{key: "mongooseVersion", value: this.$store.state.sist2Info.mongooseVersion},
|
||||||
|
];
|
||||||
|
|
||||||
|
if (!this.uiSqliteMode) {
|
||||||
|
items.push(
|
||||||
{key: "esVersion", value: this.$store.state.sist2Info.esVersion},
|
{key: "esVersion", value: this.$store.state.sist2Info.esVersion},
|
||||||
{key: "esVersionSupported", value: this.$store.state.sist2Info.esVersionSupported},
|
{key: "esVersionSupported", value: this.$store.state.sist2Info.esVersionSupported},
|
||||||
{key: "esVersionLegacy", value: this.$store.state.sist2Info.esVersionLegacy},
|
{key: "esVersionLegacy", value: this.$store.state.sist2Info.esVersionLegacy},
|
||||||
]
|
{key: "esVersionHasKnn", value: this.$store.state.sist2Info.esVersionHasKnn},
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
return items;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -16,7 +16,7 @@
|
|||||||
<!-- Audio player-->
|
<!-- Audio player-->
|
||||||
<audio v-if="doc._props.isAudio" ref="audio" preload="none" class="audio-fit fit" controls
|
<audio v-if="doc._props.isAudio" ref="audio" preload="none" class="audio-fit fit" controls
|
||||||
:type="doc._source.mime"
|
:type="doc._source.mime"
|
||||||
:src="`f/${doc._id}`"
|
:src="`f/${sid(doc)}`"
|
||||||
@play="onAudioPlay()"></audio>
|
@play="onAudioPlay()"></audio>
|
||||||
|
|
||||||
<b-card-body class="padding-03">
|
<b-card-body class="padding-03">
|
||||||
@@ -24,6 +24,7 @@
|
|||||||
<!-- Title line -->
|
<!-- Title line -->
|
||||||
<div style="display: flex">
|
<div style="display: flex">
|
||||||
<span class="info-icon" @click="onInfoClick()"></span>
|
<span class="info-icon" @click="onInfoClick()"></span>
|
||||||
|
<MLIcon v-if="doc._source.embedding" clickable @click="onEmbeddingClick()"></MLIcon>
|
||||||
<DocFileTitle :doc="doc"></DocFileTitle>
|
<DocFileTitle :doc="doc"></DocFileTitle>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
@@ -42,17 +43,19 @@
|
|||||||
</template>
|
</template>
|
||||||
|
|
||||||
<script>
|
<script>
|
||||||
import {ext, humanFileSize, humanTime} from "@/util";
|
import {ext, humanFileSize, humanTime, sid} from "@/util";
|
||||||
import TagContainer from "@/components/TagContainer.vue";
|
import TagContainer from "@/components/TagContainer.vue";
|
||||||
import DocFileTitle from "@/components/DocFileTitle.vue";
|
import DocFileTitle from "@/components/DocFileTitle.vue";
|
||||||
import DocInfoModal from "@/components/DocInfoModal.vue";
|
import DocInfoModal from "@/components/DocInfoModal.vue";
|
||||||
import ContentDiv from "@/components/ContentDiv.vue";
|
import ContentDiv from "@/components/ContentDiv.vue";
|
||||||
import FullThumbnail from "@/components/FullThumbnail";
|
import FullThumbnail from "@/components/FullThumbnail";
|
||||||
import FeaturedFieldsLine from "@/components/FeaturedFieldsLine";
|
import FeaturedFieldsLine from "@/components/FeaturedFieldsLine";
|
||||||
|
import MLIcon from "@/components/icons/MlIcon.vue";
|
||||||
|
import Sist2Api from "@/Sist2Api";
|
||||||
|
|
||||||
|
|
||||||
export default {
|
export default {
|
||||||
components: {FeaturedFieldsLine, FullThumbnail, ContentDiv, DocInfoModal, DocFileTitle, TagContainer},
|
components: {MLIcon, FeaturedFieldsLine, FullThumbnail, ContentDiv, DocInfoModal, DocFileTitle, TagContainer},
|
||||||
props: ["doc", "width"],
|
props: ["doc", "width"],
|
||||||
data() {
|
data() {
|
||||||
return {
|
return {
|
||||||
@@ -66,17 +69,25 @@ export default {
|
|||||||
}
|
}
|
||||||
},
|
},
|
||||||
methods: {
|
methods: {
|
||||||
|
sid: sid,
|
||||||
humanFileSize: humanFileSize,
|
humanFileSize: humanFileSize,
|
||||||
humanTime: humanTime,
|
humanTime: humanTime,
|
||||||
onInfoClick() {
|
onInfoClick() {
|
||||||
this.showInfo = true;
|
this.showInfo = true;
|
||||||
},
|
},
|
||||||
|
onEmbeddingClick() {
|
||||||
|
Sist2Api.getEmbeddings(sid(this.doc), this.$store.state.embeddingsModel).then(embeddings => {
|
||||||
|
this.$store.commit("setEmbeddingText", "");
|
||||||
|
this.$store.commit("setEmbedding", embeddings);
|
||||||
|
this.$store.commit("setEmbeddingDoc", this.doc);
|
||||||
|
})
|
||||||
|
},
|
||||||
async onThumbnailClick() {
|
async onThumbnailClick() {
|
||||||
this.$store.commit("setUiLightboxSlide", this.doc._seq);
|
this.$store.commit("setUiLightboxSlide", this.doc._seq);
|
||||||
await this.$store.dispatch("showLightbox");
|
await this.$store.dispatch("showLightbox");
|
||||||
},
|
},
|
||||||
onAudioPlay() {
|
onAudioPlay() {
|
||||||
document.getElementsByTagName("audio").forEach((el) => {
|
Array.prototype.slice.call(document.getElementsByTagName("audio")).forEach((el) => {
|
||||||
if (el !== this.$refs["audio"]) {
|
if (el !== this.$refs["audio"]) {
|
||||||
el.pause();
|
el.pause();
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -5,11 +5,12 @@
|
|||||||
@append="append"
|
@append="append"
|
||||||
@layout-complete="$emit('layout-complete')"
|
@layout-complete="$emit('layout-complete')"
|
||||||
>
|
>
|
||||||
<DocCard v-for="doc in docs" :key="doc._id" :doc="doc" :width="width"></DocCard>
|
<DocCard v-for="doc in docs" :key="sid(doc)" :doc="doc" :width="width"></DocCard>
|
||||||
</GridLayout>
|
</GridLayout>
|
||||||
</template>
|
</template>
|
||||||
|
|
||||||
<script>
|
<script>
|
||||||
|
import {sid} from "@/util";
|
||||||
import Vue from "vue";
|
import Vue from "vue";
|
||||||
import DocCard from "@/components/DocCard";
|
import DocCard from "@/components/DocCard";
|
||||||
|
|
||||||
@@ -39,6 +40,9 @@ export default Vue.extend({
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
methods: {
|
||||||
|
sid,
|
||||||
|
},
|
||||||
computed: {
|
computed: {
|
||||||
colCount() {
|
colCount() {
|
||||||
const columns = this.$store.getters["optColumns"];
|
const columns = this.$store.getters["optColumns"];
|
||||||
|
|||||||
@@ -1,17 +1,19 @@
|
|||||||
<template>
|
<template>
|
||||||
<a :href="`f/${doc._id}`" class="file-title-anchor" target="_blank">
|
<a :href="`f/${sid(doc)}`"
|
||||||
|
:class="doc._source.embedding ? 'file-title-anchor-with-embedding' : 'file-title-anchor'" target="_blank">
|
||||||
<div class="file-title" :title="doc._source.path + '/' + doc._source.name + ext(doc)"
|
<div class="file-title" :title="doc._source.path + '/' + doc._source.name + ext(doc)"
|
||||||
v-html="fileName() + ext(doc)"></div>
|
v-html="fileName() + ext(doc)"></div>
|
||||||
</a>
|
</a>
|
||||||
</template>
|
</template>
|
||||||
|
|
||||||
<script>
|
<script>
|
||||||
import {ext} from "@/util";
|
import {ext, sid} from "@/util";
|
||||||
|
|
||||||
export default {
|
export default {
|
||||||
name: "DocFileTitle",
|
name: "DocFileTitle",
|
||||||
props: ["doc"],
|
props: ["doc"],
|
||||||
methods: {
|
methods: {
|
||||||
|
sid: sid,
|
||||||
ext: ext,
|
ext: ext,
|
||||||
fileName() {
|
fileName() {
|
||||||
if (!this.doc.highlight) {
|
if (!this.doc.highlight) {
|
||||||
@@ -34,8 +36,13 @@ export default {
|
|||||||
max-width: calc(100% - 1.2rem);
|
max-width: calc(100% - 1.2rem);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.file-title-anchor-with-embedding {
|
||||||
|
max-width: calc(100% - 2.2rem);
|
||||||
|
}
|
||||||
|
|
||||||
.file-title {
|
.file-title {
|
||||||
width: 100%;
|
width: 100%;
|
||||||
|
max-width: 100%;
|
||||||
line-height: 1rem;
|
line-height: 1rem;
|
||||||
height: 1.1rem;
|
height: 1.1rem;
|
||||||
white-space: nowrap;
|
white-space: nowrap;
|
||||||
@@ -49,6 +56,7 @@ export default {
|
|||||||
.theme-black .file-title {
|
.theme-black .file-title {
|
||||||
color: #ddd;
|
color: #ddd;
|
||||||
}
|
}
|
||||||
|
|
||||||
.theme-black .file-title:hover {
|
.theme-black .file-title:hover {
|
||||||
color: #fff;
|
color: #fff;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -8,16 +8,16 @@
|
|||||||
</h5>
|
</h5>
|
||||||
</template>
|
</template>
|
||||||
|
|
||||||
<img v-if="doc._props.hasThumbnail" :src="`t/${doc._source.index}/${doc._id}`" alt="" class="fit card-img-top">
|
<img v-if="doc._props.hasThumbnail" :src="`t/${sid(doc)}`" alt="" class="fit card-img-top">
|
||||||
|
|
||||||
<InfoTable :doc="doc"></InfoTable>
|
<InfoTable :doc="doc"></InfoTable>
|
||||||
|
|
||||||
<LazyContentDiv :doc-id="doc._id"></LazyContentDiv>
|
<LazyContentDiv :sid="sid(doc)"></LazyContentDiv>
|
||||||
</b-modal>
|
</b-modal>
|
||||||
</template>
|
</template>
|
||||||
|
|
||||||
<script>
|
<script>
|
||||||
import {ext} from "@/util";
|
import {ext, sid} from "@/util";
|
||||||
import InfoTable from "@/components/InfoTable";
|
import InfoTable from "@/components/InfoTable";
|
||||||
import LazyContentDiv from "@/components/LazyContentDiv";
|
import LazyContentDiv from "@/components/LazyContentDiv";
|
||||||
|
|
||||||
@@ -27,6 +27,7 @@ export default {
|
|||||||
props: ["doc", "show"],
|
props: ["doc", "show"],
|
||||||
methods: {
|
methods: {
|
||||||
ext: ext,
|
ext: ext,
|
||||||
|
sid: sid
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
</script>
|
</script>
|
||||||
|
|||||||
@@ -1,10 +1,11 @@
|
|||||||
<template>
|
<template>
|
||||||
<b-list-group class="mt-3">
|
<b-list-group class="mt-3">
|
||||||
<DocListItem v-for="doc in docs" :key="doc._id" :doc="doc"></DocListItem>
|
<DocListItem v-for="doc in docs" :key="sid(doc)" :doc="doc"></DocListItem>
|
||||||
</b-list-group>
|
</b-list-group>
|
||||||
</template>
|
</template>
|
||||||
|
|
||||||
<script lang="ts">
|
<script>
|
||||||
|
import {sid} from "@/util";
|
||||||
import DocListItem from "@/components/DocListItem.vue";
|
import DocListItem from "@/components/DocListItem.vue";
|
||||||
import Vue from "vue";
|
import Vue from "vue";
|
||||||
|
|
||||||
@@ -21,6 +22,9 @@ export default Vue.extend({
|
|||||||
this.append();
|
this.append();
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
},
|
||||||
|
methods: {
|
||||||
|
sid: sid
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
</script>
|
</script>
|
||||||
|
|||||||
@@ -17,10 +17,10 @@
|
|||||||
</div>
|
</div>
|
||||||
|
|
||||||
<img v-if="doc._props.isPlayableImage || doc._props.isPlayableVideo"
|
<img v-if="doc._props.isPlayableImage || doc._props.isPlayableVideo"
|
||||||
:src="(doc._props.isGif && hover) ? `f/${doc._id}` : `t/${doc._source.index}/${doc._id}`"
|
:src="(doc._props.isGif && hover) ? `f/${sid(doc)}` : `t/${sid(doc)}`"
|
||||||
alt=""
|
alt=""
|
||||||
class="pointer fit-sm" @click="onThumbnailClick()">
|
class="pointer fit-sm" @click="onThumbnailClick()">
|
||||||
<img v-else :src="`t/${doc._source.index}/${doc._id}`" alt=""
|
<img v-else :src="`t/${sid(doc)}`" alt=""
|
||||||
class="fit-sm">
|
class="fit-sm">
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
@@ -32,6 +32,7 @@
|
|||||||
<div class="doc-line ml-3">
|
<div class="doc-line ml-3">
|
||||||
<div style="display: flex">
|
<div style="display: flex">
|
||||||
<span class="info-icon" @click="showInfo = true"></span>
|
<span class="info-icon" @click="showInfo = true"></span>
|
||||||
|
<MLIcon v-if="doc._source.embedding" clickable @click="onEmbeddingClick()"></MLIcon>
|
||||||
<DocFileTitle :doc="doc"></DocFileTitle>
|
<DocFileTitle :doc="doc"></DocFileTitle>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
@@ -67,10 +68,13 @@ import DocInfoModal from "@/components/DocInfoModal";
|
|||||||
import ContentDiv from "@/components/ContentDiv";
|
import ContentDiv from "@/components/ContentDiv";
|
||||||
import FileIcon from "@/components/icons/FileIcon";
|
import FileIcon from "@/components/icons/FileIcon";
|
||||||
import FeaturedFieldsLine from "@/components/FeaturedFieldsLine";
|
import FeaturedFieldsLine from "@/components/FeaturedFieldsLine";
|
||||||
|
import MLIcon from "@/components/icons/MlIcon.vue";
|
||||||
|
import Sist2Api from "@/Sist2Api";
|
||||||
|
import {sid} from "@/util";
|
||||||
|
|
||||||
export default {
|
export default {
|
||||||
name: "DocListItem",
|
name: "DocListItem",
|
||||||
components: {FileIcon, ContentDiv, DocInfoModal, DocFileTitle, TagContainer, FeaturedFieldsLine},
|
components: {MLIcon, FileIcon, ContentDiv, DocInfoModal, DocFileTitle, TagContainer, FeaturedFieldsLine},
|
||||||
props: ["doc"],
|
props: ["doc"],
|
||||||
data() {
|
data() {
|
||||||
return {
|
return {
|
||||||
@@ -79,10 +83,18 @@ export default {
|
|||||||
}
|
}
|
||||||
},
|
},
|
||||||
methods: {
|
methods: {
|
||||||
|
sid: sid,
|
||||||
async onThumbnailClick() {
|
async onThumbnailClick() {
|
||||||
this.$store.commit("setUiLightboxSlide", this.doc._seq);
|
this.$store.commit("setUiLightboxSlide", this.doc._seq);
|
||||||
await this.$store.dispatch("showLightbox");
|
await this.$store.dispatch("showLightbox");
|
||||||
},
|
},
|
||||||
|
onEmbeddingClick() {
|
||||||
|
Sist2Api.getEmbeddings(sid(this.doc), this.$store.state.embeddingsModel).then(embeddings => {
|
||||||
|
this.$store.commit("setEmbeddingText", "");
|
||||||
|
this.$store.commit("setEmbedding", embeddings);
|
||||||
|
this.$store.commit("setEmbeddingDoc", this.doc);
|
||||||
|
})
|
||||||
|
},
|
||||||
path() {
|
path() {
|
||||||
if (!this.doc.highlight) {
|
if (!this.doc.highlight) {
|
||||||
return this.doc._source.path + "/"
|
return this.doc._source.path + "/"
|
||||||
|
|||||||
155
sist2-vue/src/components/EmbeddingsSearchBar.vue
Normal file
@@ -0,0 +1,155 @@
|
|||||||
|
<template>
|
||||||
|
<div>
|
||||||
|
<b-progress v-if="modelLoading && [0, 1].includes(modelLoadingProgress)" max="1" class="mb-1" variant="primary"
|
||||||
|
striped animated :value="1">
|
||||||
|
</b-progress>
|
||||||
|
<b-progress v-else-if="modelLoading" :value="modelLoadingProgress" max="1" class="mb-1" variant="warning"
|
||||||
|
show-progress>
|
||||||
|
</b-progress>
|
||||||
|
<div style="display: flex">
|
||||||
|
<b-select :options="modelOptions()" class="mr-2 input-prepend" :value="modelName"
|
||||||
|
@change="onModelChange($event)"></b-select>
|
||||||
|
|
||||||
|
<b-input-group>
|
||||||
|
<b-form-input :value="embeddingText"
|
||||||
|
:placeholder="$store.state.embeddingDoc ? ' ' : $t('embeddingsSearchPlaceholder')"
|
||||||
|
@input="onInput($event)"
|
||||||
|
:disabled="modelLoading"
|
||||||
|
:style="{'pointer-events': $store.state.embeddingDoc ? 'none' : undefined}"
|
||||||
|
></b-form-input>
|
||||||
|
<b-badge v-if="$store.state.embeddingDoc" pill variant="primary" class="overlay-badge" href="#"
|
||||||
|
@click="onBadgeClick()">{{ docName }}
|
||||||
|
</b-badge>
|
||||||
|
|
||||||
|
<template #prepend>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<template #append>
|
||||||
|
<b-input-group-text>
|
||||||
|
<MLIcon class="ml-append" big></MLIcon>
|
||||||
|
</b-input-group-text>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
</b-input-group>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
import {mapGetters, mapMutations} from "vuex";
|
||||||
|
import {CLIPTransformerModel} from "@/ml/CLIPTransformerModel"
|
||||||
|
import _debounce from "lodash/debounce";
|
||||||
|
import MLIcon from "@/components/icons/MlIcon.vue";
|
||||||
|
import Sist2AdminApi from "@/Sist2Api";
|
||||||
|
|
||||||
|
export default {
|
||||||
|
components: {MLIcon},
|
||||||
|
data() {
|
||||||
|
return {
|
||||||
|
modelLoading: false,
|
||||||
|
modelLoadingProgress: 0,
|
||||||
|
modelLoaded: false,
|
||||||
|
model: null,
|
||||||
|
modelName: null
|
||||||
|
}
|
||||||
|
},
|
||||||
|
computed: {
|
||||||
|
...mapGetters({
|
||||||
|
optQueryMode: "optQueryMode",
|
||||||
|
embeddingText: "embeddingText",
|
||||||
|
fuzzy: "fuzzy",
|
||||||
|
}),
|
||||||
|
docName() {
|
||||||
|
const ext = this.$store.state.embeddingDoc._source.extension;
|
||||||
|
return this.$store.state.embeddingDoc._source.name +
|
||||||
|
(ext ? "." + ext : "")
|
||||||
|
}
|
||||||
|
},
|
||||||
|
mounted() {
|
||||||
|
// Set default model
|
||||||
|
this.modelName = Sist2AdminApi.models()[0].name;
|
||||||
|
this.onModelChange(this.modelName);
|
||||||
|
|
||||||
|
this.onInput = _debounce(this._onInput, 450, {leading: false});
|
||||||
|
},
|
||||||
|
methods: {
|
||||||
|
...mapMutations({
|
||||||
|
setEmbeddingText: "setEmbeddingText",
|
||||||
|
setEmbedding: "setEmbedding",
|
||||||
|
setEmbeddingModel: "setEmbeddingsModel",
|
||||||
|
}),
|
||||||
|
async loadModel() {
|
||||||
|
this.modelLoading = true;
|
||||||
|
|
||||||
|
await this.model.init(async progress => {
|
||||||
|
this.modelLoadingProgress = progress;
|
||||||
|
});
|
||||||
|
this.modelLoading = false;
|
||||||
|
this.modelLoaded = true;
|
||||||
|
},
|
||||||
|
async _onInput(text) {
|
||||||
|
try {
|
||||||
|
|
||||||
|
if (!this.modelLoaded) {
|
||||||
|
await this.loadModel();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (text.length === 0) {
|
||||||
|
this.setEmbeddingText("");
|
||||||
|
this.setEmbedding(null);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const embeddings = await this.model.predict(text);
|
||||||
|
|
||||||
|
this.setEmbeddingText(text);
|
||||||
|
this.setEmbedding(embeddings);
|
||||||
|
} catch (e) {
|
||||||
|
alert(e)
|
||||||
|
}
|
||||||
|
},
|
||||||
|
modelOptions() {
|
||||||
|
return Sist2AdminApi.models().map(model => model.name);
|
||||||
|
},
|
||||||
|
onModelChange(name) {
|
||||||
|
this.modelLoaded = false;
|
||||||
|
this.modelLoadingProgress = 0;
|
||||||
|
|
||||||
|
const modelInfo = Sist2AdminApi.models().find(m => m.name === name);
|
||||||
|
|
||||||
|
if (modelInfo.name === "CLIP") {
|
||||||
|
const tokenizerUrl = new URL("./tokenizer.json", modelInfo.url).href;
|
||||||
|
this.model = new CLIPTransformerModel(modelInfo.url, tokenizerUrl)
|
||||||
|
this.setEmbeddingModel(modelInfo.id);
|
||||||
|
} else {
|
||||||
|
throw new Error("Unknown model: " + name);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
onBadgeClick() {
|
||||||
|
this.$store.commit("setEmbedding", null);
|
||||||
|
this.$store.commit("setEmbeddingDoc", null);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
<style>
|
||||||
|
.overlay-badge {
|
||||||
|
position: absolute;
|
||||||
|
z-index: 1;
|
||||||
|
left: 0.375rem;
|
||||||
|
top: 8px;
|
||||||
|
line-height: 1.1rem;
|
||||||
|
overflow: hidden;
|
||||||
|
max-width: 200px;
|
||||||
|
text-overflow: ellipsis;
|
||||||
|
}
|
||||||
|
|
||||||
|
.input-prepend {
|
||||||
|
max-width: 100px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.theme-black .ml-append {
|
||||||
|
filter: brightness(0.95) !important;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
@@ -16,6 +16,10 @@ export default {
|
|||||||
props: ["doc"],
|
props: ["doc"],
|
||||||
computed: {
|
computed: {
|
||||||
featuredLineHtml() {
|
featuredLineHtml() {
|
||||||
|
if (this.$store.getters.optFeaturedFields === undefined) {
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
const scope = {doc: this.doc._source, humanDate: humanDate, humanFileSize: humanFileSize};
|
const scope = {doc: this.doc._source, humanDate: humanDate, humanFileSize: humanFileSize};
|
||||||
|
|
||||||
return this.$store.getters.optFeaturedFields
|
return this.$store.getters.optFeaturedFields
|
||||||
|
|||||||
@@ -40,7 +40,7 @@
|
|||||||
</template>
|
</template>
|
||||||
|
|
||||||
<script>
|
<script>
|
||||||
import {humanTime} from "@/util";
|
import {humanTime, sid} from "@/util";
|
||||||
import ThumbnailProgressBar from "@/components/ThumbnailProgressBar";
|
import ThumbnailProgressBar from "@/components/ThumbnailProgressBar";
|
||||||
|
|
||||||
export default {
|
export default {
|
||||||
@@ -67,15 +67,16 @@ export default {
|
|||||||
},
|
},
|
||||||
},
|
},
|
||||||
methods: {
|
methods: {
|
||||||
|
sid: sid,
|
||||||
getThumbnailSrc(thumbnailNum) {
|
getThumbnailSrc(thumbnailNum) {
|
||||||
const doc = this.doc;
|
const doc = this.doc;
|
||||||
const props = doc._props;
|
const props = doc._props;
|
||||||
if (props.isGif && this.hover) {
|
if (props.isGif && this.hover) {
|
||||||
return `f/${doc._id}`;
|
return `f/${sid(doc)}`;
|
||||||
}
|
}
|
||||||
return (this.currentThumbnailNum === 0)
|
return (this.currentThumbnailNum === 0)
|
||||||
? `t/${doc._source.index}/${doc._id}`
|
? `t/${sid(doc)}`
|
||||||
: `t/${doc._source.index}/${doc._id}/${String(thumbnailNum).padStart(4, "0")}`;
|
: `t/${sid(doc)}/${String(thumbnailNum).padStart(4, "0")}`;
|
||||||
},
|
},
|
||||||
humanTime: humanTime,
|
humanTime: humanTime,
|
||||||
onThumbnailClick() {
|
onThumbnailClick() {
|
||||||
|
|||||||
@@ -27,11 +27,16 @@
|
|||||||
@click.shift="shiftClick(idx, $event)"
|
@click.shift="shiftClick(idx, $event)"
|
||||||
class="d-flex justify-content-between align-items-center list-group-item-action pointer"
|
class="d-flex justify-content-between align-items-center list-group-item-action pointer"
|
||||||
:class="{active: lastClickIndex === idx}"
|
:class="{active: lastClickIndex === idx}"
|
||||||
|
:key="idx.id"
|
||||||
>
|
>
|
||||||
<div class="d-flex">
|
<div class="d-flex">
|
||||||
<b-checkbox style="pointer-events: none" :checked="isSelected(idx)"></b-checkbox>
|
<b-checkbox style="pointer-events: none" :checked="isSelected(idx)"></b-checkbox>
|
||||||
{{ idx.name }}
|
{{ idx.name }}
|
||||||
<span class="text-muted timestamp-text ml-2">{{ formatIdxDate(idx.timestamp) }}</span>
|
<div v-if="hasEmbeddings(idx)" style="vertical-align: center; margin-left: 5px">
|
||||||
|
<MLIcon small style="top: -1px; position: relative"></MLIcon>
|
||||||
|
</div>
|
||||||
|
<span class="text-muted timestamp-text ml-2"
|
||||||
|
style="top: 1px; position: relative">{{ formatIdxDate(idx.timestamp) }}</span>
|
||||||
</div>
|
</div>
|
||||||
<b-badge class="version-badge">v{{ idx.version }}</b-badge>
|
<b-badge class="version-badge">v{{ idx.version }}</b-badge>
|
||||||
</b-list-group-item>
|
</b-list-group-item>
|
||||||
@@ -39,14 +44,16 @@
|
|||||||
</div>
|
</div>
|
||||||
</template>
|
</template>
|
||||||
|
|
||||||
<script lang="ts">
|
<script>
|
||||||
import SmallBadge from "./SmallBadge.vue"
|
import SmallBadge from "./SmallBadge.vue"
|
||||||
import {mapActions, mapGetters} from "vuex";
|
import {mapActions, mapGetters} from "vuex";
|
||||||
import Vue from "vue";
|
import Vue from "vue";
|
||||||
import {format} from "date-fns";
|
import {format} from "date-fns";
|
||||||
|
import MLIcon from "@/components/icons/MlIcon.vue";
|
||||||
|
|
||||||
export default Vue.extend({
|
export default Vue.extend({
|
||||||
components: {
|
components: {
|
||||||
|
MLIcon,
|
||||||
SmallBadge
|
SmallBadge
|
||||||
},
|
},
|
||||||
data() {
|
data() {
|
||||||
@@ -105,7 +112,7 @@ export default Vue.extend({
|
|||||||
onSelect(value) {
|
onSelect(value) {
|
||||||
this.setSelectedIndices(this.indices.filter(idx => value.includes(idx.id)));
|
this.setSelectedIndices(this.indices.filter(idx => value.includes(idx.id)));
|
||||||
},
|
},
|
||||||
formatIdxDate(timestamp: number): string {
|
formatIdxDate(timestamp) {
|
||||||
return format(new Date(timestamp * 1000), "yyyy-MM-dd");
|
return format(new Date(timestamp * 1000), "yyyy-MM-dd");
|
||||||
},
|
},
|
||||||
toggleIndex(index, e) {
|
toggleIndex(index, e) {
|
||||||
@@ -115,14 +122,17 @@ export default Vue.extend({
|
|||||||
|
|
||||||
this.lastClickIndex = index;
|
this.lastClickIndex = index;
|
||||||
if (this.isSelected(index)) {
|
if (this.isSelected(index)) {
|
||||||
this.setSelectedIndices(this.selectedIndices.filter(idx => idx.id != index.id));
|
this.setSelectedIndices(this.selectedIndices.filter(idx => idx.id !== index.id));
|
||||||
} else {
|
} else {
|
||||||
this.setSelectedIndices([index, ...this.selectedIndices]);
|
this.setSelectedIndices([index, ...this.selectedIndices]);
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
isSelected(index) {
|
isSelected(index) {
|
||||||
return this.selectedIndices.find(idx => idx.id == index.id) != null;
|
return this.selectedIndices.find(idx => idx.id === index.id) != null;
|
||||||
}
|
},
|
||||||
|
hasEmbeddings(index) {
|
||||||
|
return index.models.length > 0;
|
||||||
|
},
|
||||||
},
|
},
|
||||||
})
|
})
|
||||||
</script>
|
</script>
|
||||||
@@ -170,18 +180,18 @@ export default Vue.extend({
|
|||||||
}
|
}
|
||||||
|
|
||||||
.theme-black .list-group-item {
|
.theme-black .list-group-item {
|
||||||
border: 1px solid rgba(255,255,255, 0.1);
|
border: 1px solid rgba(255, 255, 255, 0.1);
|
||||||
}
|
}
|
||||||
|
|
||||||
.theme-black .list-group-item:first-child {
|
.theme-black .list-group-item:first-child {
|
||||||
border: 1px solid rgba(255,255,255, 0.05);
|
border: 1px solid rgba(255, 255, 255, 0.05);
|
||||||
}
|
}
|
||||||
|
|
||||||
.theme-black .list-group-item.active {
|
.theme-black .list-group-item.active {
|
||||||
z-index: 2;
|
z-index: 2;
|
||||||
background-color: inherit;
|
background-color: inherit;
|
||||||
color: inherit;
|
color: inherit;
|
||||||
border: 1px solid rgba(255,255,255, 0.3);
|
border: 1px solid rgba(255, 255, 255, 0.3);
|
||||||
border-radius: 0;
|
border-radius: 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -59,7 +59,7 @@ export default {
|
|||||||
|
|
||||||
const fields = [
|
const fields = [
|
||||||
"title", "duration", "audioc", "videoc",
|
"title", "duration", "audioc", "videoc",
|
||||||
"bitrate", "artist", "album", "album_artist", "genre", "font_name", "author",
|
"bitrate", "artist", "album", "album_artist", "genre", "font_name", "author", "media_comment",
|
||||||
"modified_by", "pages", "tag",
|
"modified_by", "pages", "tag",
|
||||||
"exif_make", "exif_software", "exif_exposure_time", "exif_fnumber", "exif_focal_length",
|
"exif_make", "exif_software", "exif_exposure_time", "exif_fnumber", "exif_focal_length",
|
||||||
"exif_user_comment", "exif_iso_speed_ratings", "exif_model", "exif_datetime",
|
"exif_user_comment", "exif_iso_speed_ratings", "exif_model", "exif_datetime",
|
||||||
|
|||||||
@@ -1,80 +1,103 @@
|
|||||||
<template>
|
<template>
|
||||||
<Preloader v-if="loading"></Preloader>
|
<Preloader v-if="loading"></Preloader>
|
||||||
<div v-else-if="content" class="content-div" v-html="content"></div>
|
<div v-else-if="content">
|
||||||
|
<b-form inline class="my-2" v-if="ModelsRepo.getOptions().length > 0">
|
||||||
|
<b-checkbox class="ml-auto mr-2" :checked="optAutoAnalyze"
|
||||||
|
@input="setOptAutoAnalyze($event); $store.dispatch('updateConfiguration')">
|
||||||
|
{{ $t("ml.auto") }}
|
||||||
|
</b-checkbox>
|
||||||
|
<b-button :disabled="mlPredictionsLoading || mlLoading" @click="mlAnalyze" variant="primary"
|
||||||
|
>{{ $t("ml.analyzeText") }}
|
||||||
|
</b-button>
|
||||||
|
<b-select :disabled="mlPredictionsLoading || mlLoading" class="ml-2" v-model="nerModel">
|
||||||
|
<b-select-option :value="opt.value" v-for="opt of ModelsRepo.getOptions()" :key="opt.value">{{ opt.text }}
|
||||||
|
</b-select-option>
|
||||||
|
</b-select>
|
||||||
|
</b-form>
|
||||||
|
|
||||||
|
<b-progress v-if="mlLoading" variant="warning" show-progress :max="1" class="mb-3"
|
||||||
|
>
|
||||||
|
<b-progress-bar :value="modelLoadingProgress">
|
||||||
|
<strong>{{ ((modelLoadingProgress * modelSize) / (1024 * 1024)).toFixed(1) }}MB / {{
|
||||||
|
(modelSize / (1024 * 1024)).toFixed(1)
|
||||||
|
}}MB</strong>
|
||||||
|
</b-progress-bar>
|
||||||
|
</b-progress>
|
||||||
|
|
||||||
|
<b-progress v-if="mlPredictionsLoading" variant="primary" :value="modelPredictionProgress"
|
||||||
|
:max="content.length" class="mb-3"></b-progress>
|
||||||
|
|
||||||
|
<AnalyzedContentSpansContainer v-if="analyzedContentSpans.length > 0"
|
||||||
|
:spans="analyzedContentSpans" :text="rawContent"></AnalyzedContentSpansContainer>
|
||||||
|
<div v-else class="content-div" v-html="content"></div>
|
||||||
|
</div>
|
||||||
</template>
|
</template>
|
||||||
|
|
||||||
<script>
|
<script>
|
||||||
import Sist2Api from "@/Sist2Api";
|
import Sist2Api from "@/Sist2Api";
|
||||||
import Preloader from "@/components/Preloader";
|
import Preloader from "@/components/Preloader";
|
||||||
import Sist2Query from "@/Sist2Query";
|
import Sist2Query from "@/Sist2ElasticsearchQuery";
|
||||||
import store from "@/store";
|
import store from "@/store";
|
||||||
|
import BertNerModel from "@/ml/BertNerModel";
|
||||||
|
import AnalyzedContentSpansContainer from "@/components/AnalyzedContentSpanContainer.vue";
|
||||||
|
import ModelsRepo from "@/ml/modelsRepo";
|
||||||
|
import {mapGetters, mapMutations} from "vuex";
|
||||||
|
|
||||||
export default {
|
export default {
|
||||||
name: "LazyContentDiv",
|
name: "LazyContentDiv",
|
||||||
components: {Preloader},
|
components: {AnalyzedContentSpansContainer, Preloader},
|
||||||
props: ["docId"],
|
props: ["sid"],
|
||||||
data() {
|
data() {
|
||||||
return {
|
return {
|
||||||
|
ModelsRepo,
|
||||||
content: "",
|
content: "",
|
||||||
loading: true
|
rawContent: "",
|
||||||
|
loading: true,
|
||||||
|
modelLoadingProgress: 0,
|
||||||
|
modelPredictionProgress: 0,
|
||||||
|
mlPredictionsLoading: false,
|
||||||
|
mlLoading: false,
|
||||||
|
nerModel: null,
|
||||||
|
analyzedContentSpans: []
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
mounted() {
|
mounted() {
|
||||||
const query = Sist2Query.searchQuery();
|
|
||||||
|
|
||||||
if (this.$store.state.optHighlight) {
|
if (this.$store.getters.optMlDefaultModel) {
|
||||||
|
this.nerModel = this.$store.getters.optMlDefaultModel
|
||||||
const fields = this.$store.state.fuzzy
|
|
||||||
? {"content.nGram": {}}
|
|
||||||
: {content: {}};
|
|
||||||
|
|
||||||
query.highlight = {
|
|
||||||
pre_tags: ["<mark>"],
|
|
||||||
post_tags: ["</mark>"],
|
|
||||||
number_of_fragments: 0,
|
|
||||||
fields,
|
|
||||||
};
|
|
||||||
|
|
||||||
if (!store.state.sist2Info.esVersionLegacy) {
|
|
||||||
query.highlight.max_analyzed_offset = 999_999;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if ("function_score" in query.query) {
|
|
||||||
query.query = query.query.function_score.query;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!("must" in query.query.bool)) {
|
|
||||||
query.query.bool.must = [];
|
|
||||||
} else if (!Array.isArray(query.query.bool.must)) {
|
|
||||||
query.query.bool.must = [query.query.bool.must];
|
|
||||||
}
|
|
||||||
|
|
||||||
query.query.bool.must.push({match: {_id: this.docId}});
|
|
||||||
|
|
||||||
delete query["sort"];
|
|
||||||
delete query["aggs"];
|
|
||||||
delete query["search_after"];
|
|
||||||
delete query.query["function_score"];
|
|
||||||
|
|
||||||
query._source = {
|
|
||||||
includes: ["content", "name", "path", "extension"]
|
|
||||||
}
|
|
||||||
|
|
||||||
query.size = 1;
|
|
||||||
|
|
||||||
Sist2Api.esQuery(query).then(resp => {
|
|
||||||
this.loading = false;
|
|
||||||
if (resp.hits.hits.length === 1) {
|
|
||||||
this.content = this.getContent(resp.hits.hits[0]);
|
|
||||||
} else {
|
} else {
|
||||||
console.log("FIXME: could not get content")
|
this.nerModel = ModelsRepo.getDefaultModel();
|
||||||
console.log(resp)
|
}
|
||||||
|
|
||||||
|
Sist2Api
|
||||||
|
.getDocument(this.sid, this.$store.state.optHighlight, this.$store.state.fuzzy)
|
||||||
|
.then(doc => {
|
||||||
|
this.loading = false;
|
||||||
|
|
||||||
|
if (doc) {
|
||||||
|
this.content = this.getContent(doc)
|
||||||
|
}
|
||||||
|
|
||||||
|
if (this.optAutoAnalyze) {
|
||||||
|
this.mlAnalyze();
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
},
|
},
|
||||||
|
computed: {
|
||||||
|
...mapGetters(["optAutoAnalyze"]),
|
||||||
|
modelSize() {
|
||||||
|
const modelData = ModelsRepo.data[this.nerModel];
|
||||||
|
if (!modelData) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
return modelData.size;
|
||||||
|
}
|
||||||
|
},
|
||||||
methods: {
|
methods: {
|
||||||
|
...mapMutations(["setOptAutoAnalyze"]),
|
||||||
getContent(doc) {
|
getContent(doc) {
|
||||||
|
this.rawContent = doc._source.content;
|
||||||
|
|
||||||
if (!doc.highlight) {
|
if (!doc.highlight) {
|
||||||
return doc._source.content;
|
return doc._source.content;
|
||||||
}
|
}
|
||||||
@@ -85,10 +108,60 @@ export default {
|
|||||||
if (doc.highlight.content) {
|
if (doc.highlight.content) {
|
||||||
return doc.highlight.content[0];
|
return doc.highlight.content[0];
|
||||||
}
|
}
|
||||||
|
},
|
||||||
|
async getMlModel() {
|
||||||
|
if (this.$store.getters.nerModel.name !== this.nerModel) {
|
||||||
|
this.mlLoading = true;
|
||||||
|
this.modelLoadingProgress = 0;
|
||||||
|
const modelInfo = ModelsRepo.data[this.nerModel];
|
||||||
|
|
||||||
|
const model = new BertNerModel(
|
||||||
|
modelInfo.vocabUrl,
|
||||||
|
modelInfo.modelUrl,
|
||||||
|
modelInfo.id2label,
|
||||||
|
)
|
||||||
|
|
||||||
|
await model.init(progress => this.modelLoadingProgress = progress);
|
||||||
|
this.$store.commit("setNerModel", {model, name: this.nerModel});
|
||||||
|
|
||||||
|
this.mlLoading = false;
|
||||||
|
return model
|
||||||
|
}
|
||||||
|
|
||||||
|
return this.$store.getters.nerModel.model;
|
||||||
|
},
|
||||||
|
async mlAnalyze() {
|
||||||
|
if (!this.content) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const modelInfo = ModelsRepo.data[this.nerModel];
|
||||||
|
if (modelInfo === undefined) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
this.$store.commit("setOptMlDefaultModel", this.nerModel);
|
||||||
|
await this.$store.dispatch("updateConfiguration");
|
||||||
|
|
||||||
|
const model = await this.getMlModel();
|
||||||
|
|
||||||
|
this.analyzedContentSpans = [];
|
||||||
|
|
||||||
|
this.mlPredictionsLoading = true;
|
||||||
|
|
||||||
|
await model.predict(this.rawContent, results => {
|
||||||
|
results.forEach(result => result.label = modelInfo.humanLabels[result.label]);
|
||||||
|
this.analyzedContentSpans.push(...results);
|
||||||
|
this.modelPredictionProgress = results[results.length - 1].wordIndex;
|
||||||
|
});
|
||||||
|
this.mlPredictionsLoading = false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
</script>
|
</script>
|
||||||
|
|
||||||
<style scoped>
|
<style>
|
||||||
|
.progress-bar {
|
||||||
|
transition: none;
|
||||||
|
}
|
||||||
</style>
|
</style>
|
||||||
@@ -77,6 +77,7 @@ export default {
|
|||||||
return listener(e);
|
return listener(e);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
},
|
},
|
||||||
methods: {
|
methods: {
|
||||||
keyDownListener(e) {
|
keyDownListener(e) {
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ import InspireTreeDOM from "inspire-tree-dom";
|
|||||||
import "inspire-tree-dom/dist/inspire-tree-light.min.css";
|
import "inspire-tree-dom/dist/inspire-tree-light.min.css";
|
||||||
import {getSelectedTreeNodes, getTreeNodeAttributes} from "@/util";
|
import {getSelectedTreeNodes, getTreeNodeAttributes} from "@/util";
|
||||||
import Sist2Api from "@/Sist2Api";
|
import Sist2Api from "@/Sist2Api";
|
||||||
import Sist2Query from "@/Sist2Query";
|
import Sist2Query from "@/Sist2ElasticsearchQuery";
|
||||||
|
|
||||||
export default {
|
export default {
|
||||||
name: "MimePicker",
|
name: "MimePicker",
|
||||||
|
|||||||
@@ -9,8 +9,8 @@
|
|||||||
|
|
||||||
<span class="badge badge-pill version" v-if="$store && $store.state.sist2Info">
|
<span class="badge badge-pill version" v-if="$store && $store.state.sist2Info">
|
||||||
v{{ sist2Version() }}<span v-if="isDebug()">-dbg</span><span v-if="isLegacy() && !hideLegacy()">-<a
|
v{{ sist2Version() }}<span v-if="isDebug()">-dbg</span><span v-if="isLegacy() && !hideLegacy()">-<a
|
||||||
href="https://github.com/simon987/sist2/blob/master/docs/USAGE.md#elasticsearch"
|
href="https://github.com/sist2app/sist2/blob/master/docs/USAGE.md#elasticsearch"
|
||||||
target="_blank">legacyES</a></span>
|
target="_blank">legacyES</a></span><span v-if="$store.state.uiSqliteMode">-SQLite</span>
|
||||||
</span>
|
</span>
|
||||||
|
|
||||||
<span v-if="$store && $store.state.sist2Info" class="tagline" v-html="tagline()"></span>
|
<span v-if="$store && $store.state.sist2Info" class="tagline" v-html="tagline()"></span>
|
||||||
|
|||||||
@@ -20,6 +20,7 @@
|
|||||||
:list="suggestPath"
|
:list="suggestPath"
|
||||||
:max-suggestions="0"
|
:max-suggestions="0"
|
||||||
:placeholder="$t('pathBar.placeholder')"
|
:placeholder="$t('pathBar.placeholder')"
|
||||||
|
:debounce="200"
|
||||||
>
|
>
|
||||||
<!-- Suggestion item template-->
|
<!-- Suggestion item template-->
|
||||||
<div slot="suggestion-item" slot-scope="{ suggestion, query }">
|
<div slot="suggestion-item" slot-scope="{ suggestion, query }">
|
||||||
@@ -101,22 +102,8 @@ export default {
|
|||||||
},
|
},
|
||||||
async getPathChoices() {
|
async getPathChoices() {
|
||||||
return new Promise(getPaths => {
|
return new Promise(getPaths => {
|
||||||
const q = {
|
Sist2Api.getPathSuggestions(this.getPathText).then(getPaths);
|
||||||
suggest: {
|
});
|
||||||
path: {
|
|
||||||
prefix: this.getPathText,
|
|
||||||
completion: {
|
|
||||||
field: "suggest-path",
|
|
||||||
skip_duplicates: true,
|
|
||||||
size: 10000
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
Sist2Api.esQuery(q)
|
|
||||||
.then(resp => getPaths(resp["suggest"]["path"][0]["options"].map(opt => opt["_source"]["path"])));
|
|
||||||
})
|
|
||||||
},
|
},
|
||||||
async suggestPath(term) {
|
async suggestPath(term) {
|
||||||
if (!this.$store.state.optSuggestPath) {
|
if (!this.$store.state.optSuggestPath) {
|
||||||
@@ -136,64 +123,33 @@ export default {
|
|||||||
return matches.sort((a, b) => a.length - b.length);
|
return matches.sort((a, b) => a.length - b.length);
|
||||||
},
|
},
|
||||||
getNextDepth(node) {
|
getNextDepth(node) {
|
||||||
const q = {
|
return Sist2Api
|
||||||
query: {
|
.searchPaths(node.index, node.depth + 1, node.depth + 3, node.depth > 0 ? node.id : null)
|
||||||
bool: {
|
.then(buckets => {
|
||||||
filter: [
|
|
||||||
{term: {index: node.index}},
|
|
||||||
{range: {_depth: {gte: node.depth + 1, lte: node.depth + 3}}},
|
|
||||||
]
|
|
||||||
}
|
|
||||||
},
|
|
||||||
aggs: {
|
|
||||||
paths: {
|
|
||||||
terms: {
|
|
||||||
field: "path",
|
|
||||||
size: 10000
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
size: 0
|
|
||||||
};
|
|
||||||
|
|
||||||
if (node.depth > 0) {
|
|
||||||
q.query.bool.must = {
|
|
||||||
prefix: {
|
|
||||||
path: node.id,
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
return Sist2Api.esQuery(q).then(resp => {
|
|
||||||
const buckets = resp["aggregations"]["paths"]["buckets"];
|
|
||||||
if (!buckets) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
const paths = [];
|
const paths = [];
|
||||||
|
|
||||||
return buckets
|
return buckets
|
||||||
.filter(bucket => bucket.key.length > node.id.length || node.id.startsWith("/"))
|
.filter(bucket => bucket.path.length > node.id.length || node.id.startsWith("/"))
|
||||||
.sort((a, b) => a.key > b.key)
|
.sort((a, b) => a.path > b.path ? 1 : -1)
|
||||||
.map(bucket => {
|
.map(bucket => {
|
||||||
|
if (paths.some(n => bucket.path.startsWith(n))) {
|
||||||
if (paths.some(n => bucket.key.startsWith(n))) {
|
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
|
||||||
const name = node.id.startsWith("/") ? bucket.key : bucket.key.slice(node.id.length + 1);
|
const name = node.id.startsWith("/") ? bucket.path : bucket.path.slice(node.id.length + 1);
|
||||||
|
|
||||||
paths.push(bucket.key);
|
paths.push(bucket.path);
|
||||||
|
|
||||||
return {
|
return {
|
||||||
id: bucket.key,
|
id: bucket.path,
|
||||||
text: `${name}/ (${bucket.doc_count})`,
|
text: `${name}/ (${bucket.count})`,
|
||||||
depth: node.depth + 1,
|
depth: node.depth + 1,
|
||||||
index: node.index,
|
index: node.index,
|
||||||
values: [bucket.key],
|
values: [bucket.path],
|
||||||
children: true,
|
children: true,
|
||||||
}
|
}
|
||||||
}).filter(x => x !== null)
|
})
|
||||||
|
.filter(bucket => bucket !== null);
|
||||||
});
|
});
|
||||||
},
|
},
|
||||||
handleTreeClick(e, node, handler) {
|
handleTreeClick(e, node, handler) {
|
||||||
|
|||||||
@@ -17,12 +17,16 @@
|
|||||||
|
|
||||||
<b-collapse id="collapse-1" class="pt-2" style="clear:both;">
|
<b-collapse id="collapse-1" class="pt-2" style="clear:both;">
|
||||||
<b-card>
|
<b-card>
|
||||||
<b-table :items="tableItems" small borderless bordered thead-class="hidden" class="mb-0"></b-table>
|
<b-table :items="tableItems" small borderless thead-class="hidden" class="mb-0"></b-table>
|
||||||
|
|
||||||
|
<template v-if="!$store.state.uiSqliteMode">
|
||||||
|
|
||||||
<br/>
|
<br/>
|
||||||
<h4>
|
<h4>
|
||||||
{{$t("mimeTypes")}}
|
{{ $t("mimeTypes") }}
|
||||||
<b-button size="sm" variant="primary" class="float-right" @click="onCopyClick"><ClipboardIcon/></b-button>
|
<b-button size="sm" variant="primary" class="float-right" @click="onCopyClick">
|
||||||
|
<ClipboardIcon/>
|
||||||
|
</b-button>
|
||||||
</h4>
|
</h4>
|
||||||
<Preloader v-if="$store.state.uiDetailsMimeAgg == null"></Preloader>
|
<Preloader v-if="$store.state.uiDetailsMimeAgg == null"></Preloader>
|
||||||
<b-table
|
<b-table
|
||||||
@@ -30,22 +34,24 @@
|
|||||||
sort-by="doc_count"
|
sort-by="doc_count"
|
||||||
:sort-desc="true"
|
:sort-desc="true"
|
||||||
thead-class="hidden"
|
thead-class="hidden"
|
||||||
:items="$store.state.uiDetailsMimeAgg" small bordered class="mb-0"
|
bordered
|
||||||
|
:items="$store.state.uiDetailsMimeAgg" small class="mb-0"
|
||||||
></b-table>
|
></b-table>
|
||||||
|
</template>
|
||||||
</b-card>
|
</b-card>
|
||||||
</b-collapse>
|
</b-collapse>
|
||||||
</b-card>
|
</b-card>
|
||||||
</template>
|
</template>
|
||||||
|
|
||||||
<script lang="ts">
|
<script>
|
||||||
import Sist2Api, {EsResult} from "@/Sist2Api";
|
|
||||||
import Vue from "vue";
|
import Vue from "vue";
|
||||||
import {humanFileSize} from "@/util";
|
import {humanFileSize} from "@/util";
|
||||||
import DisplayModeToggle from "@/components/DisplayModeToggle.vue";
|
import DisplayModeToggle from "@/components/DisplayModeToggle.vue";
|
||||||
import SortSelect from "@/components/SortSelect.vue";
|
import SortSelect from "@/components/SortSelect.vue";
|
||||||
import Preloader from "@/components/Preloader.vue";
|
import Preloader from "@/components/Preloader.vue";
|
||||||
import Sist2Query from "@/Sist2Query";
|
import Sist2Query from "@/Sist2ElasticsearchQuery";
|
||||||
import ClipboardIcon from "@/components/icons/ClipboardIcon.vue";
|
import ClipboardIcon from "@/components/icons/ClipboardIcon.vue";
|
||||||
|
import Sist2Api from "@/Sist2Api";
|
||||||
|
|
||||||
export default Vue.extend({
|
export default Vue.extend({
|
||||||
name: "ResultsCard",
|
name: "ResultsCard",
|
||||||
@@ -58,13 +64,14 @@ export default Vue.extend({
|
|||||||
return this.$store.state.lastQueryResults != null;
|
return this.$store.state.lastQueryResults != null;
|
||||||
},
|
},
|
||||||
hitCount() {
|
hitCount() {
|
||||||
return (this.$store.state.lastQueryResults as EsResult).aggregations.total_count.value;
|
return (this.$store.state.firstQueryResults).aggregations.total_count.value;
|
||||||
},
|
},
|
||||||
tableItems() {
|
tableItems() {
|
||||||
const items = [];
|
const items = [];
|
||||||
|
|
||||||
|
if (!this.$store.state.uiSqliteMode) {
|
||||||
items.push({key: this.$t("queryTime"), value: this.took()});
|
items.push({key: this.$t("queryTime"), value: this.took()});
|
||||||
|
}
|
||||||
items.push({key: this.$t("totalSize"), value: this.totalSize()});
|
items.push({key: this.$t("totalSize"), value: this.totalSize()});
|
||||||
|
|
||||||
return items;
|
return items;
|
||||||
@@ -72,15 +79,19 @@ export default Vue.extend({
|
|||||||
},
|
},
|
||||||
methods: {
|
methods: {
|
||||||
took() {
|
took() {
|
||||||
return (this.$store.state.lastQueryResults as EsResult).took + "ms";
|
return (this.$store.state.lastQueryResults).took + "ms";
|
||||||
},
|
},
|
||||||
totalSize() {
|
totalSize() {
|
||||||
return humanFileSize((this.$store.state.lastQueryResults as EsResult).aggregations.total_size.value);
|
return humanFileSize((this.$store.state.firstQueryResults).aggregations.total_size.value);
|
||||||
},
|
},
|
||||||
onToggle() {
|
onToggle() {
|
||||||
const show = !document.getElementById("collapse-1").classList.contains("show");
|
const show = !document.getElementById("collapse-1").classList.contains("show");
|
||||||
this.$store.commit("setUiShowDetails", show);
|
this.$store.commit("setUiShowDetails", show);
|
||||||
|
|
||||||
|
if (this.$store.state.uiSqliteMode) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
if (show && this.$store.state.uiDetailsMimeAgg == null && !this.$store.state.optUpdateMimeMap) {
|
if (show && this.$store.state.uiDetailsMimeAgg == null && !this.$store.state.optUpdateMimeMap) {
|
||||||
// Mime aggs are not updated automatically, update now
|
// Mime aggs are not updated automatically, update now
|
||||||
this.forceUpdateMimeAgg();
|
this.forceUpdateMimeAgg();
|
||||||
@@ -88,7 +99,7 @@ export default Vue.extend({
|
|||||||
},
|
},
|
||||||
onCopyClick() {
|
onCopyClick() {
|
||||||
let tsvString = "";
|
let tsvString = "";
|
||||||
this.$store.state.uiDetailsMimeAgg.slice().sort((a,b) => b["doc_count"] - a["doc_count"]).forEach(row => {
|
this.$store.state.uiDetailsMimeAgg.slice().sort((a, b) => b["doc_count"] - a["doc_count"]).forEach(row => {
|
||||||
tsvString += `${row["key"]}\t${row["doc_count"]}\n`;
|
tsvString += `${row["key"]}\t${row["doc_count"]}\n`;
|
||||||
});
|
});
|
||||||
|
|
||||||
|
|||||||
@@ -6,7 +6,7 @@
|
|||||||
@input="setSearchText($event)"></b-form-input>
|
@input="setSearchText($event)"></b-form-input>
|
||||||
|
|
||||||
<template #prepend>
|
<template #prepend>
|
||||||
<b-input-group-text>
|
<b-input-group-text v-if="!$store.state.uiSqliteMode">
|
||||||
<b-form-checkbox :checked="fuzzy" title="Toggle fuzzy searching" @change="setFuzzy($event)">
|
<b-form-checkbox :checked="fuzzy" title="Toggle fuzzy searching" @change="setFuzzy($event)">
|
||||||
{{ $t("searchBar.fuzzy") }}
|
{{ $t("searchBar.fuzzy") }}
|
||||||
</b-form-checkbox>
|
</b-form-checkbox>
|
||||||
|
|||||||
@@ -5,8 +5,7 @@
|
|||||||
<script>
|
<script>
|
||||||
import noUiSlider from 'nouislider';
|
import noUiSlider from 'nouislider';
|
||||||
import 'nouislider/dist/nouislider.css';
|
import 'nouislider/dist/nouislider.css';
|
||||||
import {humanFileSize} from "@/util";
|
import {humanFileSize, mergeTooltips} from "@/util";
|
||||||
import {mergeTooltips} from "@/util-js";
|
|
||||||
|
|
||||||
export default {
|
export default {
|
||||||
name: "SizeSlider",
|
name: "SizeSlider",
|
||||||
|
|||||||