app.service.ts 6.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279
  1. import { Injectable } from '@nestjs/common';
  2. import { ConfigService } from './config.service';
  3. import { DatasetsService } from './datasets.service';
  4. import { DbService } from './db.service';
  5. import { HandbrakeService } from './handbrake.service';
  6. import { MaintenanceService } from './maintenance.service';
  7. import { TaskQueueService } from './task-queue.service';
  8. import { WatcherService } from './watcher.service';
  9. @Injectable()
  10. export class AppService {
  11. constructor(
  12. private readonly db: DbService,
  13. private readonly watcher: WatcherService,
  14. private readonly config: ConfigService,
  15. private readonly maintenance: MaintenanceService,
  16. private readonly handbrake: HandbrakeService,
  17. private readonly datasets: DatasetsService,
  18. private readonly taskQueue: TaskQueueService,
  19. ) {}
  20. listDatasets() {
  21. return this.datasets.getEnabledDatasetPaths();
  22. }
  23. listAllDatasets() {
  24. return this.datasets.getAllDatasetNames();
  25. }
  26. getDatasetConfig() {
  27. return this.datasets.getDatasetConfig();
  28. }
  29. getTotalSuccessfulFiles() {
  30. const datasetNames = this.datasets.getAllDatasetNames();
  31. let total = 0;
  32. for (const datasetName of datasetNames) {
  33. if (datasetName) {
  34. const files = this.db.getAllFiles(datasetName);
  35. total += files.length;
  36. }
  37. }
  38. return total;
  39. }
  40. getTotalProcessedFiles() {
  41. const datasetNames = this.datasets.getAllDatasetNames();
  42. let total = 0;
  43. for (const datasetName of datasetNames) {
  44. if (datasetName) {
  45. const files = this.db.getAllFiles(datasetName);
  46. total += files.length;
  47. }
  48. }
  49. return total;
  50. }
  51. deleteExpiredFiles(days?: number) {
  52. return this.db.deleteExpiredFiles(days);
  53. }
  54. migrateJsonToSqlite(opts?: { datasets?: string[]; dataDir?: string }) {
  55. return this.db.migrateJsonToSqlite(opts);
  56. }
  57. processWithHandbrake(input: string, output: string, preset: string) {
  58. return this.handbrake.processWithHandbrake(input, output, preset);
  59. }
  60. // Task queue methods
  61. createTask(taskData: {
  62. dataset: string;
  63. input: string;
  64. output: string;
  65. preset: string;
  66. priority?: number;
  67. status?: string;
  68. isRequeue?: boolean;
  69. }) {
  70. return this.taskQueue.createTask(taskData);
  71. }
  72. getQueueStatus() {
  73. return this.taskQueue.getQueueStatus();
  74. }
  75. getQueueSettings() {
  76. return this.taskQueue.getQueueSettings();
  77. }
  78. updateQueueSettings(settings: any) {
  79. return this.taskQueue.updateQueueSettings(settings);
  80. }
  81. getAllTasks() {
  82. return this.db.getAllTasks();
  83. }
  84. getTaskById(id: number) {
  85. return this.db.getTaskById(id);
  86. }
  87. updateTask(id: number, updates: any) {
  88. return this.db.updateTask(id, updates);
  89. }
  90. deleteTask(id: number) {
  91. return this.db.deleteTask(id);
  92. }
  93. getHandbrakePresets() {
  94. return this.handbrake.getPresetList();
  95. }
  96. cleanup(file: string, dirs: string[]) {
  97. return this.maintenance.cleanup(file, dirs);
  98. }
  99. purge(dirs: string[], dayMs?: number, cleanerMs?: number) {
  100. return this.maintenance.purge(dirs, dayMs, cleanerMs);
  101. }
  102. prune(dirs: string[]) {
  103. return this.maintenance.prune(dirs);
  104. }
  105. getSettings(key?: string, defaultValue?: any) {
  106. return this.config.getSettings(key, defaultValue);
  107. }
  108. setSettings(settings: Record<string, any>) {
  109. return this.config.setSettings(settings);
  110. }
  111. deleteSetting(key: string) {
  112. return this.config.deleteSetting(key);
  113. }
  114. getConfigFile(name: string) {
  115. return this.config.getConfigFile(name);
  116. }
  117. listConfigs() {
  118. return this.config.listConfigs();
  119. }
  120. startWatcher(watches: string[], options?: any) {
  121. return this.watcher.start(watches, options);
  122. }
  123. async stopWatcher() {
  124. return this.watcher.stop();
  125. }
  126. watcherStatus() {
  127. return this.watcher.status();
  128. }
  129. startTaskProcessing() {
  130. return this.taskQueue.start();
  131. }
  132. stopTaskProcessing(graceful = true) {
  133. return this.taskQueue.stop(graceful);
  134. }
  135. taskProcessingStatus() {
  136. return this.taskQueue.getQueueStatus();
  137. }
  138. // Task maintenance methods
  139. cleanupTasksByStatus(status: string, olderThanDays?: number) {
  140. return this.db.deleteTasksByStatus(status, olderThanDays);
  141. }
  142. cleanupOldTasks(days: number) {
  143. return this.db.deleteTasksOlderThan(days);
  144. }
  145. getTaskStats() {
  146. return this.db.getTaskStats();
  147. }
  148. archiveOldTasks(daysOld: number = 30): { changes?: number } {
  149. return this.db.archiveOldTasks(daysOld);
  150. }
  151. // Purge all tasks
  152. purgeAllTasks() {
  153. return this.db.purgeAllTasks();
  154. }
  155. // Duplicate management
  156. async scanDuplicateFiles(resetExisting?: boolean) {
  157. return await this.maintenance.findDuplicateFiles({ resetExisting });
  158. }
  159. listDuplicateGroups(status?: string, dataset?: string) {
  160. return this.db.listDuplicateGroups(status, dataset);
  161. }
  162. markDuplicateGroup(
  163. id: number,
  164. status: 'pending' | 'reviewed' | 'purged',
  165. note?: string,
  166. ) {
  167. return this.db.markDuplicateGroup(id, status, note);
  168. }
  169. purgeDuplicateFiles(id: number, files: string[], note?: string) {
  170. return this.maintenance.purgeDuplicateFiles(id, files, note);
  171. }
  172. // Destination file indexing
  173. async indexDestinationFiles(
  174. dataset: string,
  175. destination: string,
  176. options?: { reindex?: boolean; batchSize?: number },
  177. ) {
  178. return await this.maintenance.indexDestinationFiles(
  179. dataset,
  180. destination,
  181. options || {},
  182. );
  183. }
  184. async indexSingleFile(dataset: string, filePath: string) {
  185. return await this.maintenance.indexSingleFile(dataset, filePath);
  186. }
  187. async getIndexedDuplicateStats(dataset?: string) {
  188. return await this.maintenance.getIndexedDuplicateStats(dataset);
  189. }
  190. getDestinationFileCount(dataset: string, destination?: string) {
  191. return this.db.getDestinationFileCount(dataset, destination);
  192. }
  193. clearDestinationFiles(dataset: string, destination?: string) {
  194. return this.db.clearDestinationFiles(dataset, destination);
  195. }
  196. removeDuplicateFileEntries() {
  197. return this.db.removeDuplicateFileEntries();
  198. }
  199. // Scheduled maintenance
  200. scheduledTaskCleanup() {
  201. return this.maintenance.scheduledTaskCleanup();
  202. }
  203. getHello(): string {
  204. return 'Hello World!';
  205. }
  206. findFile(dataset: string, file: string) {
  207. return this.db.findFile(dataset, file);
  208. }
  209. setFile(dataset: string, file: string, payload: any) {
  210. return this.db.setFile(dataset, file, payload);
  211. }
  212. removeFile(dataset: string, file: string, soft = true) {
  213. return this.db.removeFile(dataset, file, soft);
  214. }
  215. clearAllFiles() {
  216. return this.db.clearAllFiles();
  217. }
  218. getAllFiles(dataset: string) {
  219. return this.db.getAllFiles(dataset);
  220. }
  221. getDeletedOlderThan(dataset: string, isoDate: string) {
  222. return this.db.getDeletedOlderThan(dataset, isoDate);
  223. }
  224. }