Compare commits
25 Commits
comicvine-
...
airdcpp-au
| Author | SHA1 | Date | |
|---|---|---|---|
| 68907a96a2 | |||
| aff4467b8b | |||
| e7341b553d | |||
| a4ccc78fe8 | |||
| 2247411ac8 | |||
| e61ecb1143 | |||
| e01421f17b | |||
| cc271021e0 | |||
| cc772504ae | |||
| 8dcb17a6a0 | |||
| a06896ffcc | |||
| 03f6623ed0 | |||
| 66f9d63b44 | |||
| a936df3144 | |||
| dc9dabca48 | |||
| 4680fd0875 | |||
| 323548c0ff | |||
| f4563c12c6 | |||
| 1b0cada848 | |||
| 750a74cd9f | |||
| 402ee4d81b | |||
| 1fa35ac0e3 | |||
| 680594e67c | |||
| 5593fcb4a0 | |||
| d7f3d3a7cf |
66
Dockerfile
66
Dockerfile
@@ -1,40 +1,62 @@
|
|||||||
FROM alpine:3.14
|
# Use a base image with Node.js 22.1.0
|
||||||
|
FROM node:22.1.0
|
||||||
|
|
||||||
|
# Set metadata for contact
|
||||||
LABEL maintainer="Rishi Ghan <rishi.ghan@gmail.com>"
|
LABEL maintainer="Rishi Ghan <rishi.ghan@gmail.com>"
|
||||||
|
|
||||||
# Show all node logs
|
# Set environment variables
|
||||||
ENV NPM_CONFIG_LOGLEVEL warn
|
ENV NPM_CONFIG_LOGLEVEL warn
|
||||||
ENV NODE_ENV=production
|
ENV NODE_ENV=production
|
||||||
|
|
||||||
|
# Set the working directory
|
||||||
WORKDIR /core-services
|
WORKDIR /core-services
|
||||||
|
|
||||||
|
# Install required packages
|
||||||
|
RUN apt-get update && apt-get install -y \
|
||||||
|
libvips-tools \
|
||||||
|
wget \
|
||||||
|
imagemagick \
|
||||||
|
python3 \
|
||||||
|
xvfb \
|
||||||
|
xz-utils \
|
||||||
|
curl \
|
||||||
|
bash \
|
||||||
|
software-properties-common
|
||||||
|
|
||||||
RUN apk add --update \
|
# Install p7zip
|
||||||
--repository http://nl.alpinelinux.org/alpine/v3.14/main \
|
RUN apt-get update && apt-get install -y p7zip
|
||||||
vips-tools \
|
|
||||||
wget \
|
|
||||||
imagemagick \
|
|
||||||
python3 \
|
|
||||||
unrar \
|
|
||||||
p7zip \
|
|
||||||
nodejs \
|
|
||||||
npm \
|
|
||||||
xvfb \
|
|
||||||
xz
|
|
||||||
|
|
||||||
|
# Install unrar directly from RARLAB
|
||||||
|
RUN wget https://www.rarlab.com/rar/rarlinux-x64-621.tar.gz \
|
||||||
|
&& tar -zxvf rarlinux-x64-621.tar.gz \
|
||||||
|
&& cp rar/unrar /usr/bin/ \
|
||||||
|
&& rm -rf rarlinux-x64-621.tar.gz rar
|
||||||
|
|
||||||
|
# Clean up package lists
|
||||||
|
RUN rm -rf /var/lib/apt/lists/*
|
||||||
|
|
||||||
|
# Verify Node.js installation
|
||||||
|
RUN node -v && npm -v
|
||||||
|
|
||||||
|
# Copy application configuration files
|
||||||
COPY package.json package-lock.json ./
|
COPY package.json package-lock.json ./
|
||||||
COPY moleculer.config.ts ./
|
COPY moleculer.config.ts ./
|
||||||
COPY tsconfig.json ./
|
COPY tsconfig.json ./
|
||||||
|
COPY scripts ./scripts
|
||||||
|
RUN chmod +x ./scripts/*
|
||||||
|
|
||||||
RUN npm i
|
# Install application dependencies
|
||||||
# Install Dependncies
|
RUN npm install
|
||||||
RUN npm install -g typescript ts-node
|
RUN npm install -g typescript ts-node
|
||||||
|
|
||||||
|
# Copy the rest of the application files
|
||||||
COPY . .
|
COPY . .
|
||||||
|
|
||||||
# Build and cleanup
|
# clean up
|
||||||
RUN npm run build \
|
RUN npm prune
|
||||||
&& npm prune
|
|
||||||
|
|
||||||
|
|
||||||
|
# Expose the application's port
|
||||||
EXPOSE 3000
|
EXPOSE 3000
|
||||||
# Start server
|
|
||||||
CMD ["npm", "start"]
|
# Command to run the application
|
||||||
|
CMD ["npm", "start"]
|
||||||
|
|||||||
@@ -1,10 +1,30 @@
|
|||||||
import { createClient } from "redis";
|
// Import the Redis library
|
||||||
const redisURL = new URL(process.env.REDIS_URI);
|
import IORedis from "ioredis";
|
||||||
|
|
||||||
const pubClient = createClient({ url: `redis://${redisURL.hostname}:6379` });
|
// Environment variable for Redis URI
|
||||||
(async () => {
|
const redisURI = process.env.REDIS_URI || "redis://localhost:6379";
|
||||||
await pubClient.connect();
|
console.log(`process.env.REDIS_URI is ${process.env.REDIS_URI}`);
|
||||||
})();
|
// Creating the publisher client
|
||||||
const subClient = pubClient.duplicate();
|
const pubClient = new IORedis(redisURI);
|
||||||
|
|
||||||
export { subClient, pubClient };
|
// Creating the subscriber client
|
||||||
|
const subClient = new IORedis(redisURI);
|
||||||
|
|
||||||
|
// Handle connection events for the publisher
|
||||||
|
pubClient.on("connect", () => {
|
||||||
|
console.log("Publisher client connected to Redis.");
|
||||||
|
});
|
||||||
|
pubClient.on("error", (err) => {
|
||||||
|
console.error("Publisher client failed to connect to Redis:", err);
|
||||||
|
});
|
||||||
|
|
||||||
|
// Handle connection events for the subscriber
|
||||||
|
subClient.on("connect", () => {
|
||||||
|
console.log("Subscriber client connected to Redis.");
|
||||||
|
});
|
||||||
|
subClient.on("error", (err) => {
|
||||||
|
console.error("Subscriber client failed to connect to Redis:", err);
|
||||||
|
});
|
||||||
|
|
||||||
|
// Export the clients for use in other parts of the application
|
||||||
|
export { pubClient, subClient };
|
||||||
|
|||||||
103
dependencies.docker-compose.yml
Normal file
103
dependencies.docker-compose.yml
Normal file
@@ -0,0 +1,103 @@
|
|||||||
|
services:
|
||||||
|
zoo1:
|
||||||
|
image: confluentinc/cp-zookeeper:7.3.2
|
||||||
|
hostname: zoo1
|
||||||
|
container_name: zoo1
|
||||||
|
ports:
|
||||||
|
- "2181:2181"
|
||||||
|
environment:
|
||||||
|
ZOOKEEPER_CLIENT_PORT: 2181
|
||||||
|
ZOOKEEPER_SERVER_ID: 1
|
||||||
|
ZOOKEEPER_SERVERS: zoo1:2888:3888
|
||||||
|
networks:
|
||||||
|
- kafka-net
|
||||||
|
|
||||||
|
kafka1:
|
||||||
|
image: confluentinc/cp-kafka:7.3.2
|
||||||
|
hostname: kafka1
|
||||||
|
container_name: kafka1
|
||||||
|
ports:
|
||||||
|
- "9092:9092"
|
||||||
|
- "29092:29092"
|
||||||
|
- "9999:9999"
|
||||||
|
environment:
|
||||||
|
KAFKA_ADVERTISED_LISTENERS: INTERNAL://kafka1:19092,EXTERNAL://${DOCKER_HOST_IP:-127.0.0.1}:9092,DOCKER://host.docker.internal:29092
|
||||||
|
KAFKA_LISTENER_SECURITY_PROTOCOL_MAP: INTERNAL:PLAINTEXT,EXTERNAL:PLAINTEXT,DOCKER:PLAINTEXT
|
||||||
|
KAFKA_INTER_BROKER_LISTENER_NAME: INTERNAL
|
||||||
|
KAFKA_ZOOKEEPER_CONNECT: "zoo1:2181"
|
||||||
|
KAFKA_BROKER_ID: 1
|
||||||
|
KAFKA_LOG4J_LOGGERS: "kafka.controller=INFO,kafka.producer.async.DefaultEventHandler=INFO,state.change.logger=INFO"
|
||||||
|
KAFKA_OFFSETS_TOPIC_REPLICATION_FACTOR: 1
|
||||||
|
KAFKA_TRANSACTION_STATE_LOG_REPLICATION_FACTOR: 1
|
||||||
|
KAFKA_TRANSACTION_STATE_LOG_MIN_ISR: 1
|
||||||
|
KAFKA_JMX_PORT: 9999
|
||||||
|
KAFKA_JMX_HOSTNAME: ${DOCKER_HOST_IP:-127.0.0.1}
|
||||||
|
KAFKA_AUTHORIZER_CLASS_NAME: kafka.security.authorizer.AclAuthorizer
|
||||||
|
KAFKA_ALLOW_EVERYONE_IF_NO_ACL_FOUND: "true"
|
||||||
|
depends_on:
|
||||||
|
- zoo1
|
||||||
|
networks:
|
||||||
|
- kafka-net
|
||||||
|
|
||||||
|
kafka-ui:
|
||||||
|
container_name: kafka-ui
|
||||||
|
image: provectuslabs/kafka-ui:latest
|
||||||
|
ports:
|
||||||
|
- 8087:8080
|
||||||
|
environment:
|
||||||
|
DYNAMIC_CONFIG_ENABLED: true
|
||||||
|
volumes:
|
||||||
|
- /Users/rishi/work/config/kafka-ui/config.yml:/etc/kafkaui/dynamic_config.yaml
|
||||||
|
depends_on:
|
||||||
|
- kafka1
|
||||||
|
- zoo1
|
||||||
|
networks:
|
||||||
|
- kafka-net
|
||||||
|
|
||||||
|
db:
|
||||||
|
image: "mongo:latest"
|
||||||
|
container_name: database
|
||||||
|
networks:
|
||||||
|
- kafka-net
|
||||||
|
ports:
|
||||||
|
- "127.0.0.1:27017:27017"
|
||||||
|
volumes:
|
||||||
|
- "mongodb_data:/bitnami/mongodb"
|
||||||
|
|
||||||
|
redis:
|
||||||
|
image: "bitnami/redis:latest"
|
||||||
|
container_name: queue
|
||||||
|
environment:
|
||||||
|
ALLOW_EMPTY_PASSWORD: "yes"
|
||||||
|
networks:
|
||||||
|
- kafka-net
|
||||||
|
ports:
|
||||||
|
- "127.0.0.1:6379:6379"
|
||||||
|
|
||||||
|
elasticsearch:
|
||||||
|
image: docker.elastic.co/elasticsearch/elasticsearch:7.16.2
|
||||||
|
container_name: elasticsearch
|
||||||
|
environment:
|
||||||
|
- "discovery.type=single-node"
|
||||||
|
- "ES_JAVA_OPTS=-Xms512m -Xmx512m"
|
||||||
|
- "xpack.security.enabled=true"
|
||||||
|
- "xpack.security.authc.api_key.enabled=true"
|
||||||
|
- "ELASTIC_PASSWORD=password"
|
||||||
|
ulimits:
|
||||||
|
memlock:
|
||||||
|
soft: -1
|
||||||
|
hard: -1
|
||||||
|
ports:
|
||||||
|
- "127.0.0.1:9200:9200"
|
||||||
|
networks:
|
||||||
|
- kafka-net
|
||||||
|
|
||||||
|
networks:
|
||||||
|
kafka-net:
|
||||||
|
driver: bridge
|
||||||
|
|
||||||
|
volumes:
|
||||||
|
mongodb_data:
|
||||||
|
driver: local
|
||||||
|
elasticsearch:
|
||||||
|
driver: local
|
||||||
@@ -3,7 +3,15 @@ LOGGER=true
|
|||||||
LOGLEVEL=info
|
LOGLEVEL=info
|
||||||
SERVICEDIR=dist/services
|
SERVICEDIR=dist/services
|
||||||
|
|
||||||
TRANSPORTER=nats://nats:4222
|
VITE_UNDERLYING_HOST=localhost
|
||||||
|
COMICS_DIRECTORY=/Users/rishi/work/threetwo-core-service/comics
|
||||||
|
USERDATA_DIRECTORY=/Users/rishi/work/threetwo-core-service/userdata
|
||||||
|
REDIS_URI=redis://redis:6379
|
||||||
|
KAFKA_BROKER=kafka1:9092
|
||||||
|
ELASTICSEARCH_URI=http://elasticsearch:9200
|
||||||
|
MONGO_URI=mongodb://db:27017/threetwo
|
||||||
|
UNRAR_BIN_PATH=/opt/homebrew/bin/unrar
|
||||||
|
SEVENZ_BINARY_PATH=/opt/homebrew/bin/7za
|
||||||
|
|
||||||
CACHER=Memory
|
CACHER=Memory
|
||||||
|
|
||||||
|
|||||||
@@ -1,58 +1,125 @@
|
|||||||
version: "3.3"
|
x-userdata-volume: &userdata-volume
|
||||||
|
type: bind
|
||||||
|
source: ${USERDATA_DIRECTORY}
|
||||||
|
target: /userdata
|
||||||
|
|
||||||
|
x-comics-volume: &comics-volume
|
||||||
|
type: bind
|
||||||
|
source: ${COMICS_DIRECTORY}
|
||||||
|
target: /comics
|
||||||
|
|
||||||
services:
|
services:
|
||||||
|
core-services:
|
||||||
api:
|
|
||||||
build:
|
build:
|
||||||
context: .
|
# context: https://github.com/rishighan/threetwo-core-service.git
|
||||||
image: threetwo-library-service
|
context: ./
|
||||||
env_file: docker-compose.env
|
dockerfile: Dockerfile
|
||||||
environment:
|
image: frishi/threetwo-core-service
|
||||||
SERVICES: api
|
container_name: core-services
|
||||||
PORT: 3000
|
|
||||||
depends_on:
|
|
||||||
- nats
|
|
||||||
labels:
|
|
||||||
- "traefik.enable=true"
|
|
||||||
- "traefik.http.routers.api-gw.rule=PathPrefix(`/`)"
|
|
||||||
- "traefik.http.services.api-gw.loadbalancer.server.port=3000"
|
|
||||||
networks:
|
|
||||||
- internal
|
|
||||||
|
|
||||||
greeter:
|
|
||||||
build:
|
|
||||||
context: .
|
|
||||||
image: threetwo-library-service
|
|
||||||
env_file: docker-compose.env
|
|
||||||
environment:
|
|
||||||
SERVICES: greeter
|
|
||||||
depends_on:
|
|
||||||
- nats
|
|
||||||
networks:
|
|
||||||
- internal
|
|
||||||
|
|
||||||
nats:
|
|
||||||
image: nats:2
|
|
||||||
networks:
|
|
||||||
- internal
|
|
||||||
|
|
||||||
traefik:
|
|
||||||
image: traefik:v2.1
|
|
||||||
command:
|
|
||||||
- "--api.insecure=true" # Don't do that in production!
|
|
||||||
- "--providers.docker=true"
|
|
||||||
- "--providers.docker.exposedbydefault=false"
|
|
||||||
ports:
|
ports:
|
||||||
- 3000:80
|
- "3000:3000"
|
||||||
- 3001:8080
|
- "3001:3001"
|
||||||
|
depends_on:
|
||||||
|
- db
|
||||||
|
- redis
|
||||||
|
- elasticsearch
|
||||||
|
- kafka1
|
||||||
|
- zoo1
|
||||||
|
environment:
|
||||||
|
name: core-services
|
||||||
|
SERVICES: api,library,imagetransformation,opds,search,settings,jobqueue,socket,torrentjobs
|
||||||
|
env_file: docker-compose.env
|
||||||
volumes:
|
volumes:
|
||||||
- /var/run/docker.sock:/var/run/docker.sock:ro
|
- *comics-volume
|
||||||
|
- *userdata-volume
|
||||||
networks:
|
networks:
|
||||||
- internal
|
- proxy
|
||||||
- default
|
|
||||||
|
zoo1:
|
||||||
|
image: confluentinc/cp-zookeeper:7.3.2
|
||||||
|
hostname: zoo1
|
||||||
|
container_name: zoo1
|
||||||
|
ports:
|
||||||
|
- "2181:2181"
|
||||||
|
environment:
|
||||||
|
ZOOKEEPER_CLIENT_PORT: 2181
|
||||||
|
ZOOKEEPER_SERVER_ID: 1
|
||||||
|
ZOOKEEPER_SERVERS: zoo1:2888:3888
|
||||||
|
networks:
|
||||||
|
- proxy
|
||||||
|
|
||||||
|
kafka1:
|
||||||
|
image: confluentinc/cp-kafka:7.3.2
|
||||||
|
hostname: kafka1
|
||||||
|
container_name: kafka1
|
||||||
|
ports:
|
||||||
|
- "9092:9092"
|
||||||
|
- "29092:29092"
|
||||||
|
- "9999:9999"
|
||||||
|
environment:
|
||||||
|
KAFKA_ADVERTISED_LISTENERS: INTERNAL://kafka1:19092,EXTERNAL://${DOCKER_HOST_IP:-127.0.0.1} :9092,DOCKER://host.docker.internal:29092
|
||||||
|
KAFKA_LISTENER_SECURITY_PROTOCOL_MAP: INTERNAL:PLAINTEXT,EXTERNAL:PLAINTEXT,DOCKER:PLAINTEXT
|
||||||
|
KAFKA_INTER_BROKER_LISTENER_NAME: INTERNAL
|
||||||
|
KAFKA_ZOOKEEPER_CONNECT: "zoo1:2181"
|
||||||
|
KAFKA_BROKER_ID: 1
|
||||||
|
KAFKA_LOG4J_LOGGERS: "kafka.controller=INFO,kafka.producer.async.DefaultEventHandler=INFO,state. change.logger=INFO"
|
||||||
|
KAFKA_OFFSETS_TOPIC_REPLICATION_FACTOR: 1
|
||||||
|
KAFKA_TRANSACTION_STATE_LOG_REPLICATION_FACTOR: 1
|
||||||
|
KAFKA_TRANSACTION_STATE_LOG_MIN_ISR: 1
|
||||||
|
KAFKA_JMX_PORT: 9999
|
||||||
|
KAFKA_JMX_HOSTNAME: ${DOCKER_HOST_IP:-127.0.0.1}
|
||||||
|
KAFKA_AUTHORIZER_CLASS_NAME: kafka.security.authorizer.AclAuthorizer
|
||||||
|
KAFKA_ALLOW_EVERYONE_IF_NO_ACL_FOUND: "true"
|
||||||
|
depends_on:
|
||||||
|
- zoo1
|
||||||
|
networks:
|
||||||
|
- proxy
|
||||||
|
|
||||||
|
db:
|
||||||
|
image: "mongo:latest"
|
||||||
|
container_name: database
|
||||||
|
networks:
|
||||||
|
- proxy
|
||||||
|
ports:
|
||||||
|
- "27017:27017"
|
||||||
|
volumes:
|
||||||
|
- "mongodb_data:/bitnami/mongodb"
|
||||||
|
|
||||||
|
redis:
|
||||||
|
image: "bitnami/redis:latest"
|
||||||
|
container_name: redis
|
||||||
|
hostname: redis
|
||||||
|
environment:
|
||||||
|
ALLOW_EMPTY_PASSWORD: "yes"
|
||||||
|
networks:
|
||||||
|
- proxy
|
||||||
|
ports:
|
||||||
|
- "6379:6379"
|
||||||
|
|
||||||
|
elasticsearch:
|
||||||
|
image: docker.elastic.co/elasticsearch/elasticsearch:7.16.2
|
||||||
|
container_name: elasticsearch
|
||||||
|
environment:
|
||||||
|
- "discovery.type=single-node"
|
||||||
|
- "ES_JAVA_OPTS=-Xms512m -Xmx512m"
|
||||||
|
- "xpack.security.enabled=true"
|
||||||
|
- "xpack.security.authc.api_key.enabled=true"
|
||||||
|
- "ELASTIC_PASSWORD=password"
|
||||||
|
ulimits:
|
||||||
|
memlock:
|
||||||
|
soft: -1
|
||||||
|
hard: -1
|
||||||
|
ports:
|
||||||
|
- 9200:9200
|
||||||
|
networks:
|
||||||
|
- proxy
|
||||||
|
|
||||||
networks:
|
networks:
|
||||||
internal:
|
proxy:
|
||||||
|
external: true
|
||||||
|
|
||||||
volumes:
|
volumes:
|
||||||
data:
|
mongodb_data:
|
||||||
|
driver: local
|
||||||
|
elasticsearch:
|
||||||
|
driver: local
|
||||||
|
|||||||
@@ -2,21 +2,60 @@ const path = require("path");
|
|||||||
const mkdir = require("mkdirp").sync;
|
const mkdir = require("mkdirp").sync;
|
||||||
const DbService = require("moleculer-db");
|
const DbService = require("moleculer-db");
|
||||||
|
|
||||||
|
|
||||||
export const DbMixin = (collection, model) => {
|
export const DbMixin = (collection, model) => {
|
||||||
if (process.env.MONGO_URI) {
|
if (!process.env.MONGO_URI) {
|
||||||
const MongooseAdapter = require("moleculer-db-adapter-mongoose");
|
console.log("MONGO_URI not provided, initializing local storage...");
|
||||||
return {
|
mkdir(path.resolve("./data"));
|
||||||
mixins: [DbService],
|
return { mixins: [DbService] }; // Handle case where no DB URI is provided
|
||||||
adapter: new MongooseAdapter(process.env.MONGO_URI, {
|
|
||||||
user: process.env.MONGO_INITDB_ROOT_USERNAME,
|
|
||||||
pass: process.env.MONGO_INITDB_ROOT_PASSWORD,
|
|
||||||
keepAlive: true,
|
|
||||||
useUnifiedTopology: true,
|
|
||||||
family: 4,
|
|
||||||
}),
|
|
||||||
model,
|
|
||||||
};
|
|
||||||
}
|
}
|
||||||
mkdir(path.resolve("./data"));
|
|
||||||
|
const MongooseAdapter = require("moleculer-db-adapter-mongoose");
|
||||||
|
const adapter = new MongooseAdapter(process.env.MONGO_URI, {
|
||||||
|
user: process.env.MONGO_INITDB_ROOT_USERNAME,
|
||||||
|
pass: process.env.MONGO_INITDB_ROOT_PASSWORD,
|
||||||
|
keepAlive: true,
|
||||||
|
useNewUrlParser: true,
|
||||||
|
useUnifiedTopology: true,
|
||||||
|
});
|
||||||
|
|
||||||
|
const connectWithRetry = async (
|
||||||
|
adapter,
|
||||||
|
maxRetries = 5,
|
||||||
|
interval = 5000
|
||||||
|
) => {
|
||||||
|
for (let retry = 0; retry < maxRetries; retry++) {
|
||||||
|
try {
|
||||||
|
await adapter.connect();
|
||||||
|
console.log("MongoDB connected successfully!");
|
||||||
|
return;
|
||||||
|
} catch (err) {
|
||||||
|
console.error("MongoDB connection error:", err);
|
||||||
|
console.log(
|
||||||
|
`Retrying MongoDB connection in ${
|
||||||
|
interval / 1000
|
||||||
|
} seconds...`
|
||||||
|
);
|
||||||
|
await new Promise((resolve) => setTimeout(resolve, interval));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
console.error("Failed to connect to MongoDB after several attempts.");
|
||||||
|
};
|
||||||
|
|
||||||
|
return {
|
||||||
|
mixins: [DbService],
|
||||||
|
adapter,
|
||||||
|
model,
|
||||||
|
collection,
|
||||||
|
async started() {
|
||||||
|
await connectWithRetry(this.adapter);
|
||||||
|
},
|
||||||
|
async stopped() {
|
||||||
|
try {
|
||||||
|
await this.adapter.disconnect();
|
||||||
|
console.log("MongoDB disconnected");
|
||||||
|
} catch (err) {
|
||||||
|
console.error("MongoDB disconnection error:", err);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
};
|
||||||
};
|
};
|
||||||
|
|||||||
@@ -54,30 +54,38 @@ const DirectConnectBundleSchema = mongoose.Schema({
|
|||||||
name: String,
|
name: String,
|
||||||
size: String,
|
size: String,
|
||||||
type: {},
|
type: {},
|
||||||
|
_id: false,
|
||||||
});
|
});
|
||||||
const wantedSchema = new mongoose.Schema({
|
const wantedSchema = mongoose.Schema(
|
||||||
source: { type: String, default: null },
|
{
|
||||||
markEntireVolumeWanted: Boolean,
|
source: { type: String, default: null },
|
||||||
issues: {
|
markEntireVolumeWanted: Boolean,
|
||||||
type: [
|
issues: {
|
||||||
{
|
type: [
|
||||||
|
{
|
||||||
|
_id: false, // Disable automatic ObjectId creation for each issue
|
||||||
|
id: Number,
|
||||||
|
url: String,
|
||||||
|
image: { type: Array, default: [] },
|
||||||
|
coverDate: String,
|
||||||
|
issueNumber: String,
|
||||||
|
},
|
||||||
|
],
|
||||||
|
default: null,
|
||||||
|
},
|
||||||
|
volume: {
|
||||||
|
type: {
|
||||||
|
_id: false, // Disable automatic ObjectId creation for volume
|
||||||
id: Number,
|
id: Number,
|
||||||
url: String,
|
url: String,
|
||||||
image: { type: Array, default: [] },
|
image: { type: Array, default: [] },
|
||||||
|
name: String,
|
||||||
},
|
},
|
||||||
],
|
default: null,
|
||||||
default: null, // Set default to null for issues
|
|
||||||
},
|
|
||||||
volume: {
|
|
||||||
type: {
|
|
||||||
id: Number,
|
|
||||||
url: String,
|
|
||||||
image: { type: Array, default: [] },
|
|
||||||
name: String,
|
|
||||||
},
|
},
|
||||||
default: null, // Set default to null for volume
|
|
||||||
},
|
},
|
||||||
});
|
{ _id: false }
|
||||||
|
); // Disable automatic ObjectId creation for the wanted object itself
|
||||||
|
|
||||||
const ComicSchema = mongoose.Schema(
|
const ComicSchema = mongoose.Schema(
|
||||||
{
|
{
|
||||||
@@ -93,18 +101,12 @@ const ComicSchema = mongoose.Schema(
|
|||||||
},
|
},
|
||||||
sourcedMetadata: {
|
sourcedMetadata: {
|
||||||
comicInfo: { type: mongoose.Schema.Types.Mixed, default: {} },
|
comicInfo: { type: mongoose.Schema.Types.Mixed, default: {} },
|
||||||
comicvine: {
|
comicvine: { type: mongoose.Schema.Types.Mixed, default: {} }, // Set as a freeform object
|
||||||
type: Object,
|
|
||||||
es_indexed: true,
|
|
||||||
default: {},
|
|
||||||
},
|
|
||||||
shortboxed: {},
|
|
||||||
locg: {
|
locg: {
|
||||||
type: LOCGSchema,
|
type: LOCGSchema,
|
||||||
es_indexed: true,
|
es_indexed: true,
|
||||||
default: {},
|
default: {},
|
||||||
},
|
},
|
||||||
gcd: {},
|
|
||||||
},
|
},
|
||||||
rawFileDetails: {
|
rawFileDetails: {
|
||||||
type: RawFileDetailsSchema,
|
type: RawFileDetailsSchema,
|
||||||
|
|||||||
@@ -5,6 +5,7 @@ import {
|
|||||||
MetricRegistry,
|
MetricRegistry,
|
||||||
ServiceBroker,
|
ServiceBroker,
|
||||||
} from "moleculer";
|
} from "moleculer";
|
||||||
|
const RedisTransporter = require("moleculer").Transporters.Redis;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Moleculer ServiceBroker configuration file
|
* Moleculer ServiceBroker configuration file
|
||||||
@@ -90,7 +91,7 @@ const brokerConfig: BrokerOptions = {
|
|||||||
// More info: https://moleculer.services/docs/0.14/networking.html
|
// More info: https://moleculer.services/docs/0.14/networking.html
|
||||||
// Note: During the development, you don't need to define it because all services will be loaded locally.
|
// Note: During the development, you don't need to define it because all services will be loaded locally.
|
||||||
// In production you can set it via `TRANSPORTER=nats://localhost:4222` environment variable.
|
// In production you can set it via `TRANSPORTER=nats://localhost:4222` environment variable.
|
||||||
transporter: process.env.REDIS_URI || "redis://localhost:6379",
|
transporter: new RedisTransporter(process.env.REDIS_URI),
|
||||||
|
|
||||||
// Define a cacher.
|
// Define a cacher.
|
||||||
// More info: https://moleculer.services/docs/0.14/caching.html
|
// More info: https://moleculer.services/docs/0.14/caching.html
|
||||||
|
|||||||
2875
package-lock.json
generated
2875
package-lock.json
generated
File diff suppressed because it is too large
Load Diff
22
package.json
22
package.json
@@ -4,8 +4,8 @@
|
|||||||
"description": "Endpoints for common operations in ThreeTwo",
|
"description": "Endpoints for common operations in ThreeTwo",
|
||||||
"scripts": {
|
"scripts": {
|
||||||
"build": "tsc --build tsconfig.json",
|
"build": "tsc --build tsconfig.json",
|
||||||
"dev": "ts-node ./node_modules/moleculer/bin/moleculer-runner.js --hot --repl --config moleculer.config.ts services/**/*.service.ts",
|
"dev": "./scripts/start.sh dev",
|
||||||
"start": "moleculer-runner --config dist/moleculer.config.js",
|
"start": "npm run build && ./scripts/start.sh prod",
|
||||||
"cli": "moleculer connect NATS",
|
"cli": "moleculer connect NATS",
|
||||||
"ci": "jest --watch",
|
"ci": "jest --watch",
|
||||||
"test": "jest --coverage",
|
"test": "jest --coverage",
|
||||||
@@ -27,6 +27,7 @@
|
|||||||
"eslint-plugin-import": "^2.20.2",
|
"eslint-plugin-import": "^2.20.2",
|
||||||
"eslint-plugin-prefer-arrow": "^1.2.2",
|
"eslint-plugin-prefer-arrow": "^1.2.2",
|
||||||
"install": "^0.13.0",
|
"install": "^0.13.0",
|
||||||
|
"ioredis": "^5.4.1",
|
||||||
"jest": "^29.5.0",
|
"jest": "^29.5.0",
|
||||||
"jest-cli": "^29.5.0",
|
"jest-cli": "^29.5.0",
|
||||||
"moleculer-repl": "^0.7.0",
|
"moleculer-repl": "^0.7.0",
|
||||||
@@ -39,7 +40,7 @@
|
|||||||
},
|
},
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"@bluelovers/fast-glob": "https://github.com/rishighan/fast-glob-v2-api.git",
|
"@bluelovers/fast-glob": "https://github.com/rishighan/fast-glob-v2-api.git",
|
||||||
"@elastic/elasticsearch": "^8.6.0",
|
"@elastic/elasticsearch": "^8.13.1",
|
||||||
"@jorgeferrero/stream-to-buffer": "^2.0.6",
|
"@jorgeferrero/stream-to-buffer": "^2.0.6",
|
||||||
"@npcz/magic": "^1.3.14",
|
"@npcz/magic": "^1.3.14",
|
||||||
"@root/walk": "^1.1.0",
|
"@root/walk": "^1.1.0",
|
||||||
@@ -48,7 +49,8 @@
|
|||||||
"@types/mkdirp": "^1.0.0",
|
"@types/mkdirp": "^1.0.0",
|
||||||
"@types/node": "^13.9.8",
|
"@types/node": "^13.9.8",
|
||||||
"@types/string-similarity": "^4.0.0",
|
"@types/string-similarity": "^4.0.0",
|
||||||
"axios": "^0.25.0",
|
"airdcpp-apisocket": "^2.4.4",
|
||||||
|
"axios": "^1.6.8",
|
||||||
"axios-retry": "^3.2.4",
|
"axios-retry": "^3.2.4",
|
||||||
"bree": "^7.1.5",
|
"bree": "^7.1.5",
|
||||||
"calibre-opds": "^1.0.7",
|
"calibre-opds": "^1.0.7",
|
||||||
@@ -67,22 +69,22 @@
|
|||||||
"mkdirp": "^0.5.5",
|
"mkdirp": "^0.5.5",
|
||||||
"moleculer-bullmq": "^3.0.0",
|
"moleculer-bullmq": "^3.0.0",
|
||||||
"moleculer-db": "^0.8.23",
|
"moleculer-db": "^0.8.23",
|
||||||
"moleculer-db-adapter-mongoose": "^0.9.2",
|
"moleculer-db-adapter-mongoose": "^0.9.4",
|
||||||
"moleculer-io": "^2.2.0",
|
"moleculer-io": "^2.2.0",
|
||||||
"moleculer-web": "^0.10.5",
|
"moleculer-web": "^0.10.7",
|
||||||
"mongoosastic-ts": "^6.0.3",
|
"mongoosastic-ts": "^6.0.3",
|
||||||
"mongoose": "^6.10.4",
|
"mongoose": "^6.10.4",
|
||||||
"mongoose-paginate-v2": "^1.3.18",
|
"mongoose-paginate-v2": "^1.3.18",
|
||||||
"nats": "^1.3.2",
|
"nats": "^1.3.2",
|
||||||
"opds-extra": "^3.0.9",
|
"opds-extra": "^3.0.10",
|
||||||
"p7zip-threetwo": "^1.0.4",
|
"p7zip-threetwo": "^1.0.4",
|
||||||
"redis": "^4.6.5",
|
"redis": "^4.6.14",
|
||||||
"sanitize-filename-ts": "^1.0.2",
|
"sanitize-filename-ts": "^1.0.2",
|
||||||
"sharp": "^0.30.4",
|
"sharp": "^0.33.3",
|
||||||
"threetwo-ui-typings": "^1.0.14",
|
"threetwo-ui-typings": "^1.0.14",
|
||||||
"through2": "^4.0.2",
|
"through2": "^4.0.2",
|
||||||
"unrar": "^0.2.0",
|
"unrar": "^0.2.0",
|
||||||
"xml2js": "^0.4.23"
|
"xml2js": "^0.6.2"
|
||||||
},
|
},
|
||||||
"engines": {
|
"engines": {
|
||||||
"node": ">= 18.x.x"
|
"node": ">= 18.x.x"
|
||||||
|
|||||||
26
scripts/start.sh
Executable file
26
scripts/start.sh
Executable file
@@ -0,0 +1,26 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
echo "Starting script with mode: $MODE"
|
||||||
|
|
||||||
|
# Extract the host and port from MONGO_URI
|
||||||
|
HOST_PORT=$(echo $MONGO_URI | sed -e 's/mongodb:\/\///' -e 's/\/.*$//')
|
||||||
|
|
||||||
|
# Assuming the script is called from the project root
|
||||||
|
PROJECT_ROOT=$(pwd)
|
||||||
|
echo "Project root: $PROJECT_ROOT"
|
||||||
|
|
||||||
|
CONFIG_PATH="$PROJECT_ROOT/moleculer.config.ts"
|
||||||
|
echo "Configuration path: $CONFIG_PATH"
|
||||||
|
|
||||||
|
# Set the correct path for moleculer-runner based on the mode
|
||||||
|
if [ "$MODE" == "dev" ]; then
|
||||||
|
# For development: use ts-node
|
||||||
|
MOLECULER_RUNNER="ts-node $PROJECT_ROOT/node_modules/moleculer/bin/moleculer-runner.js --hot --repl --config $CONFIG_PATH $PROJECT_ROOT/services/**/*.service.ts"
|
||||||
|
echo "Moleculer Runner for dev: $MOLECULER_RUNNER"
|
||||||
|
else
|
||||||
|
# For production: direct node execution of the compiled JavaScript
|
||||||
|
MOLECULER_RUNNER="moleculer-runner --config $PROJECT_ROOT/dist/moleculer.config.js $PROJECT_ROOT/dist/services/**/*.service.js"
|
||||||
|
echo "Moleculer Runner for prod: $MOLECULER_RUNNER"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Run wait-for-it, then start the application
|
||||||
|
./scripts/wait-for-it.sh $HOST_PORT -- $MOLECULER_RUNNER
|
||||||
190
scripts/wait-for-it.sh
Executable file
190
scripts/wait-for-it.sh
Executable file
@@ -0,0 +1,190 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
# Use this script to test if a given TCP host/port are available
|
||||||
|
|
||||||
|
WAITFORIT_cmdname=${0##*/}
|
||||||
|
if [[ $OSTYPE == 'darwin'* ]]; then
|
||||||
|
if ! command -v gtimeout &> /dev/null
|
||||||
|
then
|
||||||
|
echo "missing gtimeout (`brew install coreutils`)"
|
||||||
|
exit
|
||||||
|
fi
|
||||||
|
alias timeout=gtimeout
|
||||||
|
fi
|
||||||
|
|
||||||
|
echoerr() { if [[ $WAITFORIT_QUIET -ne 1 ]]; then echo "$@" 1>&2; fi }
|
||||||
|
|
||||||
|
usage()
|
||||||
|
{
|
||||||
|
cat << USAGE >&2
|
||||||
|
Usage:
|
||||||
|
$WAITFORIT_cmdname host:port [-s] [-t timeout] [-- command args]
|
||||||
|
-h HOST | --host=HOST Host or IP under test
|
||||||
|
-p PORT | --port=PORT TCP port under test
|
||||||
|
Alternatively, you specify the host and port as host:port
|
||||||
|
-s | --strict Only execute subcommand if the test succeeds
|
||||||
|
-q | --quiet Don't output any status messages
|
||||||
|
-t TIMEOUT | --timeout=TIMEOUT
|
||||||
|
Timeout in seconds, zero for no timeout
|
||||||
|
-- COMMAND ARGS Execute command with args after the test finishes
|
||||||
|
USAGE
|
||||||
|
exit 1
|
||||||
|
}
|
||||||
|
|
||||||
|
wait_for()
|
||||||
|
{
|
||||||
|
if [[ $WAITFORIT_TIMEOUT -gt 0 ]]; then
|
||||||
|
echoerr "$WAITFORIT_cmdname: waiting $WAITFORIT_TIMEOUT seconds for $WAITFORIT_HOST:$WAITFORIT_PORT"
|
||||||
|
else
|
||||||
|
echoerr "$WAITFORIT_cmdname: waiting for $WAITFORIT_HOST:$WAITFORIT_PORT without a timeout"
|
||||||
|
fi
|
||||||
|
WAITFORIT_start_ts=$(date +%s)
|
||||||
|
while :
|
||||||
|
do
|
||||||
|
if [[ $WAITFORIT_ISBUSY -eq 1 ]]; then
|
||||||
|
nc -z $WAITFORIT_HOST $WAITFORIT_PORT
|
||||||
|
WAITFORIT_result=$?
|
||||||
|
else
|
||||||
|
(echo -n > /dev/tcp/$WAITFORIT_HOST/$WAITFORIT_PORT) >/dev/null 2>&1
|
||||||
|
WAITFORIT_result=$?
|
||||||
|
fi
|
||||||
|
if [[ $WAITFORIT_result -eq 0 ]]; then
|
||||||
|
WAITFORIT_end_ts=$(date +%s)
|
||||||
|
echoerr "$WAITFORIT_cmdname: $WAITFORIT_HOST:$WAITFORIT_PORT is available after $((WAITFORIT_end_ts - WAITFORIT_start_ts)) seconds"
|
||||||
|
break
|
||||||
|
fi
|
||||||
|
sleep 1
|
||||||
|
done
|
||||||
|
return $WAITFORIT_result
|
||||||
|
}
|
||||||
|
|
||||||
|
wait_for_wrapper()
|
||||||
|
{
|
||||||
|
# In order to support SIGINT during timeout: http://unix.stackexchange.com/a/57692
|
||||||
|
if [[ $WAITFORIT_QUIET -eq 1 ]]; then
|
||||||
|
timeout $WAITFORIT_BUSYTIMEFLAG $WAITFORIT_TIMEOUT $0 --quiet --child --host=$WAITFORIT_HOST --port=$WAITFORIT_PORT --timeout=$WAITFORIT_TIMEOUT &
|
||||||
|
else
|
||||||
|
timeout $WAITFORIT_BUSYTIMEFLAG $WAITFORIT_TIMEOUT $0 --child --host=$WAITFORIT_HOST --port=$WAITFORIT_PORT --timeout=$WAITFORIT_TIMEOUT &
|
||||||
|
fi
|
||||||
|
WAITFORIT_PID=$!
|
||||||
|
trap "kill -INT -$WAITFORIT_PID" INT
|
||||||
|
wait $WAITFORIT_PID
|
||||||
|
WAITFORIT_RESULT=$?
|
||||||
|
if [[ $WAITFORIT_RESULT -ne 0 ]]; then
|
||||||
|
echoerr "$WAITFORIT_cmdname: timeout occurred after waiting $WAITFORIT_TIMEOUT seconds for $WAITFORIT_HOST:$WAITFORIT_PORT"
|
||||||
|
fi
|
||||||
|
return $WAITFORIT_RESULT
|
||||||
|
}
|
||||||
|
|
||||||
|
# process arguments
|
||||||
|
while [[ $# -gt 0 ]]
|
||||||
|
do
|
||||||
|
case "$1" in
|
||||||
|
*:* )
|
||||||
|
WAITFORIT_hostport=(${1//:/ })
|
||||||
|
WAITFORIT_HOST=${WAITFORIT_hostport[0]}
|
||||||
|
WAITFORIT_PORT=${WAITFORIT_hostport[1]}
|
||||||
|
shift 1
|
||||||
|
;;
|
||||||
|
--child)
|
||||||
|
WAITFORIT_CHILD=1
|
||||||
|
shift 1
|
||||||
|
;;
|
||||||
|
-q | --quiet)
|
||||||
|
WAITFORIT_QUIET=1
|
||||||
|
shift 1
|
||||||
|
;;
|
||||||
|
-s | --strict)
|
||||||
|
WAITFORIT_STRICT=1
|
||||||
|
shift 1
|
||||||
|
;;
|
||||||
|
-h)
|
||||||
|
WAITFORIT_HOST="$2"
|
||||||
|
if [[ $WAITFORIT_HOST == "" ]]; then break; fi
|
||||||
|
shift 2
|
||||||
|
;;
|
||||||
|
--host=*)
|
||||||
|
WAITFORIT_HOST="${1#*=}"
|
||||||
|
shift 1
|
||||||
|
;;
|
||||||
|
-p)
|
||||||
|
WAITFORIT_PORT="$2"
|
||||||
|
if [[ $WAITFORIT_PORT == "" ]]; then break; fi
|
||||||
|
shift 2
|
||||||
|
;;
|
||||||
|
--port=*)
|
||||||
|
WAITFORIT_PORT="${1#*=}"
|
||||||
|
shift 1
|
||||||
|
;;
|
||||||
|
-t)
|
||||||
|
WAITFORIT_TIMEOUT="$2"
|
||||||
|
if [[ $WAITFORIT_TIMEOUT == "" ]]; then break; fi
|
||||||
|
shift 2
|
||||||
|
;;
|
||||||
|
--timeout=*)
|
||||||
|
WAITFORIT_TIMEOUT="${1#*=}"
|
||||||
|
shift 1
|
||||||
|
;;
|
||||||
|
--)
|
||||||
|
shift
|
||||||
|
WAITFORIT_CLI=("$@")
|
||||||
|
break
|
||||||
|
;;
|
||||||
|
--help)
|
||||||
|
usage
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
echoerr "Unknown argument: $1"
|
||||||
|
usage
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
done
|
||||||
|
|
||||||
|
if [[ "$WAITFORIT_HOST" == "" || "$WAITFORIT_PORT" == "" ]]; then
|
||||||
|
echoerr "Error: you need to provide a host and port to test."
|
||||||
|
usage
|
||||||
|
fi
|
||||||
|
|
||||||
|
WAITFORIT_TIMEOUT=${WAITFORIT_TIMEOUT:-15}
|
||||||
|
WAITFORIT_STRICT=${WAITFORIT_STRICT:-0}
|
||||||
|
WAITFORIT_CHILD=${WAITFORIT_CHILD:-0}
|
||||||
|
WAITFORIT_QUIET=${WAITFORIT_QUIET:-0}
|
||||||
|
|
||||||
|
# Check to see if timeout is from busybox?
|
||||||
|
WAITFORIT_TIMEOUT_PATH=$(type -p timeout)
|
||||||
|
WAITFORIT_TIMEOUT_PATH=$(realpath $WAITFORIT_TIMEOUT_PATH 2>/dev/null || readlink -f $WAITFORIT_TIMEOUT_PATH)
|
||||||
|
|
||||||
|
WAITFORIT_BUSYTIMEFLAG=""
|
||||||
|
if [[ $WAITFORIT_TIMEOUT_PATH =~ "busybox" ]]; then
|
||||||
|
WAITFORIT_ISBUSY=1
|
||||||
|
# Check if busybox timeout uses -t flag
|
||||||
|
# (recent Alpine versions don't support -t anymore)
|
||||||
|
if timeout &>/dev/stdout | grep -q -e '-t '; then
|
||||||
|
WAITFORIT_BUSYTIMEFLAG="-t"
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
WAITFORIT_ISBUSY=0
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ $WAITFORIT_CHILD -gt 0 ]]; then
|
||||||
|
wait_for
|
||||||
|
WAITFORIT_RESULT=$?
|
||||||
|
exit $WAITFORIT_RESULT
|
||||||
|
else
|
||||||
|
if [[ $WAITFORIT_TIMEOUT -gt 0 ]]; then
|
||||||
|
wait_for_wrapper
|
||||||
|
WAITFORIT_RESULT=$?
|
||||||
|
else
|
||||||
|
wait_for
|
||||||
|
WAITFORIT_RESULT=$?
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ $WAITFORIT_CLI != "" ]]; then
|
||||||
|
if [[ $WAITFORIT_RESULT -ne 0 && $WAITFORIT_STRICT -eq 1 ]]; then
|
||||||
|
echoerr "$WAITFORIT_cmdname: strict mode, refusing to execute subprocess"
|
||||||
|
exit $WAITFORIT_RESULT
|
||||||
|
fi
|
||||||
|
exec "${WAITFORIT_CLI[@]}"
|
||||||
|
else
|
||||||
|
exit $WAITFORIT_RESULT
|
||||||
|
fi
|
||||||
156
services/airdcpp.service.ts
Normal file
156
services/airdcpp.service.ts
Normal file
@@ -0,0 +1,156 @@
|
|||||||
|
"use strict";
|
||||||
|
import {
|
||||||
|
Context,
|
||||||
|
Service,
|
||||||
|
ServiceBroker,
|
||||||
|
ServiceSchema,
|
||||||
|
Errors,
|
||||||
|
} from "moleculer";
|
||||||
|
import axios from "axios";
|
||||||
|
import AirDCPPSocket from "../shared/airdcpp.socket";
|
||||||
|
|
||||||
|
export default class AirDCPPService extends Service {
|
||||||
|
// @ts-ignore
|
||||||
|
public constructor(
|
||||||
|
public broker: ServiceBroker,
|
||||||
|
schema: ServiceSchema<{}> = { name: "airdcpp" }
|
||||||
|
) {
|
||||||
|
super(broker);
|
||||||
|
this.parseServiceSchema({
|
||||||
|
name: "airdcpp",
|
||||||
|
mixins: [],
|
||||||
|
hooks: {},
|
||||||
|
actions: {
|
||||||
|
initialize: {
|
||||||
|
rest: "POST /initialize",
|
||||||
|
handler: async (
|
||||||
|
ctx: Context<{
|
||||||
|
host: {
|
||||||
|
hostname: string;
|
||||||
|
port: string;
|
||||||
|
protocol: string;
|
||||||
|
username: string;
|
||||||
|
password: string;
|
||||||
|
};
|
||||||
|
}>
|
||||||
|
) => {
|
||||||
|
try {
|
||||||
|
const {
|
||||||
|
host: {
|
||||||
|
hostname,
|
||||||
|
protocol,
|
||||||
|
port,
|
||||||
|
username,
|
||||||
|
password,
|
||||||
|
},
|
||||||
|
} = ctx.params;
|
||||||
|
const airDCPPSocket = new AirDCPPSocket({
|
||||||
|
protocol,
|
||||||
|
hostname: `${hostname}:${port}`,
|
||||||
|
username,
|
||||||
|
password,
|
||||||
|
});
|
||||||
|
return await airDCPPSocket.connect();
|
||||||
|
} catch (err) {
|
||||||
|
console.error(err);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
},
|
||||||
|
getHubs: {
|
||||||
|
rest: "POST /getHubs",
|
||||||
|
timeout: 70000,
|
||||||
|
handler: async (
|
||||||
|
ctx: Context<{
|
||||||
|
host: {
|
||||||
|
hostname: string;
|
||||||
|
port: string;
|
||||||
|
protocol: string;
|
||||||
|
username: string;
|
||||||
|
password: string;
|
||||||
|
};
|
||||||
|
}>
|
||||||
|
) => {
|
||||||
|
const {
|
||||||
|
host: {
|
||||||
|
hostname,
|
||||||
|
port,
|
||||||
|
protocol,
|
||||||
|
username,
|
||||||
|
password,
|
||||||
|
},
|
||||||
|
} = ctx.params;
|
||||||
|
try {
|
||||||
|
const airDCPPSocket = new AirDCPPSocket({
|
||||||
|
protocol,
|
||||||
|
hostname: `${hostname}:${port}`,
|
||||||
|
username,
|
||||||
|
password,
|
||||||
|
});
|
||||||
|
await airDCPPSocket.connect();
|
||||||
|
return await airDCPPSocket.get(`hubs`);
|
||||||
|
} catch (err) {
|
||||||
|
throw err;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
},
|
||||||
|
search: {
|
||||||
|
rest: "POST /search",
|
||||||
|
timeout: 20000,
|
||||||
|
handler: async (
|
||||||
|
ctx: Context<{
|
||||||
|
host: {
|
||||||
|
hostname;
|
||||||
|
port;
|
||||||
|
protocol;
|
||||||
|
username;
|
||||||
|
password;
|
||||||
|
};
|
||||||
|
dcppSearchQuery;
|
||||||
|
}>
|
||||||
|
) => {
|
||||||
|
try {
|
||||||
|
const {
|
||||||
|
host: {
|
||||||
|
hostname,
|
||||||
|
port,
|
||||||
|
protocol,
|
||||||
|
username,
|
||||||
|
password,
|
||||||
|
},
|
||||||
|
dcppSearchQuery,
|
||||||
|
} = ctx.params;
|
||||||
|
const airDCPPSocket = new AirDCPPSocket({
|
||||||
|
protocol,
|
||||||
|
hostname: `${hostname}:${port}`,
|
||||||
|
username,
|
||||||
|
password,
|
||||||
|
});
|
||||||
|
await airDCPPSocket.connect();
|
||||||
|
const searchInstance = await airDCPPSocket.post(
|
||||||
|
`search`
|
||||||
|
);
|
||||||
|
|
||||||
|
// Post the search
|
||||||
|
const searchInfo = await airDCPPSocket.post(
|
||||||
|
`search/${searchInstance.id}/hub_search`,
|
||||||
|
dcppSearchQuery
|
||||||
|
);
|
||||||
|
await this.sleep(10000);
|
||||||
|
const results = await airDCPPSocket.get(
|
||||||
|
`search/${searchInstance.id}/results/0/5`
|
||||||
|
);
|
||||||
|
return results;
|
||||||
|
} catch (err) {
|
||||||
|
throw err;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
methods: {
|
||||||
|
sleep: (ms: number) => {
|
||||||
|
return new Promise((resolve) => setTimeout(resolve, ms));
|
||||||
|
},
|
||||||
|
},
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -14,7 +14,6 @@ import { pubClient } from "../config/redis.config";
|
|||||||
import path from "path";
|
import path from "path";
|
||||||
const { MoleculerError } = require("moleculer").Errors;
|
const { MoleculerError } = require("moleculer").Errors;
|
||||||
|
|
||||||
console.log(process.env.REDIS_URI);
|
|
||||||
export default class JobQueueService extends Service {
|
export default class JobQueueService extends Service {
|
||||||
public constructor(public broker: ServiceBroker) {
|
public constructor(public broker: ServiceBroker) {
|
||||||
super(broker);
|
super(broker);
|
||||||
@@ -22,9 +21,10 @@ export default class JobQueueService extends Service {
|
|||||||
name: "jobqueue",
|
name: "jobqueue",
|
||||||
hooks: {},
|
hooks: {},
|
||||||
mixins: [DbMixin("comics", Comic), BullMqMixin],
|
mixins: [DbMixin("comics", Comic), BullMqMixin],
|
||||||
|
|
||||||
settings: {
|
settings: {
|
||||||
bullmq: {
|
bullmq: {
|
||||||
client: process.env.REDIS_URI,
|
client: pubClient,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
actions: {
|
actions: {
|
||||||
@@ -57,20 +57,24 @@ export default class JobQueueService extends Service {
|
|||||||
handler: async (
|
handler: async (
|
||||||
ctx: Context<{ action: string; description: string }>
|
ctx: Context<{ action: string; description: string }>
|
||||||
) => {
|
) => {
|
||||||
const { action, description } = ctx.params;
|
try {
|
||||||
// Enqueue the job
|
const { action, description } = ctx.params;
|
||||||
const job = await this.localQueue(
|
// Enqueue the job
|
||||||
ctx,
|
const job = await this.localQueue(
|
||||||
action,
|
ctx,
|
||||||
ctx.params,
|
action,
|
||||||
{
|
{},
|
||||||
priority: 10,
|
{
|
||||||
}
|
priority: 10,
|
||||||
);
|
}
|
||||||
console.log(`Job ${job.id} enqueued`);
|
);
|
||||||
console.log(`${description}`);
|
console.log(`Job ${job.id} enqueued`);
|
||||||
|
console.log(`${description}`);
|
||||||
|
|
||||||
return job.id;
|
return job.id;
|
||||||
|
} catch (error) {
|
||||||
|
console.error("Failed to enqueue job:", error);
|
||||||
|
}
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
|
|
||||||
|
|||||||
@@ -58,9 +58,11 @@ import klaw from "klaw";
|
|||||||
import path from "path";
|
import path from "path";
|
||||||
import { COMICS_DIRECTORY, USERDATA_DIRECTORY } from "../constants/directories";
|
import { COMICS_DIRECTORY, USERDATA_DIRECTORY } from "../constants/directories";
|
||||||
|
|
||||||
console.log(`MONGO -> ${process.env.MONGO_URI}`);
|
export default class LibraryService extends Service {
|
||||||
export default class ImportService extends Service {
|
public constructor(
|
||||||
public constructor(public broker: ServiceBroker) {
|
public broker: ServiceBroker,
|
||||||
|
schema: ServiceSchema<{}> = { name: "library" }
|
||||||
|
) {
|
||||||
super(broker);
|
super(broker);
|
||||||
this.parseServiceSchema({
|
this.parseServiceSchema({
|
||||||
name: "library",
|
name: "library",
|
||||||
@@ -163,78 +165,51 @@ export default class ImportService extends Service {
|
|||||||
},
|
},
|
||||||
newImport: {
|
newImport: {
|
||||||
rest: "POST /newImport",
|
rest: "POST /newImport",
|
||||||
// params: {},
|
async handler(ctx) {
|
||||||
async handler(
|
const { sessionId } = ctx.params;
|
||||||
ctx: Context<{
|
|
||||||
extractionOptions?: any;
|
|
||||||
sessionId: string;
|
|
||||||
}>
|
|
||||||
) {
|
|
||||||
try {
|
try {
|
||||||
// Get params to be passed to the import jobs
|
// Initialize Redis counters once at the start of the import
|
||||||
const { sessionId } = ctx.params;
|
await pubClient.set("completedJobCount", 0);
|
||||||
// 1. Walk the Source folder
|
await pubClient.set("failedJobCount", 0);
|
||||||
klaw(path.resolve(COMICS_DIRECTORY))
|
|
||||||
// 1.1 Filter on .cb* extensions
|
// Convert klaw to use a promise-based approach for better flow control
|
||||||
.pipe(
|
const files = await this.getComicFiles(
|
||||||
through2.obj(function (item, enc, next) {
|
process.env.COMICS_DIRECTORY
|
||||||
let fileExtension = path.extname(
|
);
|
||||||
item.path
|
for (const file of files) {
|
||||||
);
|
console.info(
|
||||||
if (
|
"Found a file at path:",
|
||||||
[".cbz", ".cbr", ".cb7"].includes(
|
file.path
|
||||||
fileExtension
|
);
|
||||||
)
|
const comicExists = await Comic.exists({
|
||||||
) {
|
"rawFileDetails.name": path.basename(
|
||||||
this.push(item);
|
file.path,
|
||||||
}
|
path.extname(file.path)
|
||||||
next();
|
),
|
||||||
})
|
|
||||||
)
|
|
||||||
// 1.2 Pipe filtered results to the next step
|
|
||||||
// Enqueue the job in the queue
|
|
||||||
.on("data", async (item) => {
|
|
||||||
console.info(
|
|
||||||
"Found a file at path: %s",
|
|
||||||
item.path
|
|
||||||
);
|
|
||||||
let comicExists = await Comic.exists({
|
|
||||||
"rawFileDetails.name": `${path.basename(
|
|
||||||
item.path,
|
|
||||||
path.extname(item.path)
|
|
||||||
)}`,
|
|
||||||
});
|
|
||||||
if (!comicExists) {
|
|
||||||
// 2.1 Reset the job counters in Redis
|
|
||||||
await pubClient.set(
|
|
||||||
"completedJobCount",
|
|
||||||
0
|
|
||||||
);
|
|
||||||
await pubClient.set(
|
|
||||||
"failedJobCount",
|
|
||||||
0
|
|
||||||
);
|
|
||||||
// 2.2 Send the extraction job to the queue
|
|
||||||
this.broker.call("jobqueue.enqueue", {
|
|
||||||
fileObject: {
|
|
||||||
filePath: item.path,
|
|
||||||
fileSize: item.stats.size,
|
|
||||||
},
|
|
||||||
sessionId,
|
|
||||||
importType: "new",
|
|
||||||
action: "enqueue.async",
|
|
||||||
});
|
|
||||||
} else {
|
|
||||||
console.log(
|
|
||||||
"Comic already exists in the library."
|
|
||||||
);
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.on("end", () => {
|
|
||||||
console.log("All files traversed.");
|
|
||||||
});
|
});
|
||||||
|
if (!comicExists) {
|
||||||
|
// Send the extraction job to the queue
|
||||||
|
await this.broker.call("jobqueue.enqueue", {
|
||||||
|
fileObject: {
|
||||||
|
filePath: file.path,
|
||||||
|
fileSize: file.stats.size,
|
||||||
|
},
|
||||||
|
sessionId,
|
||||||
|
importType: "new",
|
||||||
|
action: "enqueue.async",
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
console.log(
|
||||||
|
"Comic already exists in the library."
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
console.log("All files traversed.");
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
console.log(error);
|
console.error(
|
||||||
|
"Error during newImport processing:",
|
||||||
|
error
|
||||||
|
);
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@@ -248,10 +223,7 @@ export default class ImportService extends Service {
|
|||||||
payload: {
|
payload: {
|
||||||
_id?: string;
|
_id?: string;
|
||||||
sourcedMetadata: {
|
sourcedMetadata: {
|
||||||
comicvine?: {
|
comicvine?: any;
|
||||||
volume: { api_detail_url: string };
|
|
||||||
volumeInformation: {};
|
|
||||||
};
|
|
||||||
locg?: {};
|
locg?: {};
|
||||||
};
|
};
|
||||||
inferredMetadata: {
|
inferredMetadata: {
|
||||||
@@ -262,7 +234,7 @@ export default class ImportService extends Service {
|
|||||||
};
|
};
|
||||||
wanted: {
|
wanted: {
|
||||||
issues: [];
|
issues: [];
|
||||||
volume: {};
|
volume: { id: number };
|
||||||
source: string;
|
source: string;
|
||||||
markEntireVolumeWanted: Boolean;
|
markEntireVolumeWanted: Boolean;
|
||||||
};
|
};
|
||||||
@@ -275,41 +247,139 @@ export default class ImportService extends Service {
|
|||||||
}>
|
}>
|
||||||
) {
|
) {
|
||||||
try {
|
try {
|
||||||
let volumeDetails;
|
console.log(
|
||||||
const comicMetadata = ctx.params.payload;
|
JSON.stringify(ctx.params.payload, null, 4)
|
||||||
|
);
|
||||||
|
const { payload } = ctx.params;
|
||||||
|
const { wanted } = payload;
|
||||||
|
|
||||||
console.log("Saving to Mongo...");
|
console.log("Saving to Mongo...");
|
||||||
console.log(
|
|
||||||
`Import type: [${ctx.params.importType}]`
|
if (
|
||||||
);
|
!wanted ||
|
||||||
switch (ctx.params.importType) {
|
!wanted.volume ||
|
||||||
case "new":
|
!wanted.volume.id
|
||||||
console.log(comicMetadata);
|
) {
|
||||||
return await Comic.create(comicMetadata);
|
console.log(
|
||||||
case "update":
|
"No valid identifier for upsert. Attempting to create a new document with minimal data..."
|
||||||
return await Comic.findOneAndUpdate(
|
);
|
||||||
{
|
const newDocument = new Comic(payload); // Using the entire payload for the new document
|
||||||
"acquisition.directconnect.downloads.bundleId":
|
|
||||||
ctx.params.bundleId,
|
await newDocument.save();
|
||||||
},
|
return {
|
||||||
comicMetadata,
|
success: true,
|
||||||
{
|
message:
|
||||||
upsert: true,
|
"New document created due to lack of valid identifiers.",
|
||||||
new: true,
|
data: newDocument,
|
||||||
}
|
};
|
||||||
);
|
|
||||||
default:
|
|
||||||
return false;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
let condition = {
|
||||||
|
"wanted.volume.id": wanted.volume.id,
|
||||||
|
};
|
||||||
|
|
||||||
|
let update: any = {
|
||||||
|
// Using 'any' to bypass strict type checks; alternatively, define a more accurate type
|
||||||
|
$set: {
|
||||||
|
rawFileDetails: payload.rawFileDetails,
|
||||||
|
inferredMetadata: payload.inferredMetadata,
|
||||||
|
sourcedMetadata: payload.sourcedMetadata,
|
||||||
|
},
|
||||||
|
$setOnInsert: {
|
||||||
|
"wanted.source": payload.wanted.source,
|
||||||
|
"wanted.markEntireVolumeWanted":
|
||||||
|
payload.wanted.markEntireVolumeWanted,
|
||||||
|
"wanted.volume": payload.wanted.volume,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
if (wanted.issues && wanted.issues.length > 0) {
|
||||||
|
update.$addToSet = {
|
||||||
|
"wanted.issues": { $each: wanted.issues },
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
const options = {
|
||||||
|
upsert: true,
|
||||||
|
new: true,
|
||||||
|
};
|
||||||
|
|
||||||
|
const result = await Comic.findOneAndUpdate(
|
||||||
|
condition,
|
||||||
|
update,
|
||||||
|
options
|
||||||
|
);
|
||||||
|
console.log(
|
||||||
|
"Operation completed. Document updated or inserted:",
|
||||||
|
result
|
||||||
|
);
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
message: "Document successfully upserted.",
|
||||||
|
data: result,
|
||||||
|
};
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
console.log(error);
|
console.log(error);
|
||||||
throw new Errors.MoleculerError(
|
throw new Errors.MoleculerError(
|
||||||
"Import failed.",
|
"Operation failed.",
|
||||||
500
|
500
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
|
getComicsMarkedAsWanted: {
|
||||||
|
rest: "GET /getComicsMarkedAsWanted",
|
||||||
|
params: {
|
||||||
|
page: { type: "number", default: 1 },
|
||||||
|
limit: { type: "number", default: 100 },
|
||||||
|
},
|
||||||
|
handler: async (
|
||||||
|
ctx: Context<{ page: number; limit: number }>
|
||||||
|
) => {
|
||||||
|
const { page, limit } = ctx.params;
|
||||||
|
this.logger.info(
|
||||||
|
`Requesting page ${page} with limit ${limit}`
|
||||||
|
);
|
||||||
|
try {
|
||||||
|
const options = {
|
||||||
|
page,
|
||||||
|
limit,
|
||||||
|
lean: true,
|
||||||
|
};
|
||||||
|
|
||||||
|
const result = await Comic.paginate(
|
||||||
|
{
|
||||||
|
wanted: { $exists: true },
|
||||||
|
$or: [
|
||||||
|
{
|
||||||
|
"wanted.markEntireVolumeWanted":
|
||||||
|
true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"wanted.issues": {
|
||||||
|
$not: { $size: 0 },
|
||||||
|
},
|
||||||
|
},
|
||||||
|
],
|
||||||
|
},
|
||||||
|
options
|
||||||
|
);
|
||||||
|
|
||||||
|
// Log the raw result from the database
|
||||||
|
this.logger.info(
|
||||||
|
"Paginate result:",
|
||||||
|
JSON.stringify(result, null, 2)
|
||||||
|
);
|
||||||
|
|
||||||
|
return result.docs; // Return just the docs array
|
||||||
|
} catch (error) {
|
||||||
|
this.logger.error("Error finding comics:", error);
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
},
|
||||||
|
|
||||||
applyComicVineMetadata: {
|
applyComicVineMetadata: {
|
||||||
rest: "POST /applyComicVineMetadata",
|
rest: "POST /applyComicVineMetadata",
|
||||||
params: {},
|
params: {},
|
||||||
@@ -754,7 +824,35 @@ export default class ImportService extends Service {
|
|||||||
},
|
},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
methods: {},
|
methods: {
|
||||||
|
// Method to walk the directory and filter comic files
|
||||||
|
getComicFiles: (directory) => {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const files = [];
|
||||||
|
klaw(directory)
|
||||||
|
.pipe(
|
||||||
|
through2.obj(function (item, enc, next) {
|
||||||
|
const fileExtension = path.extname(
|
||||||
|
item.path
|
||||||
|
);
|
||||||
|
if (
|
||||||
|
[".cbz", ".cbr", ".cb7"].includes(
|
||||||
|
fileExtension
|
||||||
|
)
|
||||||
|
) {
|
||||||
|
this.push(item);
|
||||||
|
}
|
||||||
|
next();
|
||||||
|
})
|
||||||
|
)
|
||||||
|
.on("data", (item) => {
|
||||||
|
files.push(item);
|
||||||
|
})
|
||||||
|
.on("end", () => resolve(files))
|
||||||
|
.on("error", (err) => reject(err));
|
||||||
|
});
|
||||||
|
},
|
||||||
|
},
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -76,6 +76,7 @@ export default class SettingsService extends Service {
|
|||||||
}>
|
}>
|
||||||
) {
|
) {
|
||||||
try {
|
try {
|
||||||
|
console.log(ctx.params);
|
||||||
let query = {};
|
let query = {};
|
||||||
const { settingsKey, settingsObjectId } =
|
const { settingsKey, settingsObjectId } =
|
||||||
ctx.params;
|
ctx.params;
|
||||||
|
|||||||
@@ -1,13 +1,13 @@
|
|||||||
"use strict";
|
"use strict";
|
||||||
import { Service, ServiceBroker, ServiceSchema, Context } from "moleculer";
|
import { Service, ServiceBroker, ServiceSchema, Context } from "moleculer";
|
||||||
import { JobType } from "moleculer-bullmq";
|
import { JobType } from "moleculer-bullmq";
|
||||||
import { createClient } from "redis";
|
|
||||||
import { createAdapter } from "@socket.io/redis-adapter";
|
import { createAdapter } from "@socket.io/redis-adapter";
|
||||||
import Session from "../models/session.model";
|
import Session from "../models/session.model";
|
||||||
import { pubClient, subClient } from "../config/redis.config";
|
import { pubClient, subClient } from "../config/redis.config";
|
||||||
const { MoleculerError } = require("moleculer").Errors;
|
const { MoleculerError } = require("moleculer").Errors;
|
||||||
const SocketIOService = require("moleculer-io");
|
const SocketIOService = require("moleculer-io");
|
||||||
const { v4: uuidv4 } = require("uuid");
|
const { v4: uuidv4 } = require("uuid");
|
||||||
|
import AirDCPPSocket from "../shared/airdcpp.socket";
|
||||||
|
|
||||||
export default class SocketService extends Service {
|
export default class SocketService extends Service {
|
||||||
// @ts-ignore
|
// @ts-ignore
|
||||||
@@ -23,10 +23,12 @@ export default class SocketService extends Service {
|
|||||||
port: process.env.PORT || 3001,
|
port: process.env.PORT || 3001,
|
||||||
io: {
|
io: {
|
||||||
namespaces: {
|
namespaces: {
|
||||||
"/": {
|
"/automated": {
|
||||||
events: {
|
events: {
|
||||||
call: {
|
call: {
|
||||||
whitelist: ["socket.*"],
|
whitelist: [
|
||||||
|
"socket.*", // Allow 'search' in the automated namespace
|
||||||
|
],
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@@ -114,9 +116,243 @@ export default class SocketService extends Service {
|
|||||||
// {}
|
// {}
|
||||||
// );
|
// );
|
||||||
},
|
},
|
||||||
|
// AirDCPP Socket actions
|
||||||
|
|
||||||
|
search: {
|
||||||
|
params: {
|
||||||
|
query: "object",
|
||||||
|
config: "object",
|
||||||
|
namespace: "string",
|
||||||
|
},
|
||||||
|
async handler(ctx) {
|
||||||
|
const { query, config, namespace } = ctx.params;
|
||||||
|
const namespacedInstance = this.io.of(namespace || "/");
|
||||||
|
const ADCPPSocket = new AirDCPPSocket(config);
|
||||||
|
try {
|
||||||
|
await ADCPPSocket.connect();
|
||||||
|
const instance = await ADCPPSocket.post(
|
||||||
|
"search",
|
||||||
|
query
|
||||||
|
);
|
||||||
|
|
||||||
|
// Send the instance to the client
|
||||||
|
await namespacedInstance.emit("searchInitiated", {
|
||||||
|
instance,
|
||||||
|
});
|
||||||
|
|
||||||
|
// Setting up listeners
|
||||||
|
await ADCPPSocket.addListener(
|
||||||
|
`search`,
|
||||||
|
`search_result_added`,
|
||||||
|
(data) => {
|
||||||
|
namespacedInstance.emit(
|
||||||
|
"searchResultAdded",
|
||||||
|
{
|
||||||
|
groupedResult: data,
|
||||||
|
instanceId: instance.id,
|
||||||
|
}
|
||||||
|
);
|
||||||
|
},
|
||||||
|
instance.id
|
||||||
|
);
|
||||||
|
|
||||||
|
await ADCPPSocket.addListener(
|
||||||
|
`search`,
|
||||||
|
`search_result_updated`,
|
||||||
|
(data) => {
|
||||||
|
console.log({
|
||||||
|
updatedResult: data,
|
||||||
|
instanceId: instance.id,
|
||||||
|
});
|
||||||
|
namespacedInstance.emit(
|
||||||
|
"searchResultUpdated",
|
||||||
|
{
|
||||||
|
updatedResult: data,
|
||||||
|
instanceId: instance.id,
|
||||||
|
}
|
||||||
|
);
|
||||||
|
},
|
||||||
|
instance.id
|
||||||
|
);
|
||||||
|
|
||||||
|
await ADCPPSocket.addListener(
|
||||||
|
`search`,
|
||||||
|
`search_hub_searches_sent`,
|
||||||
|
async (searchInfo) => {
|
||||||
|
await this.sleep(5000);
|
||||||
|
const currentInstance =
|
||||||
|
await ADCPPSocket.get(
|
||||||
|
`search/${instance.id}`
|
||||||
|
);
|
||||||
|
console.log(
|
||||||
|
JSON.stringify(currentInstance, null, 4)
|
||||||
|
);
|
||||||
|
// Send the instance to the client
|
||||||
|
await namespacedInstance.emit(
|
||||||
|
"searchesSent",
|
||||||
|
{
|
||||||
|
searchInfo,
|
||||||
|
}
|
||||||
|
);
|
||||||
|
|
||||||
|
if (currentInstance.result_count === 0) {
|
||||||
|
console.log("No more search results.");
|
||||||
|
namespacedInstance.emit(
|
||||||
|
"searchComplete",
|
||||||
|
{
|
||||||
|
message:
|
||||||
|
"No more search results.",
|
||||||
|
currentInstance,
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
instance.id
|
||||||
|
);
|
||||||
|
|
||||||
|
// Perform the actual search
|
||||||
|
await ADCPPSocket.post(
|
||||||
|
`search/${instance.id}/hub_search`,
|
||||||
|
query
|
||||||
|
);
|
||||||
|
} catch (error) {
|
||||||
|
await namespacedInstance.emit(
|
||||||
|
"searchError",
|
||||||
|
error.message
|
||||||
|
);
|
||||||
|
throw new MoleculerError(
|
||||||
|
"Search failed",
|
||||||
|
500,
|
||||||
|
"SEARCH_FAILED",
|
||||||
|
{ error }
|
||||||
|
);
|
||||||
|
} finally {
|
||||||
|
await ADCPPSocket.disconnect();
|
||||||
|
}
|
||||||
|
},
|
||||||
|
},
|
||||||
|
download: {
|
||||||
|
// params: {
|
||||||
|
// searchInstanceId: "string",
|
||||||
|
// resultId: "string",
|
||||||
|
// comicObjectId: "string",
|
||||||
|
// name: "string",
|
||||||
|
// size: "number",
|
||||||
|
// type: "any", // Define more specific type if possible
|
||||||
|
// config: "object",
|
||||||
|
// },
|
||||||
|
async handler(ctx) {
|
||||||
|
console.log(ctx.params);
|
||||||
|
const {
|
||||||
|
searchInstanceId,
|
||||||
|
resultId,
|
||||||
|
config,
|
||||||
|
comicObjectId,
|
||||||
|
name,
|
||||||
|
size,
|
||||||
|
type,
|
||||||
|
} = ctx.params;
|
||||||
|
const ADCPPSocket = new AirDCPPSocket(config);
|
||||||
|
try {
|
||||||
|
await ADCPPSocket.connect();
|
||||||
|
const downloadResult = await ADCPPSocket.post(
|
||||||
|
`search/${searchInstanceId}/results/${resultId}/download`
|
||||||
|
);
|
||||||
|
|
||||||
|
if (downloadResult && downloadResult.bundle_info) {
|
||||||
|
// Assume bundle_info is part of the response and contains the necessary details
|
||||||
|
const bundleDBImportResult = await ctx.call(
|
||||||
|
"library.applyAirDCPPDownloadMetadata",
|
||||||
|
{
|
||||||
|
bundleId: downloadResult.bundle_info.id,
|
||||||
|
comicObjectId,
|
||||||
|
name,
|
||||||
|
size,
|
||||||
|
type,
|
||||||
|
}
|
||||||
|
);
|
||||||
|
|
||||||
|
this.logger.info(
|
||||||
|
"Download and metadata update successful",
|
||||||
|
bundleDBImportResult
|
||||||
|
);
|
||||||
|
this.broker.emit(
|
||||||
|
"downloadCompleted",
|
||||||
|
bundleDBImportResult
|
||||||
|
);
|
||||||
|
return bundleDBImportResult;
|
||||||
|
} else {
|
||||||
|
throw new Error(
|
||||||
|
"Failed to download or missing download result information"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
this.broker.emit("downloadError", error.message);
|
||||||
|
throw new MoleculerError(
|
||||||
|
"Download failed",
|
||||||
|
500,
|
||||||
|
"DOWNLOAD_FAILED",
|
||||||
|
{ error }
|
||||||
|
);
|
||||||
|
} finally {
|
||||||
|
// await ADCPPSocket.disconnect();
|
||||||
|
}
|
||||||
|
},
|
||||||
|
},
|
||||||
|
|
||||||
|
listenBundleTick: {
|
||||||
|
async handler(ctx) {
|
||||||
|
const { config } = ctx.params;
|
||||||
|
const ADCPPSocket = new AirDCPPSocket(config);
|
||||||
|
|
||||||
|
try {
|
||||||
|
await ADCPPSocket.connect();
|
||||||
|
console.log("Connected to AirDCPP successfully.");
|
||||||
|
|
||||||
|
ADCPPSocket.addListener(
|
||||||
|
"queue",
|
||||||
|
"queue_bundle_tick",
|
||||||
|
(tickData) => {
|
||||||
|
console.log(
|
||||||
|
"Received tick data: ",
|
||||||
|
tickData
|
||||||
|
);
|
||||||
|
this.io.emit("bundleTickUpdate", tickData);
|
||||||
|
},
|
||||||
|
null
|
||||||
|
); // Assuming no specific ID is needed here
|
||||||
|
} catch (error) {
|
||||||
|
console.error(
|
||||||
|
"Error connecting to AirDCPP or setting listener:",
|
||||||
|
error
|
||||||
|
);
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
methods: {
|
||||||
|
sleep: (ms: number): Promise<NodeJS.Timeout> => {
|
||||||
|
return new Promise((resolve) => setTimeout(resolve, ms));
|
||||||
|
},
|
||||||
},
|
},
|
||||||
methods: {},
|
|
||||||
async started() {
|
async started() {
|
||||||
|
this.logger.info("Starting Socket Service...");
|
||||||
|
this.logger.debug("pubClient:", pubClient);
|
||||||
|
this.logger.debug("subClient:", subClient);
|
||||||
|
if (!pubClient || !subClient) {
|
||||||
|
this.logger.error("Redis clients are not initialized!");
|
||||||
|
throw new Error("Redis clients are not initialized!");
|
||||||
|
}
|
||||||
|
|
||||||
|
// Additional checks or logic if necessary
|
||||||
|
if (pubClient.status !== "ready") {
|
||||||
|
await pubClient.connect();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (subClient.status !== "ready") {
|
||||||
|
await subClient.connect();
|
||||||
|
}
|
||||||
this.io.on("connection", async (socket) => {
|
this.io.on("connection", async (socket) => {
|
||||||
console.log(
|
console.log(
|
||||||
`socket.io server connected to client with session ID: ${socket.id}`
|
`socket.io server connected to client with session ID: ${socket.id}`
|
||||||
|
|||||||
@@ -9,13 +9,14 @@ import {
|
|||||||
import { DbMixin } from "../mixins/db.mixin";
|
import { DbMixin } from "../mixins/db.mixin";
|
||||||
import Comic from "../models/comic.model";
|
import Comic from "../models/comic.model";
|
||||||
import BullMqMixin from "moleculer-bullmq";
|
import BullMqMixin from "moleculer-bullmq";
|
||||||
|
import { pubClient } from "../config/redis.config";
|
||||||
const { MoleculerError } = require("moleculer").Errors;
|
const { MoleculerError } = require("moleculer").Errors;
|
||||||
|
|
||||||
export default class ImageTransformation extends Service {
|
export default class ImageTransformation extends Service {
|
||||||
// @ts-ignore
|
// @ts-ignore
|
||||||
public constructor(
|
public constructor(
|
||||||
public broker: ServiceBroker,
|
public broker: ServiceBroker,
|
||||||
schema: ServiceSchema<{}> = { name: "imagetransformation" }
|
schema: ServiceSchema<{}> = { name: "torrentjobs" }
|
||||||
) {
|
) {
|
||||||
super(broker);
|
super(broker);
|
||||||
this.parseServiceSchema({
|
this.parseServiceSchema({
|
||||||
@@ -23,7 +24,7 @@ export default class ImageTransformation extends Service {
|
|||||||
mixins: [DbMixin("comics", Comic), BullMqMixin],
|
mixins: [DbMixin("comics", Comic), BullMqMixin],
|
||||||
settings: {
|
settings: {
|
||||||
bullmq: {
|
bullmq: {
|
||||||
client: process.env.REDIS_URI,
|
client: pubClient,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
hooks: {},
|
hooks: {},
|
||||||
@@ -77,7 +78,7 @@ export default class ImageTransformation extends Service {
|
|||||||
"qbittorrent.getTorrentRealTimeStats",
|
"qbittorrent.getTorrentRealTimeStats",
|
||||||
{ infoHashes }
|
{ infoHashes }
|
||||||
);
|
);
|
||||||
// 4. Emit the LS_COVER_EXTRACTION_FAILED event with the necessary details
|
// 4.
|
||||||
await this.broker.call("socket.broadcast", {
|
await this.broker.call("socket.broadcast", {
|
||||||
namespace: "/",
|
namespace: "/",
|
||||||
event: "AS_TORRENT_DATA",
|
event: "AS_TORRENT_DATA",
|
||||||
|
|||||||
73
shared/airdcpp.socket.ts
Normal file
73
shared/airdcpp.socket.ts
Normal file
@@ -0,0 +1,73 @@
|
|||||||
|
const WebSocket = require("ws");
|
||||||
|
const { Socket } = require("airdcpp-apisocket");
|
||||||
|
|
||||||
|
class AirDCPPSocket {
|
||||||
|
// Explicitly declare properties
|
||||||
|
options; // Holds configuration options
|
||||||
|
socketInstance; // Instance of the AirDCPP Socket
|
||||||
|
|
||||||
|
constructor(configuration: any) {
|
||||||
|
let socketProtocol = configuration.protocol === "https" ? "wss" : "ws";
|
||||||
|
this.options = {
|
||||||
|
url: `${socketProtocol}://${configuration.hostname}/api/v1/`,
|
||||||
|
autoReconnect: true,
|
||||||
|
reconnectInterval: 5000, // milliseconds
|
||||||
|
logLevel: "verbose",
|
||||||
|
ignoredListenerEvents: [
|
||||||
|
"transfer_statistics",
|
||||||
|
"hash_statistics",
|
||||||
|
"hub_counts_updated",
|
||||||
|
],
|
||||||
|
username: configuration.username,
|
||||||
|
password: configuration.password,
|
||||||
|
};
|
||||||
|
// Initialize the socket instance using the configured options and WebSocket
|
||||||
|
this.socketInstance = Socket(this.options, WebSocket);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Method to ensure the socket connection is established if required by the library or implementation logic
|
||||||
|
async connect() {
|
||||||
|
// Here we'll check if a connect method exists and call it
|
||||||
|
if (
|
||||||
|
this.socketInstance &&
|
||||||
|
typeof this.socketInstance.connect === "function"
|
||||||
|
) {
|
||||||
|
const sessionInformation = await this.socketInstance.connect();
|
||||||
|
return sessionInformation;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Method to ensure the socket is disconnected properly if required by the library or implementation logic
|
||||||
|
async disconnect() {
|
||||||
|
// Similarly, check if a disconnect method exists and call it
|
||||||
|
if (
|
||||||
|
this.socketInstance &&
|
||||||
|
typeof this.socketInstance.disconnect === "function"
|
||||||
|
) {
|
||||||
|
await this.socketInstance.disconnect();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Method to post data to an endpoint
|
||||||
|
async post(endpoint: any, data: any = {}) {
|
||||||
|
// Call post on the socket instance, assuming post is a valid method of the socket instance
|
||||||
|
return await this.socketInstance.post(endpoint, data);
|
||||||
|
}
|
||||||
|
async get(endpoint: any, data: any = {}) {
|
||||||
|
// Call post on the socket instance, assuming post is a valid method of the socket instance
|
||||||
|
return await this.socketInstance.get(endpoint, data);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Method to add listeners to the socket instance for handling real-time updates or events
|
||||||
|
async addListener(event: any, handlerName: any, callback: any, id: any) {
|
||||||
|
// Attach a listener to the socket instance
|
||||||
|
return await this.socketInstance.addListener(
|
||||||
|
event,
|
||||||
|
handlerName,
|
||||||
|
callback,
|
||||||
|
id
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export default AirDCPPSocket;
|
||||||
Reference in New Issue
Block a user