[ { "path": ".editorconfig", "content": "# Editor configuration, see http://editorconfig.org\nroot = true\n\n[*]\ncharset = utf-8\nindent_style = space\nindent_size = 2\ninsert_final_newline = true\ntrim_trailing_whitespace = true\n\n[*.md]\nmax_line_length = off\ntrim_trailing_whitespace = false\n" }, { "path": ".eslintignore", "content": "# Add files here to ignore them in eslint\n\nnode_modules\n*.js\n*.jsx\n*.spec.ts \n*.spec.tsx\n" }, { "path": ".eslintrc.js", "content": "// @ts-check\n/** @type {import('eslint').ESLint.ConfigData} */\nconst config = {\n root: true,\n ignorePatterns: ['**/*'],\n extends: [\n 'airbnb',\n 'airbnb-typescript',\n 'plugin:jest/recommended',\n 'plugin:@nrwl/nx/typescript',\n 'eslint-config-prettier',\n ],\n plugins: ['@nrwl/nx', 'jest'],\n parserOptions: { project: './tsconfig.base.json' },\n overrides: [\n {\n files: ['*.tsx'],\n plugins: ['testing-library'],\n extends: ['plugin:testing-library/react', 'airbnb/hooks'],\n },\n ],\n rules: {\n '@nrwl/nx/enforce-module-boundaries': [\n 'error',\n {\n enforceBuildableLibDependency: true,\n\n // We allow it because we need to use import from\n // postybirb-ui to make jump-to definition in the\n // FieldType.label\n allow: [\n '@postybirb/form-builder',\n '@postybirb/translations',\n '@postybirb/types',\n ],\n\n depConstraints: [{ sourceTag: '*', onlyDependOnLibsWithTags: ['*'] }],\n },\n ],\n '@typescript-eslint/lines-between-class-members': 'off',\n 'no-plusplus': 'off',\n 'no-nested-ternary': 'off',\n 'no-continue': 'off',\n 'no-await-in-loop': 'off',\n 'no-restricted-syntax': 'off',\n 'class-methods-use-this': 'off',\n\n 'import/export': 'off',\n 'import/no-cycle': 'off',\n 'import/prefer-default-export': 'off',\n\n 'react/jsx-props-no-spreading': 'off',\n 'react/react-in-jsx-scope': 'off',\n 'react/no-unescaped-entities': 'off',\n 'react/sort-comp': 'off',\n 'react/prop-types': 'off',\n\n '@typescript-eslint/ban-types': 'warn',\n '@typescript-eslint/no-unused-vars': 'off',\n '@typescript-eslint/no-use-before-define': 'off',\n\n 'import/extension': 'off',\n 'import/no-extraneous-dependencies': [\n 'warn',\n {\n // These dependencies will be considered\n // as bundled and no warning will be shown\n bundledDependencies: ['electron'],\n },\n ],\n '@typescript-eslint/naming-convention': [\n 'error',\n {\n selector: 'variable',\n format: ['camelCase', 'PascalCase', 'UPPER_CASE'],\n\n // Allow const { _ } = useLingui()\n leadingUnderscore: 'allow',\n },\n {\n selector: 'function',\n format: ['camelCase', 'PascalCase'],\n },\n {\n selector: 'typeLike',\n format: ['PascalCase'],\n },\n ],\n },\n};\n\nmodule.exports = config;\n" }, { "path": ".github/workflows/build.yml", "content": "name: Build/Release\n\non:\n workflow_dispatch:\n\njobs:\n build:\n name: ${{ matrix.name }}\n runs-on: ${{ matrix.os }}\n\n strategy:\n fail-fast: false\n matrix:\n include:\n - name: macOS (x64)\n os: macos-latest\n platform: mac\n arch: x64\n - name: macOS (arm64)\n os: macos-latest\n platform: mac\n arch: arm64\n - name: Linux (x64)\n os: ubuntu-latest\n platform: linux\n arch: x64\n - name: Linux (arm64)\n os: ubuntu-24.04-arm\n platform: linux\n arch: arm64\n - name: Windows (x64)\n os: windows-latest\n platform: win\n arch: x64\n\n steps:\n - name: Checkout repository\n uses: actions/checkout@v4\n with:\n fetch-depth: 0\n\n - name: Setup Node.js\n uses: actions/setup-node@v4\n with:\n node-version: \"22\"\n\n - name: Enable corepack to use Yarn v4\n run: |\n corepack enable\n corepack prepare --activate\n\n - name: Cache Electron binaries\n uses: actions/cache@v4\n with:\n path: |\n ~/.cache/electron\n ~/.cache/electron-builder\n node_modules/.cache/electron\n key: ${{ runner.os }}-${{ matrix.arch }}-electron-cache-${{ hashFiles('**/yarn.lock') }}\n restore-keys: |\n ${{ runner.os }}-${{ matrix.arch }}-electron-cache-\n\n - name: Install dependencies\n run: yarn install --immutable\n env:\n CYPRESS_INSTALL_BINARY: 0\n ELECTRON_CACHE: ~/.cache/electron\n ELECTRON_BUILDER_CACHE: ~/.cache/electron-builder\n\n # This is needed to prevent ids showing instead of non translated i18n texts in production builds\n - name: Extract messages\n run: yarn lingui:extract\n\n - name: Patch electron-builder.yml for fork repositories\n if: github.event.repository.fork\n uses: actions/github-script@v8\n with:\n script: |\n const fs = require('fs');\n const yaml = require('js-yaml');\n\n const configPath = './electron-builder.yml';\n const configContent = fs.readFileSync(configPath, 'utf8');\n let config = yaml.load(configContent);\n\n config.publish = {\n provider: 'github',\n owner: context.repo.owner,\n repo: context.repo.repo\n };\n\n if (config.linux && config.linux.target) {\n config.linux.target = [{ target:\"AppImage\" }]\n }\n\n // Remove snap configuration entirely for forks\n delete config.snap;\n\n // Remove deb and rpm configurations for forks\n delete config.deb;\n delete config.rpm;\n\n // Write the modified config back\n fs.writeFileSync(configPath, yaml.dump(config));\n\n console.log('✅ Patched electron-builder.yml for fork repository');\n console.log(`📦 Publish target: ${context.repo.owner}/${context.repo.repo}`);\n console.log(yaml.dump(config))\n\n - name: Install snapcraft\n if: matrix.platform == 'linux' && !github.event.repository.fork\n run: |\n sudo apt-get update\n sudo apt-get install -y snapd\n sudo snap install snapcraft --classic\n\n - name: Install Ruby + fpm (system) for Linux packaging\n if: matrix.platform == 'linux' && !github.event.repository.fork\n run: |\n sudo apt-get update\n sudo apt-get install -y ruby ruby-dev build-essential rpm\n sudo gem install --no-document fpm\n # Ensure gem bindir is on PATH for subsequent steps\n echo \"$(ruby -e 'print Gem.bindir')\" >> \"$GITHUB_PATH\"\n echo \"Ruby:\"; ruby -v\n echo \"fpm:\"; fpm --version\n\n - name: Inject Application Insights connection string\n env:\n APP_INSIGHTS_KEY: ${{ secrets.APP_INSIGHTS_KEY }}\n run: corepack yarn inject:app-insights\n\n - name: Build application\n run: corepack yarn build:prod\n env:\n NODE_ENV: production\n\n - name: Build Electron app (fork)\n if: github.event.repository.fork\n run: corepack yarn electron-builder --${{ matrix.platform }} --${{ matrix.arch }} --publish=always\n env:\n GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}\n\n - name: Build Electron app\n if: (!github.event.repository.fork)\n run: corepack yarn electron-builder --${{ matrix.platform }} --${{ matrix.arch }} --publish=always\n env:\n GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}\n APPLE_ID: ${{ secrets.APPLE_ID }}\n APPLE_APP_SPECIFIC_PASSWORD: ${{ secrets.APPLE_APP_SPECIFIC_PASSWORD }}\n APPLE_TEAM_ID: ${{ secrets.APPLE_TEAM_ID }}\n CSC_LINK: ${{ secrets.CSC_LINK }}\n CSC_KEY_PASSWORD: ${{ secrets.CSC_KEY_PASSWORD }}\n CSC_IDENTITY_AUTO_DISCOVERY: true\n SNAPCRAFT_STORE_CREDENTIALS: ${{ secrets.SNAPCRAFT_STORE_CREDENTIALS }}\n SNAPCRAFT_BUILD_ENVIRONMENT: host\n # ✅ Use system fpm for Linux only\n USE_SYSTEM_FPM: ${{ matrix.platform == 'linux' && 'true' || '' }}\n\n - name: Extract version from package.json for docker tag\n uses: actions/github-script@v8\n id: packageVersion\n with:\n script: |\n const fs = require('fs');\n const pkg = JSON.parse(fs.readFileSync(\"package.json\", \"utf8\"))\n return pkg.version\n\n - name: Build and publish a Docker image for '${{github.repository}}'\n uses: macbre/push-to-ghcr@master\n if: matrix.platform == 'linux' && matrix.arch == 'x64'\n with:\n image_name: \"${{github.repository}}\"\n extra_args: --tag ghcr.io/${{github.repository}}:${{steps.packageVersion.outputs.result}}\n github_token: ${{ secrets.GITHUB_TOKEN }}\n dockerfile: \"./Dockerfile\"\n" }, { "path": ".github/workflows/ci.yml", "content": "name: CI\non:\n workflow_dispatch:\n pull_request:\n push:\n branches:\n - main\n\n# Needed for nx-set-shas when run on the main branch\npermissions:\n actions: read\n contents: read\n\njobs:\n ci:\n name: ${{ matrix.ci.name }}\n runs-on: ubuntu-latest\n if: github.head_ref != 'weblate-postybirb-postybirb'\n strategy:\n matrix:\n ci:\n - name: Lint\n command: \"yarn nx affected -t lint --cache --cache-strategy content\n --cache-location .eslint\"\n\n - name: TypeCheck\n command: \"yarn nx affected -t typecheck\"\n\n # xvfb-run is needed because electron still tries to get $DISPLAY env even if it does not launches the window\n # so to run tests on electron runner we need to add this command\n # grep filters out useless spam in console about not able to access display.\n # pipefail needed because otherwise grep returns exit code 0 and failed tests don't display in ci as failed\n - name: Test\n command: \"set -o pipefail && xvfb-run --auto-servernum --server-args=\\\"-screen 0\n 1280x960x24\\\" -- yarn nx affected -t test | grep -v -E\n \\\"ERROR:viz_main_impl\\\\.cc\\\\(183\\\\)|ERROR:object_proxy\\\\.cc\\\\(576\\\n \\\\)|ERROR:bus\\\\.cc\\\\(408\\\\)|ERROR:gles2_cmd_decoder_passthrough\\\\\\\n .cc\\\\(1094\\\\)|ERROR:gl_utils\\\\.cc\\\\(431\\\\)|ERROR:browser_main_loop\\\n \\\\.cc\\\\(276\\\\)\\\"\"\n\n steps:\n - name: Check out Git repository\n uses: actions/checkout@v4\n with:\n # We need to fetch all branches and commits so that Nx affected has a base to compare against.\n fetch-depth: 0\n\n - name: Use eslint, jest and tsc cache\n uses: actions/cache@v4\n with:\n # out-tsc needed because we use incremental in typecheck which uses files from tsc output\n path: |\n .eslint\n .jest\n dist/out-tsc\n\n key: ${{ runner.os }}-ci\n # This is needed to be run before node setup, because node setup uses yarn to get cache dir\n - name: Enable corepack to use Yarn v4\n run: corepack enable\n\n - name: Install Node.js and Yarn\n uses: actions/setup-node@v4\n with:\n node-version: \"22\"\n cache: yarn\n # These flags needed to speed up ci\n - name: Install dependencies\n run: yarn install --immutable\n env:\n CYPRESS_INSTALL_BINARY: 0 # Skip downloading Cypress binary (not needed for CI)\n # Used to calculate affected\n - name: Setup SHAs\n uses: nrwl/nx-set-shas@v4\n\n - name: Run ${{ matrix.ci.name }}\n run: ${{ matrix.ci.command }}\n timeout-minutes: 30\n" }, { "path": ".github/workflows/i18n.yml.disabled", "content": "name: i18n CI\non:\n push:\n branches:\n - main\n\npermissions:\n actions: read\n contents: write\n checks: write\n pull-requests: write\n\njobs:\n main:\n runs-on: ubuntu-latest\n if: github.head_ref != 'weblate-postybirb-postybirb'\n steps:\n - name: Check out Git repository\n uses: actions/checkout@v4\n\n - name: Install Node.js, NPM and Yarn\n uses: actions/setup-node@v4\n with:\n cache: yarn\n\n # Note: These flags needed to speed up ci\n - name: Install dependencies\n run: yarn install --frozen-lockfile --prefer-offline\n\n # Extracts i18n strings from code\n - name: Extract i18n keys\n run: yarn lingui:extract\n\n # It commits extracted i18n string and eslint/prettier fixes\n - name: Commit changes\n run: |\n git config --local user.email \"41898282+github-actions[bot]@users.noreply.github.com\"\n git config --local user.name \"github-actions[bot]\" \n # pull only from main branch so it will not fetch all tags\n git pull origin main\n \n git add lang/*\n # ignore when nothing to commit\n yarn exitzero \"git commit -m 'ci: extract i18n messages'\"\n\n - name: Push changes\n uses: ad-m/github-push-action@master\n with:\n github_token: ${{ secrets.GITHUB_TOKEN }}\n branch: ${{ github.ref }}\n" }, { "path": ".gitignore", "content": "# See http://help.github.com/ignore-files/ for more about ignoring files.\n\n# yarn\n.yarn/*\n!.yarn/patches\n!.yarn/plugins\n!.yarn/releases\n!.yarn/sdks\n!.yarn/versions\n\n# compiled output\n/release\n/dist\n/tmp\n/out-tsc\n/test\n/.swc\n/.jest\n/.eslint\n.jest\n.swc\n\n# dependencies\n/node_modules\nscripts/node_modules\n\n# IDEs and editors\n/.idea\n.project\n.classpath\n.c9/\n*.launch\n.settings/\n*.sublime-workspace\n\n# IDE - VSCode\n.vscode/*\n!.vscode/settings.json\n!.vscode/tasks.json\n!.vscode/launch.json\n!.vscode/extensions.json\n\n# misc\n/.sass-cache\n/connect.lock\n/coverage\n/libpeerconnection.log\nnpm-debug.log\nyarn-error.log\ntestem.log\n/typings\n\n# System Files\n.DS_Store\nThumbs.db\n\n# Public UI dependencies\n.nx/cache\n.nx/workspace-data\n" }, { "path": ".husky/commit-msg", "content": "npx --no-install commitlint --edit \"$1\"\n" }, { "path": ".husky/post-merge", "content": "yarn\n" }, { "path": ".prettierignore", "content": "# Add files here to ignore them from prettier formatting\n\n/dist/**\n/coverage/**\n/.nx/cache/**\n/.nx/workspace-data/**\n\n*.hbs" }, { "path": ".prettierrc", "content": "{\n \"singleQuote\": true,\n \"endOfLine\": \"crlf\"\n}\n" }, { "path": ".vscode/extensions.json", "content": "{\n \"recommendations\": [\n \"nrwl.angular-console\",\n \"esbenp.prettier-vscode\",\n \"firsttris.vscode-jest-runner\",\n \"dbaeumer.vscode-eslint\"\n ]\n}\n" }, { "path": ".vscode/launch.json", "content": "{\n \"version\": \"0.2.0\",\n \"configurations\": [\n {\n \"name\": \"Debug Node App\",\n \"type\": \"node\",\n \"request\": \"attach\",\n \"port\": 9229\n },\n {\n \"name\": \"Debug Jest Tests\",\n \"type\": \"node\",\n \"request\": \"launch\",\n \"program\": \"${workspaceFolder}/node_modules/.bin/jest\",\n \"args\": [\n \"--runInBand\",\n \"--no-coverage\",\n \"--no-cache\",\n \"${relativeFile}\"\n ],\n \"console\": \"integratedTerminal\",\n \"internalConsoleOptions\": \"neverOpen\",\n \"disableOptimisticBPs\": true,\n \"windows\": {\n \"program\": \"${workspaceFolder}/node_modules/jest/bin/jest\"\n },\n \"sourceMaps\": true,\n \"outFiles\": [\n \"${workspaceFolder}/**/*.js\",\n \"!**/node_modules/**\"\n ],\n \"skipFiles\": [\n \"/**\",\n \"**/node_modules/**\"\n ],\n \"env\": {\n \"NODE_ENV\": \"test\"\n }\n },\n {\n \"name\": \"Debug Current Jest Test\",\n \"type\": \"node\",\n \"request\": \"launch\",\n \"program\": \"${workspaceFolder}/node_modules/.bin/jest\",\n \"args\": [\n \"--runInBand\",\n \"--no-coverage\",\n \"--no-cache\",\n \"--testNamePattern=${input:testNamePattern}\",\n \"${relativeFile}\"\n ],\n \"console\": \"integratedTerminal\",\n \"internalConsoleOptions\": \"neverOpen\",\n \"disableOptimisticBPs\": true,\n \"windows\": {\n \"program\": \"${workspaceFolder}/node_modules/jest/bin/jest\"\n },\n \"sourceMaps\": true,\n \"outFiles\": [\n \"${workspaceFolder}/**/*.js\",\n \"!**/node_modules/**\"\n ],\n \"skipFiles\": [\n \"/**\",\n \"**/node_modules/**\"\n ],\n \"env\": {\n \"NODE_ENV\": \"test\"\n }\n }\n ],\n \"inputs\": [\n {\n \"id\": \"testNamePattern\",\n \"description\": \"Test name pattern\",\n \"default\": \".*\",\n \"type\": \"promptString\"\n }\n ]\n}\n" }, { "path": ".vscode/settings.json", "content": "{\n \"editor.codeActionsOnSave\": {\n \"source.organizeImports\": \"always\"\n }\n}\n\n" }, { "path": ".yarn/patches/@handlewithcare-prosemirror-inputrules-npm-0.1.3-897e37b56f.patch", "content": "diff --git a/package.json b/package.json\nindex 20bc41090c166b7b27756b8d8e0fb67525d48f38..a539a8e91499f831bd77b4fa43e98f3970ea57e8 100644\n--- a/package.json\n+++ b/package.json\n@@ -9,6 +9,7 @@\n \"exports\": {\n \".\": {\n \"types\": \"./dist/index.d.ts\",\n+ \"require\": \"./dist/index.js\",\n \"import\": \"./dist/index.js\"\n }\n },\n" }, { "path": ".yarn/patches/@tiptap-html-npm-3.15.3-a9641901db.patch", "content": "diff --git a/package.json b/package.json\nindex 1d1aa65a70e7f3e898be608610693a9b98152b8c..e27af24e756e7c6b76651d504304f28e7de4e084 100644\n--- a/package.json\n+++ b/package.json\n@@ -12,28 +12,6 @@\n \"type\": \"github\",\n \"url\": \"https://github.com/sponsors/ueberdosis\"\n },\n- \"exports\": {\n- \".\": {\n- \"types\": {\n- \"import\": \"./dist/index.d.ts\",\n- \"require\": \"./dist/index.d.cts\"\n- },\n- \"import\": {\n- \"browser\": \"./dist/index.js\",\n- \"node\": \"./dist/server/index.js\",\n- \"default\": \"./dist/index.js\"\n- },\n- \"require\": \"./dist/index.cjs\"\n- },\n- \"./server\": {\n- \"types\": {\n- \"import\": \"./dist/server/index.d.ts\",\n- \"require\": \"./dist/server/index.d.cts\"\n- },\n- \"import\": \"./dist/server/index.js\",\n- \"require\": \"./dist/server/index.cjs\"\n- }\n- },\n \"main\": \"dist/index.cjs\",\n \"module\": \"dist/index.js\",\n \"types\": \"dist/index.d.ts\",\n" }, { "path": ".yarn/patches/jest-snapshot-npm-29.7.0-15ef0a4ad6.patch", "content": "diff --git a/build/InlineSnapshots.js b/build/InlineSnapshots.js\nindex 3481ad99885c847156afdef148d3075dcc9c68ca..44c91da106e6111f95c75b37bc41e5cadebcc145 100644\n--- a/build/InlineSnapshots.js\n+++ b/build/InlineSnapshots.js\n@@ -149,6 +149,7 @@ const saveSnapshotsForFile = (snapshots, sourceFilePath, rootDir, prettier) => {\n filename: sourceFilePath,\n plugins,\n presets,\n+ configFile: path.join(process.cwd(),'./babel.config.js'),\n root: rootDir\n });\n } catch (error) {\n" }, { "path": ".yarn/patches/strong-log-transformer-npm-2.1.0-45addd9278.patch", "content": "diff --git a/lib/logger.js b/lib/logger.js\nindex 69218a870e6022289a73c27cb2d4f166bf1691d9..8d6554bce9dd8892c945ef0740b0e9d6ba56798c 100644\n--- a/lib/logger.js\n+++ b/lib/logger.js\n@@ -29,7 +29,7 @@ var formatters = {\n \n function Logger(options) {\n var defaults = JSON.parse(JSON.stringify(Logger.DEFAULTS));\n- options = util._extend(defaults, options || {});\n+ options = Object.assign(defaults, options || {});\n var catcher = deLiner();\n var emitter = catcher;\n var transforms = [\n" }, { "path": ".yarnrc.yml", "content": "logFilters:\n # NX packages peer dependencies make a lot of warnings that does not affect code in any way\n - code: YN0086\n level: discard\n\n # This warning doesn't make sense. This package is not referenced in code at all and wasn't updated in 2 years\n - code: YN0002\n level: discard\n text: doesn't provide @mantine/utils (p2df71b), requested by @blocknote/mantine.\n\nnodeLinker: node-modules\n\nsupportedArchitectures:\n cpu:\n - current\n - x64\n - arm64\n" }, { "path": "Dockerfile", "content": "# Multi stage build to make image smaller\nFROM node:24-bookworm-slim AS builder\n\n# For ca-certificates\nRUN apt-get update && apt-get install -y curl \n\nWORKDIR /source\n\nCOPY . .\n\n# Conditional build - only build if release/linux-unpacked doesn't exist\nRUN if [ -d \"./release/linux-unpacked\" ]; then \\\n echo \"Found existing build, copying...\"; \\\n cp -r ./release/linux-unpacked/ /app;\\\n else \\\n echo \"Building from source...\"; rm -rf .nx && \\\n CYPRESS_INSTALL_BINARY=0 corepack yarn install --inline-builds && \\\n corepack yarn dist:linux --dir && \\\n cp -r ./release/linux-unpacked/ /app;\\\n fi \n \nFROM node:24-bookworm-slim\n\nCOPY --from=builder /app /app\n\n# Install dependencies for Electron and headless display\nRUN apt-get update && apt-get install -y \\\n libgtk-3-0 \\\n libnss3 \\\n libasound2 \\\n libxss1 \\\n libgbm-dev \\\n libxshmfence-dev \\\n libdrm-dev \\\n # For ca-certificates and healthcheck\n curl \\ \n xvfb \\\n && rm -rf /var/lib/apt/lists/*\n\n\nWORKDIR /app\n\n# Contains database, submissions, tags etc\nVOLUME [ \"/root/PostyBirb\" ]\n# Contains startup options, remote config, partitions etc\nVOLUME [ \"/root/.config/postybirb\" ]\n\nENV DISPLAY=:99\n\nEXPOSE 8080\n\nHEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=5 \\\n CMD curl http://127.0.0.1:8080 || [ $? -eq 52 ] && exit 0 || exit 1\n\nCOPY ./entrypoint.sh .\n\nENTRYPOINT [ \"bash\" ]\n\nCMD [ \"entrypoint.sh\" ]\n" }, { "path": "LICENSE", "content": "BSD 3-Clause License\n\nCopyright (c) 2024, Michael DiCarlo\nAll rights reserved.\n\nRedistribution and use in source and binary forms, with or without\nmodification, are permitted provided that the following conditions are met:\n\n1. Redistributions of source code must retain the above copyright notice, this\n list of conditions and the following disclaimer.\n\n2. Redistributions in binary form must reproduce the above copyright notice,\n this list of conditions and the following disclaimer in the documentation\n and/or other materials provided with the distribution.\n\n3. Neither the name of the copyright holder nor the names of its\n contributors may be used to endorse or promote products derived from\n this software without specific prior written permission.\n\nTHIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS \"AS IS\"\nAND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE\nIMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE\nDISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE\nFOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL\nDAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR\nSERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER\nCAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,\nOR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\nOF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n" }, { "path": "README.md", "content": "# Postybirb\n\n
\n\n\"Static\n\n\n\"GitHub\n\n\n\"Translation\n\n\"GitHub\n
\n\n## About\n\nPostyBirb is an application that helps artists post art and other multimedia to\nmultiple websites more quickly.The overall goal of PostyBirb is to cut down on\nthe time it takes to post submissions to multiple websites.\n\n## V4 Initiative\n\nv4 sets out to be more flexible for adding new features and updates not easily\nsupported on v3. It also aims to be more contributor friendly and ease the\nimplementation of websites where possible.\n\n## Looking for v3 (PostyBirb+)?\n\nYou can find v3 [here](https://github.com/mvdicarlo/postybirb-plus).\n\n## Translation\n\n![Translation status badge](https://hosted.weblate.org/widget/postybirb/postybirb/287x66-black.png)\n\nPostyBirb uses [Weblate](https://hosted.weblate.org/projects/postybirb/postybirb/) as transltion service\n\nLearn more: [Translation guide](./TRANSLATION.md)\n\n## Project Setup\n\n1. Ensure your NodeJS version is 24.6.0 or higher\n2. Clone project using git\n3. `corepack enable` Make NodeJS use the yarn version specific to the project (from package.json)\n4. `yarn install` Installs dependencies\n5. If it fails with `➤ YN0009: │ better-sqlite3@npm:11.8.0 couldn't be built successfully`: \n\n
\n If you're on windows run\n\n```\nwinget install -e --id Microsoft.VisualStudio.2022.BuildTools --override \"--passive --wait --add Microsoft.VisualStudio.Workload.VCTools;includeRecommended\"\n```\n\nIf you're on linux or other OS please create an issue with log from the unsucessfull build. It will have instructions on which packages are required and we will add them there. But generally it should work out of box if you have C++ compiler installed\n\n
\n\n\n\n6. `yarn run setup` Installs hooks/husky\n7. `yarn start` Starts app\n\n### Recommended Plugins (VSCode)\n\n- Nx Console\n- Jest Runner\n- Prettier\n\n## Contributing\n\nPlease write clean code.\n\nFollow [conventional commits](https://www.conventionalcommits.org/en/v1.0.0/)\n\nTo add new website [see this guide](./contributing/add-a-website)\n\n## Primary Modules (/apps)\n\n### Client-Server\n\nThe \"Back end\" of the application. This houses all data models, user settings,\nposting logic, etc.\n\n#### Primary Technologies Used\n\n- NestJS\n- Drizzle (sqlite3)\n\n### Postybirb\n\nThe Electron part of the application that contains initialization logic and\napp setup.\n\n#### Primary Technologies Used\n\n- Electron\n\n### PostyBirb-UI\n\nThe user interface for the application that talks with Client-Server through\nweb-socket and https.\n\n#### Primary Technologies Used\n\n- React\n- Blocknote/TipTap (Text editor)\n\n---\n\nThis project was generated using [Nx](https://nx.dev).\n" }, { "path": "TRANSLATION.md", "content": "# Translation guide\n\n## How to contribute to translation\n\nGo to [PostyBirb site](https://hosted.weblate.org/projects/postybirb/postybirb/), create account and suggest translation!\n\n## Add new language\n\nTo add new language you need to add [language 2-letter code](https://www.loc.gov/standards/iso639-2/php/code_list.php) to these files:\n\n- [lingui.config.ts](./lingui.config.ts)\n- [languages.tsx](./apps/postybirb-ui/src/app/languages.tsx)\n" }, { "path": "apps/client-server/.eslintrc.json", "content": "{\n \"extends\": [\"../../.eslintrc.js\"],\n \"ignorePatterns\": [\"!**/*\"],\n \"overrides\": [\n {\n \"files\": [\"*.ts\", \"*.tsx\", \"*.js\", \"*.jsx\"],\n \"rules\": {}\n },\n {\n \"files\": [\"*.ts\", \"*.tsx\"],\n \"rules\": {}\n },\n {\n \"files\": [\"*.js\", \"*.jsx\"],\n \"rules\": {}\n }\n ]\n}\n" }, { "path": "apps/client-server/jest.config.ts", "content": "/* eslint-disable */\nexport default {\n displayName: 'client-server',\n preset: '../../jest.preset.js',\n globals: {},\n testEnvironment: 'node',\n moduleFileExtensions: ['ts', 'js', 'html'],\n coverageDirectory: '../../coverage/apps/client-server',\n runner: '@kayahr/jest-electron-runner/main',\n};\n" }, { "path": "apps/client-server/project.json", "content": "{\n \"name\": \"client-server\",\n \"$schema\": \"../../node_modules/nx/schemas/project-schema.json\",\n \"sourceRoot\": \"apps/client-server/src\",\n \"projectType\": \"application\",\n \"targets\": {\n \"build\": {\n \"executor\": \"nx-electron:build\",\n \"outputs\": [\"{options.outputPath}\"],\n \"options\": {\n \"outputPath\": \"dist/apps/client-server\",\n \"main\": \"apps/client-server/src/main.ts\",\n \"tsConfig\": \"apps/client-server/tsconfig.app.json\",\n \"assets\": [\"apps/client-server/src/assets\"]\n },\n \"configurations\": {\n \"production\": {\n \"optimization\": true,\n \"extractLicenses\": true,\n \"inspect\": false,\n \"fileReplacements\": [\n {\n \"replace\": \"apps/client-server/src/environments/environment.ts\",\n \"with\": \"apps/client-server/src/environments/environment.prod.ts\"\n }\n ]\n }\n }\n },\n \"serve\": {\n \"executor\": \"@nx/js:node\",\n \"options\": {\n \"buildTarget\": \"client-server:build\",\n \"inspect\": true,\n \"port\": 9229\n }\n },\n \"lint\": {\n \"executor\": \"@nx/eslint:lint\",\n \"outputs\": [\"{options.outputFile}\"]\n },\n \"test\": {\n \"executor\": \"@nx/jest:jest\",\n \"outputs\": [\"{workspaceRoot}/coverage/apps/client-server\"],\n \"options\": {\n \"jestConfig\": \"apps/client-server/jest.config.ts\"\n }\n },\n \"typecheck\": {\n \"executor\": \"nx:run-commands\",\n \"options\": {\n \"command\": \"tsc -b {projectRoot}/tsconfig.json --incremental --pretty\"\n }\n }\n },\n \"tags\": []\n}\n" }, { "path": "apps/client-server/src/app/account/account.controller.ts", "content": "import { Body, Controller, Get, Param, Patch, Post } from '@nestjs/common';\nimport {\n ApiBadRequestResponse,\n ApiNotFoundResponse,\n ApiOkResponse,\n ApiTags,\n} from '@nestjs/swagger';\nimport { AccountId } from '@postybirb/types';\nimport { PostyBirbController } from '../common/controller/postybirb-controller';\nimport { AccountService } from './account.service';\nimport { CreateAccountDto } from './dtos/create-account.dto';\nimport { SetWebsiteDataRequestDto } from './dtos/set-website-data-request.dto';\nimport { UpdateAccountDto } from './dtos/update-account.dto';\n\n/**\n * CRUD operations on Account data.\n * @class AccountController\n */\n@ApiTags('account')\n@Controller('account')\nexport class AccountController extends PostyBirbController<'AccountSchema'> {\n constructor(readonly service: AccountService) {\n super(service);\n }\n\n @Post()\n @ApiOkResponse({ description: 'Account created.' })\n @ApiBadRequestResponse({ description: 'Bad request made.' })\n create(@Body() createAccountDto: CreateAccountDto) {\n return this.service\n .create(createAccountDto)\n .then((account) => account.toDTO());\n }\n\n @Post('/clear/:id')\n @ApiOkResponse({ description: 'Account data cleared.' })\n @ApiBadRequestResponse({ description: 'Bad request made.' })\n async clear(@Param('id') id: AccountId) {\n await this.service.clearAccountData(id);\n try {\n this.service.manuallyExecuteOnLogin(id);\n } catch {\n // For some reason throws error that crashes app when deleting account\n }\n }\n\n @Get('/refresh/:id')\n @ApiOkResponse({ description: 'Account login check queued.' })\n async refresh(@Param('id') id: AccountId) {\n this.service.manuallyExecuteOnLogin(id);\n }\n\n @Patch(':id')\n @ApiOkResponse({ description: 'Account updated.', type: Boolean })\n @ApiNotFoundResponse({ description: 'Account Id not found.' })\n update(\n @Body() updateAccountDto: UpdateAccountDto,\n @Param('id') id: AccountId,\n ) {\n return this.service\n .update(id, updateAccountDto)\n .then((account) => account.toDTO());\n }\n\n @Post('/account-data')\n @ApiOkResponse({ description: 'Account data set.' })\n setWebsiteData(@Body() oauthRequestDto: SetWebsiteDataRequestDto) {\n return this.service.setAccountData(oauthRequestDto);\n }\n}\n" }, { "path": "apps/client-server/src/app/account/account.events.ts", "content": "import { ACCOUNT_UPDATES } from '@postybirb/socket-events';\nimport { IAccountDto } from '@postybirb/types';\nimport { WebsocketEvent } from '../web-socket/models/web-socket-event';\n\nexport type AccountEventTypes = AccountUpdateEvent;\n\nclass AccountUpdateEvent implements WebsocketEvent {\n event: string = ACCOUNT_UPDATES;\n\n data: IAccountDto[];\n}\n" }, { "path": "apps/client-server/src/app/account/account.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { WebsitesModule } from '../websites/websites.module';\nimport { AccountController } from './account.controller';\nimport { AccountService } from './account.service';\n\n@Module({\n imports: [WebsitesModule],\n providers: [AccountService],\n controllers: [AccountController],\n exports: [AccountService],\n})\nexport class AccountModule {}\n" }, { "path": "apps/client-server/src/app/account/account.service.spec.ts", "content": "import { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport { NULL_ACCOUNT_ID } from '@postybirb/types';\nimport { Account } from '../drizzle/models';\nimport { waitUntil } from '../utils/wait.util';\nimport { WebsiteImplProvider } from '../websites/implementations/provider';\nimport { WebsiteRegistryService } from '../websites/website-registry.service';\nimport { AccountService } from './account.service';\nimport { CreateAccountDto } from './dtos/create-account.dto';\n\ndescribe('AccountsService', () => {\n let service: AccountService;\n let registryService: WebsiteRegistryService;\n let module: TestingModule;\n\n // Mock objects for deleteUnregisteredAccounts tests\n let mockRepository: any;\n let mockWebsiteRegistry: any;\n let mockLogger: any;\n\n const mockRegisteredAccount = {\n id: 'account-1',\n name: 'Test Account 1',\n website: 'registered-website',\n withWebsiteInstance(websiteInstance) {\n return this;\n },\n toDTO: () => {},\n } as Account;\n\n const mockUnregisteredAccount = {\n id: 'account-2',\n name: 'Test Account 2',\n website: 'unregistered-website',\n withWebsiteInstance(websiteInstance) {\n return this;\n },\n toDTO: () => {},\n } as Account;\n\n const mockAnotherUnregisteredAccount = {\n id: 'account-3',\n name: 'Test Account 3',\n website: 'another-unregistered-website',\n withWebsiteInstance(websiteInstance) {\n return this;\n },\n toDTO: () => {},\n } as Account;\n\n beforeEach(async () => {\n clearDatabase();\n module = await Test.createTestingModule({\n providers: [AccountService, WebsiteRegistryService, WebsiteImplProvider],\n }).compile();\n\n service = module.get(AccountService);\n registryService = module.get(\n WebsiteRegistryService,\n );\n\n await service.onModuleInit();\n });\n\n afterAll(async () => {\n await module.close();\n });\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n\n it('should set and clear account data', async () => {\n const dto = new CreateAccountDto();\n dto.groups = ['test'];\n dto.name = 'test';\n dto.website = 'test';\n\n const record = await service.create(dto);\n const instance = registryService.findInstance(record);\n expect(instance).toBeDefined();\n\n await instance.login();\n const websiteData = instance.getWebsiteData();\n expect(websiteData).toEqual({\n test: 'test-mode',\n });\n\n await service.setAccountData({\n id: record.id,\n data: { test: 'test-mode-2' },\n });\n expect(instance.getWebsiteData()).toEqual({\n test: 'test-mode-2',\n });\n\n await service.clearAccountData(record.id);\n expect(instance.getWebsiteData()).toEqual({});\n }, 10000);\n\n it('should create entities', async () => {\n const dto = new CreateAccountDto();\n dto.groups = ['test'];\n dto.name = 'test';\n dto.website = 'test';\n\n const record = await service.create(dto);\n expect(registryService.findInstance(record)).toBeDefined();\n\n const groups = await service.findAll();\n await waitUntil(() => !record.websiteInstance?.getLoginState().pending, 50);\n expect(groups).toHaveLength(1);\n expect(groups[0].name).toEqual(dto.name);\n expect(groups[0].website).toEqual(dto.website);\n expect(groups[0].groups).toEqual(dto.groups);\n const recordDto = record.toDTO();\n expect(recordDto).toEqual({\n groups: dto.groups,\n name: dto.name,\n website: dto.website,\n id: record.id,\n createdAt: record.createdAt,\n updatedAt: record.updatedAt,\n state: {\n isLoggedIn: true,\n pending: false,\n username: 'TestUser',\n lastUpdated: expect.any(String),\n },\n data: {\n test: 'test-mode',\n },\n websiteInfo: {\n supports: ['MESSAGE', 'FILE'],\n websiteDisplayName: 'Test',\n },\n });\n }, 10000);\n\n it('should support crud operations', async () => {\n const createAccount: CreateAccountDto = new CreateAccountDto();\n createAccount.name = 'test';\n createAccount.website = 'test';\n\n // Create\n const account = await service.create(createAccount);\n expect(account).toBeDefined();\n expect(await service.findAll()).toHaveLength(1);\n expect(await service.findById(account.id)).toBeDefined();\n\n // Update\n await service.update(account.id, { name: 'Updated', groups: [] });\n expect(await (await service.findById(account.id)).name).toEqual('Updated');\n\n // Remove\n await service.remove(account.id);\n expect(await service.findAll()).toHaveLength(0);\n });\n\n describe('deleteUnregisteredAccounts', () => {\n beforeEach(() => {\n // Setup mock objects for testing private method\n mockRepository = {\n find: jest.fn(),\n deleteById: jest.fn(),\n schemaEntity: { id: 'id' },\n };\n\n mockWebsiteRegistry = {\n canCreate: jest.fn(),\n create: jest.fn(),\n findInstance: jest.fn(),\n getAvailableWebsites: () => [],\n markAsInitialized: jest.fn(),\n emit: jest.fn(),\n };\n\n mockLogger = {\n withMetadata: jest.fn().mockReturnThis(),\n withError: jest.fn().mockReturnThis(),\n warn: jest.fn(),\n error: jest.fn(),\n };\n\n // Replace service dependencies with mocks\n (service as any).repository = mockRepository;\n (service as any).websiteRegistry = mockWebsiteRegistry;\n (service as any).logger = mockLogger;\n\n // Setup default mock behavior\n mockRepository.find.mockResolvedValue([\n mockRegisteredAccount,\n mockUnregisteredAccount,\n mockAnotherUnregisteredAccount,\n ]);\n\n mockWebsiteRegistry.canCreate.mockImplementation((website: string) => {\n return website === 'registered-website';\n });\n\n mockRepository.deleteById.mockResolvedValue(undefined);\n });\n\n it('should delete accounts for unregistered websites', async () => {\n await (service as any).deleteUnregisteredAccounts();\n\n // Verify that find was called to get all accounts except NULL_ACCOUNT_ID\n expect(mockRepository.find).toHaveBeenCalledWith({\n where: expect.any(Object), // ne(schemaEntity.id, NULL_ACCOUNT_ID)\n });\n\n // Verify canCreate was called for each account's website\n expect(mockWebsiteRegistry.canCreate).toHaveBeenCalledWith(\n 'registered-website',\n );\n expect(mockWebsiteRegistry.canCreate).toHaveBeenCalledWith(\n 'unregistered-website',\n );\n expect(mockWebsiteRegistry.canCreate).toHaveBeenCalledWith(\n 'another-unregistered-website',\n );\n expect(mockWebsiteRegistry.canCreate).toHaveBeenCalledTimes(3);\n\n // Verify deleteById was called for unregistered accounts only\n expect(mockRepository.deleteById).toHaveBeenCalledWith(['account-2']);\n expect(mockRepository.deleteById).toHaveBeenCalledWith(['account-3']);\n expect(mockRepository.deleteById).toHaveBeenCalledTimes(2);\n\n // Verify logging\n expect(mockLogger.withMetadata).toHaveBeenCalledWith(\n mockUnregisteredAccount,\n );\n expect(mockLogger.withMetadata).toHaveBeenCalledWith(\n mockAnotherUnregisteredAccount,\n );\n expect(mockLogger.warn).toHaveBeenCalledWith(\n 'Deleting unregistered account: account-2 (Test Account 2)',\n );\n expect(mockLogger.warn).toHaveBeenCalledWith(\n 'Deleting unregistered account: account-3 (Test Account 3)',\n );\n });\n\n it('should not delete accounts for registered websites', async () => {\n await (service as any).deleteUnregisteredAccounts();\n\n // Verify the registered account was not deleted\n expect(mockRepository.deleteById).not.toHaveBeenCalledWith(['account-1']);\n });\n\n it('should handle deletion errors gracefully', async () => {\n const deleteError = new Error('Database deletion failed');\n mockRepository.deleteById\n .mockResolvedValueOnce(undefined) // First deletion succeeds\n .mockRejectedValueOnce(deleteError); // Second deletion fails\n\n await (service as any).deleteUnregisteredAccounts();\n\n // Verify both deletions were attempted\n expect(mockRepository.deleteById).toHaveBeenCalledTimes(2);\n\n // Verify error was logged for the failed deletion\n expect(mockLogger.withError).toHaveBeenCalledWith(deleteError);\n expect(mockLogger.error).toHaveBeenCalledWith(\n 'Failed to delete unregistered account: account-3',\n );\n });\n\n it('should handle empty accounts list', async () => {\n mockRepository.find.mockResolvedValue([]);\n\n await (service as any).deleteUnregisteredAccounts();\n\n expect(mockWebsiteRegistry.canCreate).not.toHaveBeenCalled();\n expect(mockRepository.deleteById).not.toHaveBeenCalled();\n expect(mockLogger.warn).not.toHaveBeenCalled();\n });\n\n it('should handle case where all accounts are registered', async () => {\n mockRepository.find.mockResolvedValue([mockRegisteredAccount]);\n\n await (service as any).deleteUnregisteredAccounts();\n\n expect(mockWebsiteRegistry.canCreate).toHaveBeenCalledWith(\n 'registered-website',\n );\n expect(mockRepository.deleteById).not.toHaveBeenCalled();\n expect(mockLogger.warn).not.toHaveBeenCalled();\n });\n\n it('should exclude NULL_ACCOUNT_ID from deletion consideration', async () => {\n const nullAccount = {\n id: NULL_ACCOUNT_ID,\n name: 'Null Account',\n website: 'null',\n } as Account;\n\n // Mock the repository.find to only return non-NULL accounts (simulating the database query)\n // The actual service uses ne(this.repository.schemaEntity.id, NULL_ACCOUNT_ID) to exclude it\n mockRepository.find.mockResolvedValue([\n mockUnregisteredAccount, // Only return the unregistered account, not the null account\n ]);\n\n // Even if null website is not registered, it shouldn't be considered for deletion\n mockWebsiteRegistry.canCreate.mockImplementation((website: string) => {\n return website !== 'null' && website !== 'unregistered-website';\n });\n\n await (service as any).deleteUnregisteredAccounts();\n\n // Verify the query excludes NULL_ACCOUNT_ID (this is tested by the repository mock)\n expect(mockRepository.find).toHaveBeenCalledWith({\n where: expect.any(Object),\n });\n\n // Only the unregistered account should be deleted, not the null account\n expect(mockRepository.deleteById).toHaveBeenCalledWith(['account-2']);\n expect(mockRepository.deleteById).toHaveBeenCalledTimes(1);\n });\n });\n});\n" }, { "path": "apps/client-server/src/app/account/account.service.ts", "content": "import {\n BadRequestException,\n Injectable,\n OnModuleInit,\n Optional,\n} from '@nestjs/common';\nimport { Cron, CronExpression } from '@nestjs/schedule';\nimport { ACCOUNT_UPDATES } from '@postybirb/socket-events';\nimport {\n AccountId,\n IWebsiteMetadata,\n NULL_ACCOUNT_ID,\n NullAccount,\n} from '@postybirb/types';\nimport { IsTestEnvironment } from '@postybirb/utils/electron';\nimport { ne } from 'drizzle-orm';\nimport { Class } from 'type-fest';\nimport { PostyBirbService } from '../common/service/postybirb-service';\nimport { Account } from '../drizzle/models';\nimport { FindOptions } from '../drizzle/postybirb-database/find-options.type';\nimport { WSGateway } from '../web-socket/web-socket-gateway';\nimport { UnknownWebsite } from '../websites/website';\nimport { WebsiteRegistryService } from '../websites/website-registry.service';\nimport { CreateAccountDto } from './dtos/create-account.dto';\nimport { SetWebsiteDataRequestDto } from './dtos/set-website-data-request.dto';\nimport { UpdateAccountDto } from './dtos/update-account.dto';\nimport { LoginStatePoller } from './login-state-poller';\n\n/**\n * Service responsible for returning Account data.\n * Also stores login refresh timers for initiating login checks.\n */\n@Injectable()\nexport class AccountService\n extends PostyBirbService<'AccountSchema'>\n implements OnModuleInit\n{\n private readonly loginRefreshTimers: Record<\n string,\n {\n timer: NodeJS.Timeout;\n websites: Class[];\n }\n > = {};\n\n private readonly loginStatePoller: LoginStatePoller;\n\n constructor(\n private readonly websiteRegistry: WebsiteRegistryService,\n @Optional() webSocket?: WSGateway,\n ) {\n super('AccountSchema', webSocket);\n this.repository.subscribe('AccountSchema', () => this.emit());\n this.loginStatePoller = new LoginStatePoller(\n this.websiteRegistry,\n () => {\n this.emit();\n this.websiteRegistry.emit();\n },\n );\n }\n\n /**\n * Initializes all website login timers and creates instances for known accounts.\n * Heavy operations are deferred to avoid blocking application startup.\n */\n async onModuleInit() {\n // Critical path: only populate null account to ensure database is ready\n await this.populateNullAccount();\n\n // Defer heavy operations to avoid blocking NestJS initialization\n setImmediate(async () => {\n await this.deleteUnregisteredAccounts();\n await this.initWebsiteRegistry();\n this.websiteRegistry.markAsInitialized();\n this.initWebsiteLoginRefreshTimers();\n\n this.emit();\n\n Object.keys(this.loginRefreshTimers).forEach((interval) =>\n this.executeOnLoginForInterval(interval),\n );\n });\n }\n\n /**\n * CRON-driven poll for login state changes.\n * Compares cached login states against live values and emits to UI on change.\n */\n @Cron(CronExpression.EVERY_SECOND)\n private pollLoginStates() {\n if (!IsTestEnvironment()) {\n this.loginStatePoller.checkForChanges();\n }\n }\n\n private async deleteUnregisteredAccounts() {\n const accounts = await this.repository.find({\n where: ne(this.repository.schemaEntity.id, NULL_ACCOUNT_ID),\n });\n const unregisteredAccounts = accounts.filter(\n (account) => !this.websiteRegistry.canCreate(account.website),\n );\n for (const account of unregisteredAccounts) {\n try {\n this.logger\n .withMetadata(account)\n .warn(\n `Deleting unregistered account: ${account.id} (${account.name})`,\n );\n await this.repository.deleteById([account.id]);\n } catch (err) {\n this.logger\n .withError(err)\n .withMetadata(account)\n .error(`Failed to delete unregistered account: ${account.id}`);\n }\n }\n }\n\n /**\n * Create the Nullable typed account.\n */\n private async populateNullAccount(): Promise {\n if (!(await this.repository.findById(NULL_ACCOUNT_ID))) {\n await this.repository.insert(new NullAccount());\n }\n }\n\n /**\n * Loads accounts into website registry.\n */\n private async initWebsiteRegistry(): Promise {\n const accounts = await this.repository.find({\n where: ne(this.repository.schemaEntity.id, NULL_ACCOUNT_ID),\n });\n await Promise.all(\n accounts.map((account) => this.websiteRegistry.create(account)),\n ).catch((err) => {\n this.logger.error(err, 'onModuleInit');\n });\n }\n\n /**\n * Creates website login check timers.\n */\n private initWebsiteLoginRefreshTimers(): void {\n const availableWebsites = this.websiteRegistry.getAvailableWebsites();\n availableWebsites.forEach((website) => {\n const interval: number =\n (website.prototype.decoratedProps.metadata as IWebsiteMetadata)\n .refreshInterval ?? 60_000 * 60;\n if (!this.loginRefreshTimers[interval]) {\n this.loginRefreshTimers[interval] = {\n websites: [],\n timer: setInterval(() => {\n this.executeOnLoginForInterval(interval);\n }, interval),\n };\n }\n\n this.loginRefreshTimers[interval].websites.push(website);\n });\n }\n\n public async emit() {\n const dtos = await this.findAll().then((accounts) =>\n accounts.map((a) => this.injectWebsiteInstance(a)),\n );\n super.emit({\n event: ACCOUNT_UPDATES,\n data: dtos.map((dto) => dto.toDTO()),\n });\n }\n\n /**\n * Runs login on all created website instances within a specific interval.\n * The mutex inside website.login() ensures only one login runs at a time\n * per instance; concurrent callers simply wait and get the fresh state.\n *\n * @param {string} interval\n */\n private async executeOnLoginForInterval(interval: string | number) {\n const { websites } = this.loginRefreshTimers[interval];\n websites.forEach((website) => {\n this.websiteRegistry.getInstancesOf(website).forEach((instance) => {\n // Fire-and-forget — the poller will detect state changes\n instance.login().catch((e) => {\n this.logger.withError(e).error(`Login failed for ${instance.id}`);\n });\n });\n });\n }\n\n /**\n * Logic that needs to be run after an account is created.\n *\n * @param {Account} account\n * @param {UnknownWebsite} website\n */\n private afterCreate(account: Account, website: UnknownWebsite) {\n // Fire-and-forget — poller picks up the state change\n website.login().catch((e) => {\n this.logger.withError(e).error(`Initial login failed for ${website.id}`);\n });\n }\n\n /**\n * Executes a login refresh initiated from an external source.\n * Waits for the result so the caller knows when it's done.\n *\n * @param {AccountId} id\n */\n async manuallyExecuteOnLogin(id: AccountId): Promise {\n const account = await this.findById(id);\n if (account) {\n const instance = this.websiteRegistry.findInstance(account);\n if (instance) {\n await instance.login();\n // Force an immediate UI update for this instance\n this.loginStatePoller.checkInstance(instance);\n }\n }\n }\n\n /**\n * Creates an Account.\n * @param {CreateAccountDto} createDto\n * @return {*} {Promise}\n */\n async create(createDto: CreateAccountDto): Promise {\n this.logger\n .withMetadata(createDto)\n .info(`Creating Account '${createDto.name}:${createDto.website}`);\n if (!this.websiteRegistry.canCreate(createDto.website)) {\n throw new BadRequestException(\n `Website ${createDto.website} is not supported.`,\n );\n }\n const account = await this.repository.insert(new Account(createDto));\n const instance = await this.websiteRegistry.create(account);\n this.afterCreate(account, instance);\n return account.withWebsiteInstance(instance);\n }\n\n public findById(id: AccountId, options?: FindOptions) {\n return this.repository\n .findById(id, options)\n .then((account) => this.injectWebsiteInstance(account));\n }\n\n public async findAll() {\n return this.repository\n .find({\n where: ne(this.repository.schemaEntity.id, NULL_ACCOUNT_ID),\n })\n .then((accounts) =>\n accounts.map((account) => this.injectWebsiteInstance(account)),\n );\n }\n\n async update(id: AccountId, update: UpdateAccountDto) {\n this.logger.withMetadata(update).info(`Updating Account '${id}'`);\n return this.repository\n .update(id, update)\n .then((account) => this.injectWebsiteInstance(account));\n }\n\n async remove(id: AccountId) {\n const account = await this.findById(id);\n if (account) {\n this.websiteRegistry.remove(account);\n }\n return super.remove(id);\n }\n\n /**\n * Clears the data and login state associated with an account.\n *\n * @param {string} id\n */\n async clearAccountData(id: AccountId) {\n this.logger.info(`Clearing Account data for '${id}'`);\n const account = await this.findById(id);\n if (account) {\n const instance = this.websiteRegistry.findInstance(account);\n await instance.clearLoginStateAndData();\n }\n }\n\n /**\n * Sets the data saved to an account's website.\n *\n * @param {SetWebsiteDataRequestDto} setWebsiteDataRequestDto\n */\n async setAccountData(setWebsiteDataRequestDto: SetWebsiteDataRequestDto) {\n this.logger.info(\n `Setting Account data for '${setWebsiteDataRequestDto.id}'`,\n );\n const account = await this.repository.findById(\n setWebsiteDataRequestDto.id,\n { failOnMissing: true },\n );\n const instance = this.websiteRegistry.findInstance(account);\n await instance.setWebsiteData(setWebsiteDataRequestDto.data);\n }\n\n private injectWebsiteInstance(account?: Account): Account | null {\n if (!account) {\n return null;\n }\n return account.withWebsiteInstance(\n this.websiteRegistry.findInstance(account),\n );\n }\n}\n" }, { "path": "apps/client-server/src/app/account/dtos/create-account.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { ICreateAccountDto } from '@postybirb/types';\nimport { IsArray, IsString, Length } from 'class-validator';\n\n/**\n * Account creation request object.\n */\nexport class CreateAccountDto implements ICreateAccountDto {\n @ApiProperty()\n @IsString()\n @Length(1, 64)\n name: string;\n\n @ApiProperty()\n @IsString()\n @Length(1)\n website: string;\n\n @ApiProperty()\n @IsArray()\n groups: string[];\n}\n" }, { "path": "apps/client-server/src/app/account/dtos/set-website-data-request.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport {\n AccountId,\n DynamicObject,\n ISetWebsiteDataRequestDto,\n} from '@postybirb/types';\nimport { IsObject, IsString } from 'class-validator';\n\nexport class SetWebsiteDataRequestDto\n implements ISetWebsiteDataRequestDto\n{\n @ApiProperty()\n @IsString()\n id: AccountId;\n\n @ApiProperty({\n type: Object,\n })\n @IsObject()\n data: DynamicObject;\n}\n" }, { "path": "apps/client-server/src/app/account/dtos/update-account.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { IUpdateAccountDto } from '@postybirb/types';\nimport { IsArray, IsString } from 'class-validator';\n\n/**\n * Account update request object.\n */\nexport class UpdateAccountDto implements IUpdateAccountDto {\n @ApiProperty()\n @IsString()\n name: string;\n\n @ApiProperty()\n @IsArray()\n groups: string[];\n}\n" }, { "path": "apps/client-server/src/app/account/login-state-poller.ts", "content": "import { Logger } from '@postybirb/logger';\nimport { AccountId, ILoginState } from '@postybirb/types';\nimport { UnknownWebsite } from '../websites/website';\nimport { WebsiteRegistryService } from '../websites/website-registry.service';\n\n/**\n * Compares all website instances' login states against a cached snapshot\n * and triggers a callback when any state has changed.\n *\n * Designed to be driven externally (e.g. by a @Cron job) rather than\n * managing its own polling interval.\n */\nexport class LoginStatePoller {\n private readonly logger = Logger(LoginStatePoller.name);\n\n /**\n * Cached snapshot of the last-known login state per account.\n * Used for diffing to detect changes.\n */\n private lastKnownStates: Record = {};\n\n constructor(\n private readonly websiteRegistry: WebsiteRegistryService,\n private readonly onStateChange: () => void,\n ) {}\n\n /**\n * Compare current login states against the cached snapshot.\n * If any account's state has changed, update the cache and fire the callback.\n */\n checkForChanges(): void {\n try {\n const instances = this.websiteRegistry.getAll();\n let hasChanged = false;\n\n const currentStates: Record = {};\n\n for (const instance of instances) {\n const { accountId } = instance;\n const state = instance.getLoginState();\n currentStates[accountId] = state;\n\n const previous = this.lastKnownStates[accountId];\n if (!previous || !this.statesEqual(previous, state)) {\n hasChanged = true;\n }\n }\n\n // Detect removed accounts\n for (const accountId of Object.keys(this.lastKnownStates)) {\n if (!(accountId in currentStates)) {\n hasChanged = true;\n }\n }\n\n if (hasChanged) {\n this.lastKnownStates = currentStates;\n this.onStateChange();\n }\n } catch (e) {\n this.logger.withError(e).error('Error during login state poll');\n }\n }\n\n /**\n * Check a single website instance for login state changes.\n * More efficient than checkForChanges() when you know exactly which instance changed.\n *\n * @param {UnknownWebsite} instance - The website instance to check\n */\n checkInstance(instance: UnknownWebsite): void {\n try {\n const { accountId } = instance;\n const state = instance.getLoginState();\n const previous = this.lastKnownStates[accountId];\n\n if (!previous || !this.statesEqual(previous, state)) {\n this.lastKnownStates[accountId] = state;\n this.onStateChange();\n }\n } catch (e) {\n this.logger.withError(e).error('Error during single instance login state check');\n }\n }\n\n /**\n * Shallow equality check for two ILoginState objects.\n */\n private statesEqual(a: ILoginState, b: ILoginState): boolean {\n return (\n a.isLoggedIn === b.isLoggedIn &&\n a.pending === b.pending &&\n a.username === b.username &&\n a.lastUpdated === b.lastUpdated\n );\n }\n}\n" }, { "path": "apps/client-server/src/app/app.controller.ts", "content": "import { Controller, Get } from '@nestjs/common';\n\nimport { AppService } from './app.service';\n\n@Controller()\nexport class AppController {\n constructor(private readonly appService: AppService) {}\n\n @Get()\n getData() {\n return this.appService.getData();\n }\n}\n" }, { "path": "apps/client-server/src/app/app.module.ts", "content": "import { MiddlewareConsumer, Module, NestModule } from '@nestjs/common';\n\nimport { ScheduleModule } from '@nestjs/schedule';\nimport { ServeStaticModule } from '@nestjs/serve-static';\nimport { join } from 'path';\nimport { AccountModule } from './account/account.module';\nimport { AppController } from './app.controller';\nimport { AppService } from './app.service';\nimport { CustomShortcutsModule } from './custom-shortcuts/custom-shortcuts.module';\nimport { DirectoryWatchersModule } from './directory-watchers/directory-watchers.module';\nimport { FileConverterModule } from './file-converter/file-converter.module';\nimport { FileModule } from './file/file.module';\nimport { FormGeneratorModule } from './form-generator/form-generator.module';\nimport { ImageProcessingModule } from './image-processing/image-processing.module';\nimport { LegacyDatabaseImporterModule } from './legacy-database-importer/legacy-database-importer.module';\nimport { LogsModule } from './logs/logs.module';\nimport { NotificationsModule } from './notifications/notifications.module';\nimport { PostParsersModule } from './post-parsers/post-parsers.module';\nimport { PostModule } from './post/post.module';\nimport { RemotePasswordMiddleware } from './remote/remote.middleware';\nimport { RemoteModule } from './remote/remote.module';\nimport { SettingsModule } from './settings/settings.module';\nimport { SubmissionModule } from './submission/submission.module';\nimport { TagConvertersModule } from './tag-converters/tag-converters.module';\nimport { TagGroupsModule } from './tag-groups/tag-groups.module';\nimport { UpdateModule } from './update/update.module';\nimport { UserConvertersModule } from './user-converters/user-converters.module';\nimport { UserSpecifiedWebsiteOptionsModule } from './user-specified-website-options/user-specified-website-options.module';\nimport { ValidationModule } from './validation/validation.module';\nimport { WebSocketModule } from './web-socket/web-socket.module';\nimport { WebsiteOptionsModule } from './website-options/website-options.module';\nimport { WebsitesModule } from './websites/websites.module';\n\n@Module({\n imports: [\n ScheduleModule.forRoot(),\n ImageProcessingModule,\n AccountModule,\n WebSocketModule,\n WebsitesModule,\n FileModule,\n SubmissionModule,\n SettingsModule,\n ServeStaticModule.forRoot({\n rootPath: join(__dirname, '..', 'postybirb-ui'),\n exclude: ['/api*'],\n }),\n FormGeneratorModule,\n WebsiteOptionsModule,\n TagGroupsModule,\n TagConvertersModule,\n UserConvertersModule,\n DirectoryWatchersModule,\n UserSpecifiedWebsiteOptionsModule,\n UpdateModule,\n PostModule,\n PostParsersModule,\n ValidationModule,\n FileConverterModule,\n NotificationsModule,\n RemoteModule,\n CustomShortcutsModule,\n LegacyDatabaseImporterModule,\n LogsModule,\n ],\n controllers: [AppController],\n providers: [AppService],\n})\nexport class AppModule implements NestModule {\n configure(consumer: MiddlewareConsumer) {\n consumer.apply(RemotePasswordMiddleware).forRoutes('*');\n }\n}\n" }, { "path": "apps/client-server/src/app/app.service.ts", "content": "import { Injectable } from '@nestjs/common';\nimport { app } from 'electron';\n\n@Injectable()\nexport class AppService {\n getData(): Record {\n return {\n message: 'pong',\n version: app.getVersion(),\n location: app.getPath('userData'),\n };\n }\n}\n" }, { "path": "apps/client-server/src/app/common/controller/postybirb-controller.ts", "content": "import { Delete, Get, Param, Query } from '@nestjs/common';\nimport { ApiOkResponse } from '@nestjs/swagger';\nimport { SchemaKey } from '@postybirb/database';\nimport { EntityId } from '@postybirb/types';\nimport { PostyBirbService } from '../service/postybirb-service';\n\n/**\n * Base PostyBirb controller logic that should be good for most rest calls.\n *\n * @class PostyBirbController\n */\nexport abstract class PostyBirbController {\n constructor(protected readonly service: PostyBirbService) {}\n\n @Get(':id')\n @ApiOkResponse({ description: 'Record by Id.' })\n findOne(@Param('id') id: EntityId) {\n return this.service\n .findById(id, { failOnMissing: true })\n .then((record) => record.toDTO());\n }\n\n @Get()\n @ApiOkResponse({ description: 'A list of all records.' })\n findAll() {\n return this.service\n .findAll()\n .then((records) => records.map((record) => record.toDTO()));\n }\n\n @Delete()\n @ApiOkResponse({\n description: 'Records removed.',\n })\n async remove(@Query('ids') ids: EntityId | EntityId[]) {\n return Promise.all(\n (Array.isArray(ids) ? ids : [ids]).map((id) => this.service.remove(id)),\n ).then(() => ({\n success: true,\n }));\n }\n}\n" }, { "path": "apps/client-server/src/app/common/service/postybirb-service.ts", "content": "import { BadRequestException, Injectable } from '@nestjs/common';\nimport { SchemaKey } from '@postybirb/database';\nimport { Logger } from '@postybirb/logger';\nimport { EntityId } from '@postybirb/types';\nimport { SQL } from 'drizzle-orm';\nimport { FindOptions } from '../../drizzle/postybirb-database/find-options.type';\nimport { PostyBirbDatabase } from '../../drizzle/postybirb-database/postybirb-database';\nimport { WSGateway } from '../../web-socket/web-socket-gateway';\nimport { WebSocketEvents } from '../../web-socket/web-socket.events';\n\n/**\n * Base class that implements simple CRUD logic\n *\n * @class PostyBirbService\n */\n@Injectable()\nexport abstract class PostyBirbService {\n protected readonly logger = Logger(this.constructor.name);\n\n protected readonly repository: PostyBirbDatabase;\n\n constructor(\n private readonly table: TSchemaKey | PostyBirbDatabase,\n private readonly webSocket?: WSGateway,\n ) {\n if (typeof table === 'string') {\n this.repository = new PostyBirbDatabase(table);\n } else {\n this.repository = table;\n }\n }\n\n /**\n * Emits events onto the websocket\n *\n * @protected\n * @param {WebSocketEvents} event\n */\n protected async emit(event: WebSocketEvents) {\n try {\n if (this.webSocket) {\n this.webSocket.emit(event);\n }\n } catch (err) {\n this.logger.error(`Error emitting websocket event: ${event.event}`, err);\n }\n }\n\n protected get schema() {\n return this.repository.schemaEntity;\n }\n\n /**\n * Throws exception if a record matching the query already exists.\n *\n * @protected\n * @param {FilterQuery} where\n */\n protected async throwIfExists(where: SQL) {\n const exists = await this.repository.select(where);\n if (exists.length) {\n this.logger\n .withMetadata(exists)\n .error(`A duplicate entity already exists`);\n throw new BadRequestException(`A duplicate entity already exists`);\n }\n }\n\n // Repository Wrappers\n\n public findById(id: EntityId, options?: FindOptions) {\n return this.repository.findById(id, options);\n }\n\n public findAll() {\n return this.repository.findAll();\n }\n\n public remove(id: EntityId) {\n this.logger.withMetadata({ id }).info(`Removing entity '${id}'`);\n return this.repository.deleteById([id]);\n }\n\n // END Repository Wrappers\n}\n" }, { "path": "apps/client-server/src/app/constants.ts", "content": "// Constant variables\nexport const WEBSITE_IMPLEMENTATIONS = 'WEBSITE_IMPLEMENTATIONS';\n" }, { "path": "apps/client-server/src/app/custom-shortcuts/custom-shortcut.events.ts", "content": "import { CUSTOM_SHORTCUT_UPDATES } from '@postybirb/socket-events';\nimport { ICustomShortcut } from '@postybirb/types';\nimport { WebsocketEvent } from '../web-socket/models/web-socket-event';\n\nexport type CustomShortcutEventTypes = CustomShortcutEvent;\n\nclass CustomShortcutEvent implements WebsocketEvent {\n event: string = CUSTOM_SHORTCUT_UPDATES;\n\n data: ICustomShortcut[];\n}\n" }, { "path": "apps/client-server/src/app/custom-shortcuts/custom-shortcuts.controller.ts", "content": "import { Body, Controller, Param, Patch, Post } from '@nestjs/common';\nimport { ApiOkResponse, ApiTags } from '@nestjs/swagger';\nimport { PostyBirbController } from '../common/controller/postybirb-controller';\nimport { CustomShortcutsService } from './custom-shortcuts.service';\nimport { CreateCustomShortcutDto } from './dtos/create-custom-shortcut.dto';\nimport { UpdateCustomShortcutDto } from './dtos/update-custom-shortcut.dto';\n\n@ApiTags('custom-shortcut')\n@Controller('custom-shortcut')\nexport class CustomShortcutsController extends PostyBirbController<'CustomShortcutSchema'> {\n constructor(readonly service: CustomShortcutsService) {\n super(service);\n }\n\n @Post()\n @ApiOkResponse({ description: 'Custom shortcut created' })\n async create(@Body() createCustomShortcutDto: CreateCustomShortcutDto) {\n return this.service\n .create(createCustomShortcutDto)\n .then((shortcut) => shortcut.toDTO());\n }\n\n @Patch(':id')\n @ApiOkResponse({ description: 'Custom shortcut updated' })\n async update(\n @Body() updateCustomShortcutDto: UpdateCustomShortcutDto,\n @Param('id') id: string,\n ) {\n return this.service\n .update(id, updateCustomShortcutDto)\n .then((shortcut) => shortcut.toDTO());\n }\n}\n" }, { "path": "apps/client-server/src/app/custom-shortcuts/custom-shortcuts.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { CustomShortcutsController } from './custom-shortcuts.controller';\nimport { CustomShortcutsService } from './custom-shortcuts.service';\n\n@Module({\n controllers: [CustomShortcutsController],\n providers: [CustomShortcutsService],\n exports: [CustomShortcutsService],\n})\nexport class CustomShortcutsModule {}\n" }, { "path": "apps/client-server/src/app/custom-shortcuts/custom-shortcuts.service.ts", "content": "import { Injectable, Optional } from '@nestjs/common';\nimport { CUSTOM_SHORTCUT_UPDATES } from '@postybirb/socket-events';\nimport { EntityId } from '@postybirb/types';\nimport { eq } from 'drizzle-orm';\nimport { PostyBirbService } from '../common/service/postybirb-service';\nimport { CustomShortcut } from '../drizzle/models/custom-shortcut.entity';\nimport { WSGateway } from '../web-socket/web-socket-gateway';\nimport { CreateCustomShortcutDto } from './dtos/create-custom-shortcut.dto';\nimport { UpdateCustomShortcutDto } from './dtos/update-custom-shortcut.dto';\n\n@Injectable()\nexport class CustomShortcutsService extends PostyBirbService<'CustomShortcutSchema'> {\n constructor(@Optional() webSocket?: WSGateway) {\n super('CustomShortcutSchema', webSocket);\n this.repository.subscribe('CustomShortcutSchema', () => this.emit());\n }\n\n public async emit() {\n const dtos = await this.findAll();\n super.emit({\n event: CUSTOM_SHORTCUT_UPDATES,\n data: dtos.map((dto) => dto.toDTO()),\n });\n }\n\n public async create(\n createCustomShortcutDto: CreateCustomShortcutDto,\n ): Promise {\n this.logger\n .withMetadata(createCustomShortcutDto)\n .info('Creating custom shortcut');\n await this.throwIfExists(\n eq(this.schema.name, createCustomShortcutDto.name),\n );\n return this.repository.insert(createCustomShortcutDto);\n }\n\n public async update(\n id: string,\n updateCustomShortcutDto: UpdateCustomShortcutDto,\n ): Promise {\n this.logger\n .withMetadata(updateCustomShortcutDto)\n .info('Updating custom shortcut');\n const existing = await this.repository.findById(id, {\n failOnMissing: true,\n });\n\n return this.repository.update(id, updateCustomShortcutDto);\n }\n\n public async remove(id: EntityId) {\n const existing = await this.repository.findById(id, {\n failOnMissing: true,\n });\n return super.remove(id);\n }\n}\n" }, { "path": "apps/client-server/src/app/custom-shortcuts/dtos/create-custom-shortcut.dto.ts", "content": "import { ICreateCustomShortcutDto } from '@postybirb/types';\nimport { IsString } from 'class-validator';\n\nexport class CreateCustomShortcutDto implements ICreateCustomShortcutDto {\n @IsString()\n name: string;\n}\n" }, { "path": "apps/client-server/src/app/custom-shortcuts/dtos/update-custom-shortcut.dto.ts", "content": "import { Description, IUpdateCustomShortcutDto } from '@postybirb/types';\nimport { IsObject, IsString } from 'class-validator';\n\nexport class UpdateCustomShortcutDto implements IUpdateCustomShortcutDto {\n @IsString()\n name: string;\n\n @IsObject()\n shortcut: Description;\n}\n" }, { "path": "apps/client-server/src/app/directory-watchers/directory-watcher.events.ts", "content": "import { DIRECTORY_WATCHER_UPDATES } from '@postybirb/socket-events';\nimport { DirectoryWatcherDto } from '@postybirb/types';\nimport { WebsocketEvent } from '../web-socket/models/web-socket-event';\n\nexport type DirectoryWatcherEventTypes = DirectoryWatcherUpdateEvent;\n\nclass DirectoryWatcherUpdateEvent\n implements WebsocketEvent\n{\n event: string = DIRECTORY_WATCHER_UPDATES;\n\n data: DirectoryWatcherDto[];\n}\n" }, { "path": "apps/client-server/src/app/directory-watchers/directory-watchers.controller.ts", "content": "import { Body, Controller, Param, Patch, Post } from '@nestjs/common';\nimport {\n ApiBadRequestResponse,\n ApiNotFoundResponse,\n ApiOkResponse,\n ApiTags,\n} from '@nestjs/swagger';\nimport { EntityId } from '@postybirb/types';\nimport { PostyBirbController } from '../common/controller/postybirb-controller';\nimport { DirectoryWatchersService } from './directory-watchers.service';\nimport { CheckPathDto } from './dtos/check-path.dto';\nimport { CreateDirectoryWatcherDto } from './dtos/create-directory-watcher.dto';\nimport { UpdateDirectoryWatcherDto } from './dtos/update-directory-watcher.dto';\n\n/**\n * CRUD operations on DirectoryWatchers.\n * @class DirectoryWatchersController\n */\n@ApiTags('directory-watchers')\n@Controller('directory-watchers')\nexport class DirectoryWatchersController extends PostyBirbController<'DirectoryWatcherSchema'> {\n constructor(readonly service: DirectoryWatchersService) {\n super(service);\n }\n\n @Post()\n @ApiOkResponse({ description: 'Entity created.' })\n @ApiBadRequestResponse({ description: 'Bad request made.' })\n create(@Body() createDto: CreateDirectoryWatcherDto) {\n return this.service.create(createDto).then((entity) => entity.toDTO());\n }\n\n @Patch(':id')\n @ApiOkResponse({ description: 'Entity updated.', type: Boolean })\n @ApiNotFoundResponse({ description: 'Entity not found.' })\n update(\n @Body() updateDto: UpdateDirectoryWatcherDto,\n @Param('id') id: EntityId,\n ) {\n return this.service.update(id, updateDto).then((entity) => entity.toDTO());\n }\n\n @Post('check-path')\n @ApiOkResponse({ description: 'Path check result.' })\n @ApiBadRequestResponse({ description: 'Bad request made.' })\n checkPath(@Body() checkPathDto: CheckPathDto) {\n return this.service.checkPath(checkPathDto.path);\n }\n}\n" }, { "path": "apps/client-server/src/app/directory-watchers/directory-watchers.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { NotificationsModule } from '../notifications/notifications.module';\nimport { SubmissionModule } from '../submission/submission.module';\nimport { DirectoryWatchersController } from './directory-watchers.controller';\nimport { DirectoryWatchersService } from './directory-watchers.service';\n\n@Module({\n imports: [SubmissionModule, NotificationsModule],\n controllers: [DirectoryWatchersController],\n providers: [DirectoryWatchersService],\n})\nexport class DirectoryWatchersModule {}\n" }, { "path": "apps/client-server/src/app/directory-watchers/directory-watchers.service.spec.ts", "content": "import { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport { DirectoryWatcherImportAction, SubmissionType } from '@postybirb/types';\nimport { mkdir, readdir, rename, writeFile } from 'fs/promises';\nimport { join } from 'path';\nimport { AccountService } from '../account/account.service';\nimport { NotificationsModule } from '../notifications/notifications.module';\nimport { CreateSubmissionDto } from '../submission/dtos/create-submission.dto';\nimport { SubmissionService } from '../submission/services/submission.service';\nimport { SubmissionModule } from '../submission/submission.module';\nimport { DirectoryWatchersService } from './directory-watchers.service';\nimport { CreateDirectoryWatcherDto } from './dtos/create-directory-watcher.dto';\nimport { UpdateDirectoryWatcherDto } from './dtos/update-directory-watcher.dto';\n\n// Mock fs/promises\njest.mock('fs/promises', () => ({\n readdir: jest.fn(),\n mkdir: jest.fn(),\n rename: jest.fn(),\n writeFile: jest.fn(),\n}));\n\ndescribe('DirectoryWatchersService', () => {\n let service: DirectoryWatchersService;\n let submissionService: SubmissionService;\n let accountService: AccountService;\n let module: TestingModule;\n\n beforeEach(async () => {\n clearDatabase();\n\n // Setup mocks\n (readdir as jest.Mock).mockResolvedValue([]);\n (mkdir as jest.Mock).mockResolvedValue(undefined);\n (rename as jest.Mock).mockResolvedValue(undefined);\n (writeFile as jest.Mock).mockResolvedValue(undefined);\n\n module = await Test.createTestingModule({\n imports: [SubmissionModule, NotificationsModule],\n providers: [DirectoryWatchersService],\n }).compile();\n\n service = module.get(DirectoryWatchersService);\n submissionService = module.get(SubmissionService);\n accountService = module.get(AccountService);\n\n await accountService.onModuleInit();\n });\n\n async function createSubmission() {\n const dto = new CreateSubmissionDto();\n dto.name = 'test';\n dto.type = SubmissionType.MESSAGE;\n dto.isTemplate = true;\n\n const record = await submissionService.create(dto);\n return record;\n }\n\n afterEach(() => {\n jest.clearAllMocks();\n });\n\n afterAll(async () => {\n await module.close();\n });\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n expect(module).toBeDefined();\n });\n\n it('should create entities and setup directory structure', async () => {\n const dto = new CreateDirectoryWatcherDto();\n dto.importAction = DirectoryWatcherImportAction.NEW_SUBMISSION;\n dto.path = 'path';\n\n await service.create(dto);\n\n // Verify directory structure was created\n expect(mkdir).toHaveBeenCalledWith(join('path', 'processing'), {\n recursive: true,\n });\n expect(mkdir).toHaveBeenCalledWith(join('path', 'completed'), {\n recursive: true,\n });\n expect(mkdir).toHaveBeenCalledWith(join('path', 'failed'), {\n recursive: true,\n });\n\n const entities = await service.findAll();\n const record = entities[0];\n expect(record.path).toBe(dto.path);\n expect(record.importAction).toBe(dto.importAction);\n });\n\n it('should update entities', async () => {\n const dto = new CreateDirectoryWatcherDto();\n dto.importAction = DirectoryWatcherImportAction.NEW_SUBMISSION;\n dto.path = 'path';\n\n const record = await service.create(dto);\n expect(record.path).toBe(dto.path);\n expect(record.importAction).toBe(dto.importAction);\n\n const updateDto = new UpdateDirectoryWatcherDto();\n updateDto.path = 'updated-path';\n const updatedRecord = await service.update(record.id, updateDto);\n expect(updatedRecord.path).toBe(updateDto.path);\n\n // Verify directory structure was created for new path\n expect(mkdir).toHaveBeenCalledWith(join('updated-path', 'processing'), {\n recursive: true,\n });\n });\n\n it('should support templates', async () => {\n const template = await createSubmission();\n const dto = new CreateDirectoryWatcherDto();\n dto.importAction = DirectoryWatcherImportAction.NEW_SUBMISSION;\n dto.path = 'path';\n const record = await service.create(dto);\n expect(record.path).toBe(dto.path);\n const updateDto = new UpdateDirectoryWatcherDto();\n updateDto.templateId = template.id;\n\n const updatedRecord = await service.update(record.id, updateDto);\n expect(updatedRecord.templateId).toBe(template.id);\n expect(updatedRecord.toDTO()).toEqual({\n createdAt: updatedRecord.createdAt,\n updatedAt: updatedRecord.updatedAt,\n id: updatedRecord.id,\n importAction: DirectoryWatcherImportAction.NEW_SUBMISSION,\n path: 'path',\n templateId: template.id,\n });\n\n await submissionService.remove(template.id);\n const rec = await service.findById(updatedRecord.id);\n expect(rec.templateId).toBe(null);\n });\n\n it('should throw error if path does not exist', async () => {\n (readdir as jest.Mock).mockRejectedValue(new Error('Path not found'));\n\n const dto = new CreateDirectoryWatcherDto();\n dto.importAction = DirectoryWatcherImportAction.NEW_SUBMISSION;\n dto.path = 'non-existent-path';\n\n await expect(service.create(dto)).rejects.toThrow(\n \"Path 'non-existent-path' does not exist or is not accessible\",\n );\n });\n\n it('should create entity without path (UI flow)', async () => {\n const dto = new CreateDirectoryWatcherDto();\n dto.importAction = DirectoryWatcherImportAction.NEW_SUBMISSION;\n // No path provided\n\n const record = await service.create(dto);\n\n // Verify entity was created without path\n expect(record.importAction).toBe(dto.importAction);\n expect(record.path).toBeNull();\n\n // Verify no directory structure was created\n expect(mkdir).not.toHaveBeenCalled();\n });\n\n it('should create directory structure when path is first set via update', async () => {\n // Create without path\n const dto = new CreateDirectoryWatcherDto();\n dto.importAction = DirectoryWatcherImportAction.NEW_SUBMISSION;\n const record = await service.create(dto);\n\n expect(record.path).toBeNull();\n\n // Clear mocks from create\n jest.clearAllMocks();\n\n // Update with path\n const updateDto = new UpdateDirectoryWatcherDto();\n updateDto.path = 'new-path';\n const updatedRecord = await service.update(record.id, updateDto);\n\n expect(updatedRecord.path).toBe('new-path');\n\n // Verify directory structure was created\n expect(mkdir).toHaveBeenCalledWith(join('new-path', 'processing'), {\n recursive: true,\n });\n expect(mkdir).toHaveBeenCalledWith(join('new-path', 'completed'), {\n recursive: true,\n });\n expect(mkdir).toHaveBeenCalledWith(join('new-path', 'failed'), {\n recursive: true,\n });\n });\n});\n" }, { "path": "apps/client-server/src/app/directory-watchers/directory-watchers.service.ts", "content": "import { BadRequestException, Injectable, Optional } from '@nestjs/common';\nimport { Cron, CronExpression } from '@nestjs/schedule';\nimport { DIRECTORY_WATCHER_UPDATES } from '@postybirb/socket-events';\nimport {\n DirectoryWatcherImportAction,\n EntityId,\n SubmissionType,\n} from '@postybirb/types';\nimport { IsTestEnvironment } from '@postybirb/utils/electron';\nimport { mkdir, readdir, rename, writeFile } from 'fs/promises';\nimport { getType } from 'mime';\nimport { join } from 'path';\nimport { PostyBirbService } from '../common/service/postybirb-service';\nimport { DirectoryWatcher } from '../drizzle/models';\nimport { MulterFileInfo } from '../file/models/multer-file-info';\nimport { NotificationsService } from '../notifications/notifications.service';\nimport { SubmissionService } from '../submission/services/submission.service';\nimport { WSGateway } from '../web-socket/web-socket-gateway';\nimport { CreateDirectoryWatcherDto } from './dtos/create-directory-watcher.dto';\nimport { UpdateDirectoryWatcherDto } from './dtos/update-directory-watcher.dto';\n\n/**\n * Directory structure for file processing:\n *\n * {watch-path}/ <- Users drop files here\n * ├── processing/ <- Files being actively processed\n * ├── completed/ <- Successfully processed files\n * └── failed/ <- Files that failed processing (with .error.txt)\n */\n\n/**\n * A directory watcher service that reads created watchers and checks\n * for new files added to the folder.\n *\n * Files are moved through different folders based on processing status,\n * eliminating the need for metadata tracking.\n *\n * @class DirectoryWatchersService\n * @extends {PostyBirbService}\n */\n/**\n * Threshold for warning users about folders with many files.\n * If a folder contains more than this number of files, a confirmation will be required.\n */\nexport const FILE_COUNT_WARNING_THRESHOLD = 10;\n\n/**\n * Result of checking a directory path for file watcher usage.\n */\nexport interface CheckPathResult {\n /** Whether the path is valid and accessible */\n valid: boolean;\n /** Number of files in the directory (excluding subfolders) */\n count: number;\n /** List of file names in the directory */\n files: string[];\n /** Error message if the path is invalid */\n error?: string;\n}\n\n@Injectable()\nexport class DirectoryWatchersService extends PostyBirbService<'DirectoryWatcherSchema'> {\n private runningWatchers = new Set();\n\n private recoveredWatchers = new Set();\n\n private readonly SUBFOLDER_PROCESSING = 'processing';\n\n private readonly SUBFOLDER_COMPLETED = 'completed';\n\n private readonly SUBFOLDER_FAILED = 'failed';\n\n constructor(\n private readonly submissionService: SubmissionService,\n private readonly notificationService: NotificationsService,\n @Optional() webSocket?: WSGateway,\n ) {\n super('DirectoryWatcherSchema', webSocket);\n this.repository.subscribe('DirectoryWatcherSchema', () =>\n this.emitUpdates(),\n );\n }\n\n protected async emitUpdates() {\n super.emit({\n event: DIRECTORY_WATCHER_UPDATES,\n data: (await this.repository.findAll()).map((entity) => entity.toDTO()),\n });\n }\n\n /**\n * CRON run read of paths.\n */\n @Cron(CronExpression.EVERY_30_SECONDS)\n private async run() {\n if (!IsTestEnvironment()) {\n const entities = await this.repository.findAll();\n entities\n .filter((e) => !!e.path)\n .forEach((e) => {\n // Recover orphaned files on first run\n if (!this.recoveredWatchers.has(e.id)) {\n this.recoverOrphanedFiles(e);\n this.recoveredWatchers.add(e.id);\n }\n\n // Process new files if not already running\n if (!this.runningWatchers.has(e.id)) {\n this.runningWatchers.add(e.id);\n this.read(e).finally(() => this.runningWatchers.delete(e.id));\n }\n });\n }\n }\n\n /**\n * Ensures all required subdirectories exist.\n * Skips if the base path doesn't exist.\n *\n * @param {string} basePath\n */\n private async ensureDirectoryStructure(basePath: string): Promise {\n // Check if base path exists, skip if it doesn't\n try {\n await readdir(basePath);\n } catch {\n return;\n }\n\n const subfolders = [\n this.SUBFOLDER_PROCESSING,\n this.SUBFOLDER_COMPLETED,\n this.SUBFOLDER_FAILED,\n ];\n\n for (const folder of subfolders) {\n await mkdir(join(basePath, folder), { recursive: true });\n }\n }\n\n /**\n * Recovers files that were left in the processing folder due to app crash/restart.\n * Moves them back to the main watch folder for reprocessing.\n *\n * @param {DirectoryWatcher} watcher\n */\n private async recoverOrphanedFiles(watcher: DirectoryWatcher): Promise {\n try {\n await this.ensureDirectoryStructure(watcher.path);\n\n const processingPath = join(watcher.path, this.SUBFOLDER_PROCESSING);\n const orphanedFiles = await readdir(processingPath);\n\n if (orphanedFiles.length > 0) {\n this.logger.info(\n `Recovering ${orphanedFiles.length} orphaned files in ${watcher.path}`,\n );\n\n for (const file of orphanedFiles) {\n const sourcePath = join(processingPath, file);\n const targetPath = join(watcher.path, file);\n\n try {\n await rename(sourcePath, targetPath);\n this.logger.info(`Recovered orphaned file: ${file}`);\n } catch (err) {\n this.logger.error(err, `Failed to recover orphaned file: ${file}`);\n }\n }\n\n this.notificationService.create({\n title: 'Directory Watcher Recovery',\n message: `Recovered ${orphanedFiles.length} orphaned files in '${watcher.path}'`,\n type: 'info',\n tags: ['directory-watcher', 'recovery'],\n data: {\n recoveredFiles: orphanedFiles,\n watcherId: watcher.id,\n },\n });\n }\n } catch (err) {\n this.logger.error(\n err,\n `Failed to recover orphaned files for watcher ${watcher.id}`,\n );\n }\n }\n\n /**\n * Reads directory for processable files.\n *\n * @param {DirectoryWatcher} watcher\n */\n private async read(watcher: DirectoryWatcher) {\n try {\n // Ensure directory structure exists\n await this.ensureDirectoryStructure(watcher.path);\n\n const allFiles = await readdir(watcher.path);\n const filesInDirectory = allFiles.filter(\n (file) =>\n file !== this.SUBFOLDER_PROCESSING &&\n file !== this.SUBFOLDER_COMPLETED &&\n file !== this.SUBFOLDER_FAILED,\n );\n\n // Only process and notify if there are files\n if (filesInDirectory.length === 0) {\n return;\n }\n\n const results = { success: [], failed: [] };\n\n // Process files sequentially\n for (const file of filesInDirectory) {\n try {\n await this.processFileWithMove(watcher, file);\n results.success.push(file);\n } catch (err) {\n this.logger.error(err, `Failed to process file ${file}`);\n results.failed.push({ file, error: err.message });\n }\n }\n\n // Create notification with success/failure breakdown\n this.notificationService.create({\n title: 'Directory Watcher',\n message: `Processed ${results.success.length} of ${filesInDirectory.length} files in '${watcher.path}'`,\n type: results.failed.length > 0 ? 'warning' : 'info',\n tags: ['directory-watcher'],\n data: {\n successCount: results.success.length,\n failedCount: results.failed.length,\n successFiles: results.success,\n failedFiles: results.failed,\n watcherId: watcher.id,\n },\n });\n } catch (e) {\n this.logger.error(e, `Failed to read directory ${watcher.path}`);\n this.notificationService.create({\n title: 'Directory Watcher Error',\n message: `Failed to read directory ${watcher.path}`,\n type: 'error',\n tags: ['directory-watcher'],\n data: {\n error: e.message,\n },\n });\n }\n }\n\n /**\n * Processes a file using the move/archive pattern.\n * Files are moved through: main folder -> processing -> completed/failed\n *\n * @param {DirectoryWatcher} watcher\n * @param {string} fileName\n */\n private async processFileWithMove(\n watcher: DirectoryWatcher,\n fileName: string,\n ): Promise {\n const sourcePath = join(watcher.path, fileName);\n const processingPath = join(\n watcher.path,\n this.SUBFOLDER_PROCESSING,\n fileName,\n );\n const completedPath = join(\n watcher.path,\n this.SUBFOLDER_COMPLETED,\n fileName,\n );\n const failedPath = join(watcher.path, this.SUBFOLDER_FAILED, fileName);\n\n let currentLocation = sourcePath;\n let submissionId: EntityId | null = null;\n\n try {\n // Step 1: Move to processing folder (atomic operation)\n await rename(sourcePath, processingPath);\n currentLocation = processingPath;\n this.logger.info(`Processing file ${fileName}`);\n\n // Step 2: Process the file\n const multerInfo: MulterFileInfo = {\n fieldname: '',\n origin: 'directory-watcher',\n originalname: fileName,\n encoding: '',\n mimetype: getType(fileName),\n size: 0,\n destination: '',\n filename: fileName,\n path: processingPath, // Use processing path\n };\n\n switch (watcher.importAction) {\n case DirectoryWatcherImportAction.NEW_SUBMISSION: {\n const submission = await this.submissionService.create(\n {\n name: fileName,\n type: SubmissionType.FILE,\n },\n multerInfo,\n );\n submissionId = submission.id;\n\n if (watcher.template) {\n await this.submissionService.applyOverridingTemplate(\n submission.id,\n watcher.template?.id,\n );\n }\n break;\n }\n\n default:\n break;\n }\n\n // Step 3: Move to completed folder\n await rename(processingPath, completedPath);\n this.logger.info(\n `Successfully processed file ${fileName} (submission: ${submissionId})`,\n );\n } catch (err) {\n this.logger.error(err, `Failed to process file ${fileName}`);\n\n // Cleanup submission if it was created\n if (submissionId) {\n await this.submissionService\n .remove(submissionId)\n .catch((cleanupErr) => {\n this.logger.error(\n cleanupErr,\n `Failed to cleanup submission ${submissionId}`,\n );\n });\n }\n\n // Move to failed folder and create error file\n try {\n await rename(currentLocation, failedPath);\n\n // Create error details file\n const errorFilePath = join(\n watcher.path,\n this.SUBFOLDER_FAILED,\n `${fileName}.error.txt`,\n );\n const errorDetails = [\n `File: ${fileName}`,\n `Failed at: ${new Date().toISOString()}`,\n `Error: ${err.message}`,\n `Stack: ${err.stack || 'N/A'}`,\n ].join('\\n');\n\n await writeFile(errorFilePath, errorDetails);\n } catch (moveErr) {\n this.logger.error(\n moveErr,\n `Failed to move file to failed folder: ${fileName}`,\n );\n }\n\n throw err;\n }\n }\n\n async create(\n createDto: CreateDirectoryWatcherDto,\n ): Promise {\n // Validate path exists and is accessible (only if path is provided)\n if (createDto.path) {\n try {\n await readdir(createDto.path);\n } catch (err) {\n throw new BadRequestException(\n `Path '${createDto.path}' does not exist or is not accessible`,\n );\n }\n\n // Create directory structure\n await this.ensureDirectoryStructure(createDto.path);\n }\n\n return this.repository.insert(createDto);\n }\n\n async update(id: EntityId, update: UpdateDirectoryWatcherDto) {\n this.logger.withMetadata(update).info(`Updating DirectoryWatcher '${id}'`);\n const entity = await this.repository.findById(id, { failOnMissing: true });\n\n // Validate path if being updated\n if (update.path && update.path !== entity.path) {\n try {\n await readdir(update.path);\n } catch (err) {\n throw new BadRequestException(\n `Path '${update.path}' does not exist or is not accessible`,\n );\n }\n\n // Create directory structure for new path\n await this.ensureDirectoryStructure(update.path);\n }\n\n const template = update.templateId\n ? await this.submissionService.findById(update.templateId, {\n failOnMissing: true,\n })\n : null;\n if (template && !template.isTemplate) {\n throw new BadRequestException('Template Id provided is not a template.');\n }\n\n const updatedEntity = await this.repository.update(id, {\n importAction: update.importAction ?? entity.importAction,\n path: update.path ?? entity.path,\n templateId: update.templateId ?? entity.templateId,\n });\n\n // If this is the first time setting a path (from null/empty to valid path), ensure directory structure\n if (!entity.path && updatedEntity.path) {\n await this.ensureDirectoryStructure(updatedEntity.path);\n }\n\n return updatedEntity;\n }\n\n /**\n * Checks a directory path for validity and returns file count information.\n * Used to warn users before selecting folders with many files.\n *\n * @param {string} path - The directory path to check\n * @returns {Promise} Information about the directory\n */\n async checkPath(path: string): Promise {\n try {\n const allFiles = await readdir(path);\n const files = allFiles.filter(\n (file) =>\n file !== this.SUBFOLDER_PROCESSING &&\n file !== this.SUBFOLDER_COMPLETED &&\n file !== this.SUBFOLDER_FAILED,\n );\n\n return {\n valid: true,\n count: files.length,\n files,\n };\n } catch (err) {\n return {\n valid: false,\n count: 0,\n files: [],\n error: `Path '${path}' does not exist or is not accessible`,\n };\n }\n }\n}\n" }, { "path": "apps/client-server/src/app/directory-watchers/dtos/check-path.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { IsNotEmpty, IsString } from 'class-validator';\n\nexport class CheckPathDto {\n @ApiProperty({ description: 'The directory path to check' })\n @IsNotEmpty()\n @IsString()\n path: string;\n}\n" }, { "path": "apps/client-server/src/app/directory-watchers/dtos/create-directory-watcher.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport {\n DirectoryWatcherImportAction,\n ICreateDirectoryWatcherDto,\n} from '@postybirb/types';\nimport { IsEnum, IsOptional, IsString } from 'class-validator';\n\nexport class CreateDirectoryWatcherDto implements ICreateDirectoryWatcherDto {\n @ApiProperty()\n @IsOptional()\n @IsString()\n path: string;\n\n @ApiProperty({\n enum: DirectoryWatcherImportAction,\n })\n @IsEnum(DirectoryWatcherImportAction)\n importAction: DirectoryWatcherImportAction;\n}\n" }, { "path": "apps/client-server/src/app/directory-watchers/dtos/update-directory-watcher.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport {\n DirectoryWatcherImportAction,\n IUpdateDirectoryWatcherDto,\n SubmissionId,\n} from '@postybirb/types';\nimport { IsEnum, IsOptional, IsString } from 'class-validator';\n\nexport class UpdateDirectoryWatcherDto implements IUpdateDirectoryWatcherDto {\n @ApiProperty({\n enum: DirectoryWatcherImportAction,\n required: false,\n })\n @IsOptional()\n @IsEnum(DirectoryWatcherImportAction)\n importAction?: DirectoryWatcherImportAction;\n\n @ApiProperty({ required: false })\n @IsOptional()\n @IsString()\n path?: string;\n\n @ApiProperty({ required: false })\n @IsOptional()\n @IsString()\n templateId?: SubmissionId;\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/account.entity.ts", "content": "import { IAccount, IAccountDto } from '@postybirb/types';\nimport { Exclude, instanceToPlain, Type } from 'class-transformer';\nimport { UnknownWebsite } from '../../websites/website';\nimport { DatabaseEntity } from './database-entity';\nimport { WebsiteData } from './website-data.entity';\n\nexport class Account extends DatabaseEntity implements IAccount {\n name: string;\n\n website: string;\n\n groups: string[] = [];\n\n /**\n * we don't want to pass this down to users unless filtered\n * by the website instance.\n */\n @Exclude()\n @Type(() => WebsiteData)\n websiteData: WebsiteData;\n\n @Exclude()\n websiteInstance?: UnknownWebsite;\n\n // eslint-disable-next-line @typescript-eslint/no-useless-constructor\n constructor(entity: Partial) {\n super(entity);\n Object.assign(this, entity);\n }\n\n toObject(): IAccount {\n return instanceToPlain(this, {\n enableCircularCheck: true,\n }) as IAccount;\n }\n\n toDTO(): IAccountDto {\n const dto: IAccountDto = {\n ...(this.toObject() as unknown as IAccountDto),\n data: this.websiteInstance?.getWebsiteData() ?? {},\n state: this.websiteInstance?.getLoginState() ?? {\n isLoggedIn: false,\n username: '',\n pending: false,\n lastUpdated: null,\n },\n websiteInfo: {\n websiteDisplayName:\n this.websiteInstance?.decoratedProps.metadata.displayName ?? '',\n supports: this.websiteInstance?.getSupportedTypes() ?? [],\n },\n };\n return dto;\n }\n\n withWebsiteInstance(websiteInstance: UnknownWebsite): this {\n this.websiteInstance = websiteInstance;\n return this;\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/custom-shortcut.entity.ts", "content": "import {\n DefaultDescription,\n Description,\n ICustomShortcut,\n ICustomShortcutDto,\n} from '@postybirb/types';\nimport { instanceToPlain } from 'class-transformer';\nimport { DatabaseEntity } from './database-entity';\n\nexport class CustomShortcut extends DatabaseEntity implements ICustomShortcut {\n name: string;\n\n shortcut: Description = DefaultDescription();\n\n constructor(entity: Partial) {\n super(entity);\n Object.assign(this, entity);\n }\n\n public toObject(): ICustomShortcut {\n return instanceToPlain(this, {\n enableCircularCheck: true,\n }) as ICustomShortcut;\n }\n\n public toDTO(): ICustomShortcutDto {\n return instanceToPlain(this, {\n enableCircularCheck: true,\n }) as ICustomShortcutDto;\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/database-entity.spec.ts", "content": "import { IEntity, IEntityDto } from '@postybirb/types';\nimport { instanceToPlain } from 'class-transformer';\nimport 'reflect-metadata';\nimport { DatabaseEntity, fromDatabaseRecord } from './database-entity';\n\nclass TestEntity extends DatabaseEntity {\n public testField: string;\n\n constructor(entity: Partial) {\n super(entity);\n Object.assign(this, entity);\n }\n\n toObject(): IEntity {\n return instanceToPlain(this, {\n enableCircularCheck: true,\n }) as unknown as IEntity;\n }\n\n toDTO(): IEntityDto {\n return this.toObject() as unknown as IEntityDto;\n }\n}\n\ndescribe('DatabaseEntity', () => {\n let entity: TestEntity;\n\n beforeEach(() => {\n entity = new TestEntity({\n id: 'id',\n testField: 'test',\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n });\n });\n\n it('should create an instance', () => {\n expect(entity).toBeTruthy();\n });\n\n it('should convert class to object', () => {\n const obj = entity.toObject();\n expect(obj).toBeTruthy();\n });\n\n it('should succeed fromDatabaseObject', () => {\n const dbObj = {\n id: 'id',\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n testField: 'test',\n };\n const obj = fromDatabaseRecord(TestEntity, dbObj);\n expect(obj).toBeTruthy();\n expect(obj.id).toBe(dbObj.id);\n expect(obj.createdAt).toEqual(dbObj.createdAt);\n expect(obj.updatedAt).toEqual(dbObj.updatedAt);\n expect(obj.testField).toBe(dbObj.testField);\n });\n\n it('should convert toObject', () => {\n const obj = entity.toObject();\n expect(obj).toBeTruthy();\n expect(obj.id).toBe(entity.id);\n expect(obj.createdAt).toBe(entity.createdAt);\n expect(obj.updatedAt).toBe(entity.updatedAt);\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n expect((obj as any).testField).toBe(entity.testField);\n });\n});\n" }, { "path": "apps/client-server/src/app/drizzle/models/database-entity.ts", "content": "/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { EntityId, IEntity, IEntityDto } from '@postybirb/types';\nimport {\n ClassConstructor,\n plainToClass,\n plainToInstance,\n} from 'class-transformer';\nimport { v4 } from 'uuid';\n\nexport function fromDatabaseRecord(\n entity: ClassConstructor,\n record: any[],\n): TEntity[];\nexport function fromDatabaseRecord(\n entity: ClassConstructor,\n record: any,\n): TEntity;\nexport function fromDatabaseRecord(\n entity: ClassConstructor,\n record: any | any[],\n): TEntity | TEntity[] {\n if (Array.isArray(record)) {\n return record.map((r) =>\n plainToInstance(entity, r, { enableCircularCheck: true }),\n ) as TEntity[];\n }\n return plainToClass(entity, record, {\n enableCircularCheck: true,\n }) as TEntity;\n}\n\nexport abstract class DatabaseEntity implements IEntity {\n public readonly id: EntityId;\n\n public createdAt: string;\n\n public updatedAt: string;\n\n constructor(entity: Partial) {\n Object.assign(this, entity);\n if (!this.id) {\n this.id = v4();\n }\n }\n\n public abstract toObject(): IEntity;\n\n public abstract toDTO(): IEntityDto;\n\n public toJSON(): string {\n return JSON.stringify(this.toDTO());\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/directory-watcher.entity.ts", "content": "import {\n DirectoryWatcherDto,\n DirectoryWatcherImportAction,\n IDirectoryWatcher,\n SubmissionId,\n} from '@postybirb/types';\nimport { instanceToPlain, Type } from 'class-transformer';\nimport { DatabaseEntity } from './database-entity';\nimport { Submission } from './submission.entity';\n\nexport class DirectoryWatcher\n extends DatabaseEntity\n implements IDirectoryWatcher\n{\n path?: string;\n\n templateId: SubmissionId;\n\n importAction: DirectoryWatcherImportAction;\n\n @Type(() => Submission)\n template: Submission;\n\n toObject(): IDirectoryWatcher {\n return instanceToPlain(this, {\n enableCircularCheck: true,\n }) as IDirectoryWatcher;\n }\n\n toDTO(): DirectoryWatcherDto {\n return this.toObject() as unknown as DirectoryWatcherDto;\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/file-buffer.entity.ts", "content": "import { EntityId, FileBufferDto, IFileBuffer } from '@postybirb/types';\nimport { Exclude, instanceToPlain, Type } from 'class-transformer';\nimport { DatabaseEntity } from './database-entity';\n\nexport class FileBuffer extends DatabaseEntity implements IFileBuffer {\n submissionFileId: EntityId;\n\n @Type(() => Buffer)\n @Exclude({ toPlainOnly: true })\n buffer: Buffer;\n\n fileName: string;\n\n mimeType: string;\n\n size: number;\n\n width: number;\n\n height: number;\n\n constructor(entity: Partial) {\n super(entity);\n Object.assign(this, entity);\n }\n\n toObject(): IFileBuffer {\n return instanceToPlain(this, {\n enableCircularCheck: true,\n }) as IFileBuffer;\n }\n\n toDTO(): FileBufferDto {\n return this.toObject() as unknown as FileBufferDto;\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/index.ts", "content": "export * from './account.entity';\nexport * from './database-entity';\nexport * from './directory-watcher.entity';\nexport * from './file-buffer.entity';\nexport * from './post-event.entity';\nexport * from './post-queue-record.entity';\nexport * from './post-record.entity';\nexport * from './settings.entity';\nexport * from './submission-file.entity';\nexport * from './submission.entity';\nexport * from './tag-converter.entity';\nexport * from './tag-group.entity';\nexport * from './user-converter.entity';\nexport * from './user-specified-website-options.entity';\nexport * from './website-data.entity';\nexport * from './website-options.entity';\n\n" }, { "path": "apps/client-server/src/app/drizzle/models/notification.entity.ts", "content": "import { INotification } from '@postybirb/types';\nimport { instanceToPlain } from 'class-transformer';\nimport { DatabaseEntity } from './database-entity';\n\nexport class Notification extends DatabaseEntity implements INotification {\n title: string;\n\n message: string;\n\n tags: string[];\n\n data: Record;\n\n isRead: boolean;\n\n hasEmitted: boolean;\n\n type: 'warning' | 'error' | 'info' | 'success';\n\n toObject(): INotification {\n return instanceToPlain(this, {}) as INotification;\n }\n\n toDTO(): INotification {\n return this.toObject() as unknown as INotification;\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/post-event.entity.ts", "content": "import {\n AccountId,\n EntityId,\n IPostEvent,\n IPostEventError,\n IPostEventMetadata,\n PostEventDto,\n PostEventType,\n} from '@postybirb/types';\nimport { instanceToPlain, Type } from 'class-transformer';\nimport { Account } from './account.entity';\nimport { DatabaseEntity } from './database-entity';\nimport { PostRecord } from './post-record.entity';\n\nexport class PostEvent extends DatabaseEntity implements IPostEvent {\n postRecordId: EntityId;\n\n accountId?: AccountId;\n\n eventType: PostEventType;\n\n fileId?: EntityId;\n\n sourceUrl?: string;\n\n error?: IPostEventError;\n\n metadata?: IPostEventMetadata;\n\n @Type(() => PostRecord)\n postRecord: PostRecord;\n\n @Type(() => Account)\n account?: Account;\n\n constructor(entity: Partial) {\n super(entity);\n Object.assign(this, entity);\n }\n\n toObject(): IPostEvent {\n return instanceToPlain(this, {\n enableCircularCheck: true,\n }) as IPostEvent;\n }\n\n toDTO(): PostEventDto {\n const dto: PostEventDto = {\n ...this.toObject(),\n account: this.account?.toDTO(),\n };\n return dto;\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/post-queue-record.entity.ts", "content": "import {\n EntityId,\n IPostQueueRecord,\n PostQueueRecordDto,\n SubmissionId\n} from '@postybirb/types';\nimport { instanceToPlain, Type } from 'class-transformer';\nimport { DatabaseEntity } from './database-entity';\nimport { PostRecord } from './post-record.entity';\nimport { Submission } from './submission.entity';\n\nexport class PostQueueRecord\n extends DatabaseEntity\n implements IPostQueueRecord\n{\n postRecordId: EntityId;\n\n submissionId: SubmissionId;\n\n @Type(() => PostRecord)\n postRecord: PostRecord;\n\n @Type(() => Submission)\n submission: Submission;\n\n toObject(): IPostQueueRecord {\n return instanceToPlain(this, {\n enableCircularCheck: true,\n }) as IPostQueueRecord;\n }\n\n toDTO(): PostQueueRecordDto {\n return this.toObject() as unknown as PostQueueRecordDto;\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/post-record.entity.ts", "content": "import {\n EntityId,\n IPostRecord,\n PostRecordDto,\n PostRecordResumeMode,\n PostRecordState,\n SubmissionId,\n} from '@postybirb/types';\nimport { instanceToPlain, Type } from 'class-transformer';\nimport { DatabaseEntity } from './database-entity';\nimport { PostEvent } from './post-event.entity';\nimport { PostQueueRecord } from './post-queue-record.entity';\nimport { Submission } from './submission.entity';\n\nexport class PostRecord extends DatabaseEntity implements IPostRecord {\n postQueueRecordId: EntityId;\n\n submissionId: SubmissionId;\n\n @Type(() => Submission)\n submission: Submission;\n\n /**\n * Reference to the originating NEW PostRecord for this chain.\n * null for NEW records (they ARE the origin).\n */\n originPostRecordId?: EntityId;\n\n /**\n * The originating NEW PostRecord (resolved relation).\n */\n @Type(() => PostRecord)\n origin?: PostRecord;\n\n /**\n * All CONTINUE/RETRY PostRecords that chain to this origin.\n */\n @Type(() => PostRecord)\n chainedRecords?: PostRecord[];\n\n completedAt?: string;\n\n state: PostRecordState;\n\n resumeMode: PostRecordResumeMode;\n\n // eslint-disable-next-line @typescript-eslint/no-useless-constructor\n constructor(entity: Partial) {\n super(entity);\n }\n\n @Type(() => PostEvent)\n events: PostEvent[];\n\n @Type(() => PostQueueRecord)\n postQueueRecord: PostQueueRecord;\n\n toObject(): IPostRecord {\n return instanceToPlain(this, {\n enableCircularCheck: true,\n }) as IPostRecord;\n }\n\n toDTO(): PostRecordDto {\n const dto: PostRecordDto = {\n ...this.toObject(),\n events: this.events?.map((event) => event.toDTO()),\n postQueueRecord: this.postQueueRecord?.toDTO(),\n };\n return dto;\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/settings.entity.ts", "content": "import {\n ISettings,\n ISettingsOptions,\n SettingsConstants,\n SettingsDto,\n} from '@postybirb/types';\nimport { instanceToPlain } from 'class-transformer';\nimport { DatabaseEntity } from './database-entity';\n\nexport class Settings extends DatabaseEntity implements ISettings {\n profile: string;\n\n settings: ISettingsOptions = { ...SettingsConstants.DEFAULT_SETTINGS };\n\n toObject(): ISettings {\n return instanceToPlain(this, {\n enableCircularCheck: true,\n }) as ISettings;\n }\n\n toDTO(): SettingsDto {\n return this.toObject() as unknown as SettingsDto;\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/submission-file.entity.ts", "content": "import {\n EntityId,\n FileSubmissionMetadata,\n ISubmissionFile,\n ISubmissionFileDto,\n SubmissionFileMetadata,\n} from '@postybirb/types';\nimport { instanceToPlain, Type } from 'class-transformer';\nimport { PostyBirbDatabase } from '../postybirb-database/postybirb-database';\nimport { DatabaseEntity } from './database-entity';\nimport { FileBuffer } from './file-buffer.entity';\nimport { Submission } from './submission.entity';\n\nexport class SubmissionFile extends DatabaseEntity implements ISubmissionFile {\n submissionId: EntityId;\n\n primaryFileId: EntityId;\n\n altFileId: EntityId;\n\n thumbnailId: EntityId;\n\n @Type(() => Submission)\n submission: Submission;\n\n fileName: string;\n\n hash: string;\n\n mimeType: string;\n\n @Type(() => FileBuffer)\n file: FileBuffer;\n\n @Type(() => FileBuffer)\n thumbnail?: FileBuffer;\n\n @Type(() => FileBuffer)\n altFile?: FileBuffer;\n\n hasThumbnail: boolean;\n\n hasAltFile: boolean;\n\n hasCustomThumbnail: boolean;\n\n size: number;\n\n width: number;\n\n height: number;\n\n metadata: SubmissionFileMetadata;\n\n order: number;\n\n constructor(entity: Partial) {\n super(entity);\n Object.assign(this, entity);\n }\n\n toObject(): ISubmissionFile {\n return instanceToPlain(this, {\n enableCircularCheck: true,\n }) as ISubmissionFile;\n }\n\n toDTO(): ISubmissionFileDto {\n return this.toObject() as unknown as ISubmissionFileDto;\n }\n\n /**\n * Load the submission file from the database\n * More of a workaround for the lack of proper ORM support with\n * blob relations loading from nested with queries.\n */\n public async load(fileTarget?: 'file' | 'thumbnail' | 'alt') {\n const db = new PostyBirbDatabase('FileBufferSchema');\n\n if (fileTarget) {\n switch (fileTarget) {\n case 'file':\n this.file = await db.findById(this.primaryFileId);\n break;\n case 'thumbnail':\n this.thumbnail = await db.findById(this.thumbnailId);\n break;\n case 'alt':\n this.altFile = await db.findById(this.altFileId);\n break;\n default:\n throw new Error('Invalid file target');\n }\n return;\n }\n\n this.file = await db.findById(this.primaryFileId);\n if (this.thumbnailId) {\n this.thumbnail = await db.findById(this.thumbnailId);\n }\n if (this.altFileId) {\n this.altFile = await db.findById(this.altFileId);\n }\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/submission.entity.ts", "content": "import {\n ISubmission,\n ISubmissionDto,\n ISubmissionMetadata,\n ISubmissionScheduleInfo,\n ScheduleType,\n SubmissionType,\n} from '@postybirb/types';\nimport { instanceToPlain, Type } from 'class-transformer';\nimport { DatabaseEntity } from './database-entity';\nimport { PostQueueRecord } from './post-queue-record.entity';\nimport { PostRecord } from './post-record.entity';\nimport { SubmissionFile } from './submission-file.entity';\nimport { WebsiteOptions } from './website-options.entity';\n\nexport class Submission\n extends DatabaseEntity\n implements ISubmission\n{\n type: SubmissionType;\n\n @Type(() => WebsiteOptions)\n options: WebsiteOptions[];\n\n @Type(() => PostQueueRecord)\n postQueueRecord?: PostQueueRecord;\n\n isScheduled = false;\n\n isTemplate = false;\n\n isMultiSubmission = false;\n\n isArchived = false;\n\n isInitialized = false;\n\n schedule: ISubmissionScheduleInfo = {\n scheduleType: ScheduleType.NONE,\n };\n\n @Type(() => SubmissionFile)\n files: SubmissionFile[];\n\n metadata: T;\n\n @Type(() => PostRecord)\n posts: PostRecord[];\n\n order: number;\n\n constructor(entity: Partial>) {\n super(entity);\n Object.assign(this, entity);\n }\n\n toObject(): ISubmission {\n return instanceToPlain(this, {\n enableCircularCheck: true,\n }) as ISubmission;\n }\n\n toDTO(): ISubmissionDto {\n const dto: ISubmissionDto = {\n ...this.toObject(),\n files: this.files?.map((file) => file.toDTO()),\n options: this.options?.map((option) => option.toDTO()),\n posts: this.posts?.map((post) => post.toDTO()),\n postQueueRecord: this.postQueueRecord?.toDTO(),\n validations: [],\n };\n\n return dto;\n }\n\n getSubmissionName(): string {\n if (this.options?.length) {\n return this.options.find((o) => o.isDefault)?.data.title;\n }\n return 'Unknown';\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/tag-converter.entity.ts", "content": "import { ITagConverter, TagConverterDto } from '@postybirb/types';\nimport { instanceToPlain } from 'class-transformer';\nimport { DatabaseEntity } from './database-entity';\n\nexport class TagConverter extends DatabaseEntity implements ITagConverter {\n public readonly tag: string;\n\n public convertTo: Record = {};\n\n constructor(entity: ITagConverter) {\n super(entity);\n Object.assign(this, entity);\n }\n\n toObject(): ITagConverter {\n return instanceToPlain(this, {\n enableCircularCheck: true,\n }) as ITagConverter;\n }\n\n toDTO(): TagConverterDto {\n return this.toObject() as unknown as TagConverterDto;\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/tag-group.entity.ts", "content": "import { ITagGroup, TagGroupDto } from '@postybirb/types';\nimport { instanceToPlain } from 'class-transformer';\nimport { DatabaseEntity } from './database-entity';\n\nexport class TagGroup extends DatabaseEntity implements ITagGroup {\n name: string;\n\n tags: string[] = [];\n\n toObject(): ITagGroup {\n return instanceToPlain(this, {\n enableCircularCheck: true,\n }) as ITagGroup;\n }\n\n toDTO(): TagGroupDto {\n return this.toObject() as unknown as TagGroupDto;\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/user-converter.entity.ts", "content": "import { IUserConverter, UserConverterDto } from '@postybirb/types';\nimport { instanceToPlain } from 'class-transformer';\nimport { DatabaseEntity } from './database-entity';\n\nexport class UserConverter extends DatabaseEntity implements IUserConverter {\n public readonly username: string;\n\n public convertTo: Record = {};\n\n constructor(entity: IUserConverter) {\n super(entity);\n Object.assign(this, entity);\n }\n\n toObject(): IUserConverter {\n return instanceToPlain(this, {\n enableCircularCheck: true,\n }) as IUserConverter;\n }\n\n toDTO(): UserConverterDto {\n return this.toObject() as unknown as UserConverterDto;\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/user-specified-website-options.entity.ts", "content": "import {\n AccountId,\n DynamicObject,\n IUserSpecifiedWebsiteOptions,\n SubmissionType,\n UserSpecifiedWebsiteOptionsDto\n} from '@postybirb/types';\nimport { instanceToPlain, Type } from 'class-transformer';\nimport { Account } from './account.entity';\nimport { DatabaseEntity } from './database-entity';\n\nexport class UserSpecifiedWebsiteOptions\n extends DatabaseEntity\n implements IUserSpecifiedWebsiteOptions\n{\n accountId: AccountId;\n\n @Type(() => Account)\n account: Account;\n\n type: SubmissionType;\n\n options: DynamicObject;\n\n toObject(): IUserSpecifiedWebsiteOptions {\n return instanceToPlain(this, {\n enableCircularCheck: true,\n }) as IUserSpecifiedWebsiteOptions;\n }\n\n toDTO(): UserSpecifiedWebsiteOptionsDto {\n return this.toObject() as unknown as UserSpecifiedWebsiteOptionsDto;\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/website-data.entity.ts", "content": "import {\n DynamicObject,\n IWebsiteData,\n IWebsiteDataDto\n} from '@postybirb/types';\nimport { instanceToPlain, Type } from 'class-transformer';\nimport { Account } from './account.entity';\nimport { DatabaseEntity } from './database-entity';\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport class WebsiteData\n extends DatabaseEntity\n implements IWebsiteData\n{\n data: T = {} as T;\n\n @Type(() => Account)\n account: Account;\n\n toObject(): IWebsiteData {\n return instanceToPlain(this) as IWebsiteData;\n }\n\n toDTO(): IWebsiteDataDto {\n return this.toObject() as unknown as IWebsiteDataDto;\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/models/website-options.entity.ts", "content": "import {\n AccountId,\n IWebsiteFormFields,\n IWebsiteOptions,\n SubmissionId,\n WebsiteOptionsDto,\n} from '@postybirb/types';\nimport { instanceToPlain, Type } from 'class-transformer';\nimport { BaseWebsiteOptions } from '../../websites/models/base-website-options';\nimport { Account } from './account.entity';\nimport { DatabaseEntity } from './database-entity';\nimport { Submission } from './submission.entity';\n\nexport class WebsiteOptions extends DatabaseEntity implements IWebsiteOptions {\n accountId: AccountId;\n\n @Type(() => Account)\n account: Account;\n\n submissionId: SubmissionId;\n\n @Type(() => Submission)\n submission: Submission;\n\n data: IWebsiteFormFields = new BaseWebsiteOptions();\n\n isDefault: boolean;\n\n constructor(entity: Partial) {\n super(entity);\n Object.assign(this, entity);\n }\n\n toObject(): IWebsiteOptions {\n return instanceToPlain(this, {\n enableCircularCheck: true,\n }) as IWebsiteOptions;\n }\n\n toDTO(): WebsiteOptionsDto {\n return {\n ...this.toObject(),\n submission: this.submission?.toDTO(),\n } as unknown as WebsiteOptionsDto;\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/postybirb-database/find-options.type.ts", "content": "export type FindOptions = {\n failOnMissing?: boolean;\n};\n" }, { "path": "apps/client-server/src/app/drizzle/postybirb-database/postybirb-database.spec.ts", "content": "import { clearDatabase, Schemas } from '@postybirb/database';\nimport { eq as equals } from 'drizzle-orm';\nimport 'reflect-metadata';\nimport { PostyBirbDatabase } from './postybirb-database';\n\ndescribe('PostyBirbDatabase', () => {\n let service: PostyBirbDatabase<'AccountSchema'>;\n\n beforeEach(() => {\n clearDatabase();\n service = new PostyBirbDatabase('AccountSchema');\n });\n\n it('should be created', () => {\n expect(service).toBeTruthy();\n });\n\n it('should insert and delete', async () => {\n const account = await service.insert({\n name: 'test',\n website: 'test',\n groups: [],\n });\n\n const accounts = await service.findAll();\n expect(accounts).toHaveLength(1);\n expect(accounts[0].id).toBe(account.id);\n expect(accounts[0].name).toBe('test');\n expect(accounts[0].website).toBe('test');\n\n await service.deleteById([account.id]);\n\n const accountsAfterDelete = await service.findAll();\n expect(accountsAfterDelete).toHaveLength(0);\n });\n\n it('should find by id', async () => {\n const account = await service.insert({\n name: 'test',\n website: 'test',\n groups: [],\n });\n\n const foundAccount = await service.findById(account.id, {\n failOnMissing: true,\n });\n expect(foundAccount).toBeTruthy();\n expect(foundAccount.id).toBe(account.id);\n expect(foundAccount.name).toBe('test');\n expect(foundAccount.website).toBe('test');\n\n const notFoundAccount = await service.findById('not-found', {\n failOnMissing: false,\n });\n expect(notFoundAccount).toBeNull();\n });\n\n it('should throw on find by id not found', async () => {\n await expect(\n service.findById('not-found', { failOnMissing: true }),\n ).rejects.toThrow('Record with id not-found not found');\n });\n\n it('should update', async () => {\n const account = await service.insert({\n name: 'test',\n website: 'test',\n groups: [],\n });\n\n await service.update(account.id, {\n name: 'test2',\n });\n\n const updatedAccount = await service.findById(account.id, {\n failOnMissing: true,\n });\n expect(updatedAccount).toBeTruthy();\n expect(updatedAccount.id).toBe(account.id);\n expect(updatedAccount.name).toBe('test2');\n });\n\n it('should find one', async () => {\n await service.insert({\n name: 'test',\n website: 'test',\n groups: [],\n });\n\n const foundAccount = await service.findOne({\n where: (account, { eq }) => eq(account.name, 'test'),\n });\n\n expect(foundAccount).toBeTruthy();\n expect(foundAccount?.name).toBe('test');\n });\n\n it('should find many', async () => {\n await service.insert({\n name: 'test',\n website: 'test',\n groups: [],\n });\n\n await service.insert({\n name: 'test2',\n website: 'test',\n groups: [],\n });\n\n const foundAccounts = await service.find({\n where: (account, { eq }) => eq(account.website, 'test'),\n });\n\n expect(foundAccounts).toHaveLength(2);\n });\n\n it('should return empty array on find many not found', async () => {\n const foundAccounts = await service.find({\n where: (account, { eq }) => eq(account.website, 'test'),\n });\n\n expect(foundAccounts).toHaveLength(0);\n });\n\n it('should notify subscribers on create', async () => {\n const subscriber = jest.fn();\n service.subscribe('AccountSchema', subscriber);\n\n const entity = await service.insert({\n name: 'test',\n website: 'test',\n groups: [],\n });\n\n expect(subscriber).toHaveBeenCalledWith([entity.id], 'insert');\n });\n\n it('should select', async () => {\n await service.insert({\n name: 'test',\n website: 'test',\n groups: [],\n });\n\n const foundAccounts = await service.select(\n equals(Schemas.AccountSchema.name, 'test'),\n );\n\n expect(foundAccounts).toHaveLength(1);\n });\n});\n" }, { "path": "apps/client-server/src/app/drizzle/postybirb-database/postybirb-database.ts", "content": "/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { NotFoundException } from '@nestjs/common';\nimport {\n getDatabase,\n Insert,\n PostyBirbDatabaseType,\n SchemaKey,\n Schemas,\n Select,\n} from '@postybirb/database';\nimport { EntityId, NULL_ACCOUNT_ID } from '@postybirb/types';\nimport { eq, inArray, KnownKeysOnly, SQL } from 'drizzle-orm';\nimport {\n DBQueryConfig,\n ExtractTablesWithRelations,\n} from 'drizzle-orm/relations';\nimport { fromDatabaseRecord } from '../models';\nimport { FindOptions } from './find-options.type';\nimport {\n DatabaseSchemaEntityMap,\n DatabaseSchemaEntityMapConst,\n} from './schema-entity-map';\n\ntype ExtractedRelations = ExtractTablesWithRelations;\n\ntype Relation =\n PostyBirbDatabaseType['_']['schema'][TSchemaKey];\n\nexport type Action = 'delete' | 'insert' | 'update';\n\ntype SubscribeCallback = (ids: EntityId[], action: Action) => void;\n\nexport class PostyBirbDatabase<\n TSchemaKey extends SchemaKey,\n TEntityClass = DatabaseSchemaEntityMap[TSchemaKey],\n> {\n public readonly db: PostyBirbDatabaseType;\n\n private static readonly subscribers: Record<\n SchemaKey,\n Array\n > = {\n AccountSchema: [],\n DirectoryWatcherSchema: [],\n FileBufferSchema: [],\n PostEventSchema: [],\n PostQueueRecordSchema: [],\n PostRecordSchema: [],\n SettingsSchema: [],\n SubmissionFileSchema: [],\n SubmissionSchema: [],\n TagConverterSchema: [],\n TagGroupSchema: [],\n UserConverterSchema: [],\n UserSpecifiedWebsiteOptionsSchema: [],\n WebsiteDataSchema: [],\n WebsiteOptionsSchema: [],\n NotificationSchema: [],\n CustomShortcutSchema: [],\n };\n\n constructor(\n private readonly schemaKey: TSchemaKey,\n private readonly load?: DBQueryConfig<\n 'many',\n true,\n ExtractedRelations,\n Relation\n >['with'],\n ) {\n this.db = getDatabase();\n }\n\n public subscribe(key: SchemaKey[], callback: SubscribeCallback): this;\n public subscribe(key: SchemaKey, callback: SubscribeCallback): this;\n public subscribe(\n key: SchemaKey | SchemaKey[],\n callback: SubscribeCallback,\n ): this {\n if (Array.isArray(key)) {\n key.forEach((k) => this.subscribe(k, callback));\n return this;\n }\n if (!PostyBirbDatabase.subscribers[key]) {\n PostyBirbDatabase.subscribers[key] = [];\n }\n PostyBirbDatabase.subscribers[key].push(callback);\n return this;\n }\n\n private notify(ids: EntityId[], action: Action) {\n PostyBirbDatabase.subscribers[this.schemaKey].forEach((callback) =>\n callback(ids, action),\n );\n }\n\n /**\n * Forcefully calls notify.\n * To be used where database updates occur outside of normal db calls\n * such as during transaction and direct db mutations.\n *\n * @param {EntityId[]} ids\n * @param {Action} action\n */\n public forceNotify(ids: EntityId[], action: Action) {\n this.notify(ids, action);\n }\n\n /**\n * Static method to notify subscribers for a given schema.\n * Used by TransactionContext to trigger notifications after commit.\n *\n * @param {SchemaKey} schemaKey - The schema key to notify subscribers for\n * @param {EntityId[]} ids - The entity IDs that were affected\n * @param {Action} action - The action that occurred (insert, update, delete)\n */\n public static notifySubscribers(\n schemaKey: SchemaKey,\n ids: EntityId[],\n action: Action,\n ) {\n PostyBirbDatabase.subscribers[schemaKey]?.forEach((callback) =>\n callback(ids, action),\n );\n }\n\n public get EntityClass() {\n return DatabaseSchemaEntityMapConst[this.schemaKey];\n }\n\n public get schemaEntity() {\n return Schemas[this.schemaKey];\n }\n\n private classConverter(value: any[]): TEntityClass[];\n private classConverter(value: any): TEntityClass;\n private classConverter(value: any | any[]): TEntityClass | TEntityClass[] {\n if (Array.isArray(value)) {\n return fromDatabaseRecord(this.EntityClass, value);\n }\n return fromDatabaseRecord(this.EntityClass, value);\n }\n\n public async insert(value: Insert): Promise;\n public async insert(value: Insert[]): Promise;\n public async insert(\n value: Insert | Insert[],\n ): Promise {\n const insertQuery = this.db\n .insert(this.schemaEntity)\n .values(value)\n .returning();\n // After calling .returning(), the result is always an array of the selected fields\n const inserts = (await insertQuery) as Array>;\n this.notify(\n inserts.map((insert) => insert.id),\n 'insert',\n );\n const result = await Promise.all(\n inserts.map((insert) => this.findById(insert.id)),\n );\n return Array.isArray(value) ? result : result[0];\n }\n\n public async deleteById(ids: EntityId[]) {\n if (ids.find((id) => id === NULL_ACCOUNT_ID)) {\n throw new Error('Cannot delete the null account');\n }\n const result = await this.db\n .delete(this.schemaEntity)\n .where(inArray(this.schemaEntity.id, ids));\n this.notify(ids, 'delete');\n return result;\n }\n\n public async findById(\n id: EntityId,\n options?: FindOptions,\n load?: DBQueryConfig<\n 'many',\n true,\n ExtractedRelations,\n Relation\n >['with'],\n ): Promise {\n const record = await this.db.query[this.schemaKey].findFirst({\n where: eq(this.schemaEntity.id, id),\n with: {\n ...(load ?? this.load ?? {}),\n },\n });\n\n if (!record && options?.failOnMissing) {\n throw new NotFoundException(`Record with id ${id} not found`);\n }\n\n return record ? (this.classConverter(record) as TEntityClass) : null;\n }\n\n public async select(query: SQL): Promise {\n const records = await this.db.select().from(this.schemaEntity).where(query);\n return this.classConverter(records);\n }\n\n public async find<\n TConfig extends DBQueryConfig<\n 'many',\n true,\n ExtractedRelations,\n Relation\n >,\n >(\n query: KnownKeysOnly<\n TConfig,\n DBQueryConfig<'many', true, ExtractedRelations, Relation>\n >,\n ): Promise {\n const record: any[] =\n (await this.db.query[\n this.schemaKey as keyof PostyBirbDatabaseType\n ].findMany({\n ...query,\n with: query.with\n ? query.with\n : {\n ...(this.load ?? {}),\n },\n })) ?? [];\n return this.classConverter(record);\n }\n\n public async findOne<\n TSelection extends Omit<\n DBQueryConfig<'many', true, ExtractedRelations, Relation>,\n 'limit'\n >,\n >(\n query: KnownKeysOnly<\n TSelection,\n Omit<\n DBQueryConfig<'many', true, ExtractedRelations, Relation>,\n 'limit'\n >\n >,\n ): Promise {\n const record = await this.db.query[\n this.schemaKey as keyof PostyBirbDatabaseType\n ].findFirst({\n ...query,\n with: query.with\n ? query.with\n : {\n ...(this.load ?? {}),\n },\n });\n return record ? this.classConverter(record) : null;\n }\n\n public async findAll(): Promise {\n const records: object[] = await this.db.query[this.schemaKey].findMany({\n with: {\n ...(this.load ?? {}),\n },\n });\n return this.classConverter(records);\n }\n\n public async update(\n id: EntityId,\n set: Partial>,\n ): Promise {\n await this.findById(id, { failOnMissing: true });\n\n await this.db\n // eslint-disable-next-line testing-library/await-async-query\n .update(this.schemaEntity)\n .set(set)\n .where(eq(this.schemaEntity.id, id));\n this.notify([id], 'update');\n return this.findById(id);\n }\n\n public count(filter?: SQL): Promise {\n return this.db.$count(this.schemaEntity, filter);\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/postybirb-database/postybirb-database.util.ts", "content": "/* eslint-disable no-param-reassign */\nimport { SchemaKey, Schemas } from '@postybirb/database';\nimport { DatabaseEntity } from '../models';\nimport { PostyBirbDatabase } from './postybirb-database';\n\nexport class PostyBirbDatabaseUtil {\n static async saveFromEntity(entity: T) {\n const obj = entity.toObject();\n let entitySchemaKey: SchemaKey | undefined;\n for (const schemaKey of Object.keys(Schemas)) {\n if (schemaKey === `${entity.constructor.name}Schema`) {\n entitySchemaKey = schemaKey as SchemaKey;\n break;\n }\n }\n\n if (!entitySchemaKey) {\n throw new Error(`Could not find schema for ${entity.constructor.name}`);\n }\n\n const db = new PostyBirbDatabase(entitySchemaKey);\n const exists = await db.findById(entity.id);\n if (exists) {\n if (exists.updatedAt !== entity.updatedAt) {\n throw new Error('Entity has been updated since last fetch');\n }\n const update = await db.update(entity.id, obj);\n entity.updatedAt = update.updatedAt;\n } else {\n const insert = await db.insert(obj);\n entity.createdAt = insert.createdAt;\n entity.updatedAt = insert.updatedAt;\n }\n\n return entity;\n }\n}\n" }, { "path": "apps/client-server/src/app/drizzle/postybirb-database/schema-entity-map.ts", "content": "import { SchemaKey } from '@postybirb/database';\nimport {\n Account,\n DirectoryWatcher,\n FileBuffer,\n PostEvent,\n PostQueueRecord,\n PostRecord,\n Settings,\n Submission,\n SubmissionFile,\n TagConverter,\n TagGroup,\n UserConverter,\n UserSpecifiedWebsiteOptions,\n WebsiteData,\n WebsiteOptions,\n} from '../models';\nimport { CustomShortcut } from '../models/custom-shortcut.entity';\nimport { Notification } from '../models/notification.entity';\n\nexport type DatabaseSchemaEntityMap = {\n AccountSchema: InstanceType;\n DirectoryWatcherSchema: InstanceType;\n FileBufferSchema: InstanceType;\n PostEventSchema: InstanceType;\n PostQueueRecordSchema: InstanceType;\n PostRecordSchema: InstanceType;\n SettingsSchema: InstanceType;\n SubmissionFileSchema: InstanceType;\n SubmissionSchema: InstanceType;\n TagConverterSchema: InstanceType;\n TagGroupSchema: InstanceType;\n UserConverterSchema: InstanceType;\n UserSpecifiedWebsiteOptionsSchema: InstanceType<\n typeof UserSpecifiedWebsiteOptions\n >;\n WebsiteDataSchema: InstanceType;\n WebsiteOptionsSchema: InstanceType;\n NotificationSchema: InstanceType;\n CustomShortcutSchema: InstanceType;\n};\n\nexport const DatabaseSchemaEntityMapConst: Record<\n SchemaKey,\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n InstanceType\n> = {\n AccountSchema: Account,\n DirectoryWatcherSchema: DirectoryWatcher,\n FileBufferSchema: FileBuffer,\n PostEventSchema: PostEvent,\n PostQueueRecordSchema: PostQueueRecord,\n PostRecordSchema: PostRecord,\n SettingsSchema: Settings,\n SubmissionFileSchema: SubmissionFile,\n SubmissionSchema: Submission,\n TagConverterSchema: TagConverter,\n TagGroupSchema: TagGroup,\n UserConverterSchema: UserConverter,\n UserSpecifiedWebsiteOptionsSchema: UserSpecifiedWebsiteOptions,\n WebsiteDataSchema: WebsiteData,\n WebsiteOptionsSchema: WebsiteOptions,\n NotificationSchema: Notification,\n CustomShortcutSchema: CustomShortcut,\n};\n" }, { "path": "apps/client-server/src/app/drizzle/transaction-context.ts", "content": "import type { PostyBirbDatabaseType, SchemaKey } from '@postybirb/database';\nimport { Schemas } from '@postybirb/database';\nimport { Logger } from '@postybirb/logger';\nimport { EntityId } from '@postybirb/types';\nimport { eq } from 'drizzle-orm';\nimport { PostyBirbDatabase } from './postybirb-database/postybirb-database';\n\ninterface TrackedEntity {\n schemaKey: SchemaKey;\n id: EntityId;\n}\n\n/**\n * A transaction-like wrapper that tracks created entities and provides\n * automatic cleanup on failure. This works around drizzle-orm's synchronous\n * transaction requirement with better-sqlite3.\n */\nexport class TransactionContext {\n private readonly logger = Logger();\n\n private readonly createdEntities: TrackedEntity[] = [];\n\n private readonly db: PostyBirbDatabaseType;\n\n constructor(db: PostyBirbDatabaseType) {\n this.db = db;\n }\n\n /**\n * Track an entity that was created during this operation.\n * If the operation fails, this entity will be deleted.\n */\n track(schemaKey: SchemaKey, id: EntityId): void {\n this.createdEntities.push({ schemaKey, id });\n }\n\n /**\n * Track multiple entities that were created during this operation.\n */\n trackMany(schemaKey: SchemaKey, ids: EntityId[]): void {\n ids.forEach((id) => this.track(schemaKey, id));\n }\n\n /**\n * Get the database instance for performing operations.\n */\n getDb(): PostyBirbDatabaseType {\n return this.db;\n }\n\n /**\n * Cleanup all tracked entities in reverse order (LIFO).\n * This is called automatically on failure.\n */\n async cleanup(): Promise {\n this.logger.warn(\n `Rolling back transaction: cleaning up ${this.createdEntities.length} entities`,\n );\n\n // Delete in reverse order (most recently created first)\n const entities = [...this.createdEntities].reverse();\n\n for (const { schemaKey, id } of entities) {\n try {\n const schema = Schemas[schemaKey];\n await this.db.delete(schema).where(eq(schema.id, id));\n this.logger.debug(`Cleaned up ${String(schemaKey)} entity: ${id}`);\n } catch (err) {\n this.logger.error(\n `Failed to cleanup ${String(schemaKey)} entity ${id}: ${err.message}`,\n err.stack,\n );\n }\n }\n }\n\n /**\n * Clear tracked entities and notify subscribers (called on successful completion).\n *\n * NOTE: Currently only tracks inserts. If update/delete tracking is needed in the future,\n * add trackUpdate() and trackDelete() methods that store the action type alongside the entity.\n */\n commit(): void {\n // Group tracked entities by schemaKey\n const bySchema = new Map();\n for (const { schemaKey, id } of this.createdEntities) {\n const existing = bySchema.get(schemaKey);\n if (existing) {\n existing.push(id);\n } else {\n bySchema.set(schemaKey, [id]);\n }\n }\n\n // Notify subscribers for each schema\n for (const [schemaKey, ids] of bySchema) {\n PostyBirbDatabase.notifySubscribers(schemaKey, ids, 'insert');\n }\n\n this.createdEntities.length = 0;\n }\n}\n\n/**\n * Execute an operation with automatic cleanup on failure.\n *\n * @example\n * ```typescript\n * const result = await withTransactionContext(\n * this.fileRepository.db,\n * async (ctx) => {\n * const entity = await createEntity(...);\n * ctx.track('SubmissionFileSchema', entity.id);\n *\n * const buffer = await createBuffer(...);\n * ctx.track('FileBufferSchema', buffer.id);\n *\n * return entity;\n * }\n * );\n * ```\n */\nexport async function withTransactionContext(\n db: PostyBirbDatabaseType,\n operation: (ctx: TransactionContext) => Promise,\n): Promise {\n const ctx = new TransactionContext(db);\n\n try {\n const result = await operation(ctx);\n ctx.commit();\n return result;\n } catch (err) {\n await ctx.cleanup();\n throw err;\n }\n}\n" }, { "path": "apps/client-server/src/app/file/file.controller.ts", "content": "import {\n BadRequestException,\n Controller,\n Get,\n Param,\n Res,\n} from '@nestjs/common';\nimport { ApiNotFoundResponse, ApiOkResponse, ApiTags } from '@nestjs/swagger';\nimport { EntityId, IFileBuffer } from '@postybirb/types';\nimport { SubmissionFile } from '../drizzle/models';\nimport { FileService } from './file.service';\n\n@ApiTags('file')\n@Controller('file')\nexport class FileController {\n constructor(private readonly service: FileService) {}\n\n @Get(':fileTarget/:id')\n @ApiOkResponse()\n @ApiNotFoundResponse()\n async getThumbnail(\n @Param('fileTarget') fileTarget: 'file' | 'thumbnail' | 'alt',\n @Param('id') id: EntityId,\n @Res() response,\n ) {\n const submissionFile = await this.service.findFile(id);\n await submissionFile.load(fileTarget);\n const imageProvidingEntity = this.getFileBufferForTarget(\n fileTarget,\n submissionFile,\n );\n if (!imageProvidingEntity) {\n throw new BadRequestException(`No ${fileTarget} found for file ${id}`);\n }\n response.contentType(imageProvidingEntity.mimeType);\n response.send(imageProvidingEntity.buffer);\n }\n\n private getFileBufferForTarget(\n fileTarget: 'file' | 'thumbnail' | 'alt',\n submissionFile: SubmissionFile,\n ): IFileBuffer | undefined {\n switch (fileTarget) {\n case 'file':\n return submissionFile.file;\n case 'thumbnail':\n return submissionFile.thumbnail;\n case 'alt':\n return submissionFile.altFile;\n default:\n throw new BadRequestException('Invalid file target');\n }\n }\n}\n" }, { "path": "apps/client-server/src/app/file/file.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { ImageProcessingModule } from '../image-processing/image-processing.module';\nimport { FileController } from './file.controller';\nimport { FileService } from './file.service';\nimport { CreateFileService } from './services/create-file.service';\nimport { UpdateFileService } from './services/update-file.service';\n\n@Module({\n imports: [ImageProcessingModule],\n controllers: [FileController],\n providers: [FileService, CreateFileService, UpdateFileService],\n exports: [FileService],\n})\nexport class FileModule {}\n" }, { "path": "apps/client-server/src/app/file/file.service.spec.ts", "content": "import { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport { PostyBirbDirectories, writeSync } from '@postybirb/fs';\nimport { FileSubmission, SubmissionType } from '@postybirb/types';\nimport { readFileSync } from 'fs';\nimport { join } from 'path';\nimport { AccountService } from '../account/account.service';\nimport { CustomShortcutsService } from '../custom-shortcuts/custom-shortcuts.service';\nimport { SubmissionFile } from '../drizzle/models';\nimport { PostyBirbDatabase } from '../drizzle/postybirb-database/postybirb-database';\nimport { FileConverterService } from '../file-converter/file-converter.service';\nimport { FormGeneratorService } from '../form-generator/form-generator.service';\nimport { DescriptionParserService } from '../post-parsers/parsers/description-parser.service';\nimport { TagParserService } from '../post-parsers/parsers/tag-parser.service';\nimport { TitleParser } from '../post-parsers/parsers/title-parser';\nimport { PostParsersService } from '../post-parsers/post-parsers.service';\nimport { SettingsService } from '../settings/settings.service';\nimport { CreateSubmissionDto } from '../submission/dtos/create-submission.dto';\nimport { FileSubmissionService } from '../submission/services/file-submission.service';\nimport { MessageSubmissionService } from '../submission/services/message-submission.service';\nimport { SubmissionService } from '../submission/services/submission.service';\nimport { TagConvertersService } from '../tag-converters/tag-converters.service';\nimport { UserConvertersService } from '../user-converters/user-converters.service';\nimport { UserSpecifiedWebsiteOptionsService } from '../user-specified-website-options/user-specified-website-options.service';\nimport { ValidationService } from '../validation/validation.service';\nimport { WebsiteOptionsService } from '../website-options/website-options.service';\nimport { WebsiteImplProvider } from '../websites/implementations/provider';\nimport { WebsiteRegistryService } from '../websites/website-registry.service';\nimport { FileService } from './file.service';\nimport { MulterFileInfo } from './models/multer-file-info';\nimport { CreateFileService } from './services/create-file.service';\nimport { UpdateFileService } from './services/update-file.service';\nimport { SharpInstanceManager } from '../image-processing/sharp-instance-manager';\n\ndescribe('FileService', () => {\n let testFile: Buffer | null = null;\n let testFile2: Buffer | null = null;\n let service: FileService;\n let submissionService: SubmissionService;\n let module: TestingModule;\n let fileBufferRepository: PostyBirbDatabase<'FileBufferSchema'>;\n\n async function createSubmission() {\n const dto = new CreateSubmissionDto();\n dto.name = 'test';\n dto.type = SubmissionType.MESSAGE; // Use message submission just for the sake of insertion\n\n const record = await submissionService.create(dto);\n return record;\n }\n\n function createMulterData(path: string): MulterFileInfo {\n return {\n fieldname: 'file',\n originalname: 'small_image.jpg',\n encoding: '',\n mimetype: 'image/jpeg',\n size: testFile.length,\n destination: '',\n filename: 'small_image.jpg',\n path,\n origin: undefined,\n };\n }\n\n function createMulterData2(path: string): MulterFileInfo {\n return {\n fieldname: 'file',\n originalname: 'png_with_alpha.png',\n encoding: '',\n mimetype: 'image/png',\n size: testFile2.length,\n destination: '',\n filename: 'png_with_alpha.jpg',\n path,\n origin: undefined,\n };\n }\n\n function setup(): string[] {\n const path = `${PostyBirbDirectories.DATA_DIRECTORY}/${Date.now()}.jpg`;\n const path2 = `${PostyBirbDirectories.DATA_DIRECTORY}/${Date.now()}.png`;\n\n writeSync(path, testFile);\n writeSync(path2, testFile2);\n return [path, path2];\n }\n\n beforeAll(() => {\n testFile = readFileSync(\n join(__dirname, '../../test-files/small_image.jpg'),\n );\n testFile2 = readFileSync(\n join(__dirname, '../../test-files/png_with_alpha.png'),\n );\n });\n\n beforeEach(async () => {\n clearDatabase();\n module = await Test.createTestingModule({\n providers: [\n UserSpecifiedWebsiteOptionsService,\n SubmissionService,\n CreateFileService,\n UpdateFileService,\n SharpInstanceManager,\n FileService,\n ValidationService,\n SubmissionService,\n FileSubmissionService,\n MessageSubmissionService,\n AccountService,\n WebsiteRegistryService,\n WebsiteOptionsService,\n WebsiteImplProvider,\n PostParsersService,\n TagParserService,\n DescriptionParserService,\n TitleParser,\n TagConvertersService,\n SettingsService,\n FormGeneratorService,\n FileConverterService,\n CustomShortcutsService,\n UserConvertersService,\n ],\n }).compile();\n fileBufferRepository = new PostyBirbDatabase('FileBufferSchema');\n service = module.get(FileService);\n submissionService = module.get(SubmissionService);\n\n const accountService = module.get(AccountService);\n await accountService.onModuleInit();\n });\n\n async function loadBuffers(rec: SubmissionFile) {\n // !bug - https://github.com/drizzle-team/drizzle-orm/issues/3497\n // eslint-disable-next-line no-param-reassign\n rec.file = await fileBufferRepository.findById(rec.primaryFileId);\n // eslint-disable-next-line no-param-reassign\n rec.thumbnail = rec.thumbnailId\n ? await fileBufferRepository.findById(rec.thumbnailId)\n : undefined;\n // eslint-disable-next-line no-param-reassign\n rec.altFile = rec.altFileId\n ? await fileBufferRepository.findById(rec.altFileId)\n : undefined;\n }\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n\n it('should create submission file', async () => {\n const path = setup();\n const submission = await createSubmission();\n const fileInfo = createMulterData(path[0]);\n const file = await service.create(\n fileInfo,\n submission as unknown as FileSubmission,\n );\n await loadBuffers(file);\n expect(file.file).toBeDefined();\n expect(file.thumbnail).toBeDefined();\n expect(file.thumbnail.fileName.startsWith('thumbnail_')).toBe(true);\n expect(file.fileName).toBe(fileInfo.originalname);\n expect(file.size).toBe(fileInfo.size);\n expect(file.hasThumbnail).toBe(true);\n expect(file.hasCustomThumbnail).toBe(false);\n expect(file.height).toBe(202);\n expect(file.width).toBe(138);\n expect(file.file.size).toBe(fileInfo.size);\n expect(file.file.height).toBe(202);\n expect(file.file.width).toBe(138);\n expect(file.file.submissionFileId).toEqual(file.id);\n expect(file.file.mimeType).toEqual(fileInfo.mimetype);\n expect(file.file.buffer).toEqual(testFile);\n });\n\n it('should not update submission file when hash is same', async () => {\n const path = setup();\n const submission = await createSubmission();\n const fileInfo = createMulterData(path[0]);\n const file = await service.create(\n fileInfo,\n submission as unknown as FileSubmission,\n );\n await loadBuffers(file);\n expect(file.file).toBeDefined();\n\n const path2 = setup();\n const updateFileInfo: MulterFileInfo = {\n fieldname: 'file',\n originalname: 'small_image.jpg',\n encoding: '',\n mimetype: 'image/png',\n size: testFile.length,\n destination: '',\n filename: 'small_image.jpg',\n path: path2[0],\n origin: undefined,\n };\n const updatedFile = await service.update(updateFileInfo, file.id, false);\n await loadBuffers(updatedFile);\n expect(updatedFile.file).toBeDefined();\n expect(updatedFile.thumbnail).toBeDefined();\n expect(updatedFile.fileName).toBe(updateFileInfo.originalname);\n expect(updatedFile.size).toBe(updateFileInfo.size);\n expect(updatedFile.hasThumbnail).toBe(true);\n expect(updatedFile.hasCustomThumbnail).toBe(false);\n expect(updatedFile.height).toBe(202);\n expect(updatedFile.width).toBe(138);\n expect(updatedFile.file.size).toBe(updateFileInfo.size);\n expect(updatedFile.file.height).toBe(202);\n expect(updatedFile.file.width).toBe(138);\n expect(updatedFile.file.submissionFileId).toEqual(file.id);\n expect(updatedFile.file.mimeType).not.toEqual(updateFileInfo.mimetype);\n expect(updatedFile.file.buffer).toEqual(testFile);\n });\n\n it('should update submission primary file', async () => {\n const path = setup();\n const submission = await createSubmission();\n const fileInfo = createMulterData(path[0]);\n const file = await service.create(\n fileInfo,\n submission as unknown as FileSubmission,\n );\n await loadBuffers(file);\n expect(file.file).toBeDefined();\n\n const path2 = setup();\n const updateFileInfo = createMulterData2(path2[1]);\n const updatedFile = await service.update(updateFileInfo, file.id, false);\n await loadBuffers(updatedFile);\n expect(updatedFile.file).toBeDefined();\n expect(updatedFile.thumbnail).toBeDefined();\n expect(updatedFile.fileName).toBe(updateFileInfo.filename);\n expect(updatedFile.size).toBe(updateFileInfo.size);\n expect(updatedFile.hasThumbnail).toBe(true);\n expect(updatedFile.hasCustomThumbnail).toBe(false);\n expect(updatedFile.height).toBe(600);\n expect(updatedFile.width).toBe(600);\n expect(updatedFile.file.size).toBe(updateFileInfo.size);\n expect(updatedFile.file.height).toBe(600);\n expect(updatedFile.file.width).toBe(600);\n expect(updatedFile.file.submissionFileId).toEqual(file.id);\n expect(updatedFile.file.mimeType).toEqual(updateFileInfo.mimetype);\n expect(updatedFile.file.buffer).toEqual(testFile2);\n });\n\n it('should cleanup entities on transaction failure', async () => {\n const path = setup();\n const submission = await createSubmission();\n const fileInfo = createMulterData(path[0]);\n\n // Get initial count of entities\n const initialFiles = await new PostyBirbDatabase(\n 'SubmissionFileSchema',\n ).findAll();\n const initialBuffers = await fileBufferRepository.findAll();\n const initialFileCount = initialFiles.length;\n const initialBufferCount = initialBuffers.length;\n\n // Mock a method to throw an error partway through creation\n const createFileService = module.get(CreateFileService);\n const originalMethod = createFileService.createFileBufferEntity;\n let callCount = 0;\n jest\n .spyOn(createFileService, 'createFileBufferEntity')\n .mockImplementation(async (...args) => {\n callCount++;\n // Fail on the second buffer creation (thumbnail)\n if (callCount === 2) {\n throw new Error('Simulated error during buffer creation');\n }\n return originalMethod.apply(createFileService, args);\n });\n\n // Attempt to create a file, which should fail and trigger cleanup\n await expect(\n service.create(fileInfo, submission as unknown as FileSubmission),\n ).rejects.toThrow('Simulated error during buffer creation');\n\n // Verify that entities were cleaned up\n const finalFiles = await new PostyBirbDatabase(\n 'SubmissionFileSchema',\n ).findAll();\n const finalBuffers = await fileBufferRepository.findAll();\n\n expect(finalFiles.length).toBe(initialFileCount);\n expect(finalBuffers.length).toBe(initialBufferCount);\n\n // Restore the original method\n jest.restoreAllMocks();\n });\n});\n" }, { "path": "apps/client-server/src/app/file/file.service.ts", "content": "/* eslint-disable no-param-reassign */\nimport { BadRequestException, Injectable } from '@nestjs/common';\nimport { read } from '@postybirb/fs';\nimport { Logger } from '@postybirb/logger';\nimport {\n EntityId,\n FileSubmission,\n SubmissionFileMetadata,\n} from '@postybirb/types';\nimport type { queueAsPromised } from 'fastq';\nimport fastq from 'fastq';\nimport { readFile } from 'fs/promises';\nimport { cpus } from 'os';\nimport { SubmissionFile } from '../drizzle/models';\nimport { PostyBirbDatabase } from '../drizzle/postybirb-database/postybirb-database';\nimport { ReorderSubmissionFilesDto } from '../submission/dtos/reorder-submission-files.dto';\nimport { UpdateAltFileDto } from '../submission/dtos/update-alt-file.dto';\nimport { MulterFileInfo, TaskOrigin } from './models/multer-file-info';\nimport { CreateTask, Task, UpdateTask } from './models/task';\nimport { TaskType } from './models/task-type.enum';\nimport { CreateFileService } from './services/create-file.service';\nimport { UpdateFileService } from './services/update-file.service';\n\n/**\n * Service that handles storing file data into database.\n * @todo text encoding parsing and file name conversion (no periods)\n */\n@Injectable()\nexport class FileService {\n private readonly logger = Logger();\n\n private readonly queue: queueAsPromised = fastq.promise<\n this,\n Task\n >(this, this.doTask, Math.min(cpus().length, 5));\n\n private readonly fileBufferRepository = new PostyBirbDatabase(\n 'FileBufferSchema',\n );\n\n private readonly fileRepository = new PostyBirbDatabase(\n 'SubmissionFileSchema',\n );\n\n constructor(\n private readonly createFileService: CreateFileService,\n private readonly updateFileService: UpdateFileService,\n ) {}\n\n /**\n * Deletes a file.\n *\n * @param {EntityId} id\n * @return {*}\n */\n public async remove(id: EntityId) {\n this.logger.info(id, `Removing entity '${id}'`);\n return this.fileRepository.deleteById([id]);\n }\n\n /**\n * Queues a file to create a database record.\n *\n * @param {MulterFileInfo} file\n * @param {FileSubmission} submission\n * @return {*} {Promise}\n */\n public async create(\n file: MulterFileInfo,\n submission: FileSubmission,\n ): Promise {\n return this.queue.push({ type: TaskType.CREATE, file, submission });\n }\n\n /**\n * Queues a file to update.\n *\n * @param {MulterFileInfo} file\n * @param {EntityId} submissionFileId\n * @param {boolean} forThumbnail\n * @return {*} {Promise}\n */\n public async update(\n file: MulterFileInfo,\n submissionFileId: EntityId,\n forThumbnail: boolean,\n ): Promise {\n return this.queue.push({\n type: TaskType.UPDATE,\n file,\n submissionFileId,\n target: forThumbnail ? 'thumbnail' : undefined,\n });\n }\n\n private async doTask(task: Task): Promise {\n task.file.originalname = this.sanitizeFilename(task.file.originalname);\n const buf: Buffer = await this.getFile(task.file.path, task.file.origin);\n this.logger.withMetadata(task).info('Reading File');\n switch (task.type) {\n case TaskType.CREATE:\n // eslint-disable-next-line no-case-declarations\n const ct = task as CreateTask;\n return this.createFileService.create(ct.file, ct.submission, buf);\n case TaskType.UPDATE:\n // eslint-disable-next-line no-case-declarations\n const ut = task as UpdateTask;\n return this.updateFileService.update(\n ut.file,\n ut.submissionFileId,\n buf,\n ut.target,\n );\n default:\n throw new BadRequestException(`Unknown TaskType '${task.type}'`);\n }\n }\n\n /**\n * Removes periods from the filename.\n * There is some website that doesn't like them.\n *\n * @param {string} filename\n * @return {*} {string}\n */\n private sanitizeFilename(filename: string): string {\n const nameParts = filename.split('.');\n const ext = nameParts.pop();\n return `${nameParts.join('_')}.${ext}`;\n }\n\n private async getFile(path: string, taskOrigin: TaskOrigin): Promise {\n switch (taskOrigin) {\n case 'directory-watcher':\n return readFile(path);\n default:\n // Approved location\n return read(path);\n }\n }\n\n /**\n * Returns file by Id.\n *\n * @param {EntityId} id\n */\n public async findFile(id: EntityId): Promise {\n return this.fileRepository.findById(id, { failOnMissing: true });\n }\n\n /**\n * Gets the size of an alt text file without loading the buffer.\n * @param {EntityId} id\n */\n async getAltFileSize(id: EntityId): Promise {\n const altFile = await this.fileBufferRepository.findById(id, {\n failOnMissing: false,\n });\n return altFile?.size ?? 0;\n }\n\n /**\n * Gets the raw text of an alt text file.\n * @param {EntityId} id\n */\n async getAltText(id: EntityId): Promise {\n const altFile = await this.fileBufferRepository.findById(id, {\n failOnMissing: true,\n });\n if (altFile.size) {\n return altFile.buffer.toString();\n }\n\n return '';\n }\n\n /**\n * Updates the raw text of an alt text file.\n * @param {EntityId} id\n * @param {UpdateAltFileDto} update\n */\n async updateAltText(id: EntityId, update: UpdateAltFileDto) {\n const buffer = Buffer.from(update.text ?? '');\n return this.fileBufferRepository.update(id, {\n buffer,\n mimeType: 'text/plain',\n size: buffer.length,\n });\n }\n\n async updateMetadata(id: string, update: SubmissionFileMetadata) {\n const file = await this.findFile(id);\n const merged = { ...file.metadata, ...update };\n await this.fileRepository.update(id, { metadata: merged });\n }\n\n async reorderFiles(update: ReorderSubmissionFilesDto) {\n await Promise.all(\n Object.entries(update.order).map(([id, order]) =>\n this.fileRepository.update(id, { order }),\n ),\n );\n }\n}\n" }, { "path": "apps/client-server/src/app/file/models/multer-file-info.ts", "content": "/**\n * Matches a multer file info object.\n * TypeScript is not importing multer types correctly.\n *\n * @interface MulterFileInfo\n */\nexport interface MulterFileInfo {\n fieldname: string;\n originalname: string;\n encoding: string;\n mimetype: string;\n size: number;\n destination: string;\n filename: string;\n path: string;\n buffer?: Buffer;\n /**\n * Internal origin, empty when external.\n * @type {TaskOrigin}\n */\n origin?: TaskOrigin;\n}\n\nexport type TaskOrigin = 'directory-watcher';\n" }, { "path": "apps/client-server/src/app/file/models/task-type.enum.ts", "content": "/**\n * Defines the requested TaskType\n */\nexport enum TaskType {\n CREATE = 'CREATE', // Creating a new file entity\n UPDATE = 'UPDATE', // Updating a file entity\n}\n" }, { "path": "apps/client-server/src/app/file/models/task.ts", "content": "import { EntityId, FileSubmission } from '@postybirb/types';\nimport { MulterFileInfo } from './multer-file-info';\nimport { TaskType } from './task-type.enum';\n\n// Task Type\nexport type Task = CreateTask | UpdateTask;\n\n// Defines CreateTask Params\nexport type CreateTask = {\n type: TaskType;\n /**\n * File for use.\n * @type {MulterFileInfo}\n */\n file: MulterFileInfo;\n /**\n * Submission the entities will be attached to.\n * @type {FileSubmission}\n */\n submission: FileSubmission;\n};\n\n// Defines UpdateTask Params\nexport type UpdateTask = {\n type: TaskType;\n /**\n * Updating file.\n *\n * @type {MulterFileInfo}\n */\n file: MulterFileInfo;\n /**\n * SubmissionFile being updated.\n * @type {EntityId}\n */\n submissionFileId: EntityId;\n /**\n * The target type being updates (primary on empty).\n * @type {string}\n */\n target?: 'thumbnail';\n};\n" }, { "path": "apps/client-server/src/app/file/services/create-file.service.ts", "content": "import * as rtf from '@iarna/rtf-to-html';\nimport { Injectable } from '@nestjs/common';\nimport { Insert, Select } from '@postybirb/database';\nimport { removeFile } from '@postybirb/fs';\nimport { Logger } from '@postybirb/logger';\nimport {\n DefaultSubmissionFileMetadata,\n FileSubmission,\n FileType,\n IFileBuffer,\n} from '@postybirb/types';\nimport { getFileType } from '@postybirb/utils/file-type';\nimport { eq } from 'drizzle-orm';\nimport { async as hash } from 'hasha';\nimport { htmlToText } from 'html-to-text';\nimport * as mammoth from 'mammoth';\nimport { parse } from 'path';\nimport { promisify } from 'util';\nimport { v4 as uuid } from 'uuid';\n\nimport {\n FileBuffer,\n fromDatabaseRecord,\n SubmissionFile,\n} from '../../drizzle/models';\nimport { PostyBirbDatabase } from '../../drizzle/postybirb-database/postybirb-database';\nimport {\n TransactionContext,\n withTransactionContext,\n} from '../../drizzle/transaction-context';\nimport { SharpInstanceManager } from '../../image-processing/sharp-instance-manager';\nimport { MulterFileInfo } from '../models/multer-file-info';\nimport { ImageUtil } from '../utils/image.util';\n\n/**\n * A Service that defines operations for creating a SubmissionFile.\n * @class CreateFileService\n */\n@Injectable()\nexport class CreateFileService {\n private readonly logger = Logger();\n\n private readonly fileBufferRepository = new PostyBirbDatabase(\n 'FileBufferSchema',\n );\n\n private readonly fileRepository = new PostyBirbDatabase(\n 'SubmissionFileSchema',\n );\n\n constructor(\n private readonly sharpInstanceManager: SharpInstanceManager,\n ) {}\n\n /**\n * Creates file entity and stores it.\n * @todo extra data (image resize per website)\n * @todo figure out what to do about non-image\n *\n * @param {MulterFileInfo} file\n * @param {MulterFileInfo} submission\n * @param {Buffer} buf\n * @return {*} {Promise}\n */\n public async create(\n file: MulterFileInfo,\n submission: FileSubmission,\n buf: Buffer,\n ): Promise {\n try {\n this.logger.withMetadata(file).info(`Creating SubmissionFile entity`);\n\n const newSubmission = await withTransactionContext(\n this.fileRepository.db,\n async (ctx) => {\n let entity = await this.createSubmissionFile(\n ctx,\n file,\n submission,\n buf,\n );\n\n if (ImageUtil.isImage(file.mimetype, true)) {\n this.logger.info('[Mutation] Populating as Image');\n entity = await this.populateAsImageFile(ctx, entity, file, buf);\n }\n\n if (getFileType(file.originalname) === FileType.TEXT) {\n await this.createSubmissionTextAltFile(ctx, entity, file, buf);\n }\n\n const primaryFile = await this.createFileBufferEntity(\n ctx,\n entity,\n buf,\n );\n await ctx\n .getDb()\n .update(this.fileRepository.schemaEntity)\n .set({ primaryFileId: primaryFile.id })\n .where(eq(this.fileRepository.schemaEntity.id, entity.id));\n this.logger\n .withMetadata({ id: entity.id })\n .info('SubmissionFile Created');\n\n return entity;\n },\n );\n return await this.fileRepository.findById(newSubmission.id);\n } catch (err) {\n this.logger.error(err.message, err.stack);\n throw err;\n } finally {\n if (!file.origin) {\n removeFile(file.path);\n }\n }\n }\n\n /**\n * Populates an alt file containing text data extracted from a file.\n * Currently supports docx, rtf, and plaintext.\n *\n * @param {SubmissionFile} entity\n * @param {MulterFileInfo} file\n * @param {Buffer} buf\n */\n async createSubmissionTextAltFile(\n ctx: TransactionContext,\n entity: SubmissionFile,\n file: MulterFileInfo,\n buf: Buffer,\n ) {\n // Default to empty string - all TEXT files get an alt file\n let altText = '';\n\n if (\n file.mimetype ===\n 'application/vnd.openxmlformats-officedocument.wordprocessingml.document' ||\n file.mimetype === 'application/msword' ||\n file.originalname.endsWith('.docx') ||\n file.originalname.endsWith('.doc')\n ) {\n this.logger.info('[Mutation] Creating Alt File for Text Document: DOCX');\n altText = (await mammoth.extractRawText({ buffer: buf })).value;\n } else if (\n file.mimetype === 'application/rtf' ||\n file.originalname.endsWith('.rtf')\n ) {\n this.logger.info('[Mutation] Creating Alt File for Text Document: RTF');\n const promisifiedRtf = promisify(rtf.fromString);\n const rtfHtml = await promisifiedRtf(buf.toString(), {\n template(_, __, content: string) {\n return content;\n },\n });\n altText = htmlToText(rtfHtml, { wordwrap: false });\n } else if (\n file.mimetype === 'text/plain' ||\n file.originalname.endsWith('.txt')\n ) {\n this.logger.info('[Mutation] Creating Alt File for Text Document: TXT');\n altText = buf.toString();\n } else {\n this.logger.info(\n `[Mutation] Creating empty Alt File for unsupported text format: ${file.mimetype}`,\n );\n }\n\n const prettifiedBuf = Buffer.from(altText ?? '');\n const altFile = await this.createFileBufferEntity(\n ctx,\n entity,\n prettifiedBuf,\n {\n mimeType: 'text/plain',\n fileName: `${entity.fileName}.txt`,\n },\n );\n await ctx\n .getDb()\n .update(this.fileRepository.schemaEntity)\n .set({\n altFileId: altFile.id,\n hasAltFile: true,\n })\n .where(eq(this.fileRepository.schemaEntity.id, entity.id));\n this.logger.withMetadata({ id: altFile.id }).info('Alt File Created');\n }\n\n /**\n * Creates a SubmissionFile with pre-populated fields.\n *\n * @param {MulterFileInfo} file\n * @param {FileSubmission} submission\n * @param {Buffer} buf\n * @return {*} {Promise}\n */\n private async createSubmissionFile(\n ctx: TransactionContext,\n file: MulterFileInfo,\n submission: FileSubmission,\n buf: Buffer,\n ): Promise {\n const { mimetype: mimeType, originalname, size } = file;\n const submissionFile: Insert<'SubmissionFileSchema'> = {\n submissionId: submission.id,\n mimeType,\n fileName: originalname,\n size,\n hash: await hash(buf, { algorithm: 'sha256' }),\n width: 0,\n height: 0,\n hasThumbnail: false,\n metadata: DefaultSubmissionFileMetadata(),\n order: Date.now(),\n };\n const sf = fromDatabaseRecord(\n SubmissionFile,\n await ctx\n .getDb()\n .insert(this.fileRepository.schemaEntity)\n .values(submissionFile)\n .returning(),\n );\n\n const entity = sf[0];\n ctx.track('SubmissionFileSchema', entity.id);\n return entity;\n }\n\n /**\n * Populates SubmissionFile with Image specific fields.\n * Width, Height, Thumbnail.\n *\n * @param {SubmissionFile} entity\n * @param {MulterFileInfo} file\n * @param {Buffer} buf\n * @return {*} {Promise}\n */\n private async populateAsImageFile(\n ctx: TransactionContext,\n entity: SubmissionFile,\n file: MulterFileInfo,\n buf: Buffer,\n ): Promise {\n const meta = await this.sharpInstanceManager.getMetadata(buf);\n const thumbnail = await this.createFileThumbnail(\n ctx,\n entity,\n file,\n buf,\n );\n const update: Select = {\n width: meta.width ?? 0,\n height: meta.height ?? 0,\n hasThumbnail: true,\n thumbnailId: thumbnail.id,\n metadata: {\n ...entity.metadata,\n dimensions: {\n default: {\n width: meta.width ?? 0,\n height: meta.height ?? 0,\n },\n },\n },\n };\n\n return fromDatabaseRecord(\n SubmissionFile,\n await ctx\n .getDb()\n .update(this.fileRepository.schemaEntity)\n .set(update)\n .where(eq(this.fileRepository.schemaEntity.id, entity.id))\n .returning(),\n )[0];\n }\n\n /**\n * Returns a thumbnail entity for a file.\n *\n * @param {SubmissionFile} fileEntity\n * @param {MulterFileInfo} file\n * @param {Buffer} imageBuffer - The source image buffer\n * @return {*} {Promise}\n */\n public async createFileThumbnail(\n ctx: TransactionContext,\n fileEntity: SubmissionFile,\n file: MulterFileInfo,\n imageBuffer: Buffer,\n ): Promise {\n const {\n buffer: thumbnailBuf,\n height,\n width,\n mimeType: thumbnailMimeType,\n } = await this.generateThumbnail(\n imageBuffer,\n file.mimetype,\n );\n\n // Remove existing extension and add the appropriate thumbnail extension\n const fileNameWithoutExt = parse(fileEntity.fileName).name;\n const thumbnailExt = thumbnailMimeType === 'image/jpeg' ? 'jpg' : 'png';\n\n return this.createFileBufferEntity(ctx, fileEntity, thumbnailBuf, {\n height,\n width,\n mimeType: thumbnailMimeType,\n fileName: `thumbnail_${fileNameWithoutExt}.${thumbnailExt}`,\n });\n }\n\n /**\n * Generates a thumbnail for display at specific dimension requirements.\n * Delegates to the sharp worker pool for crash isolation.\n *\n * @param {Buffer} imageBuffer - The source image buffer\n * @param {string} sourceMimeType - The mimetype of the source image\n * @param {number} [preferredDimension=400] - The preferred thumbnail dimension\n * @return {*} {Promise<{ width: number; height: number; buffer: Buffer; mimeType: string }>}\n */\n public async generateThumbnail(\n imageBuffer: Buffer,\n sourceMimeType: string,\n preferredDimension = 400,\n ): Promise<{\n width: number;\n height: number;\n buffer: Buffer;\n mimeType: string;\n }> {\n const result = await this.sharpInstanceManager.generateThumbnail(\n imageBuffer,\n sourceMimeType,\n 'thumbnail',\n preferredDimension,\n );\n\n return {\n buffer: result.buffer,\n height: result.height,\n width: result.width,\n mimeType: result.mimeType,\n };\n }\n\n /**\n * Creates a file buffer entity for storing blob data of a file.\n *\n * @param {File} fileEntity\n * @param {Buffer} buf\n * @param {string} type - thumbnail/alt/primary\n * @return {*} {IFileBuffer}\n */\n public async createFileBufferEntity(\n ctx: TransactionContext,\n fileEntity: SubmissionFile,\n buf: Buffer,\n opts: Select<'FileBufferSchema'> = {} as Select<'FileBufferSchema'>,\n ): Promise {\n const { mimeType, height, width, fileName } = fileEntity;\n const data: Insert<'FileBufferSchema'> = {\n id: uuid(),\n buffer: buf,\n submissionFileId: fileEntity.id,\n height,\n width,\n fileName,\n mimeType,\n size: buf.length,\n ...opts,\n };\n\n const result = fromDatabaseRecord(\n FileBuffer,\n await ctx\n .getDb()\n .insert(this.fileBufferRepository.schemaEntity)\n .values(data)\n .returning(),\n )[0];\n\n ctx.track('FileBufferSchema', result.id);\n return result;\n }\n}\n" }, { "path": "apps/client-server/src/app/file/services/update-file.service.ts", "content": "/* eslint-disable no-param-reassign */\nimport * as rtf from '@iarna/rtf-to-html';\nimport {\n BadRequestException,\n Injectable,\n NotFoundException,\n} from '@nestjs/common';\nimport { Logger } from '@postybirb/logger';\nimport { EntityId, FileType } from '@postybirb/types';\nimport { getFileType } from '@postybirb/utils/file-type';\nimport { eq } from 'drizzle-orm';\nimport { async as hash } from 'hasha';\nimport { htmlToText } from 'html-to-text';\nimport * as mammoth from 'mammoth';\nimport { parse } from 'path';\nimport { promisify } from 'util';\nimport { SubmissionFile } from '../../drizzle/models';\nimport { PostyBirbDatabase } from '../../drizzle/postybirb-database/postybirb-database';\nimport {\n TransactionContext,\n withTransactionContext,\n} from '../../drizzle/transaction-context';\nimport { SharpInstanceManager } from '../../image-processing/sharp-instance-manager';\nimport { MulterFileInfo } from '../models/multer-file-info';\nimport { ImageUtil } from '../utils/image.util';\nimport { CreateFileService } from './create-file.service';\n\n/**\n * A Service for updating existing SubmissionFile entities.\n */\n@Injectable()\nexport class UpdateFileService {\n private readonly logger = Logger();\n\n private readonly fileRepository = new PostyBirbDatabase(\n 'SubmissionFileSchema',\n );\n\n private readonly fileBufferRepository = new PostyBirbDatabase(\n 'FileBufferSchema',\n );\n\n constructor(\n private readonly createFileService: CreateFileService,\n private readonly sharpInstanceManager: SharpInstanceManager,\n ) {}\n\n /**\n * Creates file entity and stores it.\n *\n * @param {MulterFileInfo} file\n * @param {MulterFileInfo} submission\n * @param {Buffer} buf\n * @param {string} target\n * @return {*} {Promise}\n */\n public async update(\n file: MulterFileInfo,\n submissionFileId: EntityId,\n buf: Buffer,\n target?: 'thumbnail',\n ): Promise {\n const submissionFile = await this.findFile(submissionFileId);\n\n await withTransactionContext(this.fileRepository.db, async (ctx) => {\n if (target === 'thumbnail') {\n await this.replaceFileThumbnail(ctx, submissionFile, file, buf);\n } else {\n await this.replacePrimaryFile(ctx, submissionFile, file, buf);\n }\n });\n\n // Notify subscribers so SubmissionService emits a websocket update\n this.fileRepository.forceNotify([submissionFileId], 'update');\n this.fileBufferRepository.forceNotify([submissionFileId], 'update');\n\n // return the latest\n return this.findFile(submissionFileId);\n }\n\n private async replaceFileThumbnail(\n ctx: TransactionContext,\n submissionFile: SubmissionFile,\n file: MulterFileInfo,\n buf: Buffer,\n ) {\n const thumbnailDetails = await this.getImageDetails(file, buf);\n let { thumbnailId } = submissionFile;\n\n if (!thumbnailId) {\n // Create a new thumbnail buffer entity\n const thumbnail = await this.createFileService.createFileBufferEntity(\n ctx,\n submissionFile,\n thumbnailDetails.buffer,\n {\n width: thumbnailDetails.width,\n height: thumbnailDetails.height,\n mimeType: file.mimetype,\n },\n );\n thumbnailId = thumbnail.id;\n } else {\n // Update existing thumbnail buffer\n await ctx\n .getDb()\n .update(this.fileBufferRepository.schemaEntity)\n .set({\n buffer: thumbnailDetails.buffer,\n size: thumbnailDetails.buffer.length,\n mimeType: file.mimetype,\n width: thumbnailDetails.width,\n height: thumbnailDetails.height,\n })\n .where(eq(this.fileBufferRepository.schemaEntity.id, thumbnailId));\n }\n\n // Recompute hash from thumbnail buffer so the frontend cache-buster updates\n const thumbnailHash = await hash(thumbnailDetails.buffer, {\n algorithm: 'sha256',\n });\n\n await ctx\n .getDb()\n .update(this.fileRepository.schemaEntity)\n .set({\n thumbnailId,\n hasCustomThumbnail: true,\n hasThumbnail: true,\n hash: thumbnailHash,\n })\n .where(eq(this.fileRepository.schemaEntity.id, submissionFile.id));\n }\n\n async replacePrimaryFile(\n ctx: TransactionContext,\n submissionFile: SubmissionFile,\n file: MulterFileInfo,\n buf: Buffer,\n ) {\n return this.updateFileEntity(ctx, submissionFile, file, buf);\n }\n\n private async updateFileEntity(\n ctx: TransactionContext,\n submissionFile: SubmissionFile,\n file: MulterFileInfo,\n buf: Buffer,\n ) {\n const fileHash = await hash(buf, { algorithm: 'sha256' });\n\n // Only need to replace when unique file is given\n if (submissionFile.hash !== fileHash) {\n const fileType = getFileType(file.filename);\n if (fileType === FileType.IMAGE) {\n await this.updateImageFileProps(ctx, submissionFile, file, buf);\n }\n\n // Update submission file entity\n await ctx\n .getDb()\n .update(this.fileRepository.schemaEntity)\n .set({\n hash: fileHash,\n size: buf.length,\n fileName: file.filename,\n mimeType: file.mimetype,\n })\n .where(eq(this.fileRepository.schemaEntity.id, submissionFile.id));\n\n // Just to get the latest data\n\n // Duplicate props to primary file\n await ctx\n .getDb()\n .update(this.fileBufferRepository.schemaEntity)\n .set({\n buffer: buf,\n size: buf.length,\n fileName: file.filename,\n mimeType: file.mimetype,\n })\n .where(\n eq(\n this.fileBufferRepository.schemaEntity.id,\n submissionFile.primaryFileId,\n ),\n );\n\n if (\n getFileType(file.originalname) === FileType.TEXT &&\n submissionFile.hasAltFile\n ) {\n const altFileText =\n (await this.repopulateTextFile(file, buf)) ||\n submissionFile?.altFile?.buffer;\n if (altFileText) {\n await ctx\n .getDb()\n .update(this.fileBufferRepository.schemaEntity)\n .set({\n buffer: altFileText,\n size: altFileText.length,\n })\n .where(\n eq(\n this.fileBufferRepository.schemaEntity.id,\n submissionFile.altFile.id,\n ),\n );\n await ctx\n .getDb()\n .update(this.fileRepository.schemaEntity)\n .set({\n hasAltFile: true,\n })\n .where(eq(this.fileRepository.schemaEntity.id, submissionFile.id));\n }\n }\n }\n }\n\n async repopulateTextFile(\n file: MulterFileInfo,\n buf: Buffer,\n ): Promise {\n let altText: string;\n if (\n file.mimetype ===\n 'application/vnd.openxmlformats-officedocument.wordprocessingml.document' ||\n file.originalname.endsWith('.docx')\n ) {\n this.logger.info('[Mutation] Updating Alt File for Text Document: DOCX');\n altText = (await mammoth.extractRawText({ buffer: buf })).value;\n }\n\n if (\n file.mimetype === 'application/rtf' ||\n file.originalname.endsWith('.rtf')\n ) {\n this.logger.info('[Mutation] Updating Alt File for Text Document: RTF');\n const promisifiedRtf = promisify(rtf.fromString);\n const rtfHtml = await promisifiedRtf(buf.toString(), {\n template(_, __, content: string) {\n return content;\n },\n });\n altText = htmlToText(rtfHtml, { wordwrap: false });\n }\n\n if (file.mimetype === 'text/plain' || file.originalname.endsWith('.txt')) {\n this.logger.info('[Mutation] Updating Alt File for Text Document: TXT');\n altText = buf.toString();\n }\n\n return altText\n ? Buffer.from(altText)\n : null;\n }\n\n private async updateImageFileProps(\n ctx: TransactionContext,\n submissionFile: SubmissionFile,\n file: MulterFileInfo,\n buf: Buffer,\n ) {\n const { width, height } = await this.getImageDetails(\n file,\n buf,\n );\n await ctx\n .getDb()\n .update(this.fileRepository.schemaEntity)\n .set({\n width,\n height,\n })\n .where(eq(this.fileRepository.schemaEntity.id, submissionFile.id));\n\n await ctx\n .getDb()\n .update(this.fileBufferRepository.schemaEntity)\n .set({\n width,\n height,\n })\n .where(\n eq(\n this.fileBufferRepository.schemaEntity.id,\n submissionFile.primaryFileId,\n ),\n );\n\n // Reset metadata dimensions so they don't reference the old file size\n const updatedMetadata = { ...submissionFile.metadata };\n if (updatedMetadata.dimensions) {\n updatedMetadata.dimensions = {\n ...updatedMetadata.dimensions,\n default: { width, height },\n };\n }\n await ctx\n .getDb()\n .update(this.fileRepository.schemaEntity)\n .set({ metadata: updatedMetadata })\n .where(eq(this.fileRepository.schemaEntity.id, submissionFile.id));\n\n if (submissionFile.hasThumbnail && !submissionFile.hasCustomThumbnail) {\n // Regenerate auto-thumbnail;\n const {\n buffer: thumbnailBuf,\n width: thumbnailWidth,\n height: thumbnailHeight,\n mimeType: thumbnailMimeType,\n } = await this.createFileService.generateThumbnail(\n buf,\n file.mimetype,\n );\n\n const fileNameWithoutExt = parse(file.filename).name;\n const thumbnailExt = thumbnailMimeType === 'image/jpeg' ? 'jpg' : 'png';\n\n await ctx\n .getDb()\n .update(this.fileBufferRepository.schemaEntity)\n .set({\n buffer: thumbnailBuf,\n width: thumbnailWidth,\n height: thumbnailHeight,\n size: thumbnailBuf.length,\n mimeType: thumbnailMimeType,\n fileName: `thumbnail_${fileNameWithoutExt}.${thumbnailExt}`,\n })\n .where(\n eq(\n this.fileBufferRepository.schemaEntity.id,\n submissionFile.thumbnailId,\n ),\n );\n }\n }\n\n /**\n * Details of a multer file.\n *\n * @param {MulterFileInfo} file\n */\n private async getImageDetails(file: MulterFileInfo, buf: Buffer) {\n if (ImageUtil.isImage(file.mimetype, false)) {\n const { height, width } = await this.sharpInstanceManager.getMetadata(buf);\n return { buffer: buf, width, height };\n }\n\n throw new BadRequestException('File is not an image');\n }\n\n /**\n * Returns file by Id.\n *\n * @param {EntityId} id\n */\n private async findFile(id: EntityId): Promise {\n try {\n const entity = await this.fileRepository.findOne({\n where: (f, { eq: equals }) => equals(f.id, id),\n // !bug - https://github.com/drizzle-team/drizzle-orm/issues/3497\n // with: {\n // thumbnail: true,\n // primaryFile: true,\n // altFile: true,\n // },\n });\n\n return entity;\n } catch (e) {\n this.logger.error(e.message, e.stack);\n throw new NotFoundException(id);\n }\n }\n}\n" }, { "path": "apps/client-server/src/app/file/utils/image.util.ts", "content": "/**\n * Utility class for image-related checks.\n *\n * NOTE: Sharp image processing has been moved to SharpInstanceManager\n * which runs sharp in isolated worker threads for crash protection.\n * The load() and getMetadata() methods have been removed.\n * Use SharpInstanceManager.getMetadata() or SharpInstanceManager.resizeForPost() instead.\n */\nexport class ImageUtil {\n static isImage(mimetype: string, includeGIF = false): boolean {\n if (includeGIF && mimetype === 'image/gif') {\n return true;\n }\n\n return mimetype.startsWith('image/') && mimetype !== 'image/gif';\n }\n}\n" }, { "path": "apps/client-server/src/app/file-converter/converters/file-converter.ts", "content": "import { IFileBuffer } from '@postybirb/types';\n\nexport interface IFileConverter {\n /**\n * Determines if the file can be converted to any of the allowable output mime types.\n *\n * @param {IFileBuffer} file\n * @param {string[]} allowableOutputMimeTypes\n * @return {*} {boolean}\n */\n canConvert(file: IFileBuffer, allowableOutputMimeTypes: string[]): boolean;\n\n /**\n * Converts the file to one of the allowable output mime types.\n *\n * @param {IFileBuffer} file\n * @param {string[]} allowableOutputMimeTypes\n * @return {*} {Promise}\n */\n convert(\n file: IFileBuffer,\n allowableOutputMimeTypes: string[],\n ): Promise;\n}\n" }, { "path": "apps/client-server/src/app/file-converter/converters/text-file-converter.ts", "content": "import { IFileBuffer } from '@postybirb/types';\nimport { htmlToText } from 'html-to-text';\nimport { TurndownService } from 'turndown';\nimport { IFileConverter } from './file-converter';\n\nconst supportedInputMimeTypes = ['text/html', 'text/plain'] as const;\nconst supportedOutputMimeTypes = [\n 'text/plain',\n 'text/html',\n 'text/markdown',\n] as const;\n\ntype SupportedInputMimeTypes = (typeof supportedInputMimeTypes)[number];\ntype SupportedOutputMimeTypes = (typeof supportedOutputMimeTypes)[number];\n\ntype ConversionMap = {\n [inputMimeType in SupportedInputMimeTypes]: {\n [outputMimeType in SupportedOutputMimeTypes]: (\n file: IFileBuffer,\n ) => Promise;\n };\n};\n\ntype ConversionWeights = {\n [outputMimeType in SupportedOutputMimeTypes]: number;\n};\n\n/**\n * A class that converts text files to other text formats.\n * Largely for use when converting AltFiles (text/plain or text/html) to other desirable formats.\n * @class TextFileConverter\n * @implements {IFileConverter}\n */\nexport class TextFileConverter implements IFileConverter {\n private passThrough = async (file: IFileBuffer): Promise => ({\n ...file,\n });\n\n private convertHtmlToPlaintext = async (\n file: IFileBuffer,\n ): Promise => {\n const text = htmlToText(file.buffer.toString(), {\n wordwrap: 120,\n });\n return this.toMergedBuffer(file, text, 'text/plain');\n };\n\n private convertHtmlToMarkdown = async (\n file: IFileBuffer,\n ): Promise => {\n const turndownService = new TurndownService();\n const markdown = turndownService.turndown(file.buffer.toString());\n return this.toMergedBuffer(file, markdown, 'text/markdown');\n };\n\n /**\n * Converts plain text to HTML by wrapping lines in

tags.\n */\n private convertPlaintextToHtml = async (\n file: IFileBuffer,\n ): Promise => {\n const lines = file.buffer.toString().split(/\\n/);\n const html = lines\n .map((line) => `

${line || '
'}

`)\n .join('\\n');\n return this.toMergedBuffer(file, html, 'text/html');\n };\n\n /**\n * Plain text is valid markdown, so this is a passthrough with mime type change.\n */\n private convertPlaintextToMarkdown = async (\n file: IFileBuffer,\n ): Promise => this.toMergedBuffer(file, file.buffer.toString(), 'text/markdown');\n\n private readonly supportConversionMappers: ConversionMap = {\n 'text/html': {\n 'text/html': this.passThrough,\n 'text/plain': this.convertHtmlToPlaintext,\n 'text/markdown': this.convertHtmlToMarkdown,\n },\n 'text/plain': {\n 'text/plain': this.passThrough,\n 'text/html': this.convertPlaintextToHtml,\n 'text/markdown': this.convertPlaintextToMarkdown,\n },\n };\n\n /**\n * Defines the preference of conversion, trying to convert to the most preferred format first.\n */\n private readonly conversionWeights: ConversionWeights = {\n 'text/plain': Number.MAX_SAFE_INTEGER,\n 'text/html': 1,\n 'text/markdown': 2,\n };\n\n canConvert(file: IFileBuffer, allowableOutputMimeTypes: string[]): boolean {\n return (\n supportedInputMimeTypes.includes(\n file.mimeType as SupportedInputMimeTypes,\n ) &&\n supportedOutputMimeTypes.some((m) => allowableOutputMimeTypes.includes(m))\n );\n }\n\n async convert(\n file: IFileBuffer,\n allowableOutputMimeTypes: string[],\n ): Promise {\n const conversionMap =\n this.supportConversionMappers[file.mimeType as SupportedInputMimeTypes];\n\n const sortedOutputMimeTypes = allowableOutputMimeTypes\n .filter((mimeType) => mimeType in conversionMap)\n .sort(\n (a, b) =>\n this.conversionWeights[a as SupportedOutputMimeTypes] -\n this.conversionWeights[b as SupportedOutputMimeTypes],\n );\n\n for (const outputMimeType of sortedOutputMimeTypes) {\n const conversionFunction =\n conversionMap[outputMimeType as SupportedOutputMimeTypes];\n if (conversionFunction) {\n return conversionFunction(file);\n }\n }\n\n throw new Error(\n `Cannot convert file ${file.fileName} with mime type: ${file.mimeType}`,\n );\n }\n\n private toMergedBuffer(\n fb: IFileBuffer,\n str: string,\n mimeType: string,\n ): IFileBuffer {\n return {\n ...fb,\n buffer: Buffer.from(str),\n mimeType,\n };\n }\n}\n" }, { "path": "apps/client-server/src/app/file-converter/file-converter.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { FileConverterService } from './file-converter.service';\n\n@Module({\n providers: [FileConverterService],\n exports: [FileConverterService],\n})\nexport class FileConverterModule {}\n" }, { "path": "apps/client-server/src/app/file-converter/file-converter.service.spec.ts", "content": "import { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport { FileConverterService } from './file-converter.service';\n\ndescribe('FileConverterService', () => {\n let service: FileConverterService;\n\n beforeEach(async () => {\n clearDatabase();\n const module: TestingModule = await Test.createTestingModule({\n providers: [FileConverterService],\n }).compile();\n\n service = module.get(FileConverterService);\n });\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n});\n" }, { "path": "apps/client-server/src/app/file-converter/file-converter.service.ts", "content": "import { Injectable } from '@nestjs/common';\nimport { IFileBuffer } from '@postybirb/types';\nimport { IFileConverter } from './converters/file-converter';\nimport { TextFileConverter } from './converters/text-file-converter';\n\n@Injectable()\nexport class FileConverterService {\n private readonly converters: IFileConverter[] = [new TextFileConverter()];\n\n public async convert(\n file: T,\n allowableOutputMimeTypes: string[],\n ): Promise {\n const converter = this.converters.find((c) =>\n c.canConvert(file, allowableOutputMimeTypes),\n );\n\n if (!converter) {\n throw new Error('No converter found for file');\n }\n\n return converter.convert(file, allowableOutputMimeTypes);\n }\n\n public async canConvert(\n mimeType: string,\n allowableOutputMimeTypes: string[],\n ): Promise {\n return this.converters.some((c) =>\n c.canConvert({ mimeType } as IFileBuffer, allowableOutputMimeTypes),\n );\n }\n}\n" }, { "path": "apps/client-server/src/app/form-generator/dtos/form-generation-request.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport {\n AccountId,\n IFormGenerationRequestDto,\n SubmissionType,\n} from '@postybirb/types';\nimport { IsBoolean, IsEnum, IsOptional, IsString } from 'class-validator';\n\nexport class FormGenerationRequestDto implements IFormGenerationRequestDto {\n @ApiProperty()\n @IsString()\n accountId: AccountId;\n\n @ApiProperty({ enum: SubmissionType })\n @IsEnum(SubmissionType)\n type: SubmissionType;\n\n @ApiProperty()\n @IsOptional()\n @IsBoolean()\n isMultiSubmission?: boolean;\n}\n" }, { "path": "apps/client-server/src/app/form-generator/form-generator.controller.ts", "content": "import { Body, Controller, Post } from '@nestjs/common';\nimport { ApiResponse, ApiTags } from '@nestjs/swagger';\nimport { NULL_ACCOUNT_ID } from '@postybirb/types';\nimport { FormGenerationRequestDto } from './dtos/form-generation-request.dto';\nimport { FormGeneratorService } from './form-generator.service';\n\n@ApiTags('form-generator')\n@Controller('form-generator')\nexport class FormGeneratorController {\n constructor(private readonly service: FormGeneratorService) {}\n\n @Post()\n @ApiResponse({\n status: 200,\n description: 'Returns the generated form with default',\n })\n @ApiResponse({ status: 404, description: 'Website instance not found.' })\n @ApiResponse({\n status: 500,\n description: 'An error occurred while performing operation.',\n })\n getFormForWebsite(@Body() request: FormGenerationRequestDto) {\n return request.accountId === NULL_ACCOUNT_ID\n ? this.service.getDefaultForm(request.type, request.isMultiSubmission)\n : this.service.generateForm(request);\n }\n}\n" }, { "path": "apps/client-server/src/app/form-generator/form-generator.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { AccountModule } from '../account/account.module';\nimport { UserSpecifiedWebsiteOptionsModule } from '../user-specified-website-options/user-specified-website-options.module';\nimport { WebsitesModule } from '../websites/websites.module';\nimport { FormGeneratorController } from './form-generator.controller';\nimport { FormGeneratorService } from './form-generator.service';\n\n@Module({\n imports: [WebsitesModule, UserSpecifiedWebsiteOptionsModule, AccountModule],\n providers: [FormGeneratorService],\n controllers: [FormGeneratorController],\n exports: [FormGeneratorService],\n})\nexport class FormGeneratorModule {}\n" }, { "path": "apps/client-server/src/app/form-generator/form-generator.service.spec.ts", "content": "import { NotFoundException } from '@nestjs/common';\nimport { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport {\n NullAccount,\n SubmissionRating,\n SubmissionType,\n} from '@postybirb/types';\nimport { AccountModule } from '../account/account.module';\nimport { AccountService } from '../account/account.service';\nimport { CreateUserSpecifiedWebsiteOptionsDto } from '../user-specified-website-options/dtos/create-user-specified-website-options.dto';\nimport { UserSpecifiedWebsiteOptionsModule } from '../user-specified-website-options/user-specified-website-options.module';\nimport { UserSpecifiedWebsiteOptionsService } from '../user-specified-website-options/user-specified-website-options.service';\nimport { WebsitesModule } from '../websites/websites.module';\nimport { FormGeneratorService } from './form-generator.service';\n\ndescribe('FormGeneratorService', () => {\n let service: FormGeneratorService;\n let userSpecifiedService: UserSpecifiedWebsiteOptionsService;\n let accountService: AccountService;\n let module: TestingModule;\n\n beforeEach(async () => {\n clearDatabase();\n module = await Test.createTestingModule({\n imports: [\n AccountModule,\n WebsitesModule,\n UserSpecifiedWebsiteOptionsModule,\n ],\n providers: [FormGeneratorService],\n }).compile();\n\n service = module.get(FormGeneratorService);\n accountService = module.get(AccountService);\n userSpecifiedService = module.get(\n UserSpecifiedWebsiteOptionsService\n );\n\n await accountService.onModuleInit();\n });\n\n afterAll(async () => {\n await module.close();\n });\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n\n it('should fail on missing account', async () => {\n await expect(\n service.generateForm({ accountId: 'fake', type: SubmissionType.MESSAGE })\n ).rejects.toThrow(NotFoundException);\n });\n\n it('should return user specific defaults', async () => {\n const userSpecifiedDto = new CreateUserSpecifiedWebsiteOptionsDto();\n userSpecifiedDto.accountId = new NullAccount().id;\n userSpecifiedDto.type = SubmissionType.MESSAGE;\n userSpecifiedDto.options = { rating: SubmissionRating.ADULT };\n await userSpecifiedService.create(userSpecifiedDto);\n\n const messageForm = await service.getDefaultForm(SubmissionType.MESSAGE);\n expect(messageForm).toMatchInlineSnapshot(`\n {\n \"contentWarning\": {\n \"defaultValue\": \"\",\n \"formField\": \"input\",\n \"hidden\": false,\n \"label\": \"contentWarning\",\n \"order\": 5,\n \"responsive\": {\n \"xs\": 12,\n },\n \"section\": \"common\",\n \"span\": 12,\n \"type\": \"text\",\n },\n \"description\": {\n \"defaultValue\": {\n \"description\": {\n \"content\": [],\n \"type\": \"doc\",\n },\n \"overrideDefault\": false,\n },\n \"descriptionType\": \"html\",\n \"formField\": \"description\",\n \"label\": \"description\",\n \"order\": 4,\n \"required\": true,\n \"responsive\": {\n \"xs\": 12,\n },\n \"section\": \"common\",\n \"span\": 12,\n \"type\": \"description\",\n },\n \"rating\": {\n \"defaultValue\": \"ADULT\",\n \"formField\": \"rating\",\n \"label\": \"rating\",\n \"layout\": \"horizontal\",\n \"options\": [\n {\n \"label\": \"General\",\n \"value\": \"GENERAL\",\n },\n {\n \"label\": \"Mature\",\n \"value\": \"MATURE\",\n },\n {\n \"label\": \"Adult\",\n \"value\": \"ADULT\",\n },\n {\n \"label\": \"Extreme\",\n \"value\": \"EXTREME\",\n },\n ],\n \"order\": 1,\n \"required\": true,\n \"responsive\": {\n \"xs\": 12,\n },\n \"section\": \"common\",\n \"span\": 12,\n \"type\": \"rating\",\n },\n \"tags\": {\n \"defaultValue\": {\n \"overrideDefault\": false,\n \"tags\": [],\n },\n \"formField\": \"tag\",\n \"label\": \"tags\",\n \"minTagLength\": 1,\n \"order\": 3,\n \"responsive\": {\n \"xs\": 12,\n },\n \"section\": \"common\",\n \"spaceReplacer\": \"_\",\n \"span\": 12,\n \"type\": \"tag\",\n },\n \"title\": {\n \"defaultValue\": \"\",\n \"expectedInDescription\": false,\n \"formField\": \"input\",\n \"label\": \"title\",\n \"order\": 2,\n \"required\": true,\n \"responsive\": {\n \"xs\": 12,\n },\n \"section\": \"common\",\n \"span\": 12,\n \"type\": \"title\",\n },\n }\n `);\n });\n\n it('should return standard form', async () => {\n const messageForm = await service.getDefaultForm(SubmissionType.MESSAGE);\n expect(messageForm).toMatchInlineSnapshot(`\n {\n \"contentWarning\": {\n \"defaultValue\": \"\",\n \"formField\": \"input\",\n \"hidden\": false,\n \"label\": \"contentWarning\",\n \"order\": 5,\n \"responsive\": {\n \"xs\": 12,\n },\n \"section\": \"common\",\n \"span\": 12,\n \"type\": \"text\",\n },\n \"description\": {\n \"defaultValue\": {\n \"description\": {\n \"content\": [],\n \"type\": \"doc\",\n },\n \"overrideDefault\": false,\n },\n \"descriptionType\": \"html\",\n \"formField\": \"description\",\n \"label\": \"description\",\n \"order\": 4,\n \"required\": true,\n \"responsive\": {\n \"xs\": 12,\n },\n \"section\": \"common\",\n \"span\": 12,\n \"type\": \"description\",\n },\n \"rating\": {\n \"defaultValue\": \"GENERAL\",\n \"formField\": \"rating\",\n \"label\": \"rating\",\n \"layout\": \"horizontal\",\n \"options\": [\n {\n \"label\": \"General\",\n \"value\": \"GENERAL\",\n },\n {\n \"label\": \"Mature\",\n \"value\": \"MATURE\",\n },\n {\n \"label\": \"Adult\",\n \"value\": \"ADULT\",\n },\n {\n \"label\": \"Extreme\",\n \"value\": \"EXTREME\",\n },\n ],\n \"order\": 1,\n \"required\": true,\n \"responsive\": {\n \"xs\": 12,\n },\n \"section\": \"common\",\n \"span\": 12,\n \"type\": \"rating\",\n },\n \"tags\": {\n \"defaultValue\": {\n \"overrideDefault\": false,\n \"tags\": [],\n },\n \"formField\": \"tag\",\n \"label\": \"tags\",\n \"minTagLength\": 1,\n \"order\": 3,\n \"responsive\": {\n \"xs\": 12,\n },\n \"section\": \"common\",\n \"spaceReplacer\": \"_\",\n \"span\": 12,\n \"type\": \"tag\",\n },\n \"title\": {\n \"defaultValue\": \"\",\n \"expectedInDescription\": false,\n \"formField\": \"input\",\n \"label\": \"title\",\n \"order\": 2,\n \"required\": true,\n \"responsive\": {\n \"xs\": 12,\n },\n \"section\": \"common\",\n \"span\": 12,\n \"type\": \"title\",\n },\n }\n `);\n\n const fileForm = await service.getDefaultForm(SubmissionType.FILE);\n expect(fileForm).toMatchInlineSnapshot(`\n {\n \"contentWarning\": {\n \"defaultValue\": \"\",\n \"formField\": \"input\",\n \"hidden\": false,\n \"label\": \"contentWarning\",\n \"order\": 5,\n \"responsive\": {\n \"xs\": 12,\n },\n \"section\": \"common\",\n \"span\": 12,\n \"type\": \"text\",\n },\n \"description\": {\n \"defaultValue\": {\n \"description\": {\n \"content\": [],\n \"type\": \"doc\",\n },\n \"overrideDefault\": false,\n },\n \"descriptionType\": \"html\",\n \"formField\": \"description\",\n \"label\": \"description\",\n \"order\": 4,\n \"required\": true,\n \"responsive\": {\n \"xs\": 12,\n },\n \"section\": \"common\",\n \"span\": 12,\n \"type\": \"description\",\n },\n \"rating\": {\n \"defaultValue\": \"GENERAL\",\n \"formField\": \"rating\",\n \"label\": \"rating\",\n \"layout\": \"horizontal\",\n \"options\": [\n {\n \"label\": \"General\",\n \"value\": \"GENERAL\",\n },\n {\n \"label\": \"Mature\",\n \"value\": \"MATURE\",\n },\n {\n \"label\": \"Adult\",\n \"value\": \"ADULT\",\n },\n {\n \"label\": \"Extreme\",\n \"value\": \"EXTREME\",\n },\n ],\n \"order\": 1,\n \"required\": true,\n \"responsive\": {\n \"xs\": 12,\n },\n \"section\": \"common\",\n \"span\": 12,\n \"type\": \"rating\",\n },\n \"tags\": {\n \"defaultValue\": {\n \"overrideDefault\": false,\n \"tags\": [],\n },\n \"formField\": \"tag\",\n \"label\": \"tags\",\n \"minTagLength\": 1,\n \"order\": 3,\n \"responsive\": {\n \"xs\": 12,\n },\n \"section\": \"common\",\n \"spaceReplacer\": \"_\",\n \"span\": 12,\n \"type\": \"tag\",\n },\n \"title\": {\n \"defaultValue\": \"\",\n \"expectedInDescription\": false,\n \"formField\": \"input\",\n \"label\": \"title\",\n \"order\": 2,\n \"required\": true,\n \"responsive\": {\n \"xs\": 12,\n },\n \"section\": \"common\",\n \"span\": 12,\n \"type\": \"title\",\n },\n }\n `);\n });\n});\n" }, { "path": "apps/client-server/src/app/form-generator/form-generator.service.ts", "content": "import {\n BadRequestException,\n Injectable,\n NotFoundException,\n} from '@nestjs/common';\nimport { FormBuilderMetadata, formBuilder } from '@postybirb/form-builder';\nimport {\n AccountId,\n IWebsiteFormFields,\n NullAccount,\n SubmissionType,\n} from '@postybirb/types';\nimport { AccountService } from '../account/account.service';\nimport { UserSpecifiedWebsiteOptionsService } from '../user-specified-website-options/user-specified-website-options.service';\nimport { DefaultWebsiteOptions } from '../websites/models/default-website-options';\nimport { isFileWebsite } from '../websites/models/website-modifiers/file-website';\nimport { isMessageWebsite } from '../websites/models/website-modifiers/message-website';\nimport { WebsiteRegistryService } from '../websites/website-registry.service';\nimport { FormGenerationRequestDto } from './dtos/form-generation-request.dto';\n\n@Injectable()\nexport class FormGeneratorService {\n constructor(\n private readonly websiteRegistryService: WebsiteRegistryService,\n private readonly userSpecifiedWebsiteOptionsService: UserSpecifiedWebsiteOptionsService,\n private readonly accountService: AccountService,\n ) {}\n\n /**\n * Generates the form properties for a submission option.\n * Form properties are used for form generation in UI.\n *\n * @param {FormGenerationRequestDto} request\n */\n async generateForm(\n request: FormGenerationRequestDto,\n ): Promise {\n const account = await this.accountService.findById(request.accountId, {\n failOnMissing: true,\n });\n\n // Get instance for creation\n const instance = await this.websiteRegistryService.findInstance(account);\n\n if (!instance) {\n throw new NotFoundException('Unable to find website instance');\n }\n\n // Get data for inserting into form\n const data = instance.getFormProperties();\n\n // Get form model\n let formModel: IWebsiteFormFields = null;\n if (request.type === SubmissionType.MESSAGE && isMessageWebsite(instance)) {\n formModel = instance.createMessageModel();\n }\n\n if (request.type === SubmissionType.FILE && isFileWebsite(instance)) {\n formModel = instance.createFileModel();\n }\n\n if (!formModel) {\n throw new BadRequestException(\n `Website instance does not support ${request.type}`,\n );\n }\n\n const form = formBuilder(formModel, data);\n const formWithPopulatedDefaults = await this.populateUserDefaults(\n form,\n request.accountId,\n request.type,\n );\n\n if (request.isMultiSubmission) {\n delete formWithPopulatedDefaults.title; // Having title here just causes confusion for multi this flow\n }\n\n return formWithPopulatedDefaults;\n }\n\n /**\n * Returns the default fields form.\n * @param {SubmissionType} type\n */\n async getDefaultForm(type: SubmissionType, isMultiSubmission = false) {\n const form = await this.populateUserDefaults(\n formBuilder(new DefaultWebsiteOptions(), {}),\n new NullAccount().id,\n type,\n );\n if (isMultiSubmission) {\n delete form.title; // Having title here just causes confusion for multi this flow\n }\n return form;\n }\n\n private async populateUserDefaults(\n form: FormBuilderMetadata,\n accountId: AccountId,\n type: SubmissionType,\n ): Promise {\n const userSpecifiedDefaults =\n await this.userSpecifiedWebsiteOptionsService.findByAccountAndSubmissionType(\n accountId,\n type,\n );\n\n if (userSpecifiedDefaults) {\n Object.entries(userSpecifiedDefaults.options).forEach(([key, value]) => {\n const field = form[key];\n if (field) {\n field.defaultValue = value ?? field.defaultValue;\n }\n });\n }\n\n return form;\n }\n}\n" }, { "path": "apps/client-server/src/app/image-processing/image-processing.module.ts", "content": "import { Global, Module } from '@nestjs/common';\nimport { SharpInstanceManager } from './sharp-instance-manager';\n\n/**\n * Global module providing the SharpInstanceManager to all modules.\n * Sharp image processing is isolated in worker threads to protect\n * the main process from native libvips crashes.\n */\n@Global()\n@Module({\n providers: [SharpInstanceManager],\n exports: [SharpInstanceManager],\n})\nexport class ImageProcessingModule {}\n" }, { "path": "apps/client-server/src/app/image-processing/index.ts", "content": "export { ImageProcessingModule } from './image-processing.module';\nexport { SharpInstanceManager } from './sharp-instance-manager';\nexport type { SharpWorkerInput, SharpWorkerResult } from './sharp-instance-manager';\n" }, { "path": "apps/client-server/src/app/image-processing/sharp-instance-manager.ts", "content": "import { Injectable, OnModuleDestroy } from '@nestjs/common';\nimport { Logger } from '@postybirb/logger';\nimport { ImageResizeProps } from '@postybirb/types';\nimport { IsTestEnvironment } from '@postybirb/utils/electron';\nimport { existsSync } from 'fs';\nimport { cpus } from 'os';\nimport { join, resolve } from 'path';\nimport Piscina from 'piscina';\n\n/**\n * Input sent to the sharp worker thread.\n */\nexport interface SharpWorkerInput {\n operation: 'resize' | 'metadata' | 'thumbnail' | 'healthcheck';\n buffer: Buffer;\n resize?: ImageResizeProps;\n mimeType: string;\n fileName?: string;\n fileId?: string;\n fileWidth?: number;\n fileHeight?: number;\n thumbnailBuffer?: Buffer;\n thumbnailMimeType?: string;\n thumbnailPreferredDimension?: number;\n generateThumbnail?: boolean;\n}\n\n/**\n * Result returned from the sharp worker thread.\n */\nexport interface SharpWorkerResult {\n buffer?: Buffer;\n mimeType?: string;\n width?: number;\n height?: number;\n format?: string;\n fileName?: string;\n modified: boolean;\n thumbnailBuffer?: Buffer;\n thumbnailMimeType?: string;\n thumbnailWidth?: number;\n thumbnailHeight?: number;\n thumbnailFileName?: string;\n}\n\n/**\n * Manages a pool of worker threads that run sharp image processing.\n *\n * This isolates sharp/libvips native code from the main process so that\n * if libvips segfaults (e.g. after long idle periods), only the worker\n * dies — the main process survives and piscina spawns a replacement.\n *\n * @class SharpInstanceManager\n */\n@Injectable()\nexport class SharpInstanceManager implements OnModuleDestroy {\n private readonly logger = Logger();\n\n private pool: Piscina | null = null;\n\n /**\n * In test mode, we call the worker function directly in-process to avoid\n * thread contention issues with the electron test runner. The sharp logic\n * is still fully exercised — only the threading is bypassed.\n */\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n private workerFn: ((input: any) => Promise) | null = null;\n\n constructor() {\n const workerPath = this.resolveWorkerPath();\n\n if (IsTestEnvironment()) {\n // In tests: call the worker function directly (no threads)\n // eslint-disable-next-line @typescript-eslint/no-require-imports, import/no-dynamic-require, global-require\n this.workerFn = require(workerPath);\n } else {\n // Production: use piscina for crash isolation\n const maxThreads = Math.min(cpus().length, 4);\n\n this.logger\n .withMetadata({ workerPath, maxThreads })\n .info('Initializing sharp worker pool');\n\n this.pool = new Piscina({\n filename: workerPath,\n maxThreads,\n minThreads: 0, // Allow ALL workers to be reaped after idle\n idleTimeout: 60_000, // Kill idle workers after 60s\n });\n }\n\n // Run health check asynchronously — don't block construction,\n // but log warnings if sharp is broken on this system.\n // The .catch() is a safety net to prevent unhandled rejection\n // if something unexpected escapes the try/catch inside runHealthCheck.\n this.runHealthCheck().catch((err) => {\n this.logger\n .withError(err)\n .error('Unexpected error during sharp health check');\n });\n }\n\n /**\n * Probe the worker with a trivial sharp operation to detect\n * missing native bindings, glibc issues, or sandbox restrictions\n * at startup rather than failing silently during posting.\n */\n private async runHealthCheck(): Promise {\n try {\n const result = await this.processImage({\n operation: 'healthcheck',\n buffer: Buffer.alloc(0),\n mimeType: '',\n });\n\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n const { diagnostics } = result as any;\n if (diagnostics) {\n this.logger\n .withMetadata(diagnostics)\n .info('Sharp health check passed');\n }\n } catch (error) {\n this.logger\n .withError(error)\n .error(\n 'Sharp health check FAILED — image processing will not work. ' +\n 'This usually means native sharp bindings failed to load. ' +\n 'On Linux, ensure glibc >= 2.17 is installed. ' +\n 'On Snap/Flatpak, the sandbox may prevent loading native modules.',\n );\n }\n }\n\n async onModuleDestroy() {\n if (this.pool) {\n this.logger.info('Destroying sharp worker pool');\n try {\n await this.pool.destroy();\n } catch {\n // pool.destroy() throws for in-flight tasks — safe to ignore\n }\n }\n }\n\n /**\n * Cached resolved path — avoids repeated filesystem probing.\n */\n private static resolvedWorkerPath: string | null = null;\n\n /**\n * Resolve the path to the sharp-worker.js file.\n * Works in:\n * - Production build (webpack output): __dirname/assets/sharp-worker.js\n * - Development (nx serve): relative to source tree\n * - Test mode: also relative to source tree (for require())\n */\n private resolveWorkerPath(): string {\n if (SharpInstanceManager.resolvedWorkerPath) {\n return SharpInstanceManager.resolvedWorkerPath;\n }\n\n const candidates = [\n // Production build: assets sit next to the bundled main.js\n join(__dirname, 'assets', 'sharp-worker.js'),\n // Source tree: __dirname = apps/client-server/src/app/image-processing\n resolve(__dirname, '..', '..', '..', 'assets', 'sharp-worker.js'),\n // cwd-based (nx serve, standalone)\n join(\n process.cwd(),\n 'apps',\n 'client-server',\n 'src',\n 'assets',\n 'sharp-worker.js',\n ),\n // cwd-based (jest may cd into apps/client-server)\n join(process.cwd(), 'src', 'assets', 'sharp-worker.js'),\n ];\n\n for (const candidate of candidates) {\n if (existsSync(candidate)) {\n SharpInstanceManager.resolvedWorkerPath = candidate;\n return candidate;\n }\n }\n\n this.logger.warn(\n `Sharp worker not found in any candidate path. Checked: ${candidates.join(', ')}`,\n );\n return candidates[0];\n }\n\n /**\n * Process an image using the worker pool.\n *\n * Buffers are copied to the worker via structured clone — the caller's\n * original buffers are NOT detached and remain usable after the call.\n * This means peak memory for a single operation is roughly\n * 2× the input size (original + worker copy).\n */\n async processImage(input: SharpWorkerInput): Promise {\n try {\n // In test mode, call the worker function directly (no threads)\n if (this.workerFn) {\n return (await this.workerFn(input)) as SharpWorkerResult;\n }\n\n const result = await this.pool.run(input);\n\n // Structured clone across worker threads converts Node.js Buffers\n // into plain Uint8Arrays. Re-wrap them so downstream consumers\n // (e.g. form-data) that rely on Buffer methods/stream semantics work.\n if (result.buffer && !Buffer.isBuffer(result.buffer)) {\n result.buffer = Buffer.from(result.buffer);\n }\n if (result.thumbnailBuffer && !Buffer.isBuffer(result.thumbnailBuffer)) {\n result.thumbnailBuffer = Buffer.from(result.thumbnailBuffer);\n }\n\n return result as SharpWorkerResult;\n } catch (error) {\n this.logger\n .withError(error)\n .error('Sharp worker error — worker may have crashed');\n throw error;\n }\n }\n\n /**\n * Get metadata for an image buffer.\n * @param buffer - The image buffer\n * @returns width, height, format, mimeType\n */\n async getMetadata(buffer: Buffer): Promise<{\n width: number;\n height: number;\n format: string;\n mimeType: string;\n }> {\n const result = await this.processImage({\n operation: 'metadata',\n buffer,\n mimeType: '',\n });\n\n return {\n width: result.width ?? 0,\n height: result.height ?? 0,\n format: result.format ?? 'unknown',\n mimeType: result.mimeType ?? 'image/unknown',\n };\n }\n\n /**\n * Generate a thumbnail from an image buffer.\n * Used by CreateFileService and UpdateFileService during file upload.\n */\n async generateThumbnail(\n buffer: Buffer,\n mimeType: string,\n fileName: string,\n preferredDimension = 400,\n ): Promise<{\n buffer: Buffer;\n width: number;\n height: number;\n mimeType: string;\n fileName: string;\n }> {\n const result = await this.processImage({\n operation: 'thumbnail',\n buffer,\n mimeType,\n fileName,\n thumbnailPreferredDimension: preferredDimension,\n });\n\n return {\n buffer: result.buffer ?? buffer,\n width: result.width ?? 0,\n height: result.height ?? 0,\n mimeType: result.mimeType ?? mimeType,\n fileName: result.fileName ?? fileName,\n };\n }\n\n /**\n * Resize an image for posting. Handles format conversion, dimensional\n * resize, maxBytes scaling, and optional thumbnail generation.\n */\n async resizeForPost(input: {\n buffer: Buffer;\n resize?: ImageResizeProps;\n mimeType: string;\n fileName: string;\n fileId: string;\n fileWidth: number;\n fileHeight: number;\n thumbnailBuffer?: Buffer;\n thumbnailMimeType?: string;\n generateThumbnail: boolean;\n thumbnailPreferredDimension?: number;\n }): Promise {\n return this.processImage({\n operation: 'resize',\n buffer: input.buffer,\n resize: input.resize,\n mimeType: input.mimeType,\n fileName: input.fileName,\n fileId: input.fileId,\n fileWidth: input.fileWidth,\n fileHeight: input.fileHeight,\n thumbnailBuffer: input.thumbnailBuffer,\n thumbnailMimeType: input.thumbnailMimeType,\n generateThumbnail: input.generateThumbnail,\n thumbnailPreferredDimension: input.thumbnailPreferredDimension ?? 500,\n });\n }\n\n /**\n * Get pool statistics for monitoring.\n */\n getStats() {\n if (!this.pool) return null;\n return {\n completed: this.pool.completed,\n duration: this.pool.duration,\n utilization: this.pool.utilization,\n queueSize: this.pool.queueSize,\n };\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/converters/legacy-converter.ts", "content": "/* eslint-disable no-underscore-dangle */\nimport { SchemaKey } from '@postybirb/database';\nimport { Logger } from '@postybirb/logger';\nimport { join } from 'path';\nimport { Class } from 'type-fest';\nimport { PostyBirbDatabase } from '../../drizzle/postybirb-database/postybirb-database';\nimport { LegacyConverterEntity } from '../legacy-entities/legacy-converter-entity';\nimport { NdjsonParser } from '../utils/ndjson-parser';\n\nexport abstract class LegacyConverter {\n abstract readonly modernSchemaKey: SchemaKey;\n\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n abstract readonly LegacyEntityConstructor: Class>;\n\n abstract readonly legacyFileName: string;\n\n constructor(protected readonly databasePath: string) {}\n\n private getEntityFilePath(): string {\n return join(this.databasePath, 'data', `${this.legacyFileName}.db`);\n }\n\n private getModernDatabase() {\n return new PostyBirbDatabase(this.modernSchemaKey);\n }\n\n public async import(): Promise {\n const logger = Logger(`LegacyConverter:${this.legacyFileName}`);\n logger.info(`Starting import for ${this.legacyFileName}...`);\n\n const filePath = this.getEntityFilePath();\n logger.info(`Reading legacy data from ${filePath}...`);\n const parser = new NdjsonParser();\n const result = await parser.parseFile(\n filePath,\n this.LegacyEntityConstructor,\n );\n\n logger.info(\n `Parsed ${filePath}: ${result.records.length} records, ${result.errors.length} errors`,\n );\n if (result.errors.length > 0) {\n throw new Error(\n `Errors occurred while parsing ${this.LegacyEntityConstructor.name} data: ${result.errors\n .map((err) => `Line ${err.line}: ${err.error}`)\n .join('; ')}`,\n );\n }\n const modernDb = this.getModernDatabase();\n\n let skippedCount = 0;\n for (const legacyEntity of result.records) {\n const exists = await modernDb.findById(legacyEntity._id);\n if (exists) {\n logger.warn(\n `Entity with ID ${legacyEntity._id} already exists in modern database. Skipping.`,\n );\n continue;\n }\n\n const modernEntity = await legacyEntity.convert();\n\n // Skip null conversions (e.g., deprecated websites)\n if (modernEntity === null) {\n skippedCount++;\n continue;\n }\n\n await modernDb.insert(modernEntity);\n }\n\n if (skippedCount > 0) {\n logger.info(`Skipped ${skippedCount} records during conversion`);\n }\n\n logger.info(`Import for ${this.legacyFileName} completed successfully.`);\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/converters/legacy-custom-shortcut.converter.spec.ts", "content": "import { clearDatabase } from '@postybirb/database';\nimport { ensureDirSync, PostyBirbDirectories, writeSync } from '@postybirb/fs';\nimport { readFileSync } from 'fs';\nimport { join } from 'path';\nimport { v4 } from 'uuid';\nimport { PostyBirbDatabase } from '../../drizzle/postybirb-database/postybirb-database';\nimport { LegacyCustomShortcut } from '../legacy-entities/legacy-custom-shortcut';\nimport { LegacyCustomShortcutConverter } from './legacy-custom-shortcut.converter';\n\ndescribe('LegacyCustomShortcutConverter', () => {\n let converter: LegacyCustomShortcutConverter;\n let testDataPath: string;\n let repository: PostyBirbDatabase<'CustomShortcutSchema'>;\n const ts = Date.now();\n\n beforeEach(async () => {\n clearDatabase();\n\n // Setup test data directory\n testDataPath = `${PostyBirbDirectories.DATA_DIRECTORY}/legacy-db/${ts}/CustomShortcut-${v4()}`;\n\n // Copy test data to temp directory\n const testDataDir = join(testDataPath, 'data');\n ensureDirSync(testDataDir);\n\n const sourceFile = join(__dirname, '../test-files/data/custom-shortcut.db');\n const testFile = readFileSync(sourceFile);\n const destFile = join(testDataDir, 'custom-shortcut.db');\n\n writeSync(destFile, testFile);\n\n converter = new LegacyCustomShortcutConverter(testDataPath);\n repository = new PostyBirbDatabase('CustomShortcutSchema');\n });\n\n it('should be defined', () => {\n expect(LegacyCustomShortcutConverter).toBeDefined();\n });\n\n describe('System Shortcuts Conversion', () => {\n it('should convert {title} to titleShortcut', async () => {\n const legacyShortcut = new LegacyCustomShortcut({\n _id: 'test-title-shortcut',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n shortcut: 'titletest',\n content: '

Artwork: {title}

',\n isDynamic: false,\n });\n\n const result = await legacyShortcut.convert();\n\n expect(result.shortcut).toMatchObject({\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Artwork: ' },\n { type: 'titleShortcut', attrs: {} },\n ],\n },\n ],\n });\n });\n\n it('should convert {tags} to tagsShortcut', async () => {\n const legacyShortcut = new LegacyCustomShortcut({\n _id: 'test-tags-shortcut',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n shortcut: 'tagstest',\n content: '

Tags: {tags}

',\n isDynamic: false,\n });\n\n const result = await legacyShortcut.convert();\n\n expect(result.shortcut).toMatchObject({\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Tags: ' },\n { type: 'tagsShortcut', attrs: {} },\n ],\n },\n ],\n });\n });\n\n it('should convert {cw} to contentWarningShortcut', async () => {\n const legacyShortcut = new LegacyCustomShortcut({\n _id: 'test-cw-shortcut',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n shortcut: 'cwtest',\n content: '

Content Warning: {cw}

',\n isDynamic: false,\n });\n\n const result = await legacyShortcut.convert();\n\n expect(result.shortcut).toMatchObject({\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Content Warning: ' },\n { type: 'contentWarningShortcut', attrs: {} },\n ],\n },\n ],\n });\n });\n\n it('should convert multiple system shortcuts in a single block', async () => {\n const legacyShortcut = new LegacyCustomShortcut({\n _id: 'test-multi-system',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n shortcut: 'multisystem',\n content: '

{title} ({cw})

{tags}

',\n isDynamic: false,\n });\n\n const result = await legacyShortcut.convert();\n\n expect(result.shortcut).toMatchObject({\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'titleShortcut', attrs: {} },\n { type: 'text', text: ' (' },\n { type: 'contentWarningShortcut', attrs: {} },\n { type: 'text', text: ')' },\n ],\n },\n {\n type: 'paragraph',\n content: [{ type: 'tagsShortcut', attrs: {} }],\n },\n ],\n });\n });\n\n it('should convert system shortcuts alongside username shortcuts', async () => {\n const legacyShortcut = new LegacyCustomShortcut({\n _id: 'test-mixed-shortcuts',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n shortcut: 'mixedtest',\n content: '

{title} by {fa:myusername}

{cw}

{tags}

',\n isDynamic: false,\n });\n\n const result = await legacyShortcut.convert();\n\n expect(result.shortcut).toMatchObject({\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'titleShortcut', attrs: {} },\n { type: 'text', text: ' by ' },\n {\n type: 'username',\n attrs: expect.objectContaining({\n shortcut: 'furaffinity',\n only: '',\n username: 'myusername',\n }),\n },\n ],\n },\n {\n type: 'paragraph',\n content: [{ type: 'contentWarningShortcut', attrs: {} }],\n },\n {\n type: 'paragraph',\n content: [{ type: 'tagsShortcut', attrs: {} }],\n },\n ],\n });\n });\n\n it('should handle case-insensitive system shortcuts', async () => {\n const legacyShortcut = new LegacyCustomShortcut({\n _id: 'test-case-insensitive',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n shortcut: 'casetest',\n content: '

{TITLE} {CW} {TAGS}

',\n isDynamic: false,\n });\n\n const result = await legacyShortcut.convert();\n\n expect(result.shortcut).toMatchObject({\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'titleShortcut', attrs: {} },\n { type: 'text', text: ' ' },\n { type: 'contentWarningShortcut', attrs: {} },\n { type: 'text', text: ' ' },\n { type: 'tagsShortcut', attrs: {} },\n ],\n },\n ],\n });\n });\n });\n\n it('should import and convert legacy custom shortcut data', async () => {\n await converter.import();\n\n const records = await repository.findAll();\n expect(records).toHaveLength(2);\n\n // Verify first shortcut\n const shortcut1 = records.find(\n (r) => r.id === 'cs123456-1234-1234-1234-123456789abc',\n );\n expect(shortcut1).toBeDefined();\n expect(shortcut1!.name).toBe('myshortcut');\n\n // Verify TipTap format\n expect(shortcut1!.shortcut).toMatchObject({\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n {\n type: 'text',\n text: 'This is my custom shortcut content',\n },\n ],\n },\n ],\n });\n\n // Verify second shortcut (dynamic with HTML)\n const shortcut2 = records.find(\n (r) => r.id === 'cs234567-2345-2345-2345-234567890bcd',\n );\n expect(shortcut2).toBeDefined();\n expect(shortcut2!.name).toBe('dynamicshortcut');\n\n // Verify HTML is converted to TipTap format with bold formatting\n const textContent2 = JSON.stringify(shortcut2!.shortcut);\n expect(textContent2).toContain('Dynamic content');\n expect(textContent2).toContain('bold');\n });\n\n it('should handle custom shortcut with empty content', async () => {\n // Create test data with empty content\n const emptyContentData = {\n _id: 'test-empty-id',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n shortcut: 'empty',\n content: '',\n isDynamic: false,\n };\n\n const testDataDir = join(testDataPath, 'data');\n const emptyFile = join(testDataDir, 'custom-shortcut.db');\n writeSync(emptyFile, Buffer.from(JSON.stringify(emptyContentData) + '\\n'));\n\n await converter.import();\n\n const records = await repository.findAll();\n expect(records).toHaveLength(1);\n\n const record = records[0];\n expect(record.name).toBe('empty');\n\n // Empty content should create a doc with a single empty paragraph\n expect(record.shortcut).toMatchObject({\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n },\n ],\n });\n });\n\n it('should preserve shortcut name as the modern name field', async () => {\n await converter.import();\n\n const records = await repository.findAll();\n expect(records).toHaveLength(2);\n\n // All records should have name field matching legacy shortcut\n const names = records.map((r) => r.name);\n expect(names).toContain('myshortcut');\n expect(names).toContain('dynamicshortcut');\n });\n\n it('should convert legacy shortcuts in content to TipTap format', async () => {\n // Create test data with legacy shortcut syntax\n const shortcutData = {\n _id: 'test-shortcuts-id',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n shortcut: 'testshortcut',\n content:\n '

Hello {default} and {fa:myusername} with {customshortcut} text

',\n isDynamic: false,\n };\n\n const testDataDir = join(testDataPath, 'data');\n const shortcutFile = join(testDataDir, 'custom-shortcut.db');\n writeSync(shortcutFile, Buffer.from(JSON.stringify(shortcutData) + '\\n'));\n\n await converter.import();\n\n const records = await repository.findAll();\n expect(records).toHaveLength(1);\n\n const record = records[0];\n expect(record.name).toBe('testshortcut');\n\n const blocks = (record.shortcut as any).content;\n\n // Should have 2 blocks: defaultShortcut block + paragraph with content\n expect(blocks).toHaveLength(2);\n\n // First block should be the defaultShortcut block\n expect(blocks[0]).toMatchObject({\n type: 'defaultShortcut',\n attrs: {},\n });\n\n // Second block should be paragraph with username shortcut and customShortcut\n expect(blocks[1]).toMatchObject({\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Hello ' },\n { type: 'text', text: ' and ' },\n {\n type: 'username',\n attrs: expect.objectContaining({\n shortcut: 'furaffinity',\n only: '',\n username: 'myusername',\n }),\n },\n { type: 'text', text: ' with ' },\n {\n type: 'customShortcut',\n attrs: { id: 'customshortcut' },\n content: [{ type: 'text', text: '' }],\n },\n { type: 'text', text: ' text' },\n ],\n });\n });\n\n it('should convert multiple username shortcuts to modern format', async () => {\n // Create test data with multiple username shortcuts\n const shortcutData = {\n _id: 'test-multi-username-id',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n shortcut: 'multiusername',\n content:\n '

Follow me on {fa:furuser}, {tw:twitterhandle}, and {da:deviantartist}

',\n isDynamic: false,\n };\n\n const testDataDir = join(testDataPath, 'data');\n const shortcutFile = join(testDataDir, 'custom-shortcut.db');\n writeSync(shortcutFile, Buffer.from(JSON.stringify(shortcutData) + '\\n'));\n\n await converter.import();\n\n const records = await repository.findAll();\n expect(records).toHaveLength(1);\n\n const record = records[0];\n\n // Verify the complete structure with all username shortcuts\n expect(record.shortcut).toMatchObject({\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Follow me on ' },\n {\n type: 'username',\n attrs: expect.objectContaining({\n shortcut: 'furaffinity',\n only: '',\n username: 'furuser',\n }),\n },\n { type: 'text', text: ', ' },\n {\n type: 'username',\n attrs: expect.objectContaining({\n shortcut: 'twitter',\n only: '',\n username: 'twitterhandle',\n }),\n },\n { type: 'text', text: ', and ' },\n {\n type: 'username',\n attrs: expect.objectContaining({\n shortcut: 'deviantart',\n only: '',\n username: 'deviantartist',\n }),\n },\n ],\n },\n ],\n });\n });\n\n it('should convert {default} to block-level when alone in paragraph', async () => {\n const shortcutData = {\n _id: 'test-default-alone',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n shortcut: 'defaultalone',\n content: '

{default}

',\n isDynamic: false,\n };\n\n const testDataDir = join(testDataPath, 'data');\n const shortcutFile = join(testDataDir, 'custom-shortcut.db');\n writeSync(shortcutFile, Buffer.from(JSON.stringify(shortcutData) + '\\n'));\n\n await converter.import();\n\n const records = await repository.findAll();\n expect(records).toHaveLength(1);\n\n const record = records[0];\n\n // Should have a single defaultShortcut block\n expect(record.shortcut).toMatchObject({\n type: 'doc',\n content: [\n {\n type: 'defaultShortcut',\n attrs: {},\n },\n ],\n });\n });\n\n it('should insert defaultShortcut block before paragraph when {default} is with other content', async () => {\n const shortcutData = {\n _id: 'test-default-with-content',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n shortcut: 'defaultwithcontent',\n content: '

Hello {default} World

',\n isDynamic: false,\n };\n\n const testDataDir = join(testDataPath, 'data');\n const shortcutFile = join(testDataDir, 'custom-shortcut.db');\n writeSync(shortcutFile, Buffer.from(JSON.stringify(shortcutData) + '\\n'));\n\n await converter.import();\n\n const records = await repository.findAll();\n expect(records).toHaveLength(1);\n\n const record = records[0];\n\n // Should have 2 blocks: defaultShortcut block + paragraph with remaining text\n expect(record.shortcut).toMatchObject({\n type: 'doc',\n content: [\n {\n type: 'defaultShortcut',\n attrs: {},\n },\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Hello ' },\n { type: 'text', text: ' World' },\n ],\n },\n ],\n });\n });\n\n it('should handle multiple {default} tags correctly', async () => {\n const shortcutData = {\n _id: 'test-multiple-defaults',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n shortcut: 'multipledefaults',\n content: '

{default}

Some text {default} here

',\n isDynamic: false,\n };\n\n const testDataDir = join(testDataPath, 'data');\n const shortcutFile = join(testDataDir, 'custom-shortcut.db');\n writeSync(shortcutFile, Buffer.from(JSON.stringify(shortcutData) + '\\n'));\n\n await converter.import();\n\n const records = await repository.findAll();\n expect(records).toHaveLength(1);\n\n const record = records[0];\n\n // Should have 3 blocks: defaultShortcut (from first para), defaultShortcut (inserted), paragraph (remaining content)\n expect(record.shortcut).toMatchObject({\n type: 'doc',\n content: [\n {\n type: 'defaultShortcut',\n attrs: {},\n },\n {\n type: 'defaultShortcut',\n attrs: {},\n },\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Some text ' },\n { type: 'text', text: ' here' },\n ],\n },\n ],\n });\n });\n\n it('should handle and strip modifier blocks from shortcuts', async () => {\n const shortcutData = {\n _id: 'test-modifiers',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n shortcut: 'modifiertest',\n content:\n '

Test {fa[only=furaffinity]:testuser} and {customshortcut[modifier]} text

',\n isDynamic: false,\n };\n\n const testDataDir = join(testDataPath, 'data');\n const shortcutFile = join(testDataDir, 'custom-shortcut.db');\n writeSync(shortcutFile, Buffer.from(JSON.stringify(shortcutData) + '\\n'));\n\n await converter.import();\n\n const records = await repository.findAll();\n expect(records).toHaveLength(1);\n\n const record = records[0];\n\n // Verify the structure - modifiers should be stripped\n expect(record.shortcut).toMatchObject({\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Test ' },\n {\n type: 'username',\n attrs: expect.objectContaining({\n shortcut: 'furaffinity',\n only: '',\n username: 'testuser',\n }),\n },\n { type: 'text', text: ' and ' },\n {\n type: 'customShortcut',\n attrs: { id: 'customshortcut' },\n content: [{ type: 'text', text: '' }],\n },\n { type: 'text', text: ' text' },\n ],\n },\n ],\n });\n });\n});\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/converters/legacy-custom-shortcut.converter.ts", "content": "import { SchemaKey } from '@postybirb/database';\nimport { LegacyCustomShortcut } from '../legacy-entities/legacy-custom-shortcut';\nimport { LegacyConverter } from './legacy-converter';\n\nexport class LegacyCustomShortcutConverter extends LegacyConverter {\n modernSchemaKey: SchemaKey = 'CustomShortcutSchema';\n\n LegacyEntityConstructor = LegacyCustomShortcut;\n\n legacyFileName = 'custom-shortcut';\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/converters/legacy-tag-converter.converter.spec.ts", "content": "import { clearDatabase } from '@postybirb/database';\nimport { ensureDirSync, PostyBirbDirectories, writeSync } from '@postybirb/fs';\nimport { readFileSync } from 'fs';\nimport { join } from 'path';\nimport { v4 } from 'uuid';\nimport { PostyBirbDatabase } from '../../drizzle/postybirb-database/postybirb-database';\nimport { LegacyTagConverterConverter } from './legacy-tag-converter.converter';\n\ndescribe('LegacyTagConverterConverter', () => {\n let converter: LegacyTagConverterConverter;\n let testDataPath: string;\n let repository: PostyBirbDatabase<'TagConverterSchema'>;\n const ts = Date.now();\n\n beforeEach(async () => {\n clearDatabase();\n\n // Setup test data directory\n testDataPath = `${PostyBirbDirectories.DATA_DIRECTORY}/legacy-db/${ts}/TagConverter-${v4()}`;\n\n // Copy test data to temp directory\n const testDataDir = join(testDataPath, 'data');\n ensureDirSync(testDataDir);\n\n const sourceFile = join(__dirname, '../test-files/data/tag-converter.db');\n const testFile = readFileSync(sourceFile);\n const destFile = join(testDataDir, 'tag-converter.db');\n\n writeSync(destFile, testFile);\n\n converter = new LegacyTagConverterConverter(testDataPath);\n repository = new PostyBirbDatabase('TagConverterSchema');\n });\n\n it('should import and convert legacy tag converter data', async () => {\n await converter.import();\n\n const records = await repository.findAll();\n expect(records).toHaveLength(1);\n\n const record = records[0];\n // Verify ID is preserved\n expect(record.id).toBe('f499b833-b465-462a-bb3c-0983b35b3475');\n // Verify tag is preserved\n expect(record.tag).toBe('converter');\n // Verify legacy website IDs are mapped to modern ones (FurAffinity -> fur-affinity)\n expect(record.convertTo).toHaveProperty('fur-affinity');\n expect(record.convertTo['fur-affinity']).toBe('converted');\n });\n\n it('should handle empty conversions object', async () => {\n // Create test data with empty conversions\n const emptyConversionData = {\n _id: 'test-empty-id',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n tag: 'empty-tag',\n conversions: {},\n };\n\n const testDataDir = join(testDataPath, 'data');\n const emptyTestFile = join(testDataDir, 'tag-converter.db');\n writeSync(\n emptyTestFile,\n Buffer.from(JSON.stringify(emptyConversionData) + '\\n'),\n );\n\n await converter.import();\n\n const records = await repository.findAll();\n expect(records).toHaveLength(1);\n\n const record = records[0];\n expect(record.tag).toBe('empty-tag');\n expect(record.convertTo).toEqual({});\n });\n\n it('should map legacy website names to modern IDs', async () => {\n // Create test data with multiple legacy website names\n const multiWebsiteData = {\n _id: 'test-multi-id',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n tag: 'multi-tag',\n conversions: {\n FurAffinity: 'fa-tag',\n DeviantArt: 'da-tag',\n },\n };\n\n const testDataDir = join(testDataPath, 'data');\n const multiTestFile = join(testDataDir, 'tag-converter.db');\n writeSync(\n multiTestFile,\n Buffer.from(JSON.stringify(multiWebsiteData) + '\\n'),\n );\n\n await converter.import();\n\n const records = await repository.findAll();\n expect(records).toHaveLength(1);\n\n const record = records[0];\n expect(record.convertTo['fur-affinity']).toBe('fa-tag');\n expect(record.convertTo['deviant-art']).toBe('da-tag');\n });\n});\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/converters/legacy-tag-converter.converter.ts", "content": "import { SchemaKey } from '@postybirb/database';\nimport { LegacyTagConverter } from '../legacy-entities/legacy-tag-converter';\nimport { LegacyConverter } from './legacy-converter';\n\nexport class LegacyTagConverterConverter extends LegacyConverter {\n modernSchemaKey: SchemaKey = 'TagConverterSchema';\n\n LegacyEntityConstructor = LegacyTagConverter;\n\n legacyFileName = 'tag-converter';\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/converters/legacy-tag-group.converter.spec.ts", "content": "import { clearDatabase } from '@postybirb/database';\nimport { ensureDirSync, PostyBirbDirectories, writeSync } from '@postybirb/fs';\nimport { readFileSync } from 'fs';\nimport { join } from 'path';\nimport { v4 } from 'uuid';\nimport { PostyBirbDatabase } from '../../drizzle/postybirb-database/postybirb-database';\nimport { LegacyTagGroupConverter } from './legacy-tag-group.converter';\n\ndescribe('LegacyTagGroupConverter', () => {\n let converter: LegacyTagGroupConverter;\n let testDataPath: string;\n let repository: PostyBirbDatabase<'TagGroupSchema'>;\n const ts = Date.now();\n\n beforeEach(async () => {\n clearDatabase();\n\n // Setup test data directory\n testDataPath = `${PostyBirbDirectories.DATA_DIRECTORY}/legacy-db/${ts}/TagGroup-${v4()}`;\n\n // Copy test data to temp directory\n const testDataDir = join(testDataPath, 'data');\n ensureDirSync(testDataDir);\n\n const sourceFile = join(__dirname, '../test-files/data/tag-group.db');\n const testFile = readFileSync(sourceFile);\n const destFile = join(testDataDir, 'tag-group.db');\n\n writeSync(destFile, testFile);\n\n converter = new LegacyTagGroupConverter(testDataPath);\n repository = new PostyBirbDatabase('TagGroupSchema');\n });\n\n it('should import and convert legacy tag group data', async () => {\n await converter.import();\n\n const records = await repository.findAll();\n expect(records).toHaveLength(1);\n\n const record = records[0];\n // Verify ID is preserved\n expect(record.id).toBe('f499b833-b465-462a-bb3c-0983b35b3475');\n // Verify alias is converted to name\n expect(record.name).toBe('converter');\n // Verify tags array is preserved\n expect(record.tags).toEqual(['tag1', 'tag2']);\n });\n\n it('should handle tag group with empty tags array', async () => {\n // Create test data with empty tags\n const emptyTagsData = {\n _id: 'test-empty-tags-id',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n alias: 'empty-group',\n tags: [],\n };\n\n const testDataDir = join(testDataPath, 'data');\n const emptyTagsFile = join(testDataDir, 'tag-group.db');\n writeSync(emptyTagsFile, Buffer.from(JSON.stringify(emptyTagsData) + '\\n'));\n\n await converter.import();\n\n const records = await repository.findAll();\n expect(records).toHaveLength(1);\n\n const record = records[0];\n expect(record.name).toBe('empty-group');\n expect(record.tags).toEqual([]);\n });\n\n it('should handle tag group with single tag', async () => {\n // Create test data with single tag\n const singleTagData = {\n _id: 'test-single-tag-id',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n alias: 'single-tag-group',\n tags: ['lonely-tag'],\n };\n\n const testDataDir = join(testDataPath, 'data');\n const singleTagFile = join(testDataDir, 'tag-group.db');\n writeSync(singleTagFile, Buffer.from(JSON.stringify(singleTagData) + '\\n'));\n\n await converter.import();\n\n const records = await repository.findAll();\n expect(records).toHaveLength(1);\n\n const record = records[0];\n expect(record.name).toBe('single-tag-group');\n expect(record.tags).toHaveLength(1);\n expect(record.tags[0]).toBe('lonely-tag');\n });\n\n it('should handle tag group with special characters in name', async () => {\n // Create test data with special characters\n const specialCharsData = {\n _id: 'test-special-chars-id',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n alias: 'group-with-special!@#$%',\n tags: ['tag1', 'tag2'],\n };\n\n const testDataDir = join(testDataPath, 'data');\n const specialCharsFile = join(testDataDir, 'tag-group.db');\n writeSync(\n specialCharsFile,\n Buffer.from(JSON.stringify(specialCharsData) + '\\n'),\n );\n\n await converter.import();\n\n const records = await repository.findAll();\n expect(records).toHaveLength(1);\n\n const record = records[0];\n expect(record.name).toBe('group-with-special!@#$%');\n });\n});\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/converters/legacy-tag-group.converter.ts", "content": "import { SchemaKey } from '@postybirb/database';\nimport { LegacyTagGroup } from '../legacy-entities/legacy-tag-group';\nimport { LegacyConverter } from './legacy-converter';\n\nexport class LegacyTagGroupConverter extends LegacyConverter {\n modernSchemaKey: SchemaKey = 'TagGroupSchema';\n\n LegacyEntityConstructor = LegacyTagGroup;\n\n legacyFileName = 'tag-group';\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/converters/legacy-user-account.converter.spec.ts", "content": "import { clearDatabase } from '@postybirb/database';\nimport { ensureDirSync, PostyBirbDirectories, writeSync } from '@postybirb/fs';\nimport { readFileSync } from 'fs';\nimport { join } from 'path';\nimport { v4 } from 'uuid';\nimport { PostyBirbDatabase } from '../../drizzle/postybirb-database/postybirb-database';\nimport { LegacyUserAccountConverter } from './legacy-user-account.converter';\n\ndescribe('LegacyUserAccountConverter', () => {\n let converter: LegacyUserAccountConverter;\n let testDataPath: string;\n let repository: PostyBirbDatabase<'AccountSchema'>;\n const ts = Date.now();\n\n beforeEach(async () => {\n clearDatabase();\n\n // Setup test data directory\n testDataPath = `${PostyBirbDirectories.DATA_DIRECTORY}/legacy-db/${ts}/Account-${v4()}`;\n\n // Copy test data to temp directory\n const testDataDir = join(testDataPath, 'data');\n ensureDirSync(testDataDir);\n\n const sourceFile = join(__dirname, '../test-files/data/accounts.db');\n const testFile = readFileSync(sourceFile);\n const destFile = join(testDataDir, 'accounts.db');\n\n writeSync(destFile, testFile);\n\n converter = new LegacyUserAccountConverter(testDataPath);\n repository = new PostyBirbDatabase('AccountSchema');\n });\n\n it('should import and convert legacy user account data', async () => {\n await converter.import();\n\n const records = await repository.findAll();\n expect(records).toHaveLength(2);\n\n // Verify FurAffinity account\n const faAccount = records.find(\n (r) => r.id === 'a1b2c3d4-e5f6-7890-abcd-ef1234567890',\n );\n expect(faAccount).toBeDefined();\n expect(faAccount!.name).toBe('FurAffinity Main');\n expect(faAccount!.website).toBe('fur-affinity');\n\n // Verify DeviantArt account\n const daAccount = records.find(\n (r) => r.id === 'b2c3d4e5-f6a7-8901-bcde-f12345678901',\n );\n expect(daAccount).toBeDefined();\n expect(daAccount!.name).toBe('DeviantArt Account');\n expect(daAccount!.website).toBe('deviant-art');\n });\n\n it('should skip accounts for deprecated websites', async () => {\n // Create test data with deprecated website\n const deprecatedData = {\n _id: 'deprecated-id',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n alias: 'FurryNetwork Account',\n website: 'FurryNetwork',\n data: { username: 'test' },\n };\n\n const testDataDir = join(testDataPath, 'data');\n const deprecatedFile = join(testDataDir, 'accounts.db');\n writeSync(\n deprecatedFile,\n Buffer.from(JSON.stringify(deprecatedData) + '\\n'),\n );\n\n await converter.import();\n\n const records = await repository.findAll();\n // Should be empty because FurryNetwork is deprecated\n expect(records).toHaveLength(0);\n });\n});\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/converters/legacy-user-account.converter.ts", "content": "import { SchemaKey } from '@postybirb/database';\nimport { LegacyUserAccount } from '../legacy-entities/legacy-user-account';\nimport { LegacyConverter } from './legacy-converter';\n\nexport class LegacyUserAccountConverter extends LegacyConverter {\n modernSchemaKey: SchemaKey = 'AccountSchema';\n\n LegacyEntityConstructor = LegacyUserAccount;\n\n legacyFileName = 'accounts';\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/converters/legacy-website-data.converter.spec.ts", "content": "import { clearDatabase } from '@postybirb/database';\nimport { ensureDirSync, PostyBirbDirectories, writeSync } from '@postybirb/fs';\nimport { readFileSync } from 'fs';\nimport { join } from 'path';\nimport { v4 } from 'uuid';\nimport { PostyBirbDatabase } from '../../drizzle/postybirb-database/postybirb-database';\nimport { LegacyUserAccountConverter } from './legacy-user-account.converter';\nimport { LegacyWebsiteDataConverter } from './legacy-website-data.converter';\n\ndescribe('LegacyWebsiteDataConverter', () => {\n let accountConverter: LegacyUserAccountConverter;\n let websiteDataConverter: LegacyWebsiteDataConverter;\n let testDataPath: string;\n let accountRepository: PostyBirbDatabase<'AccountSchema'>;\n let websiteDataRepository: PostyBirbDatabase<'WebsiteDataSchema'>;\n const ts = Date.now();\n\n beforeEach(async () => {\n clearDatabase();\n\n // Setup test data directory\n testDataPath = `${PostyBirbDirectories.DATA_DIRECTORY}/legacy-db/${ts}/WebsiteData-${v4()}`;\n\n // Copy test data to temp directory\n const testDataDir = join(testDataPath, 'data');\n ensureDirSync(testDataDir);\n\n // Use the accounts-with-websitedata.db test file\n const sourceFile = join(\n __dirname,\n '../test-files/data/accounts-with-websitedata.db',\n );\n const testFile = readFileSync(sourceFile);\n const destFile = join(testDataDir, 'accounts.db');\n writeSync(destFile, testFile);\n\n accountConverter = new LegacyUserAccountConverter(testDataPath);\n websiteDataConverter = new LegacyWebsiteDataConverter(testDataPath);\n accountRepository = new PostyBirbDatabase('AccountSchema');\n websiteDataRepository = new PostyBirbDatabase('WebsiteDataSchema');\n });\n\n /**\n * Helper to run both converters in the correct order.\n * Account converter must run first due to foreign key dependency.\n */\n async function runConverters() {\n // Accounts must be created first (WebsiteData has FK reference)\n await accountConverter.import();\n await websiteDataConverter.import();\n }\n\n it('should import Twitter WebsiteData with transformed credentials', async () => {\n await runConverters();\n\n const websiteData = await websiteDataRepository.findById(\n 'twitter-test-id-001',\n );\n expect(websiteData).toBeDefined();\n expect(websiteData!.data).toEqual({\n apiKey: 'consumer_key_123',\n apiSecret: 'consumer_secret_456',\n accessToken: 'access_token_789',\n accessTokenSecret: 'access_secret_012',\n screenName: 'test_user',\n userId: '123456789',\n });\n });\n\n it('should import Discord WebsiteData with webhook config', async () => {\n await runConverters();\n\n const websiteData = await websiteDataRepository.findById(\n 'discord-test-id-002',\n );\n expect(websiteData).toBeDefined();\n expect(websiteData!.data).toEqual({\n webhook: 'https://discord.com/api/webhooks/123456789/abcdefghijklmnop',\n serverLevel: 2,\n isForum: true,\n });\n });\n\n it('should import Telegram WebsiteData with app credentials', async () => {\n await runConverters();\n\n const websiteData = await websiteDataRepository.findById(\n 'telegram-test-id-003',\n );\n expect(websiteData).toBeDefined();\n expect(websiteData!.data).toEqual({\n appId: 12345678, // Converted from string to number\n appHash: 'abcdef0123456789abcdef0123456789',\n phoneNumber: '+1234567890',\n session: undefined,\n channels: [],\n });\n });\n\n it('should import Mastodon WebsiteData with normalized instance URL', async () => {\n await runConverters();\n\n const websiteData = await websiteDataRepository.findById(\n 'mastodon-test-id-004',\n );\n expect(websiteData).toBeDefined();\n expect(websiteData!.data).toMatchObject({\n accessToken: 'mastodon_access_token_xyz',\n instanceUrl: 'mastodon.social', // Normalized (protocol stripped)\n username: 'mastodon_user',\n });\n });\n\n it('should import Bluesky WebsiteData with username and password', async () => {\n await runConverters();\n\n const websiteData = await websiteDataRepository.findById(\n 'bluesky-test-id-005',\n );\n expect(websiteData).toBeDefined();\n expect(websiteData!.data).toEqual({\n username: 'bluesky.user.bsky.social',\n password: 'app_password_123',\n });\n });\n\n it('should import e621 WebsiteData with API key', async () => {\n await runConverters();\n\n const websiteData = await websiteDataRepository.findById('e621-test-id-007');\n expect(websiteData).toBeDefined();\n expect(websiteData!.data).toEqual({\n username: 'e621_user',\n key: 'api_key_xyz789',\n });\n });\n\n it('should import Custom webhook with fixed typo', async () => {\n await runConverters();\n\n const websiteData = await websiteDataRepository.findById(\n 'custom-test-id-009',\n );\n expect(websiteData).toBeDefined();\n // Verify typo fix: thumbnaiField -> thumbnailField\n expect(websiteData!.data).toMatchObject({\n fileUrl: 'https://example.com/upload',\n descriptionField: 'description',\n headers: [{ name: 'Authorization', value: 'Bearer token123' }],\n thumbnailField: 'thumbnail', // Fixed from thumbnaiField\n });\n });\n\n it('should import Pleroma using MegalodonDataTransformer', async () => {\n await runConverters();\n\n const websiteData = await websiteDataRepository.findById(\n 'pleroma-test-id-010',\n );\n expect(websiteData).toBeDefined();\n expect(websiteData!.data).toMatchObject({\n accessToken: 'pleroma_token_abc',\n instanceUrl: 'pleroma.example.com',\n username: 'pleroma_user',\n });\n });\n\n it('should import Pixelfed and normalize instance URL', async () => {\n await runConverters();\n\n const websiteData = await websiteDataRepository.findById(\n 'pixelfed-test-id-011',\n );\n expect(websiteData).toBeDefined();\n expect(websiteData!.data).toMatchObject({\n accessToken: 'pixelfed_token_def',\n instanceUrl: 'pixelfed.social', // Protocol and trailing slash stripped\n username: 'pixelfed_user',\n });\n });\n\n it('should NOT create WebsiteData for browser-cookie websites (FurAffinity)', async () => {\n await runConverters();\n\n // FurAffinity uses browser cookies, not WebsiteData - no transformer exists\n const websiteData = await websiteDataRepository.findById(\n 'furaffinity-test-id-012',\n );\n expect(websiteData).toBeNull();\n\n // But the account should exist\n const account = await accountRepository.findById('furaffinity-test-id-012');\n expect(account).toBeDefined();\n expect(account!.website).toBe('fur-affinity');\n });\n\n it('should skip deprecated websites', async () => {\n // Create test data with deprecated website\n const testDataDir = join(testDataPath, 'data');\n const deprecatedData = {\n _id: 'deprecated-id',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n alias: 'FurryNetwork Account',\n website: 'FurryNetwork',\n data: { username: 'test' },\n };\n const deprecatedFile = join(testDataDir, 'accounts.db');\n writeSync(\n deprecatedFile,\n Buffer.from(JSON.stringify(deprecatedData) + '\\n'),\n );\n\n // Run just the website data converter (account doesn't exist)\n await websiteDataConverter.import();\n\n // Should be empty because FurryNetwork is deprecated\n const records = await websiteDataRepository.findAll();\n expect(records).toHaveLength(0);\n });\n\n it('should handle accounts with transformer but missing data', async () => {\n // Create test data with Twitter account but no data field\n const testDataDir = join(testDataPath, 'data');\n const noDataAccount = {\n _id: 'twitter-no-data',\n created: '2023-10-01T12:00:00Z',\n lastUpdated: '2023-10-01T12:00:00Z',\n alias: 'Twitter No Data',\n website: 'Twitter',\n // data field is missing\n };\n const testFile = join(testDataDir, 'accounts.db');\n writeSync(testFile, Buffer.from(JSON.stringify(noDataAccount) + '\\n'));\n\n // Create the account first\n await accountConverter.import();\n await websiteDataConverter.import();\n\n // WebsiteData should not be created\n const websiteData = await websiteDataRepository.findById('twitter-no-data');\n expect(websiteData).toBeNull();\n });\n});\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/converters/legacy-website-data.converter.ts", "content": "import { SchemaKey } from '@postybirb/database';\nimport { LegacyWebsiteData } from '../legacy-entities/legacy-website-data';\nimport { LegacyConverter } from './legacy-converter';\n\n/**\n * Converter for importing website-specific data (OAuth tokens, API keys, credentials)\n * from legacy PostyBirb Plus accounts.\n *\n * IMPORTANT: This converter must run AFTER LegacyUserAccountConverter because\n * WebsiteData records have a foreign key reference to Account records.\n * The Account must exist before its associated WebsiteData can be created.\n *\n * Only websites with registered transformers in WebsiteDataTransformerRegistry\n * will produce records. Websites using browser cookies for authentication\n * (e.g., FurAffinity, DeviantArt) will be skipped.\n */\nexport class LegacyWebsiteDataConverter extends LegacyConverter {\n modernSchemaKey: SchemaKey = 'WebsiteDataSchema';\n\n LegacyEntityConstructor = LegacyWebsiteData;\n\n /**\n * Reads from the same 'accounts' file as LegacyUserAccountConverter,\n * but only extracts and transforms the website-specific data.\n */\n legacyFileName = 'accounts';\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/dtos/legacy-import.dto.ts", "content": "import { IsBoolean, IsOptional, IsString } from 'class-validator';\n\nexport class LegacyImportDto {\n @IsBoolean()\n customShortcuts: boolean;\n\n @IsBoolean()\n tagGroups: boolean;\n\n @IsBoolean()\n accounts: boolean;\n\n @IsBoolean()\n tagConverters: boolean;\n\n @IsOptional()\n @IsString()\n customPath?: string;\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/legacy-database-importer.controller.ts", "content": "import { Body, Controller, Post } from '@nestjs/common';\nimport { LegacyImportDto } from './dtos/legacy-import.dto';\nimport { LegacyDatabaseImporterService } from './legacy-database-importer.service';\n\n@Controller('legacy-database-importer')\nexport class LegacyDatabaseImporterController {\n constructor(\n private readonly legacyDatabaseImporterService: LegacyDatabaseImporterService,\n ) {}\n\n @Post('import')\n async import(@Body() importRequest: LegacyImportDto) {\n return this.legacyDatabaseImporterService.import(importRequest);\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/legacy-database-importer.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { AccountModule } from '../account/account.module';\nimport { LegacyDatabaseImporterController } from './legacy-database-importer.controller';\nimport { LegacyDatabaseImporterService } from './legacy-database-importer.service';\n\n@Module({\n imports: [AccountModule],\n providers: [LegacyDatabaseImporterService],\n controllers: [LegacyDatabaseImporterController],\n exports: [],\n})\nexport class LegacyDatabaseImporterModule {}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/legacy-database-importer.service.ts", "content": "import { Injectable } from '@nestjs/common';\nimport { Logger } from '@postybirb/logger';\nimport { app } from 'electron';\nimport { join } from 'path';\nimport { AccountService } from '../account/account.service';\nimport { LegacyConverter } from './converters/legacy-converter';\nimport { LegacyCustomShortcutConverter } from './converters/legacy-custom-shortcut.converter';\nimport { LegacyTagConverterConverter } from './converters/legacy-tag-converter.converter';\nimport { LegacyTagGroupConverter } from './converters/legacy-tag-group.converter';\nimport { LegacyUserAccountConverter } from './converters/legacy-user-account.converter';\nimport { LegacyWebsiteDataConverter } from './converters/legacy-website-data.converter';\nimport { LegacyImportDto } from './dtos/legacy-import.dto';\n\n@Injectable()\nexport class LegacyDatabaseImporterService {\n private readonly logger = Logger(LegacyDatabaseImporterService.name);\n\n protected readonly LEGACY_POSTYBIRB_PLUS_PATH = join(\n app.getPath('documents'),\n 'PostyBirb',\n );\n\n constructor(private readonly accountService: AccountService) {}\n\n async import(importRequest: LegacyImportDto): Promise<{ errors: Error[] }> {\n const path = importRequest.customPath || this.LEGACY_POSTYBIRB_PLUS_PATH;\n\n const errors: Error[] = [];\n if (importRequest.accounts) {\n // Import user accounts\n const result = await this.processImport(\n new LegacyUserAccountConverter(path),\n );\n if (result.error) {\n errors.push(result.error);\n }\n\n // IMPORTANT: WebsiteData must be imported AFTER accounts because\n // WebsiteData records have a foreign key reference to Account records.\n // The Account must exist before its associated WebsiteData can be created.\n const websiteDataResult = await this.processImport(\n new LegacyWebsiteDataConverter(path),\n );\n if (websiteDataResult.error) {\n errors.push(websiteDataResult.error);\n }\n\n const allAccounts = await this.accountService.findAll();\n allAccounts.forEach((account) => {\n this.accountService.manuallyExecuteOnLogin(account.id);\n });\n }\n\n if (importRequest.tagGroups) {\n // Import tag groups\n const result = await this.processImport(\n new LegacyTagGroupConverter(path),\n );\n if (result.error) {\n errors.push(result.error);\n }\n }\n\n if (importRequest.tagConverters) {\n // Import tag converters\n const result = await this.processImport(\n new LegacyTagConverterConverter(path),\n );\n if (result.error) {\n errors.push(result.error);\n }\n }\n\n if (importRequest.customShortcuts) {\n // Import custom shortcuts\n const result = await this.processImport(\n new LegacyCustomShortcutConverter(path),\n );\n if (result.error) {\n errors.push(result.error);\n }\n }\n\n return { errors };\n }\n\n private async processImport(\n converter: LegacyConverter,\n ): Promise<{ error?: Error }> {\n try {\n this.logger.info(`Starting import for ${converter.legacyFileName}...`);\n await converter.import();\n return {};\n } catch (error) {\n this.logger.error(\n `Import for ${converter.legacyFileName} failed.`,\n error,\n );\n return { error };\n }\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/legacy-entities/legacy-converter-entity.ts", "content": "import { IEntity } from '@postybirb/types';\n\nexport type MinimalEntity = Omit<\n T,\n 'createdAt' | 'updatedAt'\n>;\n\nexport interface LegacyConverterEntity {\n _id: string;\n\n convert(): Promise | null>;\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/legacy-entities/legacy-custom-shortcut.ts", "content": "/* eslint-disable @typescript-eslint/no-explicit-any */\n/* eslint-disable import/no-extraneous-dependencies */\nimport { Description, ICustomShortcut, TipTapNode } from '@postybirb/types';\nimport { Blockquote } from '@tiptap/extension-blockquote';\nimport { Bold } from '@tiptap/extension-bold';\nimport { Code } from '@tiptap/extension-code';\nimport { Document } from '@tiptap/extension-document';\nimport { HardBreak } from '@tiptap/extension-hard-break';\nimport { Heading } from '@tiptap/extension-heading';\nimport { HorizontalRule } from '@tiptap/extension-horizontal-rule';\nimport { Italic } from '@tiptap/extension-italic';\nimport { Link } from '@tiptap/extension-link';\nimport { Paragraph } from '@tiptap/extension-paragraph';\nimport { Strike } from '@tiptap/extension-strike';\nimport { Text } from '@tiptap/extension-text';\nimport { Underline } from '@tiptap/extension-underline';\nimport { generateJSON } from '@tiptap/html/dist/server';\nimport {\n LegacyConverterEntity,\n MinimalEntity,\n} from './legacy-converter-entity';\n\nconst tiptapExtensions = [\n Text,\n Document,\n Paragraph,\n Bold,\n Italic,\n Strike,\n Underline,\n Code,\n HardBreak,\n Blockquote,\n Heading,\n HorizontalRule,\n Link.configure({\n openOnClick: false,\n }),\n];\n\nexport class LegacyCustomShortcut implements LegacyConverterEntity {\n _id: string;\n\n created: string;\n\n lastUpdated: string;\n\n shortcut: string;\n\n content: string;\n\n isDynamic: boolean;\n\n constructor(data: Partial) {\n Object.assign(this, data);\n }\n\n async convert(): Promise> {\n // Convert legacy format to new format\n // Legacy: { shortcut: string, content: string, isDynamic: boolean }\n // New: { name: string, shortcut: Description (TipTap format) }\n\n // Step 1: Wrap legacy shortcuts in code tags to preserve them during HTML parsing\n const contentWithWrappedShortcuts = this.wrapLegacyShortcuts(this.content);\n\n // Step 2: Parse HTML with wrapped shortcuts to TipTap JSON format\n const doc = generateJSON(\n contentWithWrappedShortcuts || '

',\n tiptapExtensions,\n ) as Description;\n\n // Step 3: Convert legacy shortcuts to modern format\n let blocks: TipTapNode[] = doc.content ?? [];\n blocks = this.convertLegacyToModernShortcut(blocks);\n\n // Step 4: Convert default shortcuts to block-level elements\n blocks = this.convertDefaultToBlock(blocks);\n\n const shortcut: Description = { type: 'doc', content: blocks };\n\n return {\n // eslint-disable-next-line no-underscore-dangle\n id: this._id,\n name: this.shortcut, // Legacy shortcut name becomes the name\n shortcut,\n };\n }\n\n /**\n * Recursively traverses TipTap tree to find code-marked text nodes that are\n * legacy shortcuts and converts them to modern format in place.\n */\n private convertLegacyToModernShortcut(blocks: TipTapNode[]): TipTapNode[] {\n // Pattern matches:\n // {word} or {word:text} or {word[modifier]:text} or {word[modifier]}\n // Captures: (1) shortcut key, (2) optional modifier (ignored), (3) optional value\n const shortcutPattern =\n /^\\{([a-zA-Z0-9]+)(?:\\[([^\\]]+)\\])?(?::([^}]+))?\\}$/;\n\n // Mapping of legacy system shortcuts to new inline shortcut types\n const systemShortcutMapping: Record = {\n cw: 'contentWarningShortcut',\n title: 'titleShortcut',\n tags: 'tagsShortcut',\n };\n\n // Mapping of legacy username shortcut keys to modern IDs\n const usernameShortcutMapping: Record = {\n ac: 'artconomy',\n bsky: 'bluesky',\n da: 'deviantart',\n db: 'derpibooru',\n e6: 'e621',\n fa: 'furaffinity',\n furb: 'furbooru',\n hf: 'h-foundry',\n ib: 'inkbunny',\n it: 'itaku',\n mb: 'manebooru',\n ng: 'newgrounds',\n pa: 'patreon',\n pf: 'pillowfort',\n ptv: 'picarto',\n pz: 'piczel',\n sf: 'sofurry',\n ss: 'subscribe-star',\n tu: 'tumblr',\n tw: 'twitter',\n ws: 'weasyl',\n };\n\n const hasCodeMark = (item: any): boolean =>\n Array.isArray(item.marks) &&\n item.marks.some((m: any) => m.type === 'code');\n\n const processInlineContent = (content: any[]): any[] => {\n const result: any[] = [];\n\n content.forEach((item: any) => {\n // Check if this is a code-marked text node (legacy shortcut)\n if (\n item.type === 'text' &&\n hasCodeMark(item) &&\n typeof item.text === 'string'\n ) {\n const match = item.text.match(shortcutPattern);\n\n if (match) {\n const shortcutKey = match[1];\n const shortcutKeyLower = shortcutKey.toLowerCase();\n\n // Check if this is a system shortcut (cw, title, tags)\n if (systemShortcutMapping[shortcutKeyLower]) {\n result.push({\n type: systemShortcutMapping[shortcutKeyLower],\n attrs: {},\n });\n return;\n }\n // match[2] is the modifier block - we ignore it\n const shortcutValue = match[3]; // Value is now in capture group 3\n\n // Check if this is a username shortcut (has a value after colon)\n if (shortcutValue) {\n const modernId =\n usernameShortcutMapping[shortcutKey.toLowerCase()];\n\n if (modernId) {\n // Convert to username shortcut format\n result.push({\n type: 'username',\n attrs: {\n shortcut: modernId,\n only: '',\n username: shortcutValue,\n },\n });\n return;\n }\n\n // Has a colon but not a username shortcut - convert to customShortcut\n result.push({\n type: 'customShortcut',\n attrs: { id: shortcutKey },\n content: [{ type: 'text', text: shortcutValue }],\n });\n return;\n }\n\n // Simple shortcut without colon - convert to customShortcut\n result.push({\n type: 'customShortcut',\n attrs: { id: shortcutKey },\n content: [{ type: 'text', text: '' }],\n });\n return;\n }\n\n // If it doesn't match shortcut pattern, return as is\n result.push(item);\n return;\n }\n\n result.push(item);\n });\n\n return result;\n };\n\n return blocks.map((block: any) => {\n if (!Array.isArray(block.content)) {\n return block;\n }\n\n // Check if content contains inline nodes (text) or block nodes (paragraph, etc.)\n const hasInlineContent = block.content.some(\n (c: any) => c.type === 'text',\n );\n\n if (hasInlineContent) {\n // Process inline content for shortcut conversion\n // eslint-disable-next-line no-param-reassign\n block.content = processInlineContent(block.content);\n } else {\n // Recursively process nested block content (e.g. blockquote > paragraph)\n // eslint-disable-next-line no-param-reassign\n block.content = this.convertLegacyToModernShortcut(block.content);\n }\n\n return block;\n });\n }\n\n /**\n * Converts {default} shortcuts to block-level elements.\n * - If default is alone in a block, convert the block to type: 'defaultShortcut'\n * - If default is with other content, remove it and insert a defaultShortcut block before\n */\n private convertDefaultToBlock(blocks: TipTapNode[]): TipTapNode[] {\n const result: any[] = [];\n\n blocks.forEach((block: any) => {\n // Recursively process nested block content (e.g. blockquote)\n if (\n Array.isArray(block.content) &&\n !block.content.some((c: any) => c.type === 'text')\n ) {\n // eslint-disable-next-line no-param-reassign\n block.content = this.convertDefaultToBlock(block.content);\n }\n\n // Check if this block has content with a default customShortcut\n if (Array.isArray(block.content)) {\n const defaultShortcutIndex = block.content.findIndex(\n (item: any) =>\n item.type === 'customShortcut' && item.attrs?.id === 'default',\n );\n\n if (defaultShortcutIndex !== -1) {\n // Found a default shortcut in this block's content\n const otherContent = block.content.filter(\n (item: any, idx: number) => idx !== defaultShortcutIndex,\n );\n\n // Check if there are other content items (text, links, etc.)\n const hasOtherContent = otherContent.some((item: any) =>\n item.type === 'text'\n ? item.text.trim().length > 0\n : item.type !== 'customShortcut' || item.attrs?.id !== 'default',\n );\n\n if (hasOtherContent) {\n // Default was with other content - insert defaultShortcut block before this one\n result.push({\n type: 'defaultShortcut',\n attrs: {},\n });\n\n // Add the current block without the default shortcut\n result.push({\n ...block,\n content: otherContent,\n });\n } else {\n // Default was alone - convert this block to defaultShortcut type\n result.push({\n type: 'defaultShortcut',\n attrs: {},\n });\n }\n return;\n }\n }\n\n // No default shortcut found, keep block as is\n result.push(block);\n });\n\n return result;\n }\n\n /**\n * Wraps legacy shortcuts in code tags to preserve them during HTML parsing.\n * Supports:\n * - Simple shortcuts: {default}, {customshortcut}\n * - Username shortcuts: {fa:username}, {tw:handle}\n * - Dynamic shortcuts: {myshortcut:text}\n * - Modifiers (stripped): {fa[only=furaffinity]:username}, {shortcut[modifier]}\n *\n * Ignores deprecated shortcuts: {cw}, {title}, {tags}\n *\n * Uses tags to mark shortcuts so TipTap will preserve them as\n * code-marked text nodes that can be detected and converted.\n */\n private wrapLegacyShortcuts(content: string): string {\n // Pattern matches:\n // {word} or {word:text} or {word[modifier]:text} or {word[modifier]}\n // where word is alphanumeric, modifier is anything except ], and text can contain anything except }\n const shortcutPattern = /\\{([a-zA-Z0-9]+)(?:\\[([^\\]]+)\\])?(?::([^}]+))?\\}/g;\n\n return content.replace(\n shortcutPattern,\n (match, key, modifier, additionalText) =>\n // Use tag which TipTap preserves as a code mark on text nodes\n // This will create a text node with a code mark that we can identify\n `${match}`,\n );\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/legacy-entities/legacy-tag-converter.ts", "content": "import { ITagConverter } from '@postybirb/types';\nimport { WebsiteNameMapper } from '../utils/website-name-mapper';\nimport {\n LegacyConverterEntity,\n MinimalEntity,\n} from './legacy-converter-entity';\n\n/**\n * Legacy tag converter entity from PostyBirb Plus\n * Converts a tag to website-specific tags\n */\nexport class LegacyTagConverter\n implements LegacyConverterEntity\n{\n _id: string;\n\n created: number;\n\n lastUpdated: number;\n\n tag: string;\n\n conversions: Record; // Legacy website ID -> converted tag\n\n constructor(data: Partial) {\n Object.assign(this, data);\n }\n\n async convert(): Promise> {\n const conversionsMap: Record = {};\n for (const [legacyWebsiteId, convertedTag] of Object.entries(\n this.conversions,\n )) {\n const newWebsiteId = WebsiteNameMapper.map(legacyWebsiteId);\n if (newWebsiteId) {\n conversionsMap[newWebsiteId] = convertedTag;\n }\n }\n\n return {\n // eslint-disable-next-line no-underscore-dangle\n id: this._id,\n tag: this.tag,\n convertTo: conversionsMap,\n };\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/legacy-entities/legacy-tag-group.ts", "content": "import { ITagGroup } from '@postybirb/types';\nimport {\n LegacyConverterEntity,\n MinimalEntity,\n} from './legacy-converter-entity';\n\n/**\n * Legacy tag group entity from PostyBirb Plus\n * Represents a group of tags that can be applied together\n */\nexport class LegacyTagGroup implements LegacyConverterEntity {\n _id: string;\n\n created: number;\n\n lastUpdated: number;\n\n alias: string;\n\n tags: string[];\n\n constructor(data: Partial) {\n Object.assign(this, data);\n }\n\n async convert(): Promise> {\n return {\n // eslint-disable-next-line no-underscore-dangle\n id: this._id,\n name: this.alias,\n tags: this.tags,\n };\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/legacy-entities/legacy-user-account.ts", "content": "import { IAccount } from '@postybirb/types';\nimport { WebsiteNameMapper } from '../utils/website-name-mapper';\nimport {\n LegacyConverterEntity,\n MinimalEntity,\n} from './legacy-converter-entity';\n\n/**\n * Legacy user account entity from PostyBirb Plus\n * Represents a user's account on a specific website\n */\nexport class LegacyUserAccount implements LegacyConverterEntity {\n _id: string;\n\n created: number;\n\n lastUpdated: number;\n\n alias: string;\n\n website: string;\n\n data: unknown; // Website-specific data (handled by LegacyWebsiteData converter)\n\n constructor(data: Partial) {\n Object.assign(this, data);\n }\n\n async convert(): Promise | null> {\n const newWebsiteId = WebsiteNameMapper.map(this.website);\n\n // Skip accounts for deprecated websites\n if (!newWebsiteId) {\n return null;\n }\n\n return {\n // eslint-disable-next-line no-underscore-dangle\n id: this._id,\n name: this.alias,\n website: newWebsiteId,\n groups: [], // Groups weren't part of legacy accounts\n };\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/legacy-entities/legacy-website-data.ts", "content": "import { Logger } from '@postybirb/logger';\nimport { IWebsiteData } from '@postybirb/types';\nimport { WebsiteDataTransformerRegistry } from '../transformers';\nimport { WebsiteNameMapper } from '../utils/website-name-mapper';\nimport { LegacyConverterEntity, MinimalEntity } from './legacy-converter-entity';\n\nconst logger = Logger('LegacyWebsiteData');\n\n/**\n * Legacy website data entity from PostyBirb Plus.\n * This reads the same legacy account records but extracts and transforms\n * the website-specific data (OAuth tokens, API keys, credentials) into\n * WebsiteData records.\n *\n * Only websites with registered transformers will produce records.\n * Websites using browser cookies for authentication (e.g., FurAffinity)\n * will return null and be skipped.\n */\nexport class LegacyWebsiteData implements LegacyConverterEntity {\n _id: string;\n\n created: number;\n\n lastUpdated: number;\n\n alias: string;\n\n website: string;\n\n data: unknown;\n\n constructor(data: Partial) {\n Object.assign(this, data);\n }\n\n async convert(): Promise | null> {\n const newWebsiteId = WebsiteNameMapper.map(this.website);\n\n // Skip accounts for deprecated websites\n if (!newWebsiteId) {\n return null;\n }\n\n // Only process websites that have a data transformer\n const transformer = WebsiteDataTransformerRegistry.getTransformer(\n this.website,\n );\n\n if (!transformer) {\n logger.debug(\n `No transformer for website \"${this.website}\" (account: ${this.alias}). ` +\n 'This website likely uses browser cookies for authentication.',\n );\n return null;\n }\n\n if (!this.data) {\n logger.warn(\n `Account \"${this.alias}\" (${this.website}) has transformer but no data to transform.`,\n );\n return null;\n }\n\n const transformedData = transformer.transform(this.data);\n\n if (!transformedData) {\n logger.warn(\n `Transformer returned null for account \"${this.alias}\" (${this.website}).`,\n );\n return null;\n }\n\n return {\n // WebsiteData uses the same ID as the Account (foreign key relationship)\n // eslint-disable-next-line no-underscore-dangle\n id: this._id,\n data: transformedData as Record,\n };\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/transformers/implementations/bluesky-data-transformer.ts", "content": "import { BlueskyAccountData } from '@postybirb/types';\nimport { LegacyWebsiteDataTransformer } from '../legacy-website-data-transformer';\n\n/**\n * Legacy Bluesky account data structure from PostyBirb Plus\n */\ninterface LegacyBlueskyAccountData {\n username: string;\n password: string;\n}\n\n/**\n * Transforms legacy Bluesky account data to modern format.\n * This is a direct passthrough as the structure is identical.\n *\n * Field mappings:\n * - username → username\n * - password → password\n */\nexport class BlueskyDataTransformer\n implements LegacyWebsiteDataTransformer\n{\n transform(legacyData: LegacyBlueskyAccountData): BlueskyAccountData | null {\n if (!legacyData) {\n return null;\n }\n\n // Must have credentials to be useful\n if (!legacyData.username || !legacyData.password) {\n return null;\n }\n\n return {\n username: legacyData.username,\n password: legacyData.password,\n };\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/transformers/implementations/custom-data-transformer.ts", "content": "import { CustomAccountData } from '@postybirb/types';\nimport { LegacyWebsiteDataTransformer } from '../legacy-website-data-transformer';\n\n/**\n * Legacy Custom account data structure from PostyBirb Plus\n * Note: Legacy has a typo \"thumbnaiField\" instead of \"thumbnailField\"\n */\ninterface LegacyCustomAccountData {\n descriptionField?: string;\n descriptionType?: 'html' | 'text' | 'md' | 'bbcode';\n fileField?: string;\n fileUrl?: string;\n headers: { name: string; value: string }[];\n notificationUrl?: string;\n ratingField?: string;\n tagField?: string;\n thumbnaiField?: string; // Typo in legacy\n titleField?: string;\n altTextField?: string;\n}\n\n/**\n * Transforms legacy Custom account data to modern format.\n *\n * Field mappings:\n * - All fields pass through directly\n * - thumbnaiField (typo) → thumbnailField\n */\nexport class CustomDataTransformer\n implements LegacyWebsiteDataTransformer\n{\n transform(legacyData: LegacyCustomAccountData): CustomAccountData | null {\n if (!legacyData) {\n return null;\n }\n\n // Must have at least file URL to be useful\n if (!legacyData.fileUrl && !legacyData.notificationUrl) {\n return null;\n }\n\n return {\n descriptionField: legacyData.descriptionField,\n descriptionType: legacyData.descriptionType,\n fileField: legacyData.fileField,\n fileUrl: legacyData.fileUrl,\n headers: legacyData.headers ?? [],\n notificationUrl: legacyData.notificationUrl,\n ratingField: legacyData.ratingField,\n tagField: legacyData.tagField,\n // Fix typo from legacy: thumbnaiField → thumbnailField\n thumbnailField: legacyData.thumbnaiField,\n titleField: legacyData.titleField,\n altTextField: legacyData.altTextField,\n };\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/transformers/implementations/discord-data-transformer.ts", "content": "import { DiscordAccountData } from '@postybirb/types';\nimport { LegacyWebsiteDataTransformer } from '../legacy-website-data-transformer';\n\n/**\n * Legacy Discord account data structure from PostyBirb Plus\n */\ninterface LegacyDiscordAccountData {\n webhook: string;\n serverBoostLevel: number;\n name: string;\n forum: boolean;\n}\n\n/**\n * Transforms legacy Discord account data to modern format.\n *\n * Field mappings:\n * - webhook → webhook\n * - serverBoostLevel → serverLevel\n * - forum → isForum\n * - name is not used in modern (account name is stored separately)\n */\nexport class DiscordDataTransformer\n implements LegacyWebsiteDataTransformer\n{\n transform(legacyData: LegacyDiscordAccountData): DiscordAccountData | null {\n if (!legacyData) {\n return null;\n }\n\n // Must have webhook URL to be useful\n if (!legacyData.webhook) {\n return null;\n }\n\n return {\n webhook: legacyData.webhook,\n serverLevel: legacyData.serverBoostLevel ?? 0,\n isForum: legacyData.forum ?? false,\n };\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/transformers/implementations/e621-data-transformer.ts", "content": "import { E621AccountData } from '@postybirb/types';\nimport { LegacyWebsiteDataTransformer } from '../legacy-website-data-transformer';\n\n/**\n * Legacy e621 account data structure from PostyBirb Plus\n */\ninterface LegacyE621AccountData {\n username: string;\n key: string; // API key\n}\n\n/**\n * Transforms legacy e621 account data to modern format.\n * This is a direct passthrough as the structure is identical.\n *\n * Field mappings:\n * - username → username\n * - key → key\n */\nexport class E621DataTransformer\n implements LegacyWebsiteDataTransformer\n{\n transform(legacyData: LegacyE621AccountData): E621AccountData | null {\n if (!legacyData) {\n return null;\n }\n\n // Must have API key to be useful\n if (!legacyData.key) {\n return null;\n }\n\n return {\n username: legacyData.username ?? '',\n key: legacyData.key,\n };\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/transformers/implementations/index.ts", "content": "export * from './bluesky-data-transformer';\nexport * from './custom-data-transformer';\nexport * from './discord-data-transformer';\nexport * from './e621-data-transformer';\nexport * from './inkbunny-data-transformer';\nexport * from './megalodon-data-transformer';\nexport * from './telegram-data-transformer';\nexport * from './twitter-data-transformer';\n\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/transformers/implementations/inkbunny-data-transformer.ts", "content": "import { InkbunnyAccountData } from '@postybirb/types';\nimport { LegacyWebsiteDataTransformer } from '../legacy-website-data-transformer';\n\n/**\n * Legacy Inkbunny account data structure from PostyBirb Plus\n */\ninterface LegacyInkbunnyAccountData {\n username: string;\n sid: string; // Session ID\n}\n\n/**\n * Transforms legacy Inkbunny account data to modern format.\n * This is mostly a direct passthrough.\n *\n * Field mappings:\n * - username → username\n * - sid → sid\n * - folders: undefined (will be fetched on login)\n */\nexport class InkbunnyDataTransformer\n implements LegacyWebsiteDataTransformer\n{\n transform(legacyData: LegacyInkbunnyAccountData): InkbunnyAccountData | null {\n if (!legacyData) {\n return null;\n }\n\n // Must have session ID to be useful\n if (!legacyData.sid) {\n return null;\n }\n\n return {\n username: legacyData.username,\n sid: legacyData.sid,\n folders: undefined, // Will be populated on login\n };\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/transformers/implementations/megalodon-data-transformer.ts", "content": "import { MegalodonAccountData } from '@postybirb/types';\nimport { LegacyWebsiteDataTransformer } from '../legacy-website-data-transformer';\n\n/**\n * Legacy Megalodon account data structure from PostyBirb Plus\n * Used by Mastodon, Pleroma, and Pixelfed\n */\ninterface LegacyMegalodonAccountData {\n token: string; // Access token\n website: string; // Instance URL (e.g., \"mastodon.social\")\n username: string;\n}\n\n/**\n * Transforms legacy Megalodon-based account data to modern format.\n * This transformer is used for Mastodon, Pleroma, and Pixelfed.\n *\n * Field mappings:\n * - token → accessToken\n * - website → instanceUrl\n * - username → username\n *\n * Note: OAuth client credentials (clientId, clientSecret) are not\n * preserved from legacy. The token should still work, but users\n * may need to re-authenticate if the token expires.\n */\nexport class MegalodonDataTransformer\n implements LegacyWebsiteDataTransformer\n{\n transform(legacyData: LegacyMegalodonAccountData): MegalodonAccountData | null {\n if (!legacyData) {\n return null;\n }\n\n // Must have token and instance to be useful\n if (!legacyData.token || !legacyData.website) {\n return null;\n }\n\n return {\n accessToken: legacyData.token,\n instanceUrl: this.normalizeInstanceUrl(legacyData.website),\n username: legacyData.username,\n // These are not available from legacy but may be populated on next login\n clientId: undefined,\n clientSecret: undefined,\n displayName: undefined,\n instanceType: undefined,\n };\n }\n\n /**\n * Normalize instance URL to consistent format (without protocol or trailing slash).\n */\n private normalizeInstanceUrl(url: string): string {\n let normalized = url.trim().toLowerCase();\n normalized = normalized.replace(/^(https?:\\/\\/)/, '');\n normalized = normalized.replace(/\\/$/, '');\n return normalized;\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/transformers/implementations/telegram-data-transformer.ts", "content": "import { TelegramAccountData } from '@postybirb/types';\nimport { LegacyWebsiteDataTransformer } from '../legacy-website-data-transformer';\n\n/**\n * Legacy Telegram account data structure from PostyBirb Plus\n */\ninterface LegacyTelegramAccountData {\n appId: string; // Note: stored as string in legacy\n appHash: string;\n phoneNumber: string;\n}\n\n/**\n * Transforms legacy Telegram account data to modern format.\n *\n * Field mappings:\n * - appId (string) → appId (number)\n * - appHash → appHash\n * - phoneNumber → phoneNumber\n * - session: undefined (must be re-authenticated)\n * - channels: [] (must be re-fetched after login)\n *\n * Note: Legacy Telegram sessions cannot be migrated as they use\n * a different session storage mechanism. Users will need to\n * re-authenticate after import.\n */\nexport class TelegramDataTransformer\n implements LegacyWebsiteDataTransformer\n{\n transform(legacyData: LegacyTelegramAccountData): TelegramAccountData | null {\n if (!legacyData) {\n return null;\n }\n\n // Must have app credentials to be useful\n if (!legacyData.appId || !legacyData.appHash) {\n return null;\n }\n\n return {\n appId: parseInt(legacyData.appId, 10),\n appHash: legacyData.appHash,\n phoneNumber: legacyData.phoneNumber ?? '',\n session: undefined, // Cannot migrate session, requires re-auth\n channels: [], // Will be populated after re-authentication\n };\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/transformers/implementations/twitter-data-transformer.ts", "content": "import { TwitterAccountData } from '@postybirb/types';\nimport { LegacyWebsiteDataTransformer } from '../legacy-website-data-transformer';\n\n/**\n * Legacy Twitter account data structure from PostyBirb Plus\n */\ninterface LegacyTwitterAccountData {\n key: string; // Consumer key (API key)\n secret: string; // Consumer secret (API secret)\n oauth_token: string; // Access token\n oauth_token_secret: string; // Access token secret\n screen_name: string; // Username\n user_id: number; // User ID\n}\n\n/**\n * Transforms legacy Twitter account data to modern format.\n *\n * Field mappings:\n * - key → apiKey\n * - secret → apiSecret\n * - oauth_token → accessToken\n * - oauth_token_secret → accessTokenSecret\n * - screen_name → screenName\n * - user_id → userId (number to string)\n */\nexport class TwitterDataTransformer\n implements LegacyWebsiteDataTransformer\n{\n transform(legacyData: LegacyTwitterAccountData): TwitterAccountData | null {\n if (!legacyData) {\n return null;\n }\n\n // Must have OAuth tokens to be useful\n if (!legacyData.oauth_token || !legacyData.oauth_token_secret) {\n return null;\n }\n\n return {\n apiKey: legacyData.key,\n apiSecret: legacyData.secret,\n accessToken: legacyData.oauth_token,\n accessTokenSecret: legacyData.oauth_token_secret,\n screenName: legacyData.screen_name,\n userId: legacyData.user_id?.toString(),\n };\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/transformers/index.ts", "content": "export * from './legacy-website-data-transformer';\nexport * from './website-data-transformer-registry';\n\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/transformers/legacy-website-data-transformer.ts", "content": "/**\n * Interface for transforming legacy website-specific account data\n * to modern WebsiteData format.\n */\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport interface LegacyWebsiteDataTransformer {\n /**\n * Transform legacy account data to modern WebsiteData format.\n * @param legacyData The legacy website-specific data from account.data\n * @returns The transformed data for modern WebsiteData.data, or null if transformation fails\n */\n transform(legacyData: TLegacy): TModern | null;\n}\n\n/**\n * Base transformer that passes through data unchanged.\n * Useful for websites where the data structure is already compatible.\n */\nexport class PassthroughTransformer>\n implements LegacyWebsiteDataTransformer\n{\n transform(legacyData: T): T | null {\n if (!legacyData) {\n return null;\n }\n return { ...legacyData };\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/transformers/website-data-transformer-registry.ts", "content": "import { BlueskyDataTransformer } from './implementations/bluesky-data-transformer';\nimport { CustomDataTransformer } from './implementations/custom-data-transformer';\nimport { DiscordDataTransformer } from './implementations/discord-data-transformer';\nimport { E621DataTransformer } from './implementations/e621-data-transformer';\nimport { InkbunnyDataTransformer } from './implementations/inkbunny-data-transformer';\nimport { MegalodonDataTransformer } from './implementations/megalodon-data-transformer';\nimport { TelegramDataTransformer } from './implementations/telegram-data-transformer';\nimport { TwitterDataTransformer } from './implementations/twitter-data-transformer';\nimport { LegacyWebsiteDataTransformer } from './legacy-website-data-transformer';\n\n/**\n * Registry that maps legacy website names to their data transformers.\n * Only websites with custom login flows that store credentials in WebsiteData\n * need transformers here.\n */\nexport class WebsiteDataTransformerRegistry {\n private static readonly transformers: Record<\n string,\n LegacyWebsiteDataTransformer\n > = {\n // OAuth/API key websites\n Twitter: new TwitterDataTransformer(),\n Discord: new DiscordDataTransformer(),\n Telegram: new TelegramDataTransformer(),\n\n // Megalodon-based fediverse websites (all use same transformer)\n Mastodon: new MegalodonDataTransformer(),\n Pleroma: new MegalodonDataTransformer(),\n Pixelfed: new MegalodonDataTransformer(),\n\n // Direct credential websites\n Bluesky: new BlueskyDataTransformer(),\n Inkbunny: new InkbunnyDataTransformer(),\n e621: new E621DataTransformer(),\n\n // Custom webhook website\n Custom: new CustomDataTransformer(),\n };\n\n /**\n * Get the transformer for a legacy website name.\n * @param legacyWebsiteName The legacy website name (e.g., \"Twitter\", \"Mastodon\")\n * @returns The transformer instance, or undefined if no transformer exists\n */\n static getTransformer(\n legacyWebsiteName: string,\n ): LegacyWebsiteDataTransformer | undefined {\n return this.transformers[legacyWebsiteName];\n }\n\n /**\n * Check if a legacy website has a data transformer.\n * Websites without transformers typically use browser cookies for auth.\n * @param legacyWebsiteName The legacy website name\n */\n static hasTransformer(legacyWebsiteName: string): boolean {\n return legacyWebsiteName in this.transformers;\n }\n\n /**\n * Get all legacy website names that have transformers.\n */\n static getTransformableWebsites(): string[] {\n return Object.keys(this.transformers);\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/utils/ndjson-parser.ts", "content": "import { Injectable } from '@nestjs/common';\nimport { Logger } from '@postybirb/logger';\nimport { promises as fs } from 'fs';\n\nexport interface ParseResult {\n records: T[];\n errors: ParseError[];\n}\n\nexport interface ParseError {\n line: number;\n content: string;\n error: string;\n}\n\n/**\n * Simple NDJSON (Newline Delimited JSON) parser for NeDB files.\n * Each line in a NeDB file is a separate JSON object.\n */\n@Injectable()\nexport class NdjsonParser {\n private readonly logger = Logger(NdjsonParser.name);\n\n /**\n * Parse an NDJSON file and instantiate objects of the specified class\n */\n async parseFile(\n filePath: string,\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n EntityClass: new (data: any) => T,\n ): Promise> {\n const records: T[] = [];\n const errors: ParseError[] = [];\n\n try {\n const content = await fs.readFile(filePath, 'utf-8');\n const lines = content.split('\\n');\n\n for (let i = 0; i < lines.length; i++) {\n const line = lines[i].trim();\n\n // Skip empty lines\n if (!line) {\n continue;\n }\n\n try {\n const parsed = JSON.parse(line);\n const instance = new EntityClass(parsed);\n records.push(instance);\n } catch (error) {\n errors.push({\n line: i + 1,\n content: line.substring(0, 100), // Truncate long lines\n error: error instanceof Error ? error.message : String(error),\n });\n }\n }\n\n if (errors.length > 0) {\n this.logger.warn(\n `Parsed ${filePath}: ${records.length} records, ${errors.length} errors`,\n );\n } else {\n this.logger.info(`Parsed ${filePath}: ${records.length} records`);\n }\n\n return { records, errors };\n } catch (error) {\n if ((error as NodeJS.ErrnoException).code === 'ENOENT') {\n this.logger.warn(`File not found: ${filePath}`);\n return { records: [], errors: [] };\n }\n\n this.logger.error(`Error reading ${filePath}: ${error}`);\n throw error;\n }\n }\n\n /**\n * Check if a file exists\n */\n async fileExists(filePath: string): Promise {\n try {\n await fs.access(filePath);\n return true;\n } catch {\n return false;\n }\n }\n}\n" }, { "path": "apps/client-server/src/app/legacy-database-importer/utils/website-name-mapper.ts", "content": "/**\n * Maps legacy PostyBirb Plus website names to new PostyBirb V4 website IDs.\n * The website ID in V4 is the value from the @WebsiteMetadata({ name }) decorator.\n * Legacy uses PascalCase names, V4 uses kebab-case IDs.\n */\nexport class WebsiteNameMapper {\n private static readonly LEGACY_TO_NEW: Record = {\n // Direct mappings (kebab-case in V4)\n Artconomy: 'artconomy',\n Aryion: 'aryion',\n Bluesky: 'bluesky',\n Custom: 'custom',\n Derpibooru: 'derpibooru',\n DeviantArt: 'deviant-art',\n Discord: 'discord',\n FurAffinity: 'fur-affinity',\n Furbooru: 'furbooru',\n FurryNetwork: null, // Deprecated - no longer exists in V4\n HentaiFoundry: 'hentai-foundry',\n Inkbunny: 'inkbunny',\n Itaku: 'itaku',\n KoFi: 'ko-fi',\n Manebooru: 'manebooru',\n Mastodon: 'mastodon',\n MissKey: 'misskey',\n Newgrounds: 'newgrounds',\n Patreon: 'patreon',\n Picarto: 'picarto',\n Piczel: 'piczel',\n Pillowfort: 'pillowfort',\n Pixelfed: 'pixelfed',\n Pixiv: 'pixiv',\n Pleroma: 'pleroma',\n SoFurry: 'sofurry',\n SubscribeStar: 'subscribe-star',\n SubscribeStarAdult: 'subscribe-star', // Maps to same base (Adult variant handled differently in V4)\n Telegram: 'telegram',\n Tumblr: 'tumblr',\n Twitter: 'twitter',\n Weasyl: 'weasyl',\n e621: 'e621',\n\n // New websites in V4 that didn't exist in Plus:\n // cara: 'cara',\n // firefish: 'firefish',\n // friendica: 'friendica',\n // gotosocial: 'gotosocial',\n // toyhouse: 'toyhouse',\n };\n\n /**\n * Map a legacy website name to the new website name\n */\n static map(legacyName: string): string | null {\n return this.LEGACY_TO_NEW[legacyName] || null;\n }\n\n /**\n * Map multiple legacy website names\n */\n static mapMany(\n legacyNames: string[],\n ): Array<{ legacy: string; new: string | null }> {\n return legacyNames.map((legacy) => ({\n legacy,\n new: this.map(legacy),\n }));\n }\n\n /**\n * Check if a legacy website name has a mapping\n */\n static hasMapping(legacyName: string): boolean {\n return legacyName in this.LEGACY_TO_NEW;\n }\n\n /**\n * Get all legacy website names that have mappings\n */\n static getAllLegacyNames(): string[] {\n return Object.keys(this.LEGACY_TO_NEW);\n }\n\n /**\n * Get all new website names\n */\n static getAllNewNames(): string[] {\n return Array.from(new Set(Object.values(this.LEGACY_TO_NEW)));\n }\n}\n" }, { "path": "apps/client-server/src/app/logs/logs.controller.ts", "content": "import { Controller, Get, Res } from '@nestjs/common';\nimport { ApiOkResponse, ApiTags } from '@nestjs/swagger';\nimport { LogsService } from './logs.service';\n\n/**\n * Controller for log file operations.\n * Provides an endpoint to download all logs as a .tar.gz archive.\n */\n@ApiTags('logs')\n@Controller('logs')\nexport class LogsController {\n constructor(private readonly service: LogsService) {}\n\n @Get('download')\n @ApiOkResponse({ description: 'Returns a .tar.gz archive of all log files.' })\n download(@Res() response) {\n const archive = this.service.getLogsArchive();\n const date = new Date().toISOString().split('T')[0];\n response.set({\n 'Content-Type': 'application/gzip',\n 'Content-Disposition': `attachment; filename=postybirb-logs-${date}.tar.gz`,\n 'Content-Length': archive.length,\n });\n response.send(archive);\n }\n}\n" }, { "path": "apps/client-server/src/app/logs/logs.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { LogsController } from './logs.controller';\nimport { LogsService } from './logs.service';\n\n@Module({\n providers: [LogsService],\n controllers: [LogsController],\n exports: [LogsService],\n})\nexport class LogsModule {}\n" }, { "path": "apps/client-server/src/app/logs/logs.service.ts", "content": "import { Injectable } from '@nestjs/common';\nimport { PostyBirbDirectories } from '@postybirb/fs';\nimport { Logger } from '@postybirb/logger';\nimport { readdirSync, readFileSync } from 'fs';\nimport { join } from 'path';\nimport { gzipSync } from 'zlib';\n\n/**\n * Service for managing log file operations.\n * Provides the ability to bundle all log files into a downloadable .tar.gz archive.\n */\n@Injectable()\nexport class LogsService {\n private readonly logger = Logger('LogsService');\n\n /**\n * Creates a gzipped tar archive of all files in the logs directory.\n * Uses Node.js built-in zlib (no external dependencies).\n *\n * @returns A Buffer containing the .tar.gz archive\n */\n getLogsArchive(): Buffer {\n const logsDir = PostyBirbDirectories.LOGS_DIRECTORY;\n this.logger.info(`Creating logs archive from: ${logsDir}`);\n\n const files = readdirSync(logsDir);\n const tarBuffer = this.createTar(logsDir, files);\n return gzipSync(tarBuffer);\n }\n\n /**\n * Creates a tar archive buffer from the given files.\n * Implements a minimal tar writer (POSIX ustar format) sufficient for\n * bundling flat log files.\n */\n private createTar(dir: string, fileNames: string[]): Buffer {\n const blocks: Buffer[] = [];\n\n for (const fileName of fileNames) {\n try {\n const filePath = join(dir, fileName);\n const content = readFileSync(filePath);\n const header = this.createTarHeader(fileName, content.length);\n blocks.push(header);\n blocks.push(content);\n\n // Pad content to 512-byte boundary\n const remainder = content.length % 512;\n if (remainder > 0) {\n blocks.push(Buffer.alloc(512 - remainder));\n }\n } catch (err) {\n this.logger.withError(err).warn(`Skipping file: ${fileName}`);\n }\n }\n\n // End-of-archive marker: two 512-byte blocks of zeros\n blocks.push(Buffer.alloc(1024));\n\n return Buffer.concat(blocks);\n }\n\n /**\n * Creates a 512-byte tar header for a single file entry.\n */\n private createTarHeader(fileName: string, fileSize: number): Buffer {\n const header = Buffer.alloc(512);\n\n // File name (0–99, 100 bytes)\n header.write(fileName.slice(0, 100), 0, 100, 'utf-8');\n\n // File mode (100–107, 8 bytes) — 0644\n header.write('0000644\\0', 100, 8, 'utf-8');\n\n // Owner UID (108–115, 8 bytes)\n header.write('0000000\\0', 108, 8, 'utf-8');\n\n // Group GID (116–123, 8 bytes)\n header.write('0000000\\0', 116, 8, 'utf-8');\n\n // File size in octal (124–135, 12 bytes)\n header.write(`${fileSize.toString(8).padStart(11, '0')}\\0`, 124, 12, 'utf-8');\n\n // Modification time in octal (136–147, 12 bytes)\n const mtime = Math.floor(Date.now() / 1000);\n header.write(`${mtime.toString(8).padStart(11, '0')}\\0`, 136, 12, 'utf-8');\n\n // Type flag (156, 1 byte) — '0' for regular file\n header.write('0', 156, 1, 'utf-8');\n\n // USTAR magic (257–262, 6 bytes) + version (263–264, 2 bytes)\n header.write('ustar\\0', 257, 6, 'utf-8');\n header.write('00', 263, 2, 'utf-8');\n\n // Checksum placeholder: fill with spaces first (148–155, 8 bytes)\n header.write(' ', 148, 8, 'utf-8');\n\n // Compute checksum (sum of all unsigned bytes in the header)\n let checksum = 0;\n for (let i = 0; i < 512; i++) {\n checksum += header[i];\n }\n // Write checksum in octal, null-terminated, space-padded\n header.write(`${checksum.toString(8).padStart(6, '0')}\\0 `, 148, 8, 'utf-8');\n\n return header;\n }\n}\n" }, { "path": "apps/client-server/src/app/notifications/dtos/create-notification.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { ICreateNotificationDto } from '@postybirb/types';\nimport { IsArray, IsObject, IsString } from 'class-validator';\n\nexport class CreateNotificationDto implements ICreateNotificationDto {\n @ApiProperty()\n @IsObject()\n data: Record = {};\n\n @ApiProperty()\n @IsString()\n title: string;\n\n @ApiProperty()\n @IsString()\n message: string;\n\n @ApiProperty()\n @IsArray()\n @IsString({ each: true })\n tags: string[] = [];\n\n @ApiProperty()\n @IsString()\n type: 'warning' | 'error' | 'info' | 'success' = 'info';\n}\n" }, { "path": "apps/client-server/src/app/notifications/dtos/update-notification.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { IUpdateNotificationDto } from '@postybirb/types';\nimport { IsBoolean } from 'class-validator';\n\nexport class UpdateNotificationDto implements IUpdateNotificationDto {\n @ApiProperty()\n @IsBoolean()\n isRead: boolean;\n\n @ApiProperty()\n @IsBoolean()\n hasEmitted: boolean;\n}\n" }, { "path": "apps/client-server/src/app/notifications/notification.events.ts", "content": "import { NOTIFICATION_UPDATES } from '@postybirb/socket-events';\nimport { INotification } from '@postybirb/types';\nimport { WebsocketEvent } from '../web-socket/models/web-socket-event';\n\nexport type NotificationEventTypes = NotificationEvent;\n\nclass NotificationEvent implements WebsocketEvent {\n event: string = NOTIFICATION_UPDATES;\n\n data: INotification[];\n}\n" }, { "path": "apps/client-server/src/app/notifications/notifications.controller.ts", "content": "import {\n Body,\n Controller,\n Delete,\n Get,\n Param,\n Patch,\n Post,\n Query,\n} from '@nestjs/common';\nimport {\n ApiBadRequestResponse,\n ApiNotFoundResponse,\n ApiOkResponse,\n ApiTags,\n} from '@nestjs/swagger';\nimport { EntityId } from '@postybirb/types';\nimport { CreateNotificationDto } from './dtos/create-notification.dto';\nimport { UpdateNotificationDto } from './dtos/update-notification.dto';\nimport { NotificationsService } from './notifications.service';\n\n/**\n * @class NotificationsController\n */\n@ApiTags('notifications')\n@Controller('notifications')\nexport class NotificationsController {\n constructor(readonly service: NotificationsService) {}\n\n @Post()\n @ApiOkResponse({ description: 'Notification created.' })\n @ApiBadRequestResponse({ description: 'Bad request made.' })\n create(@Body() createDto: CreateNotificationDto) {\n return this.service.create(createDto).then((entity) => entity.toDTO());\n }\n\n @Patch(':id')\n @ApiOkResponse({ description: 'Notification updated.' })\n @ApiNotFoundResponse({ description: 'Notification not found.' })\n update(@Param('id') id: EntityId, @Body() updateDto: UpdateNotificationDto) {\n return this.service.update(id, updateDto).then((entity) => entity.toDTO());\n }\n\n @Get()\n @ApiOkResponse({ description: 'A list of all records.' })\n findAll() {\n return this.service\n .findAll()\n .then((records) => records.map((record) => record.toDTO()));\n }\n\n @Delete()\n @ApiOkResponse({ description: 'Notification deleted.' })\n @ApiNotFoundResponse({ description: 'Notification not found.' })\n async remove(@Query('ids') ids: EntityId | EntityId[]) {\n return Promise.all(\n (Array.isArray(ids) ? ids : [ids]).map((id) => this.service.remove(id)),\n ).then(() => ({\n success: true,\n }));\n }\n}\n" }, { "path": "apps/client-server/src/app/notifications/notifications.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { AccountModule } from '../account/account.module';\nimport { SettingsModule } from '../settings/settings.module';\nimport { UserSpecifiedWebsiteOptionsModule } from '../user-specified-website-options/user-specified-website-options.module';\nimport { WebsitesModule } from '../websites/websites.module';\nimport { NotificationsController } from './notifications.controller';\nimport { NotificationsService } from './notifications.service';\n\n@Module({\n imports: [\n WebsitesModule,\n UserSpecifiedWebsiteOptionsModule,\n AccountModule,\n SettingsModule,\n ],\n providers: [NotificationsService],\n controllers: [NotificationsController],\n exports: [NotificationsService],\n})\nexport class NotificationsModule {}\n" }, { "path": "apps/client-server/src/app/notifications/notifications.service.spec.ts", "content": "import { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport { SettingsService } from '../settings/settings.service';\nimport { WSGateway } from '../web-socket/web-socket-gateway';\nimport { CreateNotificationDto } from './dtos/create-notification.dto';\nimport { UpdateNotificationDto } from './dtos/update-notification.dto';\nimport { NotificationsService } from './notifications.service';\n\ndescribe('NotificationsService', () => {\n let service: NotificationsService;\n let module: TestingModule;\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n let webSocketMock: any;\n\n beforeEach(async () => {\n clearDatabase();\n webSocketMock = {\n emit: jest.fn(),\n };\n\n module = await Test.createTestingModule({\n providers: [\n NotificationsService,\n SettingsService,\n {\n provide: WSGateway,\n useValue: webSocketMock,\n },\n ],\n }).compile();\n\n service = module.get(NotificationsService);\n });\n\n afterAll(async () => {\n await module?.close();\n });\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n\n it('should create a notification', async () => {\n const dto = new CreateNotificationDto();\n dto.title = 'Test Notification';\n dto.message = 'This is a test notification';\n dto.type = 'info';\n\n const notification = await service.create(dto);\n expect(notification).toBeDefined();\n expect(notification.title).toBe(dto.title);\n expect(notification.message).toBe(dto.message);\n expect(notification.type).toBe(dto.type);\n\n const notifications = await service.findAll();\n expect(notifications).toHaveLength(1);\n expect(notifications[0].id).toBe(notification.id);\n });\n\n it('should update a notification', async () => {\n const createDto = new CreateNotificationDto();\n createDto.title = 'Initial Title';\n createDto.message = 'Initial Message';\n createDto.type = 'info';\n\n const notification = await service.create(createDto);\n\n const updateDto = new UpdateNotificationDto();\n updateDto.isRead = true;\n\n await service.update(notification.id, updateDto);\n\n const updatedNotification = await service.findById(notification.id);\n expect(updatedNotification.message).toBe(createDto.message); // unchanged\n expect(updatedNotification.isRead).toBe(true);\n });\n\n it('should emit notification updates when changes occur', async () => {\n // Create a notification which should trigger an emit\n const dto = new CreateNotificationDto();\n dto.title = 'Test Notification';\n dto.message = 'This is a test notification';\n dto.type = 'info';\n\n await service.create(dto);\n\n // Verify websocket emit was called with the correct event\n expect(webSocketMock.emit).toHaveBeenCalled();\n const emitArgs = webSocketMock.emit.mock.calls[0];\n expect(emitArgs[0].data[0].title).toBe(dto.title);\n });\n\n it('should initialize without websocket and not throw error', () => {\n // @ts-expect-error Test case\n const serviceWithoutWebsocket = new NotificationsService();\n expect(serviceWithoutWebsocket).toBeDefined();\n });\n});\n" }, { "path": "apps/client-server/src/app/notifications/notifications.service.ts", "content": "import { Injectable, Optional } from '@nestjs/common';\nimport { Cron, CronExpression } from '@nestjs/schedule';\nimport { NOTIFICATION_UPDATES } from '@postybirb/socket-events';\nimport { EntityId } from '@postybirb/types';\nimport { Notification as ElectronNotification } from 'electron';\nimport { PostyBirbService } from '../common/service/postybirb-service';\nimport { Notification } from '../drizzle/models/notification.entity';\nimport { SettingsService } from '../settings/settings.service';\nimport { WSGateway } from '../web-socket/web-socket-gateway';\nimport { CreateNotificationDto } from './dtos/create-notification.dto';\nimport { UpdateNotificationDto } from './dtos/update-notification.dto';\n\n/**\n * Service responsible for managing application notifications.\n * Handles creation, updating, and removal of notifications, as well as\n * sending desktop notifications based on user settings.\n */\n@Injectable()\nexport class NotificationsService extends PostyBirbService<'NotificationSchema'> {\n /**\n * Creates a new instance of the NotificationsService.\n *\n * @param settingsService - Service for accessing application settings\n * @param webSocket - Optional websocket gateway for emitting events\n */\n constructor(\n private readonly settingsService: SettingsService,\n @Optional() webSocket?: WSGateway,\n ) {\n super('NotificationSchema', webSocket);\n this.repository.subscribe('NotificationSchema', () => this.emit());\n this.removeStaleNotifications();\n }\n\n /**\n * Removes notifications older than one month.\n * Runs automatically every hour via cron job.\n */\n @Cron(CronExpression.EVERY_HOUR)\n private async removeStaleNotifications() {\n const notifications = await this.repository.findAll();\n\n const aMonthAgo = new Date();\n aMonthAgo.setMonth(aMonthAgo.getMonth() - 1);\n const staleNotifications = notifications.filter(\n (notification) =>\n new Date(notification.createdAt).getTime() < aMonthAgo.getTime(),\n );\n if (staleNotifications.length) {\n await this.repository.deleteById(staleNotifications.map((n) => n.id));\n }\n }\n\n /**\n * Creates a new notification and optionally sends a desktop notification.\n *\n * @param createDto - The notification data to create\n * @param sendDesktopNotification - Whether to also send a desktop notification\n * @returns The created notification entity\n */\n async create(\n createDto: CreateNotificationDto,\n sendDesktopNotification = false,\n ): Promise {\n this.logger\n .withMetadata(createDto)\n .info(`Creating notification '${createDto.title}'`);\n if (sendDesktopNotification) {\n this.sendDesktopNotification(createDto);\n }\n\n return this.repository.insert(createDto);\n }\n\n /**\n * Trims notifications to a maximum of 250, removing the oldest first.\n * Runs every 5 minutes.\n */\n @Cron(CronExpression.EVERY_5_MINUTES)\n private async trimNotifications() {\n const notifications = await this.repository.findAll();\n if (notifications.length <= 250) {\n return;\n }\n const sorted = notifications.sort(\n (a, b) =>\n new Date(a.createdAt).getTime() - new Date(b.createdAt).getTime(),\n );\n const toRemove = sorted.slice(0, notifications.length - 250);\n await this.repository.deleteById(toRemove.map((n) => n.id));\n }\n\n /**\n * Sends a desktop notification based on user settings and notification type.\n *\n * @param notification - The notification data to display\n */\n async sendDesktopNotification(\n notification: CreateNotificationDto,\n ): Promise {\n const { settings } = await this.settingsService.getDefaultSettings();\n const { desktopNotifications } = settings;\n const { tags, title, message, type } = notification;\n if (!desktopNotifications.enabled) {\n return;\n }\n\n if (\n desktopNotifications.showOnDirectoryWatcherError &&\n tags.includes('directory-watcher') &&\n type === 'error'\n ) {\n new ElectronNotification({\n title,\n body: message,\n }).show();\n }\n\n if (\n desktopNotifications.showOnDirectoryWatcherSuccess &&\n tags.includes('directory-watcher') &&\n type === 'success'\n ) {\n new ElectronNotification({\n title,\n body: message,\n }).show();\n }\n\n if (\n desktopNotifications.showOnPostError &&\n tags.includes('post') &&\n type === 'error'\n ) {\n new ElectronNotification({\n title,\n body: message,\n }).show();\n }\n\n if (\n desktopNotifications.showOnPostSuccess &&\n tags.includes('post') &&\n type === 'success'\n ) {\n new ElectronNotification({\n title,\n body: message,\n }).show();\n }\n }\n\n /**\n * Updates an existing notification.\n *\n * @param id - The ID of the notification to update\n * @param update - The data to update\n * @returns The updated notification\n */\n update(id: EntityId, update: UpdateNotificationDto) {\n this.logger.withMetadata(update).info(`Updating notification '${id}'`);\n return this.repository.update(id, update);\n }\n\n /**\n * Emits notification updates to connected clients.\n * Converts entities to DTOs before sending.\n */\n protected async emit() {\n super.emit({\n event: NOTIFICATION_UPDATES,\n data: (await this.repository.findAll()).map((entity) => entity.toDTO()),\n });\n }\n}\n" }, { "path": "apps/client-server/src/app/post/dtos/post-queue-action.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { IPostQueueActionDto, PostRecordResumeMode } from '@postybirb/types';\nimport { ArrayNotEmpty, IsArray, IsEnum, IsOptional } from 'class-validator';\n\nexport class PostQueueActionDto implements IPostQueueActionDto {\n @ApiProperty()\n @IsArray()\n @ArrayNotEmpty()\n submissionIds: string[];\n\n @ApiProperty({ enum: PostRecordResumeMode, required: false })\n @IsOptional()\n @IsEnum(PostRecordResumeMode)\n resumeMode?: PostRecordResumeMode;\n}\n" }, { "path": "apps/client-server/src/app/post/dtos/queue-post-record.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { IQueuePostRecordRequestDto, SubmissionId } from '@postybirb/types';\nimport { ArrayNotEmpty, IsArray } from 'class-validator';\n\n/** @inheritdoc */\nexport class QueuePostRecordRequestDto implements IQueuePostRecordRequestDto {\n @ApiProperty()\n @IsArray()\n @ArrayNotEmpty()\n ids: SubmissionId[];\n}\n" }, { "path": "apps/client-server/src/app/post/errors/index.ts", "content": "export * from './invalid-post-chain.error';\n\n" }, { "path": "apps/client-server/src/app/post/errors/invalid-post-chain.error.ts", "content": "import { EntityId, PostRecordResumeMode } from '@postybirb/types';\n\n/**\n * Error thrown when attempting to create a PostRecord in an invalid state.\n *\n * This indicates a bug in the calling code - the caller should have verified\n * the submission state before requesting a PostRecord creation.\n *\n * Error reasons:\n * - no_origin: CONTINUE/RETRY requested but no prior NEW PostRecord exists\n * - origin_done: CONTINUE/RETRY requested but the origin NEW record is DONE (chain closed)\n * - in_progress: A PostRecord for this submission is already PENDING or RUNNING\n */\nexport class InvalidPostChainError extends Error {\n public readonly submissionId: EntityId;\n\n public readonly requestedResumeMode: PostRecordResumeMode;\n\n public readonly reason: 'no_origin' | 'origin_done' | 'in_progress';\n\n constructor(\n submissionId: EntityId,\n requestedResumeMode: PostRecordResumeMode,\n reason: 'no_origin' | 'origin_done' | 'in_progress',\n ) {\n let message: string;\n // eslint-disable-next-line default-case\n switch (reason) {\n case 'no_origin':\n message = `Cannot create ${requestedResumeMode} PostRecord for submission ${submissionId}: no prior NEW PostRecord found to chain from`;\n break;\n case 'origin_done':\n message = `Cannot create ${requestedResumeMode} PostRecord for submission ${submissionId}: the origin NEW PostRecord is already DONE (chain is closed)`;\n break;\n case 'in_progress':\n message = `Cannot create ${requestedResumeMode} PostRecord for submission ${submissionId}: a PostRecord is already PENDING or RUNNING`;\n break;\n }\n\n super(message);\n this.name = 'InvalidPostChainError';\n this.submissionId = submissionId;\n this.requestedResumeMode = requestedResumeMode;\n this.reason = reason;\n }\n}\n" }, { "path": "apps/client-server/src/app/post/models/cancellable-token.ts", "content": "import { CancellationError } from './cancellation-error';\n\n/**\n * CancellableToken is a simple class that can be used to cancel a task.\n * @class CancellableToken\n */\nexport class CancellableToken {\n private cancelled = false;\n\n public get isCancelled(): boolean {\n return this.cancelled;\n }\n\n public cancel(): void {\n this.cancelled = true;\n }\n\n public throwIfCancelled(): void {\n if (this.cancelled) {\n throw new CancellationError();\n }\n }\n}\n" }, { "path": "apps/client-server/src/app/post/models/cancellation-error.ts", "content": "/**\n * CancellationError is thrown when a task is cancelled.\n * @class CancellationError\n */\nexport class CancellationError extends Error {\n constructor(message = 'Task was cancelled.') {\n super(message);\n this.name = 'CancellationError';\n\n // Maintains proper stack trace for where our error was thrown (only available on V8)\n if (Error.captureStackTrace) {\n Error.captureStackTrace(this, CancellationError);\n }\n }\n}\n" }, { "path": "apps/client-server/src/app/post/models/posting-file.ts", "content": "import { FormFile } from '@postybirb/http';\nimport {\n FileType,\n IFileBuffer,\n SubmissionFileId,\n SubmissionFileMetadata,\n} from '@postybirb/types';\nimport { getFileType } from '@postybirb/utils/file-type';\nimport { parse } from 'path';\n\nexport type ThumbnailOptions = Pick<\n IFileBuffer,\n 'buffer' | 'height' | 'width' | 'mimeType' | 'fileName'\n>;\n\nexport type FormDataFileFormat = {\n value: Buffer;\n options: {\n contentType: string;\n filename: string;\n };\n};\n\nexport class PostingFile {\n public readonly id: SubmissionFileId;\n\n public readonly buffer: Buffer;\n\n public readonly mimeType: string;\n\n public readonly fileType: FileType;\n\n public readonly fileName: string;\n\n public readonly width: number;\n\n public readonly height: number;\n\n public metadata: SubmissionFileMetadata;\n\n public readonly thumbnail?: ThumbnailOptions;\n\n public constructor(\n id: SubmissionFileId,\n file: IFileBuffer,\n thumbnail?: ThumbnailOptions,\n ) {\n this.id = id;\n this.buffer = file.buffer;\n this.mimeType = file.mimeType;\n this.width = file.width;\n this.height = file.height;\n this.fileType = getFileType(file.fileName);\n this.fileName = this.normalizeFileName(file);\n this.thumbnail = thumbnail;\n }\n\n private normalizeFileName(file: IFileBuffer): string {\n const { ext } = parse(file.fileName);\n return `${file.id}${ext}`;\n }\n\n public withMetadata(metadata: SubmissionFileMetadata): PostingFile {\n this.metadata = metadata;\n return this;\n }\n\n public toPostFormat(): FormFile {\n return new FormFile(this.buffer, {\n contentType: this.mimeType,\n filename: this.fileName,\n });\n }\n\n public thumbnailToPostFormat(): FormFile | undefined {\n if (!this.thumbnail) {\n return undefined;\n }\n\n return new FormFile(this.thumbnail.buffer, {\n contentType: this.thumbnail.mimeType,\n filename: this.thumbnail.fileName,\n });\n }\n}\n" }, { "path": "apps/client-server/src/app/post/post.controller.ts", "content": "import { Controller, Get, Param } from '@nestjs/common';\nimport { ApiOkResponse, ApiTags } from '@nestjs/swagger';\nimport { EntityId } from '@postybirb/types';\nimport { PostyBirbController } from '../common/controller/postybirb-controller';\nimport { PostService } from './post.service';\n\n/**\n * Queue operations for Post data.\n * @class PostController\n */\n@ApiTags('post')\n@Controller('post')\nexport class PostController extends PostyBirbController<'PostRecordSchema'> {\n constructor(readonly service: PostService) {\n super(service);\n }\n\n /**\n * Get all events for a specific post record.\n * Returns the immutable event ledger showing all posting actions.\n *\n * @param {EntityId} id - The post record ID\n * @returns {Promise} Array of post events\n */\n @Get(':id/events')\n @ApiOkResponse({ description: 'Events for the post record.' })\n async getEvents(@Param('id') id: EntityId) {\n return this.service.getEvents(id);\n }\n}\n" }, { "path": "apps/client-server/src/app/post/post.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { FileConverterModule } from '../file-converter/file-converter.module';\nimport { NotificationsModule } from '../notifications/notifications.module';\nimport { PostParsersModule } from '../post-parsers/post-parsers.module';\nimport { SettingsModule } from '../settings/settings.module';\nimport { SubmissionModule } from '../submission/submission.module';\nimport { ValidationModule } from '../validation/validation.module';\nimport { WebsiteOptionsModule } from '../website-options/website-options.module';\nimport { WebsiteImplProvider } from '../websites/implementations/provider';\nimport { WebsitesModule } from '../websites/websites.module';\nimport { PostController } from './post.controller';\nimport { PostService } from './post.service';\nimport { PostFileResizerService } from './services/post-file-resizer/post-file-resizer.service';\nimport {\n FileSubmissionPostManager,\n MessageSubmissionPostManager,\n PostManagerRegistry,\n} from './services/post-manager-v2';\nimport { PostManagerController } from './services/post-manager/post-manager.controller';\nimport { PostQueueController } from './services/post-queue/post-queue.controller';\nimport { PostQueueService } from './services/post-queue/post-queue.service';\nimport {\n PostEventRepository,\n PostRecordFactory,\n} from './services/post-record-factory';\n\n@Module({\n imports: [\n WebsiteOptionsModule,\n WebsitesModule,\n PostParsersModule,\n ValidationModule,\n FileConverterModule,\n SettingsModule,\n SubmissionModule,\n NotificationsModule,\n ],\n controllers: [PostController, PostQueueController, PostManagerController],\n providers: [\n PostService,\n PostFileResizerService,\n WebsiteImplProvider,\n PostQueueService,\n PostEventRepository,\n PostRecordFactory,\n FileSubmissionPostManager,\n MessageSubmissionPostManager,\n PostManagerRegistry,\n ],\n exports: [PostEventRepository, PostRecordFactory, PostManagerRegistry],\n})\nexport class PostModule {}\n" }, { "path": "apps/client-server/src/app/post/post.service.ts", "content": "import { Injectable, Optional } from '@nestjs/common';\nimport { EntityId, PostEventDto } from '@postybirb/types';\nimport { PostyBirbService } from '../common/service/postybirb-service';\nimport { PostyBirbDatabase } from '../drizzle/postybirb-database/postybirb-database';\nimport { WSGateway } from '../web-socket/web-socket-gateway';\n\n/**\n * Simple entity service for post records.\n * @class PostService\n */\n@Injectable()\nexport class PostService extends PostyBirbService<'PostRecordSchema'> {\n private readonly postEventRepository = new PostyBirbDatabase(\n 'PostEventSchema',\n );\n\n constructor(@Optional() webSocket?: WSGateway) {\n super('PostRecordSchema', webSocket);\n }\n\n /**\n * Get all events for a specific post record.\n *\n * @param {EntityId} postRecordId - The post record ID\n * @returns {Promise} Array of post events\n */\n async getEvents(postRecordId: EntityId): Promise {\n const events = await this.postEventRepository.find({\n where: (event, { eq }) => eq(event.postRecordId, postRecordId),\n orderBy: (event, { asc }) => asc(event.createdAt),\n with: {\n account: true,\n },\n });\n\n return events.map((event) => event.toDTO());\n }\n}\n" }, { "path": "apps/client-server/src/app/post/services/post-file-resizer/post-file-resizer.service.spec.ts", "content": "import { Test, TestingModule } from '@nestjs/testing';\nimport {\n DefaultSubmissionFileMetadata,\n ISubmission,\n ISubmissionFile,\n} from '@postybirb/types';\nimport { readFileSync } from 'fs';\nimport { join } from 'path';\nimport { SharpInstanceManager } from '../../../image-processing/sharp-instance-manager';\nimport { PostFileResizerService } from './post-file-resizer.service';\n\ndescribe('PostFileResizerService', () => {\n let service: PostFileResizerService;\n let sharpManager: SharpInstanceManager;\n let module: TestingModule;\n let testFile: Buffer;\n let file: ISubmissionFile;\n\n function createFile(\n fileName: string,\n mimeType: string,\n height: number,\n width: number,\n buf: Buffer,\n ): ISubmissionFile {\n return {\n id: 'test',\n fileName,\n hash: 'test',\n mimeType,\n size: buf.length,\n hasThumbnail: false,\n hasCustomThumbnail: false,\n hasAltFile: false,\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n submission: {} as ISubmission,\n width,\n height,\n primaryFileId: 'test',\n submissionId: 'test',\n metadata: DefaultSubmissionFileMetadata(),\n order: 0,\n file: {\n submissionFileId: 'test',\n fileName,\n mimeType,\n id: 'test',\n buffer: buf,\n size: buf.length,\n width,\n height,\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n },\n };\n }\n\n beforeAll(async () => {\n testFile = readFileSync(\n join(__dirname, '../../../../test-files/small_image.jpg'),\n );\n file = createFile('test.jpg', 'image/jpeg', 202, 138, testFile);\n\n module = await Test.createTestingModule({\n providers: [PostFileResizerService, SharpInstanceManager],\n }).compile();\n\n service = module.get(PostFileResizerService);\n sharpManager = module.get(SharpInstanceManager);\n });\n\n afterAll(async () => {\n await module.close();\n });\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n\n it('should resize image', async () => {\n const resized = await service.resize({ file, resize: { width: 100 } });\n expect(resized.buffer.length).toBeLessThan(testFile.length);\n const metadata = await sharpManager.getMetadata(resized.buffer);\n expect(metadata.width).toBe(100);\n expect(metadata.height).toBeLessThan(202);\n expect(resized.fileName).toBe('test.jpeg');\n expect(resized.thumbnail).toBeDefined();\n });\n\n it('should not resize image', async () => {\n const resized = await service.resize({ file, resize: { width: 300 } });\n expect(resized.buffer.length).toBe(testFile.length);\n const metadata = await sharpManager.getMetadata(resized.buffer);\n expect(metadata.width).toBe(file.width);\n expect(metadata.height).toBe(file.height);\n expect(resized.fileName).toBe('test.jpg');\n });\n\n it('should scale down image', async () => {\n const resized = await service.resize({\n file,\n resize: { maxBytes: testFile.length - 1000 },\n });\n expect(resized.buffer.length).toBeLessThan(testFile.length);\n expect(resized.thumbnail?.buffer.length).toBeLessThan(testFile.length);\n expect(resized.fileName).toBe('test.jpeg');\n expect(resized.thumbnail?.fileName).toBe('test.jpg');\n expect(resized.mimeType).toBe('image/jpeg');\n });\n\n it('should fail to scale down image', async () => {\n await expect(\n service.resize({\n file,\n resize: { maxBytes: -1 },\n }),\n ).rejects.toThrow();\n });\n\n it('should not convert png thumbnail with alpha to jpeg', async () => {\n const noAlphaFile = readFileSync(\n join(__dirname, '../../../../test-files/png_with_alpha.png'),\n );\n const tf = createFile('test.png', 'image/png', 600, 600, noAlphaFile);\n const resized = await service.resize({\n file: tf,\n resize: { maxBytes: noAlphaFile.length - 1000 },\n });\n\n expect(resized.buffer.length).toBeLessThan(noAlphaFile.length);\n expect(resized.fileName).toBe('test.png');\n expect(resized.thumbnail?.buffer.length).toBeLessThan(noAlphaFile.length);\n expect(resized.thumbnail?.fileName).toBe('test.png');\n });\n});\n" }, { "path": "apps/client-server/src/app/post/services/post-file-resizer/post-file-resizer.service.ts", "content": "import { Injectable } from '@nestjs/common';\nimport { Logger } from '@postybirb/logger';\nimport {\n FileType,\n IFileBuffer,\n ImageResizeProps,\n ISubmissionFile,\n} from '@postybirb/types';\nimport { getFileType } from '@postybirb/utils/file-type';\nimport { SharpInstanceManager } from '../../../image-processing/sharp-instance-manager';\nimport { PostingFile, ThumbnailOptions } from '../../models/posting-file';\n\ntype ResizeRequest = {\n file: ISubmissionFile;\n resize?: ImageResizeProps;\n};\n\n/**\n * Responsible for resizing an image file to a smaller size before\n * posting to a website.\n *\n * All sharp/libvips work is delegated to the SharpInstanceManager,\n * which runs sharp in isolated worker threads. If libvips crashes\n * (e.g. after long idle), only the worker dies — the main process\n * survives.\n *\n * @class PostFileResizer\n */\n@Injectable()\nexport class PostFileResizerService {\n private readonly logger = Logger();\n\n constructor(\n private readonly sharpInstanceManager: SharpInstanceManager,\n ) {}\n\n public async resize(request: ResizeRequest): Promise {\n return this.process(request);\n }\n\n private async process(request: ResizeRequest): Promise {\n const { resize } = request;\n const { file } = request;\n this.logger.withMetadata({ resize }).info('Resizing image...');\n\n if (!file.file) {\n throw new Error('File buffer is missing');\n }\n\n const primaryFile = await this.processPrimaryFile(file, resize);\n const thumbnail = await this.processThumbnailFile(file);\n\n const newPostingFile = new PostingFile(\n file.id,\n primaryFile,\n thumbnail,\n );\n\n newPostingFile.metadata = file.metadata;\n return newPostingFile;\n }\n\n private async processPrimaryFile(\n file: ISubmissionFile,\n resize?: ImageResizeProps,\n ): Promise {\n if (!resize) return file.file;\n\n const result = await this.sharpInstanceManager.resizeForPost({\n buffer: file.file.buffer,\n resize,\n mimeType: file.mimeType,\n fileName: file.fileName,\n fileId: file.id,\n fileWidth: file.file.width,\n fileHeight: file.file.height,\n generateThumbnail: false,\n });\n\n if (result.modified && result.buffer) {\n return {\n ...file.file,\n fileName: result.fileName || `${file.id}.${result.format}`,\n buffer: result.buffer,\n mimeType: result.mimeType || file.mimeType,\n height: result.height || file.file.height,\n width: result.width || file.file.width,\n };\n }\n\n return file.file;\n }\n\n private async processThumbnailFile(\n file: ISubmissionFile,\n ): Promise {\n let thumb = file.thumbnail;\n const shouldProcessThumbnail =\n !!thumb || getFileType(file.fileName) === FileType.IMAGE;\n\n if (!shouldProcessThumbnail) {\n return undefined;\n }\n\n thumb = thumb ?? { ...file.file }; // Ensure file to process\n\n const result = await this.sharpInstanceManager.generateThumbnail(\n thumb.buffer,\n thumb.mimeType,\n thumb.fileName,\n 500,\n );\n\n return {\n buffer: result.buffer,\n fileName: thumb.fileName,\n mimeType: thumb.mimeType,\n height: result.height,\n width: result.width,\n };\n }\n}\n" }, { "path": "apps/client-server/src/app/post/services/post-manager/post-manager.controller.ts", "content": "import { Controller, Get, Param, Post } from '@nestjs/common';\nimport { ApiOkResponse, ApiTags } from '@nestjs/swagger';\nimport { SubmissionId, SubmissionType } from '@postybirb/types';\nimport { PostManagerRegistry } from '../post-manager-v2';\nimport { PostQueueService } from '../post-queue/post-queue.service';\n\n@ApiTags('post-manager')\n@Controller('post-manager')\nexport class PostManagerController {\n constructor(\n readonly service: PostManagerRegistry,\n private readonly postQueueService: PostQueueService,\n ) {}\n\n @Post('cancel/:id')\n @ApiOkResponse({ description: 'Post cancelled if running.' })\n async cancelIfRunning(@Param('id') id: SubmissionId) {\n // Use post-queue's dequeue which ensures all db records are properly handled\n await this.postQueueService.dequeue([id]);\n return true;\n }\n\n @Get('is-posting/:submissionType')\n @ApiOkResponse({ description: 'Check if a post is in progress.' })\n async isPosting(@Param('submissionType') submissionType: SubmissionType) {\n return { isPosting: this.service.getManager(submissionType).isPosting() };\n }\n}\n" }, { "path": "apps/client-server/src/app/post/services/post-manager-v2/base-post-manager.service.ts", "content": "import {\n Logger,\n trackEvent,\n trackException,\n trackMetric,\n} from '@postybirb/logger';\nimport {\n AccountId,\n EntityId,\n PostData,\n PostEventType,\n PostRecordState,\n PostResponse,\n SubmissionType,\n} from '@postybirb/types';\nimport {\n PostRecord,\n Submission,\n WebsiteOptions,\n} from '../../../drizzle/models';\nimport { PostyBirbDatabase } from '../../../drizzle/postybirb-database/postybirb-database';\nimport { NotificationsService } from '../../../notifications/notifications.service';\nimport { PostParsersService } from '../../../post-parsers/post-parsers.service';\nimport { ValidationService } from '../../../validation/validation.service';\nimport { UnknownWebsite, Website } from '../../../websites/website';\nimport { WebsiteRegistryService } from '../../../websites/website-registry.service';\nimport { CancellableToken } from '../../models/cancellable-token';\nimport { CancellationError } from '../../models/cancellation-error';\nimport { PostEventRepository, ResumeContext } from '../post-record-factory';\n\n/**\n * Website info for posting order.\n */\ninterface WebsiteInfo {\n accountId: AccountId;\n instance: Website;\n}\n\n/**\n * Abstract base class for PostManager implementations.\n * Handles common posting logic and event emission.\n * @abstract\n * @class BasePostManager\n */\nexport abstract class BasePostManager {\n protected readonly logger = Logger(this.constructor.name);\n\n protected readonly lastTimePostedToWebsite: Record = {};\n\n /**\n * The current post being processed.\n */\n protected currentPost: PostRecord | null = null;\n\n /**\n * The current cancel token for the current post.\n */\n protected cancelToken: CancellableToken | null = null;\n\n /**\n * Resume context from prior attempts.\n */\n protected resumeContext: ResumeContext | null = null;\n\n protected readonly postRepository: PostyBirbDatabase<'PostRecordSchema'>;\n\n constructor(\n protected readonly postEventRepository: PostEventRepository,\n protected readonly websiteRegistry: WebsiteRegistryService,\n protected readonly postParserService: PostParsersService,\n protected readonly validationService: ValidationService,\n protected readonly notificationService: NotificationsService,\n ) {\n this.postRepository = new PostyBirbDatabase('PostRecordSchema');\n }\n\n /**\n * Get the submission type this manager handles.\n * @abstract\n * @returns {SubmissionType} The submission type\n */\n abstract getSupportedType(): SubmissionType;\n\n /**\n * Cancels the current post if it is running and matches the Id.\n * @param {EntityId} submissionId - The submission ID to check\n * @returns {Promise} True if the post was cancelled\n */\n public async cancelIfRunning(submissionId: EntityId): Promise {\n if (this.currentPost && this.currentPost.submissionId === submissionId) {\n this.logger.info(`Cancelling current post`);\n this.cancelToken?.cancel();\n return true;\n }\n return false;\n }\n\n /**\n * Check if this manager is currently posting.\n * @returns {boolean} True if posting\n */\n public isPosting(): boolean {\n return !!this.currentPost;\n }\n\n /**\n * Starts a post attempt.\n * @param {PostRecord} entity - The post record to start\n * @param {ResumeContext} [resumeContext] - Optional resume context from prior attempts\n */\n public async startPost(\n entity: PostRecord,\n resumeContext?: ResumeContext,\n ): Promise {\n try {\n if (this.currentPost) {\n this.logger.warn(\n `PostManager is already posting, cannot start new post`,\n );\n return;\n }\n\n this.cancelToken = new CancellableToken();\n this.resumeContext = resumeContext || null;\n\n this.logger.withMetadata(entity.toDTO()).info(`Initializing post`);\n this.currentPost = entity;\n await this.postRepository.update(entity.id, {\n state: PostRecordState.RUNNING,\n });\n\n // Posts order occurs in batched groups\n // Standard websites first, then websites that accept external source urls\n this.logger.info(`Creating post order`);\n const postOrderGroups = this.getPostOrder(entity);\n\n this.logger.info(`Posting to websites`);\n for (const websites of postOrderGroups) {\n this.cancelToken.throwIfCancelled();\n await Promise.allSettled(\n websites.map((w) => this.postToWebsite(entity, w)),\n );\n }\n this.logger.info(`Finished posting to websites`);\n } catch (error) {\n this.logger.withError(error).error(`Error posting`);\n } finally {\n await this.finishPost(entity);\n }\n }\n\n /**\n * Post to a single website.\n * @protected\n * @param {PostRecord} entity - The post record\n * @param {WebsiteInfo} websiteInfo - The website information\n */\n protected async postToWebsite(\n entity: PostRecord,\n websiteInfo: WebsiteInfo,\n ): Promise {\n const { submission } = entity;\n const { accountId, instance } = websiteInfo;\n let data: PostData | undefined;\n const option = submission.options.find((o) => o.accountId === accountId);\n\n try {\n await this.ensureLoggedIn(instance);\n\n const supportedTypes = instance.getSupportedTypes();\n if (!supportedTypes.includes(submission.type)) {\n throw new Error(\n `Website '${instance.decoratedProps.metadata.displayName}' does not support ${submission.type}`,\n );\n }\n\n this.logger.info('Preparing post data');\n data = await this.preparePostData(submission, instance, option);\n this.logger.withMetadata(data).info('Post data prepared');\n\n // Emit POST_ATTEMPT_STARTED event with post data\n await this.emitPostAttemptStarted(\n entity.id,\n accountId,\n instance,\n data,\n option,\n );\n\n this.logger.info('Validating submission');\n const validationResult =\n await this.validationService.validateSubmission(submission);\n if (validationResult.some((v) => v.errors.length > 0)) {\n throw new Error('Submission contains validation errors');\n }\n\n await this.attemptToPost(entity, accountId, instance, data);\n\n // Emit POST_ATTEMPT_COMPLETED event\n await this.postEventRepository.insert({\n postRecordId: entity.id,\n accountId,\n eventType: PostEventType.POST_ATTEMPT_COMPLETED,\n metadata: {\n accountSnapshot: {\n name: instance.account.name,\n website: instance.decoratedProps.metadata.name,\n },\n },\n });\n\n this.lastTimePostedToWebsite[accountId] = new Date();\n\n // Track successful post in App Insights (detailed)\n const websiteName = instance.decoratedProps.metadata.name;\n trackEvent('PostSuccess', {\n website: websiteName,\n accountId,\n submissionId: entity.submissionId,\n submissionType: submission.type,\n hasSourceUrl: 'unknown', // Individual managers track this\n fileCount: '0',\n options: this.redactPostDataForLogging(data),\n });\n\n // Track success metric per website\n trackMetric(`post.success.${websiteName}`, 1, {\n website: websiteName,\n submissionType: submission.type,\n });\n } catch (error) {\n this.logger\n .withError(error)\n .error(`Error posting to website: ${instance.id}`);\n\n const errorResponse =\n error instanceof PostResponse\n ? error\n : PostResponse.fromWebsite(instance)\n .withException(error)\n .withMessage(\n `${\n instance.decoratedProps.metadata.displayName ||\n instance.decoratedProps.metadata.name\n }: ${String(error)}`,\n );\n\n await this.handlePostFailure(\n entity.id,\n accountId,\n instance,\n errorResponse,\n data,\n );\n\n // Track failure in App Insights (detailed)\n const websiteName = instance.decoratedProps.metadata.name;\n\n // Only track non-cancellation failures\n if (!(error instanceof CancellationError)) {\n trackEvent('PostFailure', {\n website: websiteName,\n accountId,\n submissionId: entity.submissionId,\n submissionType: submission.type,\n errorMessage: errorResponse.message ?? 'unknown',\n stage: errorResponse.stage ?? 'unknown',\n hasException: errorResponse.exception ? 'true' : 'false',\n fileCount: '0',\n options: data ? this.redactPostDataForLogging(data) : '',\n });\n\n // Track failure metric per website\n trackMetric(`post.failure.${websiteName}`, 1, {\n website: websiteName,\n submissionType: submission.type,\n });\n\n // Track the exception if present\n if (errorResponse.exception) {\n trackException(errorResponse.exception, {\n website: websiteName,\n accountId,\n submissionId: entity.submissionId,\n stage: errorResponse.stage ?? 'unknown',\n errorMessage: errorResponse.message ?? 'unknown',\n });\n }\n }\n }\n }\n\n /**\n * Emit POST_ATTEMPT_STARTED event.\n * @protected\n * @param {EntityId} postRecordId - The post record ID\n * @param {AccountId} accountId - The account ID\n * @param {Website} instance - The website instance\n * @param {PostData} [data] - The post data\n * @param {WebsiteOptions} [option] - The website options\n */\n protected async emitPostAttemptStarted(\n postRecordId: EntityId,\n accountId: AccountId,\n instance: Website,\n data?: PostData,\n option?: WebsiteOptions,\n ): Promise {\n await this.postEventRepository.insert({\n postRecordId,\n accountId,\n eventType: PostEventType.POST_ATTEMPT_STARTED,\n metadata: {\n accountSnapshot: {\n name: instance.account.name,\n website: instance.decoratedProps.metadata.name,\n },\n postData: data\n ? {\n parsedOptions: data.options,\n websiteOptions: option ? [] : [], // Blank for now; populate if needed later\n }\n : undefined,\n },\n });\n }\n\n /**\n * Handle post failure and emit appropriate events.\n * @protected\n * @param {EntityId} postRecordId - The post record ID\n * @param {AccountId} accountId - The account ID\n * @param {Website} instance - The website instance\n * @param {PostResponse} errorResponse - The error response\n * @param {PostData} [postData] - The post data\n */\n protected async handlePostFailure(\n postRecordId: EntityId,\n accountId: AccountId,\n instance: Website,\n errorResponse: PostResponse,\n postData?: PostData,\n ): Promise {\n await this.postEventRepository.insert({\n postRecordId,\n accountId,\n eventType: PostEventType.POST_ATTEMPT_FAILED,\n error: {\n message: errorResponse.message || 'Unknown error',\n stack: errorResponse.exception?.stack,\n stage: errorResponse.stage,\n additionalInfo: errorResponse.additionalInfo,\n },\n metadata: {\n accountSnapshot: {\n name: instance.account.name,\n website: instance.decoratedProps.metadata.name,\n },\n },\n });\n\n await this.notificationService.create({\n type: 'error',\n title: `Failed to post to ${instance.decoratedProps.metadata.displayName}`,\n message: errorResponse.message || 'Unknown error',\n tags: ['post-failure', instance.decoratedProps.metadata.name],\n data: {},\n });\n }\n\n /**\n * Attempt to post to the website based on submission type.\n * @abstract\n * @protected\n * @param {PostRecord} entity - The post record\n * @param {AccountId} accountId - The account ID\n * @param {UnknownWebsite} instance - The website instance\n * @param {PostData} data - The post data\n */\n protected abstract attemptToPost(\n entity: PostRecord,\n accountId: AccountId,\n instance: UnknownWebsite,\n data: PostData,\n ): Promise;\n\n /**\n * Prepare post data for a website.\n * @protected\n * @param {Submission} submission - The submission\n * @param {Website} instance - The website instance\n * @param {WebsiteOptions} [option] - The website options\n * @returns {Promise} The prepared post data\n */\n protected async preparePostData(\n submission: Submission,\n instance: Website,\n option?: WebsiteOptions,\n ): Promise {\n return this.postParserService.parse(submission, instance, option);\n }\n\n /**\n * Get post order for websites.\n * Groups websites into batches - standard websites first, then websites that accept external sources.\n * @protected\n * @param {PostRecord} entity - The post record\n * @returns {WebsiteInfo[][]} Batched website info\n */\n protected getPostOrder(entity: PostRecord): WebsiteInfo[][] {\n const { submission } = entity;\n const websiteInfos: WebsiteInfo[] = [];\n\n for (const option of submission.options) {\n const instance = this.websiteRegistry.findInstance(option.account);\n if (!instance) {\n this.logger.warn(`Website instance not found for ${option.accountId}`);\n continue;\n }\n\n if (!instance.getSupportedTypes().includes(submission.type)) {\n this.logger.warn(\n `Website ${instance.id} does not support ${submission.type}`,\n );\n continue;\n }\n\n // Skip if account is completed (based on resume context)\n if (\n this.resumeContext &&\n this.resumeContext.completedAccountIds.has(option.accountId)\n ) {\n this.logger.info(\n `Skipping account ${option.accountId} - already completed`,\n );\n continue;\n }\n\n websiteInfos.push({\n accountId: option.accountId,\n instance,\n });\n }\n\n // Split into batches: standard websites first, then websites that accept external sources\n const standard: WebsiteInfo[] = [];\n const acceptsExternal: WebsiteInfo[] = [];\n\n for (const info of websiteInfos) {\n // Check if website accepts source URLs by looking at fileOptions\n const acceptsSourceUrls =\n info.instance.decoratedProps.fileOptions?.acceptsExternalSourceUrls ??\n false;\n if (acceptsSourceUrls) {\n acceptsExternal.push(info);\n } else {\n standard.push(info);\n }\n }\n\n const batches: WebsiteInfo[][] = [];\n if (standard.length > 0) batches.push(standard);\n if (acceptsExternal.length > 0) batches.push(acceptsExternal);\n\n return batches;\n }\n\n /**\n * Finish the post and update the post record state.\n * @protected\n * @param {PostRecord} entity - The post record\n */\n protected async finishPost(entity: PostRecord): Promise {\n this.currentPost = null;\n this.cancelToken = null;\n this.resumeContext = null;\n\n const entityInDb = await this.postRepository.findById(entity.id);\n if (!entityInDb) {\n this.logger.error(\n `Entity ${entity.id} not found in database. It may have been deleted while posting.`,\n );\n return;\n }\n\n // Query events to determine if post was successful\n const failedEvents = await this.postEventRepository.getFailedEvents(\n entity.id,\n );\n\n // DONE only if there are zero failures; any failure (including partial) means FAILED\n const state =\n failedEvents.length > 0 ? PostRecordState.FAILED : PostRecordState.DONE;\n\n await this.postRepository.update(entity.id, {\n state,\n completedAt: new Date().toISOString(),\n });\n\n trackMetric(\n 'Post Duration',\n Date.now() - new Date(entity.createdAt).getTime(),\n {\n submissionType: entity.submission.type,\n state,\n },\n );\n\n this.logger.info(`Post ${entity.id} finished with state: ${state}`);\n }\n\n /**\n * Wait for posting wait interval to avoid rate limiting.\n * Uses the website's configured minimumPostWaitInterval.\n * @protected\n * @param {AccountId} accountId - The account ID\n * @param {Website} instance - The website instance\n */\n protected async waitForPostingWaitInterval(\n accountId: AccountId,\n instance: Website,\n ): Promise {\n const lastTime = this.lastTimePostedToWebsite[accountId];\n if (!lastTime) return;\n\n const waitInterval =\n instance.decoratedProps.metadata.minimumPostWaitInterval ?? 0;\n if (!waitInterval) return;\n\n const now = new Date();\n const timeSinceLastPost = now.getTime() - lastTime.getTime();\n\n if (timeSinceLastPost < waitInterval) {\n const waitTime = waitInterval - timeSinceLastPost;\n this.logger.info(\n `Waiting ${waitTime}ms before posting to ${instance.id}`,\n );\n await new Promise((resolve) => {\n setTimeout(resolve, waitTime);\n });\n }\n }\n\n /**\n * Ensures the website instance is logged in before posting.\n * Delegates to website.login() which is mutex-guarded:\n * - If a login is already in progress, waits for it to finish.\n * - If not logged in and no login in progress, triggers a fresh login.\n * @protected\n * @param {Website} instance - The website instance\n * @throws {Error} If the website is still not logged in after the login attempt\n */\n protected async ensureLoggedIn(instance: Website): Promise {\n const state = instance.getLoginState();\n if (state.isLoggedIn) {\n return;\n }\n\n this.logger.info(\n `Not logged in for ${instance.id}, triggering login...`,\n );\n\n const result = await instance.login();\n\n if (!result.isLoggedIn) {\n throw new Error('Not logged in');\n }\n\n this.logger.info(`Login resolved for ${instance.id} — now logged in`);\n }\n\n /**\n * Redact sensitive information from post data for logging.\n * @protected\n * @param {PostData} postData - The post data\n * @returns {string} Redacted post data as JSON string\n */\n protected redactPostDataForLogging(postData: PostData): string {\n const opts = { ...postData.options };\n // Redact sensitive information\n if (opts.description) {\n opts.description = `[REDACTED ${opts.description.length}]`;\n }\n return JSON.stringify({ options: opts });\n }\n}\n" }, { "path": "apps/client-server/src/app/post/services/post-manager-v2/file-submission-post-manager.service.spec.ts", "content": "import { clearDatabase } from '@postybirb/database';\nimport {\n AccountId,\n DefaultSubmissionFileMetadata,\n EntityId,\n FileSubmissionMetadata,\n FileType,\n IPostResponse,\n PostData,\n PostEventType,\n PostRecordState,\n SubmissionRating,\n SubmissionType,\n} from '@postybirb/types';\nimport 'reflect-metadata';\nimport {\n FileBuffer,\n PostRecord,\n Submission,\n SubmissionFile,\n} from '../../../drizzle/models';\nimport { FileConverterService } from '../../../file-converter/file-converter.service';\nimport { NotificationsService } from '../../../notifications/notifications.service';\nimport { PostParsersService } from '../../../post-parsers/post-parsers.service';\nimport { ValidationService } from '../../../validation/validation.service';\nimport {\n FileWebsite,\n ImplementedFileWebsite,\n PostBatchData,\n} from '../../../websites/models/website-modifiers/file-website';\nimport { UnknownWebsite } from '../../../websites/website';\nimport { WebsiteRegistryService } from '../../../websites/website-registry.service';\nimport { CancellableToken } from '../../models/cancellable-token';\nimport { PostingFile } from '../../models/posting-file';\nimport { PostFileResizerService } from '../post-file-resizer/post-file-resizer.service';\nimport { PostEventRepository } from '../post-record-factory';\nimport { FileSubmissionPostManager } from './file-submission-post-manager.service';\n\ndescribe('FileSubmissionPostManager', () => {\n let manager: FileSubmissionPostManager;\n let postEventRepositoryMock: jest.Mocked;\n let websiteRegistryMock: jest.Mocked;\n let postParserServiceMock: jest.Mocked;\n let validationServiceMock: jest.Mocked;\n let notificationServiceMock: jest.Mocked;\n let resizerServiceMock: jest.Mocked;\n let fileConverterServiceMock: jest.Mocked;\n\n beforeEach(() => {\n clearDatabase();\n\n postEventRepositoryMock = {\n insert: jest.fn().mockResolvedValue({}),\n findByPostRecordId: jest.fn().mockResolvedValue([]),\n getCompletedAccounts: jest.fn().mockResolvedValue([]),\n getFailedEvents: jest.fn().mockResolvedValue([]),\n getSourceUrlsFromPost: jest.fn().mockResolvedValue([]),\n } as unknown as jest.Mocked;\n\n websiteRegistryMock = {\n findInstance: jest.fn(),\n } as unknown as jest.Mocked;\n\n postParserServiceMock = {\n parse: jest.fn(),\n } as unknown as jest.Mocked;\n\n validationServiceMock = {\n validateSubmission: jest.fn().mockResolvedValue([]),\n } as unknown as jest.Mocked;\n\n notificationServiceMock = {\n create: jest.fn(),\n sendDesktopNotification: jest.fn(),\n } as unknown as jest.Mocked;\n\n resizerServiceMock = {\n resize: jest.fn(),\n } as unknown as jest.Mocked;\n\n fileConverterServiceMock = {\n convert: jest.fn(),\n canConvert: jest.fn().mockResolvedValue(false),\n } as unknown as jest.Mocked;\n\n manager = new FileSubmissionPostManager(\n postEventRepositoryMock,\n websiteRegistryMock,\n postParserServiceMock,\n validationServiceMock,\n notificationServiceMock,\n resizerServiceMock,\n fileConverterServiceMock,\n );\n });\n\n function createFileBuffer(): FileBuffer {\n return new FileBuffer({\n id: 'test-file-buffer',\n fileName: 'test.png',\n mimeType: 'image/png',\n buffer: Buffer.from('fake-image-data'),\n size: 100,\n width: 600,\n height: 600,\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n });\n }\n\n function createSubmissionFile(\n id: string = 'test-file',\n order: number = 1,\n ): SubmissionFile {\n const fileBuffer = createFileBuffer();\n return new SubmissionFile({\n id,\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n fileName: 'test.png',\n hash: 'fake-hash',\n mimeType: 'image/png',\n size: fileBuffer.size,\n width: 600,\n height: 600,\n hasAltFile: false,\n hasThumbnail: false,\n hasCustomThumbnail: false,\n file: fileBuffer,\n order,\n metadata: DefaultSubmissionFileMetadata(),\n });\n }\n\n function createSubmission(\n files: SubmissionFile[] = [],\n ): Submission {\n return new Submission({\n id: 'test-submission',\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n metadata: {},\n type: SubmissionType.FILE,\n files,\n options: [],\n isScheduled: false,\n schedule: {} as never,\n order: 1,\n posts: [] as never,\n isTemplate: false,\n isMultiSubmission: false,\n isArchived: false,\n });\n }\n\n function createPostRecord(\n submission: Submission,\n ): PostRecord {\n return new PostRecord({\n id: 'test-post-record' as EntityId,\n submission,\n state: PostRecordState.PENDING,\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n });\n }\n\n function createMockFileWebsite(accountId: AccountId): UnknownWebsite {\n return {\n id: 'test-website',\n accountId,\n account: {\n id: accountId,\n name: 'test-account',\n },\n decoratedProps: {\n metadata: {\n name: 'Test Website',\n displayName: 'Test Website',\n },\n fileOptions: {\n acceptedMimeTypes: ['image/png', 'image/jpeg'],\n fileBatchSize: 1,\n supportedFileTypes: [FileType.IMAGE],\n },\n },\n onPostFileSubmission: jest.fn(),\n calculateImageResize: jest.fn().mockReturnValue(undefined),\n supportsFile: true,\n } as unknown as UnknownWebsite;\n }\n\n function createPostingFile(file: SubmissionFile): PostingFile {\n const postingFile = new PostingFile(file.id, file.file, file.thumbnail);\n postingFile.metadata = file.metadata;\n return postingFile;\n }\n\n describe('getSupportedType', () => {\n it('should return FILE submission type', () => {\n expect(manager.getSupportedType()).toBe(SubmissionType.FILE);\n });\n });\n\n describe('attemptToPost', () => {\n let mockWebsite: UnknownWebsite;\n let postRecord: PostRecord;\n let accountId: AccountId;\n let postData: PostData;\n let cancelToken: CancellableToken;\n let submissionFile: SubmissionFile;\n\n beforeEach(() => {\n accountId = 'test-account-id' as AccountId;\n submissionFile = createSubmissionFile();\n const submission = createSubmission([submissionFile]);\n postRecord = createPostRecord(submission);\n mockWebsite = createMockFileWebsite(accountId);\n\n cancelToken = new CancellableToken();\n (manager as any).cancelToken = cancelToken;\n (manager as any).lastTimePostedToWebsite = {};\n\n postData = {\n submission,\n options: {\n title: 'Test Title',\n description: 'Test Description',\n rating: SubmissionRating.GENERAL,\n tags: ['test'],\n },\n } as PostData;\n\n // Setup resizer mock to return a PostingFile\n resizerServiceMock.resize.mockImplementation(async (request) => {\n const file = request.file as SubmissionFile;\n return createPostingFile(file);\n });\n });\n\n it('should throw error if website does not support file submissions', async () => {\n const nonFileWebsite = {\n ...mockWebsite,\n supportsFile: false,\n decoratedProps: {\n ...mockWebsite.decoratedProps,\n metadata: {\n name: 'Message Only Website',\n displayName: 'Message Only Website',\n },\n },\n } as unknown as UnknownWebsite;\n delete (nonFileWebsite as any).onPostFileSubmission;\n\n await expect(\n (manager as any).attemptToPost(\n postRecord,\n accountId,\n nonFileWebsite,\n postData,\n ),\n ).rejects.toThrow('does not support file submissions');\n });\n\n it('should successfully post a file and emit FILE_POSTED event', async () => {\n const sourceUrl = 'https://example.com/file/123';\n const responseMessage = 'File posted successfully';\n\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission = jest\n .fn()\n .mockResolvedValue({\n instanceId: 'test-website',\n sourceUrl,\n message: responseMessage,\n });\n\n await (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n );\n\n // Verify website method was called\n expect(\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission,\n ).toHaveBeenCalledWith(postData, expect.any(Array), cancelToken, {\n index: 0,\n totalBatches: 1,\n } satisfies PostBatchData);\n\n // Verify FILE_POSTED event was emitted\n expect(postEventRepositoryMock.insert).toHaveBeenCalledWith(\n expect.objectContaining({\n postRecordId: postRecord.id,\n accountId,\n eventType: PostEventType.FILE_POSTED,\n fileId: submissionFile.id,\n sourceUrl,\n metadata: expect.objectContaining({\n batchNumber: 0,\n accountSnapshot: {\n name: 'test-account',\n website: 'Test Website',\n },\n fileSnapshot: expect.objectContaining({\n fileName: submissionFile.fileName,\n mimeType: submissionFile.mimeType,\n }),\n }),\n }),\n );\n });\n\n it('should emit FILE_FAILED event and throw on error', async () => {\n const errorMessage = 'Failed to upload file';\n const exception = new Error('API Error');\n const stage = 'upload';\n const additionalInfo = { code: 'ERR_API' };\n\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission = jest\n .fn()\n .mockResolvedValue({\n instanceId: 'test-website',\n message: errorMessage,\n exception,\n stage,\n additionalInfo,\n } as IPostResponse);\n\n await expect(\n (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n ),\n ).rejects.toMatchObject({\n exception,\n message: errorMessage,\n });\n\n // Verify FILE_FAILED event was emitted\n expect(postEventRepositoryMock.insert).toHaveBeenCalledWith(\n expect.objectContaining({\n postRecordId: postRecord.id,\n accountId,\n eventType: PostEventType.FILE_FAILED,\n fileId: submissionFile.id,\n error: expect.objectContaining({\n message: errorMessage,\n stage,\n additionalInfo,\n }),\n }),\n );\n });\n\n it('should emit FILE_FAILED with unknown error when no message provided', async () => {\n const exception = new Error('Unknown API Error');\n\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission = jest\n .fn()\n .mockResolvedValue({\n instanceId: 'test-website',\n exception,\n } as IPostResponse);\n\n await expect(\n (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n ),\n ).rejects.toMatchObject({\n exception,\n });\n\n // Verify FILE_FAILED event was emitted with default message\n expect(postEventRepositoryMock.insert).toHaveBeenCalledWith(\n expect.objectContaining({\n eventType: PostEventType.FILE_FAILED,\n error: expect.objectContaining({\n message: 'Unknown error',\n }),\n }),\n );\n });\n\n it('should return early if no files to post', async () => {\n const emptySubmission = createSubmission([]);\n const emptyPostRecord = createPostRecord(emptySubmission);\n postData.submission = emptySubmission;\n\n await (manager as any).attemptToPost(\n emptyPostRecord,\n accountId,\n mockWebsite,\n postData,\n );\n\n // Website method should not have been called\n expect(\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission,\n ).not.toHaveBeenCalled();\n\n // No events should have been emitted\n expect(postEventRepositoryMock.insert).not.toHaveBeenCalled();\n });\n\n it('should check cancel token during posting', async () => {\n // Cancel immediately\n cancelToken.cancel();\n\n await expect(\n (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n ),\n ).rejects.toThrow('Task was cancelled');\n\n // Website method should not have been called\n expect(\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission,\n ).not.toHaveBeenCalled();\n });\n\n it('should filter out files ignored for this website', async () => {\n const ignoredFile = createSubmissionFile('ignored-file', 1);\n ignoredFile.metadata.ignoredWebsites = [accountId];\n\n const validFile = createSubmissionFile('valid-file', 2);\n\n const submission = createSubmission([ignoredFile, validFile]);\n const postRecordWithIgnored = createPostRecord(submission);\n postData.submission = submission;\n\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission = jest\n .fn()\n .mockResolvedValue({\n instanceId: 'test-website',\n sourceUrl: 'https://example.com/file/456',\n });\n\n await (manager as any).attemptToPost(\n postRecordWithIgnored,\n accountId,\n mockWebsite,\n postData,\n );\n\n // Should only post one file (the valid one)\n expect(\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission,\n ).toHaveBeenCalledTimes(1);\n\n // Verify FILE_POSTED event only for valid file\n expect(postEventRepositoryMock.insert).toHaveBeenCalledWith(\n expect.objectContaining({\n fileId: validFile.id,\n eventType: PostEventType.FILE_POSTED,\n }),\n );\n });\n\n it('should sort files by order before posting', async () => {\n const file1 = createSubmissionFile('file-1', 3);\n const file2 = createSubmissionFile('file-2', 1);\n const file3 = createSubmissionFile('file-3', 2);\n\n const submission = createSubmission([file1, file2, file3]);\n const postRecordWithMultiple = createPostRecord(submission);\n postData.submission = submission;\n\n // Set batch size to 3 to get all files in one batch\n (\n mockWebsite as unknown as ImplementedFileWebsite\n ).decoratedProps.fileOptions.fileBatchSize = 3;\n\n let capturedFiles: PostingFile[] = [];\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission = jest\n .fn()\n .mockImplementation((data, files) => {\n capturedFiles = files;\n return {\n instanceId: 'test-website',\n sourceUrl: 'https://example.com/file/all',\n };\n });\n\n await (manager as any).attemptToPost(\n postRecordWithMultiple,\n accountId,\n mockWebsite,\n postData,\n );\n\n // Verify files are sorted by order\n expect(capturedFiles.map((f) => f.id)).toEqual([\n 'file-2', // order 1\n 'file-3', // order 2\n 'file-1', // order 3\n ]);\n });\n });\n\n describe('file batching', () => {\n let mockWebsite: UnknownWebsite;\n let accountId: AccountId;\n let postData: PostData;\n let cancelToken: CancellableToken;\n\n beforeEach(() => {\n accountId = 'test-account-id' as AccountId;\n mockWebsite = createMockFileWebsite(accountId);\n cancelToken = new CancellableToken();\n (manager as any).cancelToken = cancelToken;\n (manager as any).lastTimePostedToWebsite = {};\n\n resizerServiceMock.resize.mockImplementation(async (request) => {\n const file = request.file as SubmissionFile;\n return createPostingFile(file);\n });\n });\n\n it('should batch files according to website batch size', async () => {\n const files = [\n createSubmissionFile('file-1', 1),\n createSubmissionFile('file-2', 2),\n createSubmissionFile('file-3', 3),\n ];\n\n const submission = createSubmission(files);\n const postRecord = createPostRecord(submission);\n\n postData = {\n submission,\n options: {\n title: 'Test',\n description: 'Test',\n rating: SubmissionRating.GENERAL,\n tags: [],\n },\n } as PostData;\n\n // Set batch size to 2\n (\n mockWebsite as unknown as ImplementedFileWebsite\n ).decoratedProps.fileOptions.fileBatchSize = 2;\n\n const batchIndices: number[] = [];\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission = jest\n .fn()\n .mockImplementation((data, files, token, { index }) => {\n batchIndices.push(index);\n return {\n instanceId: 'test-website',\n sourceUrl: `https://example.com/batch/${index}`,\n };\n });\n\n await (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n );\n\n // Should have 2 batches: [file-1, file-2] and [file-3]\n expect(\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission,\n ).toHaveBeenCalledTimes(2);\n expect(batchIndices).toEqual([0, 1]);\n });\n\n it('should use minimum batch size of 1', async () => {\n const file = createSubmissionFile();\n const submission = createSubmission([file]);\n const postRecord = createPostRecord(submission);\n\n postData = {\n submission,\n options: {\n title: 'Test',\n description: 'Test',\n rating: SubmissionRating.GENERAL,\n tags: [],\n },\n } as PostData;\n\n // Set invalid batch size\n (\n mockWebsite as unknown as ImplementedFileWebsite\n ).decoratedProps.fileOptions.fileBatchSize = 0;\n\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission = jest\n .fn()\n .mockResolvedValue({\n instanceId: 'test-website',\n sourceUrl: 'https://example.com/file',\n });\n\n await (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n );\n\n // Should still work with batch size 1\n expect(\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission,\n ).toHaveBeenCalledTimes(1);\n });\n\n it('should stop posting if a batch fails', async () => {\n const files = [\n createSubmissionFile('file-1', 1),\n createSubmissionFile('file-2', 2),\n createSubmissionFile('file-3', 3),\n ];\n\n const submission = createSubmission(files);\n const postRecord = createPostRecord(submission);\n\n postData = {\n submission,\n options: {\n title: 'Test',\n description: 'Test',\n rating: SubmissionRating.GENERAL,\n tags: [],\n },\n } as PostData;\n\n // Set batch size to 1 to have 3 separate batches\n (\n mockWebsite as unknown as ImplementedFileWebsite\n ).decoratedProps.fileOptions.fileBatchSize = 1;\n\n let callCount = 0;\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission = jest\n .fn()\n .mockImplementation(() => {\n callCount++;\n if (callCount === 2) {\n return {\n instanceId: 'test-website',\n exception: new Error('Batch 2 failed'),\n message: 'Failed on second batch',\n };\n }\n return {\n instanceId: 'test-website',\n sourceUrl: `https://example.com/file/${callCount}`,\n };\n });\n\n await expect(\n (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n ),\n ).rejects.toMatchObject({\n message: 'Failed on second batch',\n });\n\n // Should have stopped at batch 2\n expect(\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission,\n ).toHaveBeenCalledTimes(2);\n\n // Should have emitted FILE_POSTED for first file and FILE_FAILED for second\n expect(postEventRepositoryMock.insert).toHaveBeenCalledWith(\n expect.objectContaining({\n fileId: 'file-1',\n eventType: PostEventType.FILE_POSTED,\n }),\n );\n expect(postEventRepositoryMock.insert).toHaveBeenCalledWith(\n expect.objectContaining({\n fileId: 'file-2',\n eventType: PostEventType.FILE_FAILED,\n }),\n );\n });\n });\n\n describe('source URL propagation', () => {\n let mockWebsite: UnknownWebsite;\n let accountId: AccountId;\n let postData: PostData;\n let cancelToken: CancellableToken;\n\n beforeEach(() => {\n accountId = 'test-account-id' as AccountId;\n mockWebsite = createMockFileWebsite(accountId);\n cancelToken = new CancellableToken();\n (manager as any).cancelToken = cancelToken;\n (manager as any).lastTimePostedToWebsite = {};\n\n resizerServiceMock.resize.mockImplementation(async (request) => {\n const file = request.file as SubmissionFile;\n return createPostingFile(file);\n });\n });\n\n it('should include source URLs from previous posts in current batch', async () => {\n const file = createSubmissionFile();\n const submission = createSubmission([file]);\n const postRecord = createPostRecord(submission);\n\n postData = {\n submission,\n options: {\n title: 'Test',\n description: 'Test',\n rating: SubmissionRating.GENERAL,\n tags: [],\n },\n } as PostData;\n\n // Mock that another account has already posted\n postEventRepositoryMock.getSourceUrlsFromPost.mockResolvedValue([\n 'https://other-site.com/post/1',\n 'https://other-site.com/post/2',\n ]);\n\n let capturedFiles: PostingFile[] = [];\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission = jest\n .fn()\n .mockImplementation((data, files) => {\n capturedFiles = files;\n return {\n instanceId: 'test-website',\n sourceUrl: 'https://example.com/file',\n };\n });\n\n await (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n );\n\n // Verify source URLs were included in file metadata\n expect(capturedFiles[0].metadata.sourceUrls).toContain(\n 'https://other-site.com/post/1',\n );\n expect(capturedFiles[0].metadata.sourceUrls).toContain(\n 'https://other-site.com/post/2',\n );\n });\n\n it('should include source URLs from resume context', async () => {\n const file = createSubmissionFile();\n const submission = createSubmission([file]);\n const postRecord = createPostRecord(submission);\n\n postData = {\n submission,\n options: {\n title: 'Test',\n description: 'Test',\n rating: SubmissionRating.GENERAL,\n tags: [],\n },\n } as PostData;\n\n // Set resume context with source URLs from prior attempts\n const resumeContext = {\n sourceUrlsByAccount: new Map([\n ['other-account' as AccountId, ['https://prior-site.com/post/1']],\n [accountId, ['https://self-url.com/post/1']], // Should be excluded\n ]),\n postedFilesByAccount: new Map(),\n };\n (manager as any).resumeContext = resumeContext;\n\n let capturedFiles: PostingFile[] = [];\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission = jest\n .fn()\n .mockImplementation((data, files) => {\n capturedFiles = files;\n return {\n instanceId: 'test-website',\n sourceUrl: 'https://example.com/file',\n };\n });\n\n await (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n );\n\n // Verify source URLs from prior attempts were included (excluding self)\n expect(capturedFiles[0].metadata.sourceUrls).toContain(\n 'https://prior-site.com/post/1',\n );\n expect(capturedFiles[0].metadata.sourceUrls).not.toContain(\n 'https://self-url.com/post/1',\n );\n });\n\n it('should deduplicate source URLs', async () => {\n const file = createSubmissionFile();\n const submission = createSubmission([file]);\n const postRecord = createPostRecord(submission);\n\n postData = {\n submission,\n options: {\n title: 'Test',\n description: 'Test',\n rating: SubmissionRating.GENERAL,\n tags: [],\n },\n } as PostData;\n\n // Same URL from both sources\n postEventRepositoryMock.getSourceUrlsFromPost.mockResolvedValue([\n 'https://duplicate.com/post/1',\n ]);\n\n const resumeContext = {\n sourceUrlsByAccount: new Map([\n ['other-account' as AccountId, ['https://duplicate.com/post/1']],\n ]),\n postedFilesByAccount: new Map(),\n };\n (manager as any).resumeContext = resumeContext;\n\n let capturedFiles: PostingFile[] = [];\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission = jest\n .fn()\n .mockImplementation((data, files) => {\n capturedFiles = files;\n return {\n instanceId: 'test-website',\n sourceUrl: 'https://example.com/file',\n };\n });\n\n await (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n );\n\n // Verify URL appears only once\n const duplicateCount = capturedFiles[0].metadata.sourceUrls.filter(\n (url) => url === 'https://duplicate.com/post/1',\n ).length;\n expect(duplicateCount).toBe(1);\n });\n });\n\n describe('resume context handling', () => {\n let mockWebsite: UnknownWebsite;\n let accountId: AccountId;\n let postData: PostData;\n let cancelToken: CancellableToken;\n\n beforeEach(() => {\n accountId = 'test-account-id' as AccountId;\n mockWebsite = createMockFileWebsite(accountId);\n cancelToken = new CancellableToken();\n (manager as any).cancelToken = cancelToken;\n (manager as any).lastTimePostedToWebsite = {};\n\n resizerServiceMock.resize.mockImplementation(async (request) => {\n const file = request.file as SubmissionFile;\n return createPostingFile(file);\n });\n });\n\n it('should skip files that were already posted according to resume context', async () => {\n const alreadyPostedFile = createSubmissionFile('already-posted', 1);\n const newFile = createSubmissionFile('new-file', 2);\n\n const submission = createSubmission([alreadyPostedFile, newFile]);\n const postRecord = createPostRecord(submission);\n\n postData = {\n submission,\n options: {\n title: 'Test',\n description: 'Test',\n rating: SubmissionRating.GENERAL,\n tags: [],\n },\n } as PostData;\n\n // Set resume context indicating one file was already posted\n const resumeContext = {\n sourceUrlsByAccount: new Map(),\n postedFilesByAccount: new Map([\n [accountId, new Set(['already-posted'])],\n ]),\n };\n (manager as any).resumeContext = resumeContext;\n\n let capturedFiles: PostingFile[] = [];\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission = jest\n .fn()\n .mockImplementation((data, files) => {\n capturedFiles = files;\n return {\n instanceId: 'test-website',\n sourceUrl: 'https://example.com/file',\n };\n });\n\n await (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n );\n\n // Should only post the new file\n expect(capturedFiles).toHaveLength(1);\n expect(capturedFiles[0].id).toBe('new-file');\n });\n });\n\n describe('file verification', () => {\n let mockWebsite: UnknownWebsite;\n let accountId: AccountId;\n let postData: PostData;\n let cancelToken: CancellableToken;\n\n beforeEach(() => {\n accountId = 'test-account-id' as AccountId;\n mockWebsite = createMockFileWebsite(accountId);\n cancelToken = new CancellableToken();\n (manager as any).cancelToken = cancelToken;\n (manager as any).lastTimePostedToWebsite = {};\n });\n\n it('should throw error if file type is not supported after processing', async () => {\n const unsupportedFile = createSubmissionFile();\n\n const submission = createSubmission([unsupportedFile]);\n const postRecord = createPostRecord(submission);\n\n postData = {\n submission,\n options: {\n title: 'Test',\n description: 'Test',\n rating: SubmissionRating.GENERAL,\n tags: [],\n },\n } as PostData;\n\n // Website only accepts JPEG\n (\n mockWebsite as unknown as ImplementedFileWebsite\n ).decoratedProps.fileOptions.acceptedMimeTypes = ['image/jpeg'];\n\n // Resizer returns PNG (which website doesn't accept)\n resizerServiceMock.resize.mockImplementation(async (request) => {\n const file = request.file as SubmissionFile;\n return createPostingFile(file); // Returns PNG\n });\n\n await expect(\n (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n ),\n ).rejects.toThrow('does not support the file type image/png');\n });\n\n it('should not verify if no accepted mime types are specified', async () => {\n const file = createSubmissionFile();\n const submission = createSubmission([file]);\n const postRecord = createPostRecord(submission);\n\n postData = {\n submission,\n options: {\n title: 'Test',\n description: 'Test',\n rating: SubmissionRating.GENERAL,\n tags: [],\n },\n } as PostData;\n\n // Website accepts all file types\n (\n mockWebsite as unknown as ImplementedFileWebsite\n ).decoratedProps.fileOptions.acceptedMimeTypes = [];\n\n resizerServiceMock.resize.mockImplementation(async (request) => {\n const f = request.file as SubmissionFile;\n return createPostingFile(f);\n });\n\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission = jest\n .fn()\n .mockResolvedValue({\n instanceId: 'test-website',\n sourceUrl: 'https://example.com/file',\n });\n\n await expect(\n (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n ),\n ).resolves.not.toThrow();\n });\n });\n\n describe('event metadata structure', () => {\n it('should create events with proper file snapshot metadata', async () => {\n const accountId = 'metadata-account' as AccountId;\n const submissionFile = createSubmissionFile();\n submissionFile.hash = 'test-hash-123';\n\n const submission = createSubmission([submissionFile]);\n const postRecord = createPostRecord(submission);\n const mockWebsite = createMockFileWebsite(accountId);\n\n (mockWebsite as unknown as FileWebsite).onPostFileSubmission = jest\n .fn()\n .mockResolvedValue({\n instanceId: 'test-website',\n sourceUrl: 'https://example.com/file/456',\n message: 'Successfully posted file',\n });\n\n resizerServiceMock.resize.mockImplementation(async (request) => {\n const file = request.file as SubmissionFile;\n return createPostingFile(file);\n });\n\n const cancelToken = new CancellableToken();\n (manager as any).cancelToken = cancelToken;\n (manager as any).lastTimePostedToWebsite = {};\n\n const postData = {\n submission,\n options: {\n title: 'Test',\n description: 'Test',\n rating: SubmissionRating.GENERAL,\n tags: [],\n },\n } as PostData;\n\n await (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n );\n\n expect(postEventRepositoryMock.insert).toHaveBeenCalledWith({\n postRecordId: postRecord.id,\n accountId,\n eventType: PostEventType.FILE_POSTED,\n fileId: submissionFile.id,\n sourceUrl: 'https://example.com/file/456',\n metadata: {\n batchNumber: 0,\n accountSnapshot: {\n name: 'test-account',\n website: 'Test Website',\n },\n fileSnapshot: {\n fileName: submissionFile.fileName,\n mimeType: submissionFile.mimeType,\n size: submissionFile.size,\n hash: submissionFile.hash,\n },\n responseMessage: 'Successfully posted file',\n },\n });\n });\n });\n});\n" }, { "path": "apps/client-server/src/app/post/services/post-manager-v2/file-submission-post-manager.service.ts", "content": "import { Injectable } from '@nestjs/common';\nimport { Logger } from '@postybirb/logger';\nimport {\n AccountId,\n FileSubmission,\n FileSubmissionMetadata,\n FileType,\n ImageResizeProps,\n PostData,\n PostEventType,\n SubmissionFileMetadata,\n SubmissionType,\n} from '@postybirb/types';\nimport { getFileType } from '@postybirb/utils/file-type';\nimport { chunk } from 'lodash';\nimport {\n FileBuffer,\n PostRecord,\n Submission,\n SubmissionFile,\n} from '../../../drizzle/models';\nimport { FileConverterService } from '../../../file-converter/file-converter.service';\nimport { NotificationsService } from '../../../notifications/notifications.service';\nimport { PostParsersService } from '../../../post-parsers/post-parsers.service';\nimport { ValidationService } from '../../../validation/validation.service';\nimport { getSupportedFileSize } from '../../../websites/decorators/supports-files.decorator';\nimport {\n ImplementedFileWebsite,\n isFileWebsite,\n} from '../../../websites/models/website-modifiers/file-website';\nimport { UnknownWebsite } from '../../../websites/website';\nimport { WebsiteRegistryService } from '../../../websites/website-registry.service';\nimport { PostingFile } from '../../models/posting-file';\nimport { PostFileResizerService } from '../post-file-resizer/post-file-resizer.service';\nimport { PostEventRepository } from '../post-record-factory';\nimport { BasePostManager } from './base-post-manager.service';\n\n/**\n * Returns true if `mimeType` is accepted by any entry in `patterns`.\n * Handles exact matches, prefix patterns (\"image/\"), and wildcard patterns (\"image/*\").\n */\nfunction mimeTypeIsAccepted(mimeType: string, patterns: string[]): boolean {\n return patterns.some((pattern) => {\n if (pattern === mimeType) return true;\n if (pattern.endsWith('/*')) {\n return mimeType.startsWith(pattern.slice(0, -1)); // 'image/*' → prefix 'image/'\n }\n if (pattern.endsWith('/')) {\n return mimeType.startsWith(pattern); // 'image/' → matches 'image/jpeg'\n }\n return false;\n });\n}\n\n/**\n * PostManager for file submissions.\n * Handles file batching, resizing, and conversion.\n * @class FileSubmissionPostManager\n */\n@Injectable()\nexport class FileSubmissionPostManager extends BasePostManager {\n protected readonly logger = Logger(this.constructor.name);\n\n constructor(\n postEventRepository: PostEventRepository,\n websiteRegistry: WebsiteRegistryService,\n postParserService: PostParsersService,\n validationService: ValidationService,\n notificationService: NotificationsService,\n private readonly resizerService: PostFileResizerService,\n private readonly fileConverterService: FileConverterService,\n ) {\n super(\n postEventRepository,\n websiteRegistry,\n postParserService,\n validationService,\n notificationService,\n );\n }\n\n getSupportedType(): SubmissionType {\n return SubmissionType.FILE;\n }\n\n protected async attemptToPost(\n entity: PostRecord,\n accountId: AccountId,\n instance: UnknownWebsite,\n data: PostData,\n ): Promise {\n if (!isFileWebsite(instance)) {\n throw new Error(\n `Website '${instance.decoratedProps.metadata.displayName}' does not support file submissions`,\n );\n }\n\n const submission = entity.submission as Submission;\n\n // Order files based on submission order\n const fileBatchSize = Math.max(\n instance.decoratedProps.fileOptions.fileBatchSize ?? 1,\n 1,\n );\n\n const files = this.getFilesToPost(submission, accountId, instance);\n\n if (files.length === 0) {\n this.logger.info(`No files to post for account ${accountId}`);\n return;\n }\n\n // Split files into batches based on instance file batch size\n const batches = chunk(files, fileBatchSize);\n\n for (const [batchIndex, batch] of batches.entries()) {\n this.cancelToken.throwIfCancelled();\n\n // Get source URLs from other accounts for cross-website propagation\n // 1. From current post attempt (other accounts that have already posted)\n const currentPostUrls =\n await this.postEventRepository.getSourceUrlsFromPost(\n entity.id,\n accountId,\n );\n\n // 2. From prior attempts (resume context, excluding self)\n const priorUrls: string[] = [];\n if (this.resumeContext) {\n for (const [\n priorAccountId,\n urls,\n ] of this.resumeContext.sourceUrlsByAccount.entries()) {\n if (priorAccountId !== accountId) {\n priorUrls.push(...urls);\n }\n }\n }\n\n // Merge and deduplicate\n const allSourceUrls = [...new Set([...currentPostUrls, ...priorUrls])];\n\n // Resize files if necessary\n const processedFiles: PostingFile[] = (\n await Promise.all(\n batch.map((submissionFile) =>\n this.resizeOrModifyFile(submissionFile, submission, instance),\n ),\n )\n ).map((f) => {\n const fileWithMetadata = f.withMetadata(f.metadata);\n fileWithMetadata.metadata.sourceUrls = [\n ...(fileWithMetadata.metadata.sourceUrls ?? []),\n ...allSourceUrls,\n ].filter((s) => !!s?.trim());\n\n return fileWithMetadata;\n });\n\n // Verify files are supported by the website after all processing\n this.verifyPostingFiles(instance, processedFiles);\n\n // Post\n this.cancelToken.throwIfCancelled();\n const fileIds = batch.map((f) => f.id);\n this.logger\n .withMetadata({\n batchIndex,\n totalBatches: batches.length,\n totalFiles: files.length,\n totalFilesInBatch: batch.length,\n fileIds,\n })\n .info(`Posting file batch to ${instance.id}`);\n\n await this.waitForPostingWaitInterval(accountId, instance);\n this.cancelToken.throwIfCancelled();\n\n const result = await instance.onPostFileSubmission(\n data,\n processedFiles,\n this.cancelToken,\n { totalBatches: batch.length, index: batchIndex },\n );\n\n if (result.exception) {\n // Emit FILE_FAILED events for each file in the batch\n const storedBatchIndex = batchIndex; // Capture for closure\n await Promise.all(\n batch.map((file) =>\n this.postEventRepository.insert({\n postRecordId: entity.id,\n accountId,\n eventType: PostEventType.FILE_FAILED,\n fileId: file.id,\n error: {\n message: result.message || 'Unknown error',\n stack: result.exception?.stack,\n stage: result.stage,\n additionalInfo: result.additionalInfo,\n },\n metadata: {\n batchNumber: storedBatchIndex,\n accountSnapshot: {\n name: instance.account.name,\n website: instance.decoratedProps.metadata.name,\n },\n fileSnapshot: {\n fileName: file.fileName,\n mimeType: file.mimeType,\n size: file.size,\n hash: file.hash,\n },\n },\n }),\n ),\n );\n\n // Behavior is to stop posting if a batch fails\n // eslint-disable-next-line @typescript-eslint/no-throw-literal\n throw result;\n }\n\n // Emit FILE_POSTED events for each file in the batch\n const storedBatchIndex = batchIndex; // Capture for closure\n await Promise.all(\n batch.map((file) =>\n this.postEventRepository.insert({\n postRecordId: entity.id,\n accountId,\n eventType: PostEventType.FILE_POSTED,\n fileId: file.id,\n sourceUrl: result.sourceUrl,\n metadata: {\n batchNumber: storedBatchIndex,\n accountSnapshot: {\n name: instance.account.name,\n website: instance.decoratedProps.metadata.name,\n },\n fileSnapshot: {\n fileName: file.fileName,\n mimeType: file.mimeType,\n size: file.size,\n hash: file.hash,\n },\n responseMessage: result.message,\n },\n }),\n ),\n );\n\n this.logger\n .withMetadata(result)\n .info(`File batch posted to ${instance.id}`);\n }\n }\n\n /**\n * Get files to post, filtered by resume context and user settings.\n * @private\n * @param {Submission} submission - The submission\n * @param {AccountId} accountId - The account ID\n * @param {ImplementedFileWebsite} instance - The website instance\n * @returns {SubmissionFile[]} Files to post\n */\n private getFilesToPost(\n submission: Submission,\n accountId: AccountId,\n instance: ImplementedFileWebsite,\n ): SubmissionFile[] {\n return (\n submission.files\n ?.filter(\n // Filter out files that have been marked by the user as ignored for this website\n (f) => !f.metadata.ignoredWebsites?.includes(accountId),\n )\n .filter(\n // Only post files that haven't been posted (based on resume context)\n (f) => {\n if (!this.resumeContext) return true;\n const postedFiles =\n this.resumeContext.postedFilesByAccount.get(accountId);\n return !postedFiles?.has(f.id);\n },\n )\n .sort((a, b) => {\n const aOrder = a.order ?? Number.MAX_SAFE_INTEGER;\n const bOrder = b.order ?? Number.MAX_SAFE_INTEGER;\n return aOrder - bOrder;\n }) ?? []\n );\n }\n\n /**\n * Verify that files are supported by the website.\n * @private\n * @param {UnknownWebsite} websiteInstance - The website instance\n * @param {PostingFile[]} postingFiles - The posting files\n */\n private verifyPostingFiles(\n websiteInstance: UnknownWebsite,\n postingFiles: PostingFile[],\n ): void {\n const acceptedMimeTypes =\n websiteInstance.decoratedProps.fileOptions.acceptedMimeTypes ?? [];\n if (acceptedMimeTypes.length === 0) return;\n\n postingFiles.forEach((f) => {\n if (!mimeTypeIsAccepted(f.mimeType, acceptedMimeTypes)) {\n throw new Error(\n `Website '${websiteInstance.decoratedProps.metadata.displayName}' does not support the file type ${f.mimeType} and attempts to convert it did not resolve the issue`,\n );\n }\n });\n }\n\n /**\n * Resize or modify a file for posting.\n * @private\n * @param {SubmissionFile} file - The submission file\n * @param {FileSubmission} submission - The file submission\n * @param {ImplementedFileWebsite} instance - The website instance\n * @returns {Promise} The posting file\n */\n private async resizeOrModifyFile(\n file: SubmissionFile,\n submission: FileSubmission,\n instance: ImplementedFileWebsite,\n ): Promise {\n if (!file.file) {\n await file.load();\n }\n\n const fileMetadata: SubmissionFileMetadata = file.metadata;\n let resizeParams: ImageResizeProps | undefined;\n const { fileOptions } = instance.decoratedProps;\n const allowedMimeTypes = fileOptions.acceptedMimeTypes ?? [];\n const fileType = getFileType(file.mimeType);\n\n if (fileType === FileType.IMAGE) {\n if (\n await this.fileConverterService.canConvert(\n file.mimeType,\n allowedMimeTypes,\n )\n ) {\n // eslint-disable-next-line no-param-reassign\n file.file = new FileBuffer(\n await this.fileConverterService.convert(file.file, allowedMimeTypes),\n );\n }\n\n resizeParams = this.getResizeParameters(submission, instance, file);\n\n // User defined dimensions\n const userDefinedDimensions =\n // eslint-disable-next-line @typescript-eslint/dot-notation\n fileMetadata?.dimensions['default'] ??\n fileMetadata?.dimensions[instance.accountId];\n\n if (userDefinedDimensions) {\n if (userDefinedDimensions.width && userDefinedDimensions.height) {\n resizeParams = resizeParams ?? {};\n if (\n userDefinedDimensions.width > resizeParams.width &&\n userDefinedDimensions.height > resizeParams.height\n ) {\n resizeParams = {\n ...resizeParams,\n width: userDefinedDimensions.width,\n height: userDefinedDimensions.height,\n };\n }\n }\n }\n\n // We pass to resize even if no resize parameters are set\n // as it handles the bundling to PostingFile\n return this.resizerService.resize({\n file,\n resize: resizeParams,\n });\n }\n\n if (\n fileType === FileType.TEXT &&\n file.hasAltFile &&\n !mimeTypeIsAccepted(file.mimeType, allowedMimeTypes)\n ) {\n // Use alt file if it exists and is a text file\n if (\n await this.fileConverterService.canConvert(\n file.altFile.mimeType,\n allowedMimeTypes,\n )\n ) {\n // eslint-disable-next-line no-param-reassign\n file.file = new FileBuffer(\n await this.fileConverterService.convert(\n file.altFile,\n allowedMimeTypes,\n ),\n );\n }\n }\n\n return new PostingFile(file.id, file.file, file.thumbnail).withMetadata(\n file.metadata,\n );\n }\n\n /**\n * Get resize parameters for a file.\n * @private\n * @param {FileSubmission} submission - The file submission\n * @param {ImplementedFileWebsite} instance - The website instance\n * @param {SubmissionFile} file - The submission file\n * @returns {ImageResizeProps | undefined} The resize parameters\n */\n private getResizeParameters(\n submission: FileSubmission,\n instance: ImplementedFileWebsite,\n file: SubmissionFile,\n ): ImageResizeProps | undefined {\n // Use website's own calculation method\n let resizeParams = instance.calculateImageResize(file);\n\n // Apply user-defined dimensions if set\n const fileParams = file.metadata.dimensions?.[instance.accountId];\n if (fileParams) {\n if (fileParams.width) {\n if (!resizeParams) {\n resizeParams = {};\n }\n resizeParams.width = Math.min(\n file.width,\n fileParams.width,\n resizeParams.width ?? Infinity,\n );\n }\n if (fileParams.height) {\n if (!resizeParams) {\n resizeParams = {};\n }\n resizeParams.height = Math.min(\n file.height,\n fileParams.height,\n resizeParams.height ?? Infinity,\n );\n }\n }\n\n // Fall back to supported file size if needed\n if (!resizeParams?.maxBytes) {\n const supportedFileSize = getSupportedFileSize(instance, file);\n if (supportedFileSize && file.size > supportedFileSize) {\n if (!resizeParams) {\n resizeParams = {};\n }\n resizeParams.maxBytes = supportedFileSize;\n }\n }\n\n return resizeParams;\n }\n}\n" }, { "path": "apps/client-server/src/app/post/services/post-manager-v2/index.ts", "content": "export * from './base-post-manager.service';\nexport * from './file-submission-post-manager.service';\nexport * from './message-submission-post-manager.service';\nexport * from './post-manager-registry.service';\n\n" }, { "path": "apps/client-server/src/app/post/services/post-manager-v2/message-submission-post-manager.service.spec.ts", "content": "import { clearDatabase } from '@postybirb/database';\nimport {\n AccountId,\n EntityId,\n IPostResponse,\n PostData,\n PostEventType,\n PostRecordState,\n SubmissionRating,\n SubmissionType,\n} from '@postybirb/types';\nimport 'reflect-metadata';\nimport { PostRecord, Submission } from '../../../drizzle/models';\nimport { NotificationsService } from '../../../notifications/notifications.service';\nimport { PostParsersService } from '../../../post-parsers/post-parsers.service';\nimport { ValidationService } from '../../../validation/validation.service';\nimport { MessageWebsite } from '../../../websites/models/website-modifiers/message-website';\nimport { UnknownWebsite } from '../../../websites/website';\nimport { WebsiteRegistryService } from '../../../websites/website-registry.service';\nimport { CancellableToken } from '../../models/cancellable-token';\nimport { PostEventRepository } from '../post-record-factory';\nimport { MessageSubmissionPostManager } from './message-submission-post-manager.service';\n\ndescribe('MessageSubmissionPostManager', () => {\n let manager: MessageSubmissionPostManager;\n let postEventRepositoryMock: jest.Mocked;\n let websiteRegistryMock: jest.Mocked;\n let postParserServiceMock: jest.Mocked;\n let validationServiceMock: jest.Mocked;\n let notificationServiceMock: jest.Mocked;\n\n beforeEach(() => {\n clearDatabase();\n\n postEventRepositoryMock = {\n insert: jest.fn().mockResolvedValue({}),\n findByPostRecordId: jest.fn().mockResolvedValue([]),\n getCompletedAccounts: jest.fn().mockResolvedValue([]),\n getFailedEvents: jest.fn().mockResolvedValue([]),\n } as unknown as jest.Mocked;\n\n websiteRegistryMock = {\n findInstance: jest.fn(),\n } as unknown as jest.Mocked;\n\n postParserServiceMock = {\n parse: jest.fn(),\n } as unknown as jest.Mocked;\n\n validationServiceMock = {\n validateSubmission: jest.fn().mockResolvedValue([]),\n } as unknown as jest.Mocked;\n\n notificationServiceMock = {\n create: jest.fn(),\n sendDesktopNotification: jest.fn(),\n } as unknown as jest.Mocked;\n\n manager = new MessageSubmissionPostManager(\n postEventRepositoryMock,\n websiteRegistryMock,\n postParserServiceMock,\n validationServiceMock,\n notificationServiceMock,\n );\n });\n\n function createSubmission(): Submission {\n return new Submission({\n id: 'test-submission',\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n metadata: {},\n type: SubmissionType.MESSAGE,\n files: [],\n options: [],\n isScheduled: false,\n schedule: {} as never,\n order: 1,\n posts: [] as never,\n isTemplate: false,\n isMultiSubmission: false,\n isArchived: false,\n });\n }\n\n function createPostRecord(submission: Submission): PostRecord {\n return new PostRecord({\n id: 'test-post-record' as EntityId,\n submission,\n state: PostRecordState.PENDING,\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n });\n }\n\n function createMockWebsite(accountId: AccountId): UnknownWebsite {\n return {\n id: 'test-website',\n account: {\n id: accountId,\n name: 'test-account',\n },\n decoratedProps: {\n metadata: {\n name: 'Test Website',\n displayName: 'Test Website',\n },\n },\n onPostMessageSubmission: jest.fn(),\n } as unknown as UnknownWebsite;\n }\n\n describe('getSupportedType', () => {\n it('should return MESSAGE submission type', () => {\n expect(manager.getSupportedType()).toBe(SubmissionType.MESSAGE);\n });\n });\n\n describe('attemptToPost', () => {\n let mockWebsite: UnknownWebsite;\n let postRecord: PostRecord;\n let accountId: AccountId;\n let postData: PostData;\n let cancelToken: CancellableToken;\n\n beforeEach(() => {\n accountId = 'test-account-id' as AccountId;\n const submission = createSubmission();\n postRecord = createPostRecord(submission);\n mockWebsite = createMockWebsite(accountId);\n\n cancelToken = new CancellableToken();\n (manager as any).cancelToken = cancelToken;\n (manager as any).lastTimePostedToWebsite = {};\n\n postData = {\n submission,\n options: {\n title: 'Test Title',\n description: 'Test Description',\n rating: SubmissionRating.GENERAL,\n tags: ['test'],\n },\n } as PostData;\n });\n\n it('should successfully post a message and emit MESSAGE_POSTED event', async () => {\n const sourceUrl = 'https://example.com/message/123';\n const responseMessage = 'Message posted successfully';\n\n (mockWebsite as unknown as MessageWebsite).onPostMessageSubmission = jest\n .fn()\n .mockResolvedValue({\n instanceId: 'test-website',\n sourceUrl,\n message: responseMessage,\n });\n\n await (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n );\n\n // Verify website method was called\n expect(\n (mockWebsite as unknown as MessageWebsite).onPostMessageSubmission,\n ).toHaveBeenCalledWith(postData, cancelToken);\n\n // Verify MESSAGE_POSTED event was emitted\n expect(postEventRepositoryMock.insert).toHaveBeenCalledWith(\n expect.objectContaining({\n postRecordId: postRecord.id,\n accountId,\n eventType: PostEventType.MESSAGE_POSTED,\n sourceUrl,\n metadata: expect.objectContaining({\n accountSnapshot: {\n name: 'test-account',\n website: 'Test Website',\n },\n responseMessage,\n }),\n }),\n );\n });\n\n it('should emit MESSAGE_FAILED event and throw on error', async () => {\n const errorMessage = 'Failed to post message';\n const exception = new Error('API Error');\n const stage = 'upload';\n const additionalInfo = { code: 'ERR_API' };\n\n (mockWebsite as unknown as MessageWebsite).onPostMessageSubmission = jest\n .fn()\n .mockResolvedValue({\n instanceId: 'test-website',\n message: errorMessage,\n exception,\n stage,\n additionalInfo,\n } as IPostResponse);\n\n await expect(\n (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n ),\n ).rejects.toMatchObject({\n exception,\n message: errorMessage,\n });\n\n // Verify MESSAGE_FAILED event was emitted\n expect(postEventRepositoryMock.insert).toHaveBeenCalledWith(\n expect.objectContaining({\n postRecordId: postRecord.id,\n accountId,\n eventType: PostEventType.MESSAGE_FAILED,\n error: expect.objectContaining({\n message: errorMessage,\n stage,\n additionalInfo,\n }),\n }),\n );\n });\n\n it('should emit MESSAGE_FAILED event with unknown error when no message provided', async () => {\n const exception = new Error('Unknown API Error');\n\n (mockWebsite as unknown as MessageWebsite).onPostMessageSubmission = jest\n .fn()\n .mockResolvedValue({\n instanceId: 'test-website',\n exception,\n } as IPostResponse);\n\n await expect(\n (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n ),\n ).rejects.toMatchObject({\n exception,\n });\n\n // Verify MESSAGE_FAILED event was emitted with default message\n expect(postEventRepositoryMock.insert).toHaveBeenCalledWith(\n expect.objectContaining({\n eventType: PostEventType.MESSAGE_FAILED,\n error: expect.objectContaining({\n message: 'Unknown error',\n }),\n }),\n );\n });\n\n it('should not wait if no previous post to account', async () => {\n (mockWebsite as unknown as MessageWebsite).onPostMessageSubmission = jest\n .fn()\n .mockResolvedValue({\n instanceId: 'test-website',\n sourceUrl: 'https://example.com/message/123',\n });\n\n const startTime = Date.now();\n await (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n );\n const elapsed = Date.now() - startTime;\n\n // Should not wait\n expect(elapsed).toBeLessThan(1000);\n });\n\n it('should not wait if enough time has passed since last post', async () => {\n (mockWebsite as unknown as MessageWebsite).onPostMessageSubmission = jest\n .fn()\n .mockResolvedValue({\n instanceId: 'test-website',\n sourceUrl: 'https://example.com/message/123',\n });\n\n // Set last time posted to 10 seconds ago (beyond min interval)\n const now = new Date();\n const lastTime = new Date(now.getTime() - 10000);\n (manager as any).lastTimePostedToWebsite[accountId] = lastTime;\n\n const startTime = Date.now();\n await (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n );\n const elapsed = Date.now() - startTime;\n\n // Should not wait\n expect(elapsed).toBeLessThan(1000);\n });\n\n it('should check cancel token during posting', async () => {\n // Cancel immediately\n cancelToken.cancel();\n\n await expect(\n (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n ),\n ).rejects.toThrow('Task was cancelled');\n\n // Website method should not have been called\n expect(\n (mockWebsite as unknown as MessageWebsite).onPostMessageSubmission,\n ).not.toHaveBeenCalled();\n });\n });\n\n describe('event metadata structure', () => {\n it('should create events with proper metadata structure', async () => {\n const accountId = 'metadata-account' as AccountId;\n const submission = createSubmission();\n const postRecord = createPostRecord(submission);\n const mockWebsite = createMockWebsite(accountId);\n\n (mockWebsite as unknown as MessageWebsite).onPostMessageSubmission = jest\n .fn()\n .mockResolvedValue({\n instanceId: 'test-website',\n sourceUrl: 'https://example.com/message/456',\n message: 'Successfully posted message',\n });\n\n const cancelToken = new CancellableToken();\n (manager as any).cancelToken = cancelToken;\n (manager as any).lastTimePostedToWebsite = {};\n\n const postData = {\n submission,\n options: {\n title: 'Test',\n description: 'Test',\n rating: SubmissionRating.GENERAL,\n tags: [],\n },\n } as PostData;\n\n await (manager as any).attemptToPost(\n postRecord,\n accountId,\n mockWebsite,\n postData,\n );\n\n expect(postEventRepositoryMock.insert).toHaveBeenCalledWith({\n postRecordId: postRecord.id,\n accountId,\n eventType: PostEventType.MESSAGE_POSTED,\n sourceUrl: 'https://example.com/message/456',\n metadata: {\n accountSnapshot: {\n name: 'test-account',\n website: 'Test Website',\n },\n responseMessage: 'Successfully posted message',\n },\n });\n });\n });\n});\n" }, { "path": "apps/client-server/src/app/post/services/post-manager-v2/message-submission-post-manager.service.ts", "content": "import { Injectable } from '@nestjs/common';\nimport { Logger } from '@postybirb/logger';\nimport {\n AccountId,\n PostData,\n PostEventType,\n SubmissionType,\n} from '@postybirb/types';\nimport { PostRecord } from '../../../drizzle/models';\nimport { NotificationsService } from '../../../notifications/notifications.service';\nimport { PostParsersService } from '../../../post-parsers/post-parsers.service';\nimport { ValidationService } from '../../../validation/validation.service';\nimport { MessageWebsite } from '../../../websites/models/website-modifiers/message-website';\nimport { UnknownWebsite } from '../../../websites/website';\nimport { WebsiteRegistryService } from '../../../websites/website-registry.service';\nimport { PostEventRepository } from '../post-record-factory';\nimport { BasePostManager } from './base-post-manager.service';\n\n/**\n * PostManager for message submissions.\n * Handles message-only posting.\n * @class MessageSubmissionPostManager\n */\n@Injectable()\nexport class MessageSubmissionPostManager extends BasePostManager {\n protected readonly logger = Logger(this.constructor.name);\n\n // eslint-disable-next-line @typescript-eslint/no-useless-constructor\n constructor(\n postEventRepository: PostEventRepository,\n websiteRegistry: WebsiteRegistryService,\n postParserService: PostParsersService,\n validationService: ValidationService,\n notificationService: NotificationsService,\n ) {\n super(\n postEventRepository,\n websiteRegistry,\n postParserService,\n validationService,\n notificationService,\n );\n }\n\n getSupportedType(): SubmissionType {\n return SubmissionType.MESSAGE;\n }\n\n protected async attemptToPost(\n entity: PostRecord,\n accountId: AccountId,\n instance: UnknownWebsite,\n data: PostData,\n ): Promise {\n this.logger.info(`Posting message to ${instance.id}`);\n\n await this.waitForPostingWaitInterval(accountId, instance);\n this.cancelToken.throwIfCancelled();\n\n const result = await (\n instance as unknown as MessageWebsite\n ).onPostMessageSubmission(data, this.cancelToken);\n\n if (result.exception) {\n // Emit MESSAGE_FAILED event\n await this.postEventRepository.insert({\n postRecordId: entity.id,\n accountId,\n eventType: PostEventType.MESSAGE_FAILED,\n error: {\n message: result.message || 'Unknown error',\n stack: result.exception?.stack,\n stage: result.stage,\n additionalInfo: result.additionalInfo,\n },\n metadata: {\n accountSnapshot: {\n name: instance.account.name,\n website: instance.decoratedProps.metadata.name,\n },\n responseMessage: result.message,\n },\n });\n\n // eslint-disable-next-line @typescript-eslint/no-throw-literal\n throw result;\n }\n\n // Emit MESSAGE_POSTED event\n await this.postEventRepository.insert({\n postRecordId: entity.id,\n accountId,\n eventType: PostEventType.MESSAGE_POSTED,\n sourceUrl: result.sourceUrl,\n metadata: {\n accountSnapshot: {\n name: instance.account.name,\n website: instance.decoratedProps.metadata.name,\n },\n responseMessage: result.message,\n },\n });\n\n this.logger.withMetadata(result).info(`Message posted to ${instance.id}`);\n }\n}\n" }, { "path": "apps/client-server/src/app/post/services/post-manager-v2/post-manager-registry.service.ts", "content": "import { Injectable } from '@nestjs/common';\nimport { Logger } from '@postybirb/logger';\nimport { EntityId, SubmissionType } from '@postybirb/types';\nimport { PostRecord } from '../../../drizzle/models';\nimport { PostRecordFactory } from '../post-record-factory';\nimport { BasePostManager } from './base-post-manager.service';\nimport { FileSubmissionPostManager } from './file-submission-post-manager.service';\nimport { MessageSubmissionPostManager } from './message-submission-post-manager.service';\n\n/**\n * Registry for PostManager instances.\n * Maps submission types to appropriate PostManager implementations.\n * Allows multiple submissions to be posted concurrently.\n * @class PostManagerRegistry\n */\n@Injectable()\nexport class PostManagerRegistry {\n private readonly logger = Logger(this.constructor.name);\n\n private readonly managers: Map;\n\n constructor(\n private readonly fileSubmissionPostManager: FileSubmissionPostManager,\n private readonly messageSubmissionPostManager: MessageSubmissionPostManager,\n private readonly postRecordFactory: PostRecordFactory,\n ) {\n this.managers = new Map();\n this.managers.set(SubmissionType.FILE, fileSubmissionPostManager);\n this.managers.set(SubmissionType.MESSAGE, messageSubmissionPostManager);\n }\n\n /**\n * Get a PostManager for a submission type.\n * @param {SubmissionType} type - The submission type\n * @returns {BasePostManager | undefined} The PostManager instance\n */\n public getManager(type: SubmissionType): BasePostManager | undefined {\n return this.managers.get(type);\n }\n\n /**\n * Start posting for a post record.\n * Determines the appropriate PostManager and starts posting.\n * @param {PostRecord} postRecord - The post record to start\n */\n public async startPost(postRecord: PostRecord): Promise {\n const submissionType = postRecord.submission.type;\n const manager = this.getManager(submissionType);\n\n if (!manager) {\n this.logger.error(`No PostManager found for type: ${submissionType}`);\n throw new Error(`No PostManager found for type: ${submissionType}`);\n }\n\n if (manager.isPosting()) {\n this.logger.warn(\n `PostManager for ${submissionType} is already posting, cannot start new post`,\n );\n return;\n }\n\n // Build resume context for this post record\n const resumeContext = await this.postRecordFactory.buildResumeContext(\n postRecord.submissionId,\n postRecord.id,\n postRecord.resumeMode,\n );\n\n await manager.startPost(postRecord, resumeContext);\n }\n\n /**\n * Cancel posting for a submission if it's currently running.\n * Checks all managers to find the one handling this submission.\n * @param {EntityId} submissionId - The submission ID to cancel\n * @returns {Promise} True if cancelled, false if not found\n */\n public async cancelIfRunning(submissionId: EntityId): Promise {\n for (const manager of this.managers.values()) {\n if (await manager.cancelIfRunning(submissionId)) {\n return true;\n }\n }\n return false;\n }\n\n /**\n * Check if a specific submission type's manager is posting.\n * @param {SubmissionType} type - The submission type\n * @returns {boolean} True if posting\n */\n public isPostingType(type: SubmissionType): boolean {\n const manager = this.getManager(type);\n return manager?.isPosting() ?? false;\n }\n}\n" }, { "path": "apps/client-server/src/app/post/services/post-queue/post-queue.controller.ts", "content": "import { Body, Controller, Get, Post } from '@nestjs/common';\nimport { ApiOkResponse, ApiTags } from '@nestjs/swagger';\nimport { PostyBirbController } from '../../../common/controller/postybirb-controller';\nimport { PostQueueActionDto } from '../../dtos/post-queue-action.dto';\nimport { PostQueueService } from './post-queue.service';\n\n/**\n * Queue operations for Post data.\n * @class PostController\n */\n@ApiTags('post-queue')\n@Controller('post-queue')\nexport class PostQueueController extends PostyBirbController<'PostQueueRecordSchema'> {\n constructor(readonly service: PostQueueService) {\n super(service);\n }\n\n @Post('enqueue')\n @ApiOkResponse({ description: 'Post(s) queued.' })\n async enqueue(@Body() request: PostQueueActionDto) {\n return this.service.enqueue(request.submissionIds, request.resumeMode);\n }\n\n @Post('dequeue')\n @ApiOkResponse({ description: 'Post(s) dequeued.' })\n async dequeue(@Body() request: PostQueueActionDto) {\n this.service.dequeue(request.submissionIds);\n }\n\n @Get('is-paused')\n @ApiOkResponse({ description: 'Get if queue is paused.' })\n async isPaused() {\n return { paused: await this.service.isPaused() };\n }\n\n @Post('pause')\n @ApiOkResponse({ description: 'Queue paused.' })\n async pause() {\n await this.service.pause();\n return { paused: true };\n }\n\n @Post('resume')\n @ApiOkResponse({ description: 'Queue resumed.' })\n async resume() {\n await this.service.resume();\n return { paused: false };\n }\n}\n" }, { "path": "apps/client-server/src/app/post/services/post-queue/post-queue.service.spec.ts", "content": "import { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport {\n AccountId,\n DefaultDescription,\n PostRecordState,\n SubmissionId,\n SubmissionRating,\n SubmissionType,\n} from '@postybirb/types';\nimport { AccountModule } from '../../../account/account.module';\nimport { AccountService } from '../../../account/account.service';\nimport { CreateAccountDto } from '../../../account/dtos/create-account.dto';\nimport { PostyBirbDatabase } from '../../../drizzle/postybirb-database/postybirb-database';\nimport { SettingsService } from '../../../settings/settings.service';\nimport { CreateSubmissionDto } from '../../../submission/dtos/create-submission.dto';\nimport { SubmissionService } from '../../../submission/services/submission.service';\nimport { SubmissionModule } from '../../../submission/submission.module';\nimport { CreateWebsiteOptionsDto } from '../../../website-options/dtos/create-website-options.dto';\nimport { WebsiteOptionsModule } from '../../../website-options/website-options.module';\nimport { WebsiteOptionsService } from '../../../website-options/website-options.service';\nimport { WebsiteRegistryService } from '../../../websites/website-registry.service';\nimport { WebsitesModule } from '../../../websites/websites.module';\nimport { PostModule } from '../../post.module';\nimport { PostService } from '../../post.service';\nimport { PostManagerRegistry } from '../post-manager-v2';\nimport { PostQueueService } from './post-queue.service';\n\ndescribe('PostQueueService', () => {\n let service: PostQueueService;\n let module: TestingModule;\n let submissionService: SubmissionService;\n let accountService: AccountService;\n let websiteOptionsService: WebsiteOptionsService;\n let registryService: WebsiteRegistryService;\n let postService: PostService;\n let mockPostManagerRegistry: jest.Mocked;\n\n beforeEach(async () => {\n clearDatabase();\n\n // Create mock PostManagerRegistry\n mockPostManagerRegistry = {\n startPost: jest.fn().mockResolvedValue(undefined),\n cancelIfRunning: jest.fn().mockResolvedValue(true),\n isPostingType: jest.fn().mockReturnValue(false),\n getManager: jest.fn(),\n } as any;\n\n try {\n module = await Test.createTestingModule({\n imports: [\n SubmissionModule,\n AccountModule,\n WebsiteOptionsModule,\n WebsitesModule,\n PostModule,\n ],\n })\n .overrideProvider(PostManagerRegistry)\n .useValue(mockPostManagerRegistry)\n .compile();\n\n service = module.get(PostQueueService);\n submissionService = module.get(SubmissionService);\n accountService = module.get(AccountService);\n const settingsService = module.get(SettingsService);\n websiteOptionsService = module.get(\n WebsiteOptionsService,\n );\n registryService = module.get(\n WebsiteRegistryService,\n );\n postService = module.get(PostService);\n await accountService.onModuleInit();\n await settingsService.onModuleInit();\n } catch (err) {\n console.log(err);\n }\n });\n\n function createSubmissionDto(): CreateSubmissionDto {\n const dto = new CreateSubmissionDto();\n dto.name = 'Test';\n dto.type = SubmissionType.MESSAGE;\n return dto;\n }\n\n function createAccountDto(): CreateAccountDto {\n const dto = new CreateAccountDto();\n dto.name = 'Test';\n dto.website = 'test';\n return dto;\n }\n\n function createWebsiteOptionsDto(\n submissionId: SubmissionId,\n accountId: AccountId,\n ): CreateWebsiteOptionsDto {\n const dto = new CreateWebsiteOptionsDto();\n dto.submissionId = submissionId;\n dto.accountId = accountId;\n dto.data = {\n title: 'Test Title',\n tags: {\n overrideDefault: true,\n tags: ['test'],\n },\n description: {\n overrideDefault: true,\n description: DefaultDescription(),\n },\n rating: SubmissionRating.GENERAL,\n };\n return dto;\n }\n\n afterAll(async () => {\n await module.close();\n });\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n\n it('should handle pausing and resuming the queue', async () => {\n await service.pause();\n expect(await service.isPaused()).toBe(true);\n await service.resume();\n expect(await service.isPaused()).toBe(false);\n });\n\n it('should handle enqueue and dequeue of submissions', async () => {\n await service.pause(); // Just to test the function\n const submission = await submissionService.create(createSubmissionDto());\n const account = await accountService.create(createAccountDto());\n expect(registryService.findInstance(account)).toBeDefined();\n\n await websiteOptionsService.create(\n createWebsiteOptionsDto(submission.id, account.id),\n );\n\n await service.enqueue([submission.id, submission.id]);\n expect((await service.findAll()).length).toBe(1);\n const top = await service.peek();\n expect(top).toBeDefined();\n expect(top.submission.id).toBe(submission.id);\n\n await service.dequeue([submission.id]);\n expect((await service.findAll()).length).toBe(0);\n expect(await service.peek()).toBeNull();\n });\n\n it('should insert posts into the post manager', async () => {\n const submission = await submissionService.create(createSubmissionDto());\n const account = await accountService.create(createAccountDto());\n expect(registryService.findInstance(account)).toBeDefined();\n\n await websiteOptionsService.create(\n createWebsiteOptionsDto(submission.id, account.id),\n );\n\n // Enqueue now creates the PostRecord immediately\n await service.enqueue([submission.id]);\n expect((await service.findAll()).length).toBe(1);\n\n // PostRecord should already exist after enqueue\n let postRecord = (await postService.findAll())[0];\n expect(postRecord).toBeDefined();\n expect(postRecord.submissionId).toBe(submission.id);\n expect(postRecord.state).toBe(PostRecordState.PENDING);\n\n // Initially, no manager is posting (so execute will start the post)\n mockPostManagerRegistry.isPostingType.mockReturnValue(false);\n\n // Execute should start the post manager\n await service.execute();\n let queueRecord = await service.peek();\n expect(mockPostManagerRegistry.startPost).toHaveBeenCalledWith(\n expect.objectContaining({\n id: postRecord.id,\n submissionId: submission.id,\n }),\n );\n expect(queueRecord).toBeDefined();\n expect(queueRecord.postRecord).toBeDefined();\n\n // Now simulate that the manager is posting\n mockPostManagerRegistry.isPostingType.mockReturnValue(true);\n\n // Simulate cancellation\n await mockPostManagerRegistry.cancelIfRunning(submission.id);\n expect(mockPostManagerRegistry.cancelIfRunning).toHaveBeenCalledWith(\n submission.id,\n );\n\n // Simulate the post completing (with failure) - manually update the record\n const database = new PostyBirbDatabase('PostRecordSchema');\n await database.update(postRecord.id, {\n state: PostRecordState.FAILED,\n completedAt: new Date().toISOString(),\n });\n\n // Simulate posting finished\n mockPostManagerRegistry.isPostingType.mockReturnValue(false);\n\n queueRecord = await service.peek();\n expect(queueRecord).toBeDefined();\n expect(queueRecord.postRecord).toBeDefined();\n\n // We expect the post to be in a terminal state and cleanup of the record.\n // The post record should remain after the queue record is deleted.\n await service.execute();\n expect((await service.findAll()).length).toBe(0);\n postRecord = await postService.findById(postRecord.id);\n expect(postRecord.state).toBe(PostRecordState.FAILED);\n expect(postRecord.completedAt).toBeDefined();\n });\n});\n" }, { "path": "apps/client-server/src/app/post/services/post-queue/post-queue.service.ts", "content": "import {\n Injectable,\n InternalServerErrorException,\n OnModuleInit,\n Optional,\n} from '@nestjs/common';\nimport { Cron, CronExpression } from '@nestjs/schedule';\nimport {\n EntityId,\n PostRecordResumeMode,\n PostRecordState,\n ScheduleType,\n SubmissionId,\n} from '@postybirb/types';\nimport { IsTestEnvironment } from '@postybirb/utils/electron';\nimport { Mutex } from 'async-mutex';\nimport { Cron as CronGenerator } from 'croner';\nimport { PostyBirbService } from '../../../common/service/postybirb-service';\nimport { PostQueueRecord, PostRecord } from '../../../drizzle/models';\nimport { PostyBirbDatabase } from '../../../drizzle/postybirb-database/postybirb-database';\nimport { NotificationsService } from '../../../notifications/notifications.service';\nimport { SettingsService } from '../../../settings/settings.service';\nimport { SubmissionService } from '../../../submission/services/submission.service';\nimport { WSGateway } from '../../../web-socket/web-socket-gateway';\nimport { WebsiteRegistryService } from '../../../websites/website-registry.service';\nimport { PostManagerRegistry } from '../post-manager-v2';\nimport { PostRecordFactory } from '../post-record-factory';\n\n/**\n * Handles the queue of posts to be posted.\n * This service is responsible for managing the queue of posts to be posted.\n * It will create post records and start the post manager when a post is ready to be posted.\n * @class PostQueueService\n */\n@Injectable()\nexport class PostQueueService\n extends PostyBirbService<'PostQueueRecordSchema'>\n implements OnModuleInit\n{\n private readonly queueModificationMutex = new Mutex();\n\n private readonly queueMutex = new Mutex();\n\n private initTime = Date.now();\n\n private readonly postRecordRepository = new PostyBirbDatabase(\n 'PostRecordSchema',\n );\n\n private readonly submissionRepository = new PostyBirbDatabase(\n 'SubmissionSchema',\n );\n\n /**\n * Maximum time (in ms) a post can be RUNNING without any activity before being considered stuck.\n */\n private readonly maxPostIdleTime = 30 * 60 * 1000; // 30 minutes\n\n constructor(\n private readonly postManagerRegistry: PostManagerRegistry,\n private readonly postRecordFactory: PostRecordFactory,\n private readonly settingsService: SettingsService,\n private readonly notificationService: NotificationsService,\n private readonly submissionService: SubmissionService,\n private readonly websiteRegistryService: WebsiteRegistryService,\n @Optional() webSocket?: WSGateway,\n ) {\n super('PostQueueRecordSchema', webSocket);\n }\n\n /**\n * Crash recovery: Resume any PostRecords that were left in RUNNING state.\n * This handles cases where the application was forcefully shut down or crashed\n * while a post was in progress.\n *\n * Also handles orphaned records - PENDING/RUNNING records that have no queue record\n * (can happen if queue record was deleted but post record state wasn't updated).\n */\n async onModuleInit() {\n if (IsTestEnvironment()) {\n return;\n }\n\n try {\n // First, handle orphaned post records (PENDING/RUNNING without a queue record)\n await this.failOrphanedPostRecords();\n\n // Find all RUNNING post records\n const runningRecords = await this.postRecordRepository.find({\n where: (record, { eq }) => eq(record.state, PostRecordState.RUNNING),\n with: {\n submission: {\n with: {\n files: true,\n options: {\n with: {\n account: true,\n },\n },\n },\n },\n },\n });\n\n if (runningRecords.length > 0) {\n this.logger\n .withMetadata({ count: runningRecords.length })\n .info(\n 'Detected interrupted PostRecords from crash/shutdown, scheduling resume',\n );\n\n // Resume in the background so onModuleInit does not block server startup\n this.resumeInterruptedPosts(runningRecords);\n }\n } catch (error) {\n this.logger\n .withMetadata({ error })\n .error('Failed to recover interrupted posts on startup');\n // Don't throw - allow the app to start even if crash recovery fails\n }\n }\n\n /**\n * Resume interrupted posts in the background.\n * This is intentionally not awaited so it does not block server startup.\n */\n private async resumeInterruptedPosts(\n runningRecords: PostRecord[],\n ): Promise {\n try {\n this.logger.info(\n 'Waiting for website registry initialization before crash recovery...',\n );\n await this.websiteRegistryService.waitForInitialization(60_000);\n this.logger.info(\n 'Website registry initialized, proceeding with crash recovery',\n );\n\n for (const record of runningRecords) {\n this.logger\n .withMetadata({\n recordId: record.id,\n resumeMode: record.resumeMode,\n })\n .info('Resuming interrupted PostRecord');\n\n await this.postManagerRegistry.startPost(record);\n }\n } catch (error) {\n this.logger\n .withMetadata({ error })\n .error('Failed to resume interrupted posts');\n }\n }\n\n /**\n * Find and fail orphaned post records.\n * An orphaned record is one in PENDING or RUNNING state that has no corresponding queue record.\n * This can happen if the queue record was deleted (dequeue/cancel) but the post record\n * state wasn't properly updated to FAILED.\n */\n private async failOrphanedPostRecords(): Promise {\n // Find all PENDING or RUNNING post records\n const inProgressRecords = await this.postRecordRepository.find({\n where: (record, { or, eq }) =>\n or(\n eq(record.state, PostRecordState.PENDING),\n eq(record.state, PostRecordState.RUNNING),\n ),\n });\n\n if (inProgressRecords.length === 0) {\n return;\n }\n\n // Get all queue records to check which post records have a corresponding queue entry\n const allQueueRecords = await this.repository.findAll();\n const queuedPostRecordIds = new Set(\n allQueueRecords\n .map((qr) => qr.postRecordId)\n .filter((id): id is string => id != null),\n );\n\n // Find orphaned records (in-progress but not in queue)\n const orphanedRecords = inProgressRecords.filter(\n (record) => !queuedPostRecordIds.has(record.id),\n );\n\n if (orphanedRecords.length > 0) {\n this.logger\n .withMetadata({ count: orphanedRecords.length })\n .warn('Found orphaned PostRecords (PENDING/RUNNING with no queue record), marking as FAILED');\n\n for (const record of orphanedRecords) {\n this.logger\n .withMetadata({\n recordId: record.id,\n submissionId: record.submissionId,\n state: record.state,\n })\n .warn('Marking orphaned PostRecord as FAILED');\n\n await this.postRecordRepository.update(record.id, {\n state: PostRecordState.FAILED,\n completedAt: new Date().toISOString(),\n });\n }\n }\n }\n\n public async isPaused(): Promise {\n return (await this.settingsService.getDefaultSettings()).settings\n .queuePaused;\n }\n\n public async pause() {\n this.logger.info('Queue paused');\n const settings = await this.settingsService.getDefaultSettings();\n await this.settingsService.update(settings.id, {\n settings: { ...settings.settings, queuePaused: true },\n });\n }\n\n public async resume() {\n this.logger.info('Queue resumed');\n const settings = await this.settingsService.getDefaultSettings();\n await this.settingsService.update(settings.id, {\n settings: { ...settings.settings, queuePaused: false },\n });\n }\n\n public override remove(id: EntityId) {\n return this.dequeue([id]);\n }\n\n /**\n * Get the most recent terminal PostRecord for a submission.\n * @param {SubmissionId} submissionId - The submission ID\n * @returns {Promise} The most recent terminal record, or null if none\n */\n private async getMostRecentTerminalPostRecord(\n submissionId: SubmissionId,\n ): Promise {\n const records = await this.postRecordRepository.find({\n where: (record, { eq, and, inArray }) =>\n and(\n eq(record.submissionId, submissionId),\n inArray(record.state, [PostRecordState.DONE, PostRecordState.FAILED]),\n ),\n orderBy: (record, { desc }) => desc(record.createdAt),\n limit: 1,\n });\n\n return records.length > 0 ? records[0] : null;\n }\n\n /**\n * Handle terminal state for a completed post record.\n * Archives the submission if successful and non-recurring.\n * Emits appropriate notifications.\n * @param {PostRecord} record - The completed post record\n */\n private async handleTerminalState(record: PostRecord): Promise {\n const { submission } = record;\n const submissionName = submission.getSubmissionName() ?? 'Submission';\n\n if (record.state === PostRecordState.DONE) {\n this.logger\n .withMetadata({ submissionId: record.submissionId })\n .info('Post completed successfully');\n\n // Archive submission if non-recurring schedule\n if (submission.schedule.scheduleType !== ScheduleType.RECURRING) {\n await this.submissionService.archive(record.submissionId);\n this.logger\n .withMetadata({ submissionId: record.submissionId })\n .info('Submission archived after successful post');\n }\n\n // Emit success notification\n await this.notificationService.create({\n type: 'success',\n title: 'Post Completed',\n message: `Successfully posted \"${submissionName}\" to all websites`,\n tags: ['post-success'],\n data: {\n submissionId: record.submissionId,\n type: submission.type,\n },\n });\n } else if (record.state === PostRecordState.FAILED) {\n this.logger\n .withMetadata({ submissionId: record.submissionId })\n .error('Post failed');\n\n // Clear isScheduled for non-recurring submissions so they don't retry indefinitely\n if (\n submission.isScheduled &&\n submission.schedule.scheduleType !== ScheduleType.RECURRING\n ) {\n await this.submissionService.update(record.submissionId, {\n isScheduled: false,\n });\n this.logger\n .withMetadata({ submissionId: record.submissionId })\n .info('Cleared isScheduled flag after post failure');\n }\n\n // Count failed events for the message\n const failedEventCount =\n record.events?.filter((e) => e.eventType === 'POST_ATTEMPT_FAILED')\n .length ?? 0;\n\n // Emit failure notification (summary - individual failures already notified)\n await this.notificationService.create({\n type: 'warning',\n title: 'Post Incomplete',\n message:\n failedEventCount > 0\n ? `\"${submissionName}\" failed to post to ${failedEventCount} website(s)`\n : `\"${submissionName}\" failed to post`,\n tags: ['post-incomplete'],\n data: {\n submissionId: record.submissionId,\n type: submission.type,\n failedCount: failedEventCount,\n },\n });\n }\n }\n\n /**\n * Enqueue submissions for posting.\n * If resumeMode is provided and the submission has a terminal PostRecord,\n * a new PostRecord will be created using the specified resume mode.\n *\n * Smart handling: If the most recent PostRecord is DONE (successful completion),\n * we always create a fresh record (restart) regardless of the provided resumeMode,\n * since the user is starting a new posting session.\n *\n * @param {SubmissionId[]} submissionIds - The submissions to enqueue\n * @param {PostRecordResumeMode} resumeMode - Optional resume mode for re-queuing terminal records\n */\n public async enqueue(\n submissionIds: SubmissionId[],\n resumeMode?: PostRecordResumeMode,\n ) {\n if (!submissionIds.length) {\n return;\n }\n const release = await this.queueModificationMutex.acquire();\n this.logger\n .withMetadata({ submissionIds, resumeMode })\n .info('Enqueueing posts');\n\n try {\n for (const submissionId of submissionIds) {\n // Check if submission exists and is not archived before doing anything\n const submission =\n await this.submissionRepository.findById(submissionId);\n if (!submission) {\n this.logger\n .withMetadata({ submissionId })\n .warn('Submission not found, skipping enqueue');\n continue;\n }\n if (submission.isArchived) {\n this.logger\n .withMetadata({ submissionId })\n .info('Submission is archived, skipping enqueue');\n continue;\n }\n\n const existing = await this.repository.findOne({\n where: (queueRecord, { eq }) =>\n eq(queueRecord.submissionId, submissionId),\n with: {\n postRecord: true,\n },\n });\n\n if (!existing) {\n // No queue entry exists - determine resume mode based on most recent PostRecord\n const mostRecentRecord =\n await this.getMostRecentTerminalPostRecord(submissionId);\n\n let effectiveResumeMode: PostRecordResumeMode;\n\n if (!mostRecentRecord) {\n // No prior post record - fresh start\n this.logger\n .withMetadata({ submissionId })\n .info('No prior PostRecord - creating fresh');\n effectiveResumeMode = PostRecordResumeMode.NEW;\n } else if (mostRecentRecord.state === PostRecordState.DONE) {\n // Prior was successful - always restart fresh regardless of provided mode\n this.logger\n .withMetadata({ submissionId })\n .info('Prior PostRecord was DONE - creating fresh (restart)');\n effectiveResumeMode = PostRecordResumeMode.NEW;\n } else {\n // Prior was FAILED - use provided mode or default to CONTINUE\n effectiveResumeMode = resumeMode ?? PostRecordResumeMode.CONTINUE;\n this.logger\n .withMetadata({\n submissionId,\n priorRecordId: mostRecentRecord.id,\n resumeMode: effectiveResumeMode,\n })\n .info('Prior PostRecord was FAILED - using resume mode');\n }\n\n const newRecord = await this.postRecordFactory.create(\n submissionId,\n effectiveResumeMode,\n );\n\n await this.repository.insert({\n submissionId,\n postRecordId: newRecord.id,\n });\n }\n // If existing, do nothing (first-in-wins)\n }\n } catch (error) {\n this.logger.withMetadata({ error }).error('Failed to enqueue posts');\n throw new InternalServerErrorException(error.message);\n } finally {\n release();\n this.initTime -= 61_000; // Ensure queue processing starts after next cycle\n }\n }\n\n public async dequeue(submissionIds: SubmissionId[]) {\n const release = await this.queueModificationMutex.acquire();\n this.logger.withMetadata({ submissionIds }).info('Dequeueing posts');\n\n try {\n const records = await this.repository.find({\n where: (queueRecord, { inArray }) =>\n inArray(queueRecord.submissionId, submissionIds),\n });\n\n submissionIds.forEach((id) =>\n this.postManagerRegistry.cancelIfRunning(id),\n );\n\n return await this.repository.deleteById(records.map((r) => r.id));\n } catch (error) {\n this.logger.withMetadata({ error }).error('Failed to dequeue posts');\n throw new InternalServerErrorException(error.message);\n } finally {\n release();\n }\n }\n\n /**\n * CRON based enqueueing of scheduled submissions.\n */\n @Cron(CronExpression.EVERY_30_SECONDS)\n public async checkForScheduledSubmissions() {\n if (IsTestEnvironment()) {\n return;\n }\n\n const entities = await this.submissionRepository.find({\n where: (queueRecord, { eq, and }) =>\n and(\n eq(queueRecord.isScheduled, true),\n eq(queueRecord.isArchived, false),\n ),\n });\n const now = Date.now();\n const sorted = entities\n .filter((e) => new Date(e.schedule.scheduledFor).getTime() <= now) // Only those that are ready to be posted.\n .sort(\n (a, b) =>\n new Date(a.schedule.scheduledFor).getTime() -\n new Date(b.schedule.scheduledFor).getTime(),\n ); // Sort by oldest first.\n await this.enqueue(sorted.map((s) => s.id));\n sorted\n .filter((s) => s.schedule.cron)\n .forEach((s) => {\n const next = CronGenerator(s.schedule.cron).nextRun()?.toISOString();\n if (next) {\n // eslint-disable-next-line no-param-reassign\n s.schedule.scheduledFor = next;\n this.submissionRepository.update(s.id, {\n schedule: s.schedule,\n });\n }\n });\n }\n\n /**\n * This runs a check every second on the state of queue items.\n * This aims to have simple logic. Each run will either create a post record and start the post manager,\n * or remove a submission from the queue if it is in a terminal state.\n * Nothing happens if the queue is empty.\n */\n @Cron(CronExpression.EVERY_SECOND)\n public async run() {\n if (!(this.initTime + 60_000 <= Date.now())) {\n // Only run after 1 minute to allow the application to start up.\n return;\n }\n\n if (IsTestEnvironment()) {\n return;\n }\n\n await this.execute();\n }\n\n /**\n * Check if a RUNNING post record appears to be stuck (no activity for too long).\n * Uses both record update time and last event time to determine activity.\n *\n * @param {PostRecord} record - The post record to check\n * @returns {boolean} True if the record appears to be stuck\n */\n private isStuck(record: PostRecord): boolean {\n if (record.state !== PostRecordState.RUNNING) {\n return false;\n }\n\n // Find the most recent activity: either record update or last event\n const recordUpdatedAt = new Date(record.updatedAt).getTime();\n const lastEventAt = record.events?.length\n ? Math.max(...record.events.map((e) => new Date(e.createdAt).getTime()))\n : 0;\n const lastActivityAt = Math.max(recordUpdatedAt, lastEventAt);\n const idleTime = Date.now() - lastActivityAt;\n\n return idleTime > this.maxPostIdleTime;\n }\n\n /**\n * Manages the queue by peeking at the top of the queue and deciding what to do based on the\n * state of the queue.\n *\n * Made public for testing purposes.\n */\n public async execute() {\n if (this.queueMutex.isLocked()) {\n return;\n }\n\n const release = await this.queueMutex.acquire();\n\n try {\n const isPaused = await this.isPaused();\n if (isPaused) {\n this.logger.info('Queue is paused, skipping execution cycle');\n return;\n }\n\n const top = await this.peek();\n // Queue Empty\n if (!top) {\n return;\n }\n\n const { postRecord: record, submissionId, submission } = top;\n if (submission.isArchived) {\n // Submission is archived, remove from queue\n this.logger\n .withMetadata({ submissionId })\n .info('Submission is archived, removing from queue');\n await this.dequeue([submissionId]);\n return;\n }\n\n if (!record) {\n // PostRecord should always exist since enqueue() creates it\n // If missing, something is wrong - log error and remove from queue\n this.logger\n .withMetadata({ submissionId })\n .error('Queue entry has no PostRecord - removing invalid entry');\n await this.dequeue([submissionId]);\n return;\n }\n\n if (\n record.state === PostRecordState.DONE ||\n record.state === PostRecordState.FAILED\n ) {\n // Post is in a terminal state - handle completion actions and remove from queue\n await this.handleTerminalState(record);\n await this.dequeue([submissionId]);\n } else if (!this.postManagerRegistry.isPostingType(submission.type)) {\n // Post is not in a terminal state, but the manager for this type is not posting, so restart it.\n this.logger\n .withMetadata({ record })\n .info(\n 'PostManager is not posting, but record is not in terminal state. Resuming record.',\n );\n\n // Start the post - the manager will build resume context from the record\n await this.postManagerRegistry.startPost(record);\n } else if (this.isStuck(record)) {\n // Manager is posting but record shows no activity - it's stuck\n const recordUpdatedAt = new Date(record.updatedAt).getTime();\n const lastEventAt = record.events?.length\n ? Math.max(\n ...record.events.map((e) => new Date(e.createdAt).getTime()),\n )\n : 0;\n const lastActivityAt = Math.max(recordUpdatedAt, lastEventAt);\n\n this.logger\n .withMetadata({\n submissionId,\n recordId: record.id,\n idleTime: Date.now() - lastActivityAt,\n lastActivityAt: new Date(lastActivityAt).toISOString(),\n eventCount: record.events?.length ?? 0,\n })\n .warn(\n 'PostRecord has been RUNNING without activity - marking as failed',\n );\n\n await this.postRecordRepository.update(record.id, {\n state: PostRecordState.FAILED,\n completedAt: new Date().toISOString(),\n });\n // Next cycle will handle terminal state\n }\n // else: manager is actively posting and making progress - do nothing\n } catch (error) {\n this.logger.withMetadata({ error }).error('Failed to run queue');\n throw error;\n } finally {\n release();\n }\n }\n\n /**\n * Peeks at the next item in the queue.\n * Based on the createdAt date.\n */\n public async peek(): Promise {\n return this.repository.findOne({\n orderBy: (queueRecord, { asc }) => asc(queueRecord.createdAt),\n with: {\n submission: true,\n postRecord: {\n with: {\n events: true,\n submission: {\n with: {\n files: true,\n options: {\n with: {\n account: true,\n },\n },\n },\n },\n },\n },\n },\n });\n }\n}\n" }, { "path": "apps/client-server/src/app/post/services/post-record-factory/index.ts", "content": "export * from './post-event.repository';\nexport * from './post-record-factory.service';\n\n" }, { "path": "apps/client-server/src/app/post/services/post-record-factory/post-event.repository.ts", "content": "import { Injectable } from '@nestjs/common';\nimport { Insert } from '@postybirb/database';\nimport {\n AccountId,\n EntityId,\n PostEventType,\n} from '@postybirb/types';\nimport { PostEvent } from '../../../drizzle/models';\nimport { PostyBirbDatabase } from '../../../drizzle/postybirb-database/postybirb-database';\n\n/**\n * Repository for querying PostEvent records.\n * Provides specialized query methods for resume mode logic.\n * @class PostEventRepository\n */\n@Injectable()\nexport class PostEventRepository {\n private readonly repository: PostyBirbDatabase<'PostEventSchema'>;\n\n constructor() {\n this.repository = new PostyBirbDatabase('PostEventSchema', {\n account: true,\n });\n }\n\n /**\n * Get all source URLs from a post record, excluding a specific account.\n * Used for cross-website source URL propagation within the current post.\n * @param {EntityId} postRecordId - The post record ID\n * @param {AccountId} excludeAccountId - The account ID to exclude (to avoid self-referential URLs)\n * @returns {Promise} Array of source URLs from other accounts\n */\n async getSourceUrlsFromPost(\n postRecordId: EntityId,\n excludeAccountId: AccountId,\n ): Promise {\n const events = await this.repository.find({\n where: (event, { eq, and, inArray }) =>\n and(\n eq(event.postRecordId, postRecordId),\n inArray(event.eventType, [\n PostEventType.FILE_POSTED,\n PostEventType.MESSAGE_POSTED,\n ]),\n ),\n });\n\n return events\n .filter(\n (event) =>\n event.sourceUrl &&\n event.accountId &&\n event.accountId !== excludeAccountId,\n )\n .map((event) => event.sourceUrl as string);\n }\n\n /**\n * Get all error events for a post record.\n * @param {EntityId} postRecordId - The post record ID\n * @returns {Promise} All failed events\n */\n async getFailedEvents(postRecordId: EntityId): Promise {\n return this.repository.find({\n where: (event, { eq, and, inArray }) =>\n and(\n eq(event.postRecordId, postRecordId),\n inArray(event.eventType, [\n PostEventType.POST_ATTEMPT_FAILED,\n PostEventType.FILE_FAILED,\n PostEventType.MESSAGE_FAILED,\n ]),\n ),\n });\n }\n\n /**\n * Insert a new post event.\n * @param {Insert<'PostEventSchema'>} event - The event to insert\n * @returns {Promise} The inserted event\n */\n async insert(event: Insert<'PostEventSchema'>): Promise {\n return this.repository.insert(event) as Promise;\n }\n}\n" }, { "path": "apps/client-server/src/app/post/services/post-record-factory/post-record-factory.service.spec.ts", "content": "import { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport {\n EntityId,\n PostEventType,\n PostRecordResumeMode,\n PostRecordState,\n SubmissionType,\n} from '@postybirb/types';\nimport { AccountModule } from '../../../account/account.module';\nimport { AccountService } from '../../../account/account.service';\nimport { PostEvent, PostRecord } from '../../../drizzle/models';\nimport { PostyBirbDatabase } from '../../../drizzle/postybirb-database/postybirb-database';\nimport { PostParsersModule } from '../../../post-parsers/post-parsers.module';\nimport { CreateSubmissionDto } from '../../../submission/dtos/create-submission.dto';\nimport { SubmissionService } from '../../../submission/services/submission.service';\nimport { SubmissionModule } from '../../../submission/submission.module';\nimport { UserSpecifiedWebsiteOptionsModule } from '../../../user-specified-website-options/user-specified-website-options.module';\nimport { WebsitesModule } from '../../../websites/websites.module';\nimport { InvalidPostChainError } from '../../errors';\nimport { PostEventRepository } from './post-event.repository';\nimport { PostRecordFactory, ResumeContext } from './post-record-factory.service';\n\ndescribe('PostRecordFactory', () => {\n let module: TestingModule;\n let factory: PostRecordFactory;\n let submissionService: SubmissionService;\n let accountService: AccountService;\n let postRecordRepository: PostyBirbDatabase<'PostRecordSchema'>;\n let postEventRepository: PostEventRepository;\n\n beforeEach(async () => {\n clearDatabase();\n\n module = await Test.createTestingModule({\n imports: [\n SubmissionModule,\n AccountModule,\n WebsitesModule,\n UserSpecifiedWebsiteOptionsModule,\n PostParsersModule,\n ],\n providers: [PostRecordFactory, PostEventRepository],\n }).compile();\n\n factory = module.get(PostRecordFactory);\n submissionService = module.get(SubmissionService);\n accountService = module.get(AccountService);\n postEventRepository = module.get(PostEventRepository);\n postRecordRepository = new PostyBirbDatabase('PostRecordSchema');\n\n await accountService.onModuleInit();\n });\n\n afterAll(async () => {\n await module.close();\n });\n\n async function createSubmission(): Promise {\n const dto = new CreateSubmissionDto();\n dto.name = 'Test Submission';\n dto.type = SubmissionType.MESSAGE;\n const submission = await submissionService.create(dto);\n return submission.id;\n }\n\n async function createAccount(name: string): Promise {\n const dto = {\n name,\n website: 'test',\n groups: [],\n };\n const account = await accountService.create(dto as any);\n return account.id;\n }\n\n describe('create', () => {\n it('should create a PostRecord with default NEW mode and null originPostRecordId', async () => {\n const submissionId = await createSubmission();\n const record = await factory.create(submissionId);\n\n expect(record).toBeDefined();\n expect(record.submissionId).toBe(submissionId);\n expect(record.state).toBe(PostRecordState.PENDING);\n expect(record.resumeMode).toBe(PostRecordResumeMode.NEW);\n expect(record.originPostRecordId).toBeNull();\n expect(record.id).toBeDefined();\n expect(record.createdAt).toBeDefined();\n });\n\n it('should create multiple records for different submissions', async () => {\n const submission1 = await createSubmission();\n const submission2 = await createSubmission();\n\n const record1 = await factory.create(submission1);\n const record2 = await factory.create(submission2);\n\n expect(record1.submissionId).toBe(submission1);\n expect(record2.submissionId).toBe(submission2);\n expect(record1.id).not.toBe(record2.id);\n expect(record1.originPostRecordId).toBeNull();\n expect(record2.originPostRecordId).toBeNull();\n });\n\n it('should set originPostRecordId when creating CONTINUE record', async () => {\n const submissionId = await createSubmission();\n\n // Create origin NEW record first\n const originRecord = await factory.create(submissionId, PostRecordResumeMode.NEW);\n \n // Mark it as FAILED so CONTINUE is valid\n await postRecordRepository.update(originRecord.id, { state: PostRecordState.FAILED });\n\n const continueRecord = await factory.create(submissionId, PostRecordResumeMode.CONTINUE);\n\n expect(continueRecord.resumeMode).toBe(PostRecordResumeMode.CONTINUE);\n expect(continueRecord.originPostRecordId).toBe(originRecord.id);\n });\n\n it('should set originPostRecordId when creating CONTINUE_RETRY record', async () => {\n const submissionId = await createSubmission();\n\n // Create origin NEW record first\n const originRecord = await factory.create(submissionId, PostRecordResumeMode.NEW);\n \n // Mark it as FAILED so CONTINUE_RETRY is valid\n await postRecordRepository.update(originRecord.id, { state: PostRecordState.FAILED });\n\n const retryRecord = await factory.create(submissionId, PostRecordResumeMode.CONTINUE_RETRY);\n\n expect(retryRecord.resumeMode).toBe(PostRecordResumeMode.CONTINUE_RETRY);\n expect(retryRecord.originPostRecordId).toBe(originRecord.id);\n });\n\n it('should throw InvalidPostChainError for CONTINUE without any origin', async () => {\n const submissionId = await createSubmission();\n\n await expect(\n factory.create(submissionId, PostRecordResumeMode.CONTINUE),\n ).rejects.toThrow(InvalidPostChainError);\n\n try {\n await factory.create(submissionId, PostRecordResumeMode.CONTINUE);\n } catch (error) {\n expect(error).toBeInstanceOf(InvalidPostChainError);\n expect((error as InvalidPostChainError).reason).toBe('no_origin');\n }\n });\n\n it('should throw InvalidPostChainError for CONTINUE_RETRY without any origin', async () => {\n const submissionId = await createSubmission();\n\n await expect(\n factory.create(submissionId, PostRecordResumeMode.CONTINUE_RETRY),\n ).rejects.toThrow(InvalidPostChainError);\n\n try {\n await factory.create(submissionId, PostRecordResumeMode.CONTINUE_RETRY);\n } catch (error) {\n expect(error).toBeInstanceOf(InvalidPostChainError);\n expect((error as InvalidPostChainError).reason).toBe('no_origin');\n }\n });\n\n it('should throw InvalidPostChainError when origin is DONE', async () => {\n const submissionId = await createSubmission();\n\n // Create origin NEW record and mark it DONE (closed chain)\n const originRecord = await factory.create(submissionId, PostRecordResumeMode.NEW);\n await postRecordRepository.update(originRecord.id, { state: PostRecordState.DONE });\n\n await expect(\n factory.create(submissionId, PostRecordResumeMode.CONTINUE),\n ).rejects.toThrow(InvalidPostChainError);\n\n try {\n await factory.create(submissionId, PostRecordResumeMode.CONTINUE);\n } catch (error) {\n expect(error).toBeInstanceOf(InvalidPostChainError);\n expect((error as InvalidPostChainError).reason).toBe('origin_done');\n }\n });\n\n it('should chain to most recent NEW record, not older ones', async () => {\n const submissionId = await createSubmission();\n\n // Create first NEW record and mark it DONE\n const oldOrigin = await factory.create(submissionId, PostRecordResumeMode.NEW);\n await postRecordRepository.update(oldOrigin.id, { state: PostRecordState.DONE });\n\n // Create second NEW record (new chain)\n const newOrigin = await factory.create(submissionId, PostRecordResumeMode.NEW);\n await postRecordRepository.update(newOrigin.id, { state: PostRecordState.FAILED });\n\n // CONTINUE should chain to the newer origin\n const continueRecord = await factory.create(submissionId, PostRecordResumeMode.CONTINUE);\n\n expect(continueRecord.originPostRecordId).toBe(newOrigin.id);\n expect(continueRecord.originPostRecordId).not.toBe(oldOrigin.id);\n });\n\n it('should throw InvalidPostChainError when PENDING record exists for submission', async () => {\n const submissionId = await createSubmission();\n\n // Create a PENDING record (simulating one already in the queue)\n await postRecordRepository.insert({\n submissionId,\n state: PostRecordState.PENDING,\n resumeMode: PostRecordResumeMode.NEW,\n originPostRecordId: null,\n });\n\n // Attempting to create another should fail\n await expect(\n factory.create(submissionId, PostRecordResumeMode.NEW),\n ).rejects.toThrow(InvalidPostChainError);\n\n try {\n await factory.create(submissionId, PostRecordResumeMode.NEW);\n } catch (error) {\n expect(error).toBeInstanceOf(InvalidPostChainError);\n expect((error as InvalidPostChainError).reason).toBe('in_progress');\n }\n });\n\n it('should throw InvalidPostChainError when RUNNING record exists for submission', async () => {\n const submissionId = await createSubmission();\n\n // Create a RUNNING record (simulating one currently being processed)\n await postRecordRepository.insert({\n submissionId,\n state: PostRecordState.RUNNING,\n resumeMode: PostRecordResumeMode.NEW,\n originPostRecordId: null,\n });\n\n // Attempting to create another should fail\n await expect(\n factory.create(submissionId, PostRecordResumeMode.NEW),\n ).rejects.toThrow(InvalidPostChainError);\n\n try {\n await factory.create(submissionId, PostRecordResumeMode.NEW);\n } catch (error) {\n expect(error).toBeInstanceOf(InvalidPostChainError);\n expect((error as InvalidPostChainError).reason).toBe('in_progress');\n }\n });\n\n it('should allow creating NEW record when prior records are FAILED or DONE', async () => {\n const submissionId = await createSubmission();\n\n // Create a FAILED record\n await postRecordRepository.insert({\n submissionId,\n state: PostRecordState.FAILED,\n resumeMode: PostRecordResumeMode.NEW,\n originPostRecordId: null,\n });\n\n // Create a DONE record\n await postRecordRepository.insert({\n submissionId,\n state: PostRecordState.DONE,\n resumeMode: PostRecordResumeMode.NEW,\n originPostRecordId: null,\n });\n\n // Creating a NEW record should succeed (no PENDING/RUNNING blocking)\n const record = await factory.create(submissionId, PostRecordResumeMode.NEW);\n expect(record).toBeDefined();\n expect(record.state).toBe(PostRecordState.PENDING);\n });\n });\n\n describe('buildResumeContext', () => {\n async function createPostRecordWithState(\n submissionId: EntityId,\n state: PostRecordState,\n resumeMode: PostRecordResumeMode,\n originPostRecordId?: EntityId,\n ): Promise {\n const record = await postRecordRepository.insert({\n submissionId,\n state,\n resumeMode,\n originPostRecordId: originPostRecordId ?? null,\n });\n return record;\n }\n\n async function addEvent(\n postRecordId: EntityId,\n eventType: PostEventType,\n accountId?: string,\n additionalData?: Partial,\n ): Promise {\n const eventData: any = {\n postRecordId,\n eventType,\n ...additionalData,\n };\n \n // Only add accountId if provided\n if (accountId) {\n eventData.accountId = accountId as EntityId;\n }\n \n return postEventRepository.insert(eventData);\n }\n\n it('should return empty context for NEW mode', async () => {\n const submissionId = await createSubmission();\n // Create an origin NEW record first\n const originRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.FAILED,\n PostRecordResumeMode.NEW,\n );\n // Create a CONTINUE record chained to it\n const priorRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.FAILED,\n PostRecordResumeMode.CONTINUE,\n originRecord.id,\n );\n\n const context = await factory.buildResumeContext(\n submissionId,\n priorRecord.id,\n PostRecordResumeMode.NEW,\n );\n\n expect(context.completedAccountIds.size).toBe(0);\n expect(context.postedFilesByAccount.size).toBe(0);\n expect(context.sourceUrlsByAccount.size).toBe(0);\n });\n\n it('should return empty context when no prior records exist', async () => {\n const submissionId = await createSubmission();\n // Create a NEW record with no chain\n const priorRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.PENDING,\n PostRecordResumeMode.NEW,\n );\n\n const context = await factory.buildResumeContext(\n submissionId,\n priorRecord.id,\n PostRecordResumeMode.CONTINUE,\n );\n\n expect(context.completedAccountIds.size).toBe(0);\n expect(context.postedFilesByAccount.size).toBe(0);\n });\n\n it('should aggregate completed accounts', async () => {\n const submissionId = await createSubmission();\n const account1 = await createAccount('account-1');\n const account2 = await createAccount('account-2');\n \n // Create origin NEW record\n const originRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.FAILED,\n PostRecordResumeMode.NEW,\n );\n\n await addEvent(\n originRecord.id,\n PostEventType.POST_ATTEMPT_COMPLETED,\n account1,\n );\n await addEvent(\n originRecord.id,\n PostEventType.POST_ATTEMPT_COMPLETED,\n account2,\n );\n\n const context = await factory.buildResumeContext(\n submissionId,\n originRecord.id,\n PostRecordResumeMode.CONTINUE_RETRY,\n );\n\n expect(context.completedAccountIds.size).toBe(2);\n expect(context.completedAccountIds.has(account1)).toBe(true);\n expect(context.completedAccountIds.has(account2)).toBe(true);\n });\n\n it('should aggregate posted files in CONTINUE mode', async () => {\n const submissionId = await createSubmission();\n const account1 = await createAccount('account-1');\n \n // Create origin NEW record\n const originRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.FAILED,\n PostRecordResumeMode.NEW,\n );\n\n await addEvent(originRecord.id, PostEventType.FILE_POSTED, account1, {\n fileId: 'file-1' as EntityId,\n sourceUrl: 'https://example.com/1',\n });\n await addEvent(originRecord.id, PostEventType.FILE_POSTED, account1, {\n fileId: 'file-2' as EntityId,\n sourceUrl: 'https://example.com/2',\n });\n\n const context = await factory.buildResumeContext(\n submissionId,\n originRecord.id,\n PostRecordResumeMode.CONTINUE,\n );\n\n expect(context.postedFilesByAccount.size).toBe(1);\n const postedFiles = context.postedFilesByAccount.get(account1);\n expect(postedFiles?.size).toBe(2);\n expect(postedFiles?.has('file-1' as EntityId)).toBe(true);\n expect(postedFiles?.has('file-2' as EntityId)).toBe(true);\n });\n\n it('should NOT aggregate posted files in CONTINUE_RETRY mode', async () => {\n const submissionId = await createSubmission();\n const account1 = await createAccount('account-1');\n \n // Create origin NEW record\n const originRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.FAILED,\n PostRecordResumeMode.NEW,\n );\n\n await addEvent(\n originRecord.id,\n PostEventType.FILE_POSTED,\n account1,\n {\n fileId: 'file-1' as EntityId,\n },\n );\n\n const context = await factory.buildResumeContext(\n submissionId,\n originRecord.id,\n PostRecordResumeMode.CONTINUE_RETRY,\n );\n\n expect(context.postedFilesByAccount.size).toBe(0);\n });\n\n it('should aggregate source URLs from FILE_POSTED events', async () => {\n const submissionId = await createSubmission();\n const account1 = await createAccount('account-1');\n \n // Create origin NEW record\n const originRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.FAILED,\n PostRecordResumeMode.NEW,\n );\n\n await addEvent(\n originRecord.id,\n PostEventType.FILE_POSTED,\n account1,\n {\n sourceUrl: 'https://example.com/post1',\n },\n );\n\n const context = await factory.buildResumeContext(\n submissionId,\n originRecord.id,\n PostRecordResumeMode.CONTINUE,\n );\n\n const sourceUrls = context.sourceUrlsByAccount.get(account1);\n expect(sourceUrls).toHaveLength(1);\n expect(sourceUrls?.[0]).toBe('https://example.com/post1');\n });\n\n it('should aggregate source URLs from MESSAGE_POSTED events', async () => {\n const submissionId = await createSubmission();\n const account1 = await createAccount('account-1');\n \n // Create origin NEW record\n const originRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.FAILED,\n PostRecordResumeMode.NEW,\n );\n\n await addEvent(\n originRecord.id,\n PostEventType.MESSAGE_POSTED,\n account1,\n {\n sourceUrl: 'https://example.com/message1',\n },\n );\n\n const context = await factory.buildResumeContext(\n submissionId,\n originRecord.id,\n PostRecordResumeMode.CONTINUE,\n );\n\n const sourceUrls = context.sourceUrlsByAccount.get(account1);\n expect(sourceUrls).toHaveLength(1);\n expect(sourceUrls?.[0]).toBe('https://example.com/message1');\n });\n\n it('should only aggregate events from current chain (not prior DONE chain)', async () => {\n const submissionId = await createSubmission();\n const accountOld = await createAccount('account-old');\n const accountNew = await createAccount('account-new');\n\n // Create older chain that completed successfully (DONE)\n const olderOrigin = await createPostRecordWithState(\n submissionId,\n PostRecordState.DONE,\n PostRecordResumeMode.NEW,\n );\n await addEvent(\n olderOrigin.id,\n PostEventType.POST_ATTEMPT_COMPLETED,\n accountOld,\n );\n\n // Create new chain origin\n const newOrigin = await createPostRecordWithState(\n submissionId,\n PostRecordState.FAILED,\n PostRecordResumeMode.NEW,\n );\n await addEvent(\n newOrigin.id,\n PostEventType.POST_ATTEMPT_COMPLETED,\n accountNew,\n );\n\n const context = await factory.buildResumeContext(\n submissionId,\n newOrigin.id,\n PostRecordResumeMode.CONTINUE_RETRY,\n );\n\n // Should only have account-new from the current chain\n expect(context.completedAccountIds.size).toBe(1);\n expect(context.completedAccountIds.has(accountNew)).toBe(true);\n expect(context.completedAccountIds.has(accountOld)).toBe(false);\n });\n\n it('should aggregate events from origin and all chained records', async () => {\n const submissionId = await createSubmission();\n const accountRestart = await createAccount('account-restart');\n const accountNew = await createAccount('account-new');\n\n // Create NEW record (origin)\n const originRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.FAILED,\n PostRecordResumeMode.NEW,\n );\n await addEvent(\n originRecord.id,\n PostEventType.POST_ATTEMPT_COMPLETED,\n accountRestart,\n );\n\n // Create CONTINUE record chained to origin\n const continueRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.FAILED,\n PostRecordResumeMode.CONTINUE,\n originRecord.id,\n );\n await addEvent(\n continueRecord.id,\n PostEventType.POST_ATTEMPT_COMPLETED,\n accountNew,\n );\n\n const context = await factory.buildResumeContext(\n submissionId,\n continueRecord.id,\n PostRecordResumeMode.CONTINUE_RETRY,\n );\n\n // Should have both accounts from the chain\n expect(context.completedAccountIds.size).toBe(2);\n expect(context.completedAccountIds.has(accountNew)).toBe(true);\n expect(context.completedAccountIds.has(accountRestart)).toBe(true);\n });\n\n it('should aggregate events from multiple chained FAILED records', async () => {\n const submissionId = await createSubmission();\n const account1 = await createAccount('account-1');\n const account2 = await createAccount('account-2');\n const account3 = await createAccount('account-3');\n\n // Create origin NEW record\n const originRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.FAILED,\n PostRecordResumeMode.NEW,\n );\n await addEvent(\n originRecord.id,\n PostEventType.POST_ATTEMPT_COMPLETED,\n account1,\n );\n\n // Create first CONTINUE chained to origin\n const continue1 = await createPostRecordWithState(\n submissionId,\n PostRecordState.FAILED,\n PostRecordResumeMode.CONTINUE,\n originRecord.id,\n );\n await addEvent(\n continue1.id,\n PostEventType.POST_ATTEMPT_COMPLETED,\n account2,\n );\n\n // Create second CONTINUE chained to origin\n const continue2 = await createPostRecordWithState(\n submissionId,\n PostRecordState.FAILED,\n PostRecordResumeMode.CONTINUE,\n originRecord.id,\n );\n await addEvent(\n continue2.id,\n PostEventType.POST_ATTEMPT_COMPLETED,\n account3,\n );\n\n const context = await factory.buildResumeContext(\n submissionId,\n continue2.id,\n PostRecordResumeMode.CONTINUE_RETRY,\n );\n\n expect(context.completedAccountIds.size).toBe(3);\n expect(context.completedAccountIds.has(account1)).toBe(true);\n expect(context.completedAccountIds.has(account2)).toBe(true);\n expect(context.completedAccountIds.has(account3)).toBe(true);\n });\n });\n\n describe('shouldSkipAccount', () => {\n it('should return true for completed accounts', () => {\n const context: ResumeContext = {\n resumeMode: PostRecordResumeMode.CONTINUE_RETRY,\n completedAccountIds: new Set(['account-1' as EntityId]),\n postedFilesByAccount: new Map(),\n sourceUrlsByAccount: new Map(),\n };\n\n expect(factory.shouldSkipAccount(context, 'account-1' as EntityId)).toBe(\n true,\n );\n });\n\n it('should return false for non-completed accounts', () => {\n const context: ResumeContext = {\n resumeMode: PostRecordResumeMode.CONTINUE_RETRY,\n completedAccountIds: new Set(['account-1' as EntityId]),\n postedFilesByAccount: new Map(),\n sourceUrlsByAccount: new Map(),\n };\n\n expect(factory.shouldSkipAccount(context, 'account-2' as EntityId)).toBe(\n false,\n );\n });\n });\n\n describe('shouldSkipFile', () => {\n it('should return false in NEW mode', () => {\n const context: ResumeContext = {\n resumeMode: PostRecordResumeMode.NEW,\n completedAccountIds: new Set(),\n postedFilesByAccount: new Map(),\n sourceUrlsByAccount: new Map(),\n };\n\n expect(\n factory.shouldSkipFile(\n context,\n 'account-1' as EntityId,\n 'file-1' as EntityId,\n ),\n ).toBe(false);\n });\n\n it('should return false in CONTINUE_RETRY mode', () => {\n const context: ResumeContext = {\n resumeMode: PostRecordResumeMode.CONTINUE_RETRY,\n completedAccountIds: new Set(),\n postedFilesByAccount: new Map([\n ['account-1' as EntityId, new Set(['file-1' as EntityId])],\n ]),\n sourceUrlsByAccount: new Map(),\n };\n\n expect(\n factory.shouldSkipFile(\n context,\n 'account-1' as EntityId,\n 'file-1' as EntityId,\n ),\n ).toBe(false);\n });\n\n it('should return true for posted files in CONTINUE mode', () => {\n const context: ResumeContext = {\n resumeMode: PostRecordResumeMode.CONTINUE,\n completedAccountIds: new Set(),\n postedFilesByAccount: new Map([\n ['account-1' as EntityId, new Set(['file-1' as EntityId])],\n ]),\n sourceUrlsByAccount: new Map(),\n };\n\n expect(\n factory.shouldSkipFile(\n context,\n 'account-1' as EntityId,\n 'file-1' as EntityId,\n ),\n ).toBe(true);\n });\n\n it('should return false for non-posted files in CONTINUE mode', () => {\n const context: ResumeContext = {\n resumeMode: PostRecordResumeMode.CONTINUE,\n completedAccountIds: new Set(),\n postedFilesByAccount: new Map([\n ['account-1' as EntityId, new Set(['file-1' as EntityId])],\n ]),\n sourceUrlsByAccount: new Map(),\n };\n\n expect(\n factory.shouldSkipFile(\n context,\n 'account-1' as EntityId,\n 'file-2' as EntityId,\n ),\n ).toBe(false);\n });\n });\n\n describe('getSourceUrlsForAccount', () => {\n it('should return source URLs for an account', () => {\n const context: ResumeContext = {\n resumeMode: PostRecordResumeMode.CONTINUE,\n completedAccountIds: new Set(),\n postedFilesByAccount: new Map(),\n sourceUrlsByAccount: new Map([\n [\n 'account-1' as EntityId,\n ['https://example.com/1', 'https://example.com/2'],\n ],\n ]),\n };\n\n const urls = factory.getSourceUrlsForAccount(\n context,\n 'account-1' as EntityId,\n );\n expect(urls).toHaveLength(2);\n expect(urls[0]).toBe('https://example.com/1');\n expect(urls[1]).toBe('https://example.com/2');\n });\n\n it('should return empty array for account with no URLs', () => {\n const context: ResumeContext = {\n resumeMode: PostRecordResumeMode.CONTINUE,\n completedAccountIds: new Set(),\n postedFilesByAccount: new Map(),\n sourceUrlsByAccount: new Map(),\n };\n\n const urls = factory.getSourceUrlsForAccount(\n context,\n 'account-1' as EntityId,\n );\n expect(urls).toHaveLength(0);\n });\n });\n\n describe('getAllSourceUrls', () => {\n it('should return all source URLs from all accounts', () => {\n const context: ResumeContext = {\n resumeMode: PostRecordResumeMode.CONTINUE,\n completedAccountIds: new Set(),\n postedFilesByAccount: new Map(),\n sourceUrlsByAccount: new Map([\n ['account-1' as EntityId, ['https://example.com/1']],\n ['account-2' as EntityId, ['https://example.com/2']],\n ]),\n };\n\n const urls = factory.getAllSourceUrls(context);\n expect(urls).toHaveLength(2);\n expect(urls).toContain('https://example.com/1');\n expect(urls).toContain('https://example.com/2');\n });\n\n it('should return empty array when no URLs exist', () => {\n const context: ResumeContext = {\n resumeMode: PostRecordResumeMode.CONTINUE,\n completedAccountIds: new Set(),\n postedFilesByAccount: new Map(),\n sourceUrlsByAccount: new Map(),\n };\n\n const urls = factory.getAllSourceUrls(context);\n expect(urls).toHaveLength(0);\n });\n });\n\n describe('crash recovery', () => {\n /**\n * These tests verify the crash recovery behavior:\n * When PostyBirb crashes mid-post, a PostRecord is left in RUNNING state.\n * On restart, onModuleInit() finds these RUNNING records and resumes them.\n * The key behavior is that events from the RUNNING record must be preserved,\n * EVEN if the user originally requested NEW mode (fresh start).\n */\n\n async function createPostRecordWithState(\n submissionId: EntityId,\n state: PostRecordState,\n resumeMode: PostRecordResumeMode,\n originPostRecordId?: EntityId,\n ): Promise {\n const record = await postRecordRepository.insert({\n submissionId,\n state,\n resumeMode,\n originPostRecordId: originPostRecordId ?? null,\n });\n return record;\n }\n\n async function addEvent(\n postRecordId: EntityId,\n eventType: PostEventType,\n accountId?: string,\n additionalData?: Partial,\n ): Promise {\n const eventData: any = {\n postRecordId,\n eventType,\n ...additionalData,\n };\n if (accountId) {\n eventData.accountId = accountId as EntityId;\n }\n return postEventRepository.insert(eventData);\n }\n\n it('should aggregate events from RUNNING record even with NEW mode (crash recovery)', async () => {\n // Scenario: User started a NEW post, posted to 2 accounts, then crashed.\n // On restart, we must preserve those 2 completed accounts.\n const submissionId = await createSubmission();\n const account1 = await createAccount('crash-account-1');\n const account2 = await createAccount('crash-account-2');\n\n // Create a RUNNING record with NEW mode (simulates mid-post crash)\n // NEW mode records are their own origin (originPostRecordId = null)\n const runningRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.RUNNING,\n PostRecordResumeMode.NEW,\n );\n\n // Add events that occurred before crash\n await addEvent(runningRecord.id, PostEventType.POST_ATTEMPT_COMPLETED, account1);\n await addEvent(runningRecord.id, PostEventType.POST_ATTEMPT_COMPLETED, account2);\n await addEvent(runningRecord.id, PostEventType.FILE_POSTED, account1, {\n fileId: 'crash-file-1' as EntityId,\n sourceUrl: 'https://example.com/crash-1',\n });\n\n // Simulate crash recovery: buildResumeContext is called with NEW mode\n // but the record is still RUNNING (not terminal)\n const context = await factory.buildResumeContext(\n submissionId,\n runningRecord.id,\n PostRecordResumeMode.NEW,\n );\n\n // Despite NEW mode, crash recovery should preserve completed accounts\n expect(context.completedAccountIds.size).toBe(2);\n expect(context.completedAccountIds.has(account1)).toBe(true);\n expect(context.completedAccountIds.has(account2)).toBe(true);\n\n // Crash recovery with NEW mode should include posted files to avoid re-uploading\n expect(context.postedFilesByAccount.size).toBe(1);\n const postedFiles = context.postedFilesByAccount.get(account1);\n expect(postedFiles?.has('crash-file-1' as EntityId)).toBe(true);\n });\n\n it('should return empty context for NEW mode with terminal (FAILED/DONE) record', async () => {\n // Normal NEW behavior: if the prior record is terminal, start fresh\n const submissionId = await createSubmission();\n const account1 = await createAccount('terminal-account-1');\n\n // Create origin NEW record that failed\n const originRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.FAILED,\n PostRecordResumeMode.NEW,\n );\n\n // Create a CONTINUE record chained to it\n const failedRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.FAILED,\n PostRecordResumeMode.CONTINUE,\n originRecord.id,\n );\n\n await addEvent(failedRecord.id, PostEventType.POST_ATTEMPT_COMPLETED, account1);\n\n // NEW mode with a FAILED record should return empty context\n const context = await factory.buildResumeContext(\n submissionId,\n failedRecord.id,\n PostRecordResumeMode.NEW,\n );\n\n expect(context.completedAccountIds.size).toBe(0);\n expect(context.postedFilesByAccount.size).toBe(0);\n });\n\n it('should aggregate events from RUNNING record with CONTINUE mode (crash recovery)', async () => {\n // CONTINUE mode crash recovery should also work\n const submissionId = await createSubmission();\n const account1 = await createAccount('continue-crash-account-1');\n\n // Create origin NEW record\n const originRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.FAILED,\n PostRecordResumeMode.NEW,\n );\n\n // Create RUNNING CONTINUE record chained to origin\n const runningRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.RUNNING,\n PostRecordResumeMode.CONTINUE,\n originRecord.id,\n );\n\n await addEvent(runningRecord.id, PostEventType.POST_ATTEMPT_COMPLETED, account1);\n await addEvent(runningRecord.id, PostEventType.FILE_POSTED, account1, {\n fileId: 'continue-crash-file-1' as EntityId,\n });\n\n const context = await factory.buildResumeContext(\n submissionId,\n runningRecord.id,\n PostRecordResumeMode.CONTINUE,\n );\n\n // CONTINUE mode crash recovery should include both completed accounts and posted files\n expect(context.completedAccountIds.size).toBe(1);\n expect(context.completedAccountIds.has(account1)).toBe(true);\n expect(context.postedFilesByAccount.size).toBe(1);\n const postedFiles = context.postedFilesByAccount.get(account1);\n expect(postedFiles?.has('continue-crash-file-1' as EntityId)).toBe(true);\n });\n\n it('should combine RUNNING record with prior FAILED records in CONTINUE mode', async () => {\n // Scenario: Previous attempt failed, user chose CONTINUE, then crashed mid-post.\n // We need to aggregate from both the RUNNING record AND the prior FAILED record.\n const submissionId = await createSubmission();\n const account1 = await createAccount('prior-account-1');\n const account2 = await createAccount('current-account-2');\n\n // Create origin NEW record\n const originRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.FAILED,\n PostRecordResumeMode.NEW,\n );\n await addEvent(originRecord.id, PostEventType.POST_ATTEMPT_COMPLETED, account1);\n await addEvent(originRecord.id, PostEventType.FILE_POSTED, account1, {\n fileId: 'prior-file-1' as EntityId,\n });\n\n // Second attempt: CONTINUE from prior, posted to account2, then crashed\n const runningRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.RUNNING,\n PostRecordResumeMode.CONTINUE,\n originRecord.id,\n );\n await addEvent(runningRecord.id, PostEventType.POST_ATTEMPT_COMPLETED, account2);\n await addEvent(runningRecord.id, PostEventType.FILE_POSTED, account2, {\n fileId: 'current-file-1' as EntityId,\n });\n\n const context = await factory.buildResumeContext(\n submissionId,\n runningRecord.id,\n PostRecordResumeMode.CONTINUE,\n );\n\n // Should include completed accounts from both records\n expect(context.completedAccountIds.size).toBe(2);\n expect(context.completedAccountIds.has(account1)).toBe(true);\n expect(context.completedAccountIds.has(account2)).toBe(true);\n\n // Should include posted files from both records\n expect(context.postedFilesByAccount.size).toBe(2);\n expect(context.postedFilesByAccount.get(account1)?.has('prior-file-1' as EntityId)).toBe(true);\n expect(context.postedFilesByAccount.get(account2)?.has('current-file-1' as EntityId)).toBe(true);\n });\n\n it('should preserve source URLs from RUNNING record in crash recovery', async () => {\n const submissionId = await createSubmission();\n const account1 = await createAccount('url-crash-account-1');\n\n // Create RUNNING NEW record (origin with crash)\n const runningRecord = await createPostRecordWithState(\n submissionId,\n PostRecordState.RUNNING,\n PostRecordResumeMode.NEW,\n );\n\n await addEvent(runningRecord.id, PostEventType.FILE_POSTED, account1, {\n fileId: 'url-file-1' as EntityId,\n sourceUrl: 'https://example.com/posted-before-crash',\n });\n\n const context = await factory.buildResumeContext(\n submissionId,\n runningRecord.id,\n PostRecordResumeMode.NEW,\n );\n\n // Source URLs should be preserved for NEW mode crash recovery\n expect(context.sourceUrlsByAccount.size).toBe(1);\n const urls = context.sourceUrlsByAccount.get(account1);\n expect(urls).toContain('https://example.com/posted-before-crash');\n });\n });\n});\n" }, { "path": "apps/client-server/src/app/post/services/post-record-factory/post-record-factory.service.ts", "content": "import { Injectable } from '@nestjs/common';\nimport { Logger } from '@postybirb/logger';\nimport {\n AccountId,\n EntityId,\n PostEventType,\n PostRecordResumeMode,\n PostRecordState,\n} from '@postybirb/types';\nimport { PostEvent, PostRecord } from '../../../drizzle/models';\nimport { PostyBirbDatabase } from '../../../drizzle/postybirb-database/postybirb-database';\nimport { InvalidPostChainError } from '../../errors';\nimport { PostEventRepository } from './post-event.repository';\n\n/**\n * Resume context containing information from a prior post attempt.\n * Used to determine what to skip or retry when resuming.\n * @interface ResumeContext\n */\nexport interface ResumeContext {\n /**\n * The resume mode used to create this context.\n * @type {PostRecordResumeMode}\n */\n resumeMode: PostRecordResumeMode;\n\n /**\n * Account IDs that have already completed successfully.\n * For CONTINUE_RETRY mode, these accounts should be skipped entirely.\n * @type {Set}\n */\n completedAccountIds: Set;\n\n /**\n * Map of account ID to file IDs that have been successfully posted.\n * For CONTINUE mode, these files should be skipped for each account.\n * @type {Map>}\n */\n postedFilesByAccount: Map>;\n\n /**\n * Map of account ID to source URLs from prior successful posts.\n * Used for cross-website source URL propagation.\n * @type {Map}\n */\n sourceUrlsByAccount: Map;\n}\n\n/**\n * Factory service for creating PostRecord entities.\n * Handles resume mode logic by querying the event ledger.\n * @class PostRecordFactory\n */\n@Injectable()\nexport class PostRecordFactory {\n private readonly logger = Logger(this.constructor.name);\n\n private readonly postRecordRepository: PostyBirbDatabase<'PostRecordSchema'>;\n\n constructor(private readonly postEventRepository: PostEventRepository) {\n this.postRecordRepository = new PostyBirbDatabase('PostRecordSchema');\n }\n\n /**\n * Create a new PostRecord for a submission.\n *\n * For NEW mode: Creates a fresh record with originPostRecordId = null (it IS the origin).\n * For CONTINUE/RETRY: Finds the most recent NEW record for this submission and chains to it.\n *\n * @param {EntityId} submissionId - The submission ID\n * @param {PostRecordResumeMode} resumeMode - The resume mode (defaults to NEW)\n * @returns {Promise} The created post record\n * @throws {InvalidPostChainError} If CONTINUE/RETRY is requested but no valid origin exists,\n * or if a PostRecord is already PENDING or RUNNING for this submission\n */\n async create(\n submissionId: EntityId,\n resumeMode: PostRecordResumeMode = PostRecordResumeMode.NEW,\n ): Promise {\n this.logger\n .withMetadata({ submissionId, resumeMode })\n .info('Creating post record');\n\n // Guard: Prevent creating a new record if one is already in progress\n const inProgressRecord = await this.findInProgressRecord(submissionId);\n if (inProgressRecord) {\n this.logger\n .withMetadata({ existingRecordId: inProgressRecord.id, existingState: inProgressRecord.state })\n .warn('Cannot create PostRecord: submission already has an in-progress record');\n throw new InvalidPostChainError(submissionId, resumeMode, 'in_progress');\n }\n\n let originPostRecordId: EntityId | null = null;\n\n if (resumeMode !== PostRecordResumeMode.NEW) {\n // Find the most recent NEW record for this submission (no state filter)\n const originRecord = await this.findMostRecentOrigin(submissionId);\n\n if (!originRecord) {\n throw new InvalidPostChainError(submissionId, resumeMode, 'no_origin');\n }\n\n if (originRecord.state === PostRecordState.DONE) {\n throw new InvalidPostChainError(submissionId, resumeMode, 'origin_done');\n }\n\n originPostRecordId = originRecord.id;\n this.logger\n .withMetadata({ originPostRecordId })\n .debug('Chaining to origin PostRecord');\n }\n\n return this.postRecordRepository.insert({\n submissionId,\n state: PostRecordState.PENDING,\n resumeMode,\n originPostRecordId,\n });\n }\n\n /**\n * Find the most recent NEW PostRecord for a submission.\n * Used to determine the origin for CONTINUE/RETRY records.\n *\n * @param {EntityId} submissionId - The submission ID\n * @returns {Promise} The most recent NEW record, or null if none exists\n */\n private async findMostRecentOrigin(\n submissionId: EntityId,\n ): Promise {\n const records = await this.postRecordRepository.find({\n where: (record, { eq, and }) =>\n and(\n eq(record.submissionId, submissionId),\n eq(record.resumeMode, PostRecordResumeMode.NEW),\n ),\n orderBy: (record, { desc }) => desc(record.createdAt),\n limit: 1,\n });\n\n return records.length > 0 ? records[0] : null;\n }\n\n /**\n * Find any PENDING or RUNNING PostRecord for a submission.\n * Used to prevent creating a new record when one is already in progress.\n *\n * @param {EntityId} submissionId - The submission ID\n * @returns {Promise} An in-progress record, or null if none exists\n */\n private async findInProgressRecord(\n submissionId: EntityId,\n ): Promise {\n const records = await this.postRecordRepository.find({\n where: (record, { eq, and, or }) =>\n and(\n eq(record.submissionId, submissionId),\n or(\n eq(record.state, PostRecordState.PENDING),\n eq(record.state, PostRecordState.RUNNING),\n ),\n ),\n limit: 1,\n });\n\n return records.length > 0 ? records[0] : null;\n }\n\n // ========================================================================\n // Resume Context Building\n // ========================================================================\n\n /**\n * Build resume context from prior post records for the same submission.\n *\n * This method handles the chain of posting attempts correctly:\n * - DONE records represent complete \"posting sessions\" - they act as stop points\n * - FAILED records represent incomplete attempts that should be aggregated\n * - RUNNING records (crash recovery) should aggregate their own events\n *\n * Logic:\n * 1. Always include events from the current record being started (handles crash recovery)\n * 2. If the most recent terminal is DONE → return empty context (nothing to continue, start fresh)\n * 3. If the most recent terminal is FAILED → aggregate from FAILED records until we hit a DONE\n *\n * @param {EntityId} submissionId - The submission ID\n * @param {EntityId} currentRecordId - The ID of the record being started\n * @param {PostRecordResumeMode} resumeMode - The resume mode\n * @returns {Promise} The resume context\n */\n async buildResumeContext(\n submissionId: EntityId,\n currentRecordId: EntityId,\n resumeMode: PostRecordResumeMode,\n ): Promise {\n const context = this.createEmptyContext(resumeMode);\n\n // First, always fetch the specific record we're starting.\n // This handles crash recovery where the record is RUNNING (not terminal).\n const currentRecord = await this.postRecordRepository.findById(\n currentRecordId,\n undefined,\n { events: true },\n );\n\n // For NEW mode on a fresh record, return empty context\n // But for crash recovery (RUNNING state), we still need to aggregate our own events\n if (resumeMode === PostRecordResumeMode.NEW) {\n if (currentRecord?.state === PostRecordState.RUNNING) {\n // Crash recovery: aggregate events from this record regardless of resumeMode\n this.logger.debug(\n 'NEW mode but RUNNING state (crash recovery) - aggregating own events',\n );\n this.aggregateFromRecords([currentRecord], context, true);\n } else {\n this.logger.debug('NEW mode - returning empty resume context');\n }\n return context;\n }\n\n // Get terminal records to aggregate based on the chain logic\n const terminalRecords = await this.getRecordsToAggregate(submissionId);\n\n // Combine: current record (if RUNNING) + terminal records (excluding duplicates)\n const recordsToAggregate = this.combineRecordsForAggregation(\n currentRecord,\n terminalRecords,\n );\n\n if (recordsToAggregate.length === 0) {\n this.logger.debug(\n 'No records to aggregate (fresh start or most recent was DONE)',\n );\n return context;\n }\n\n // Aggregate events based on resume mode\n const includePostedFiles = resumeMode === PostRecordResumeMode.CONTINUE;\n this.aggregateFromRecords(recordsToAggregate, context, includePostedFiles);\n\n this.logger\n .withMetadata({\n completedAccountCount: context.completedAccountIds.size,\n accountsWithPostedFiles: context.postedFilesByAccount.size,\n aggregatedRecordCount: recordsToAggregate.length,\n resumeMode,\n })\n .debug('Built resume context');\n\n return context;\n }\n\n /**\n * Combine the current record (if RUNNING) with terminal records, avoiding duplicates.\n * This ensures crash recovery includes the RUNNING record's events.\n */\n private combineRecordsForAggregation(\n currentRecord: PostRecord | null | undefined,\n terminalRecords: PostRecord[],\n ): PostRecord[] {\n const result: PostRecord[] = [];\n const seenIds = new Set();\n\n // Add current record first if it's RUNNING (crash recovery case)\n if (currentRecord?.state === PostRecordState.RUNNING) {\n result.push(currentRecord);\n seenIds.add(currentRecord.id);\n }\n\n // Add terminal records that weren't already added\n for (const record of terminalRecords) {\n if (!seenIds.has(record.id)) {\n result.push(record);\n seenIds.add(record.id);\n }\n }\n\n return result;\n }\n\n /**\n * Aggregate events from a list of records into the context.\n */\n private aggregateFromRecords(\n records: PostRecord[],\n context: ResumeContext,\n includePostedFiles: boolean,\n ): void {\n this.aggregateSourceUrls(records, context);\n this.aggregateCompletedAccounts(records, context);\n\n if (includePostedFiles) {\n this.aggregatePostedFiles(records, context);\n }\n }\n\n /**\n * Create an empty resume context with default values.\n */\n private createEmptyContext(resumeMode: PostRecordResumeMode): ResumeContext {\n return {\n resumeMode,\n completedAccountIds: new Set(),\n postedFilesByAccount: new Map>(),\n sourceUrlsByAccount: new Map(),\n };\n }\n\n /**\n * Get the list of PostRecords whose events should be aggregated.\n *\n * Uses the originPostRecordId field to find all records in the same chain.\n * A chain consists of:\n * - The origin NEW record (originPostRecordId = null, resumeMode = NEW)\n * - All CONTINUE/RETRY records that reference that origin\n *\n * @param {EntityId} submissionId - The submission ID\n * @param {EntityId} [originId] - Optional origin ID to query directly\n * @returns {Promise} Records to aggregate (may be empty)\n */\n private async getRecordsToAggregate(\n submissionId: EntityId,\n originId?: EntityId,\n ): Promise {\n // If no origin provided, find the most recent origin for this submission\n let effectiveOriginId = originId;\n if (!effectiveOriginId) {\n const origin = await this.findMostRecentOrigin(submissionId);\n if (!origin) {\n this.logger.debug('No origin PostRecord found for submission');\n return [];\n }\n effectiveOriginId = origin.id;\n }\n\n // Query all records in this chain: the origin + all records referencing it\n const chainRecords = await this.postRecordRepository.find({\n where: (record, { eq, or }) =>\n or(\n eq(record.id, effectiveOriginId),\n eq(record.originPostRecordId, effectiveOriginId),\n ),\n orderBy: (record, { asc }) => asc(record.createdAt),\n with: {\n events: true,\n },\n });\n\n this.logger\n .withMetadata({\n submissionId,\n originId: effectiveOriginId,\n chainRecordCount: chainRecords.length,\n chainRecordIds: chainRecords.map((r) => r.id),\n })\n .debug('Retrieved chain records for aggregation');\n\n return chainRecords;\n }\n\n /**\n * Aggregate source URLs from events into the context.\n * Source URLs are used for cross-website propagation.\n */\n private aggregateSourceUrls(\n records: PostRecord[],\n context: ResumeContext,\n ): void {\n for (const record of records) {\n if (!record.events) continue;\n\n for (const event of record.events) {\n if (\n this.isSourceUrlEvent(event) &&\n event.accountId &&\n event.sourceUrl\n ) {\n this.addSourceUrl(context, event.accountId, event.sourceUrl);\n }\n }\n }\n }\n\n /**\n * Aggregate completed account IDs from events into the context.\n */\n private aggregateCompletedAccounts(\n records: PostRecord[],\n context: ResumeContext,\n ): void {\n for (const record of records) {\n if (!record.events) continue;\n\n for (const event of record.events) {\n if (\n event.eventType === PostEventType.POST_ATTEMPT_COMPLETED &&\n event.accountId\n ) {\n context.completedAccountIds.add(event.accountId);\n }\n }\n }\n }\n\n /**\n * Aggregate posted file IDs (per account) from events into the context.\n */\n private aggregatePostedFiles(\n records: PostRecord[],\n context: ResumeContext,\n ): void {\n for (const record of records) {\n if (!record.events) continue;\n\n for (const event of record.events) {\n if (\n event.eventType === PostEventType.FILE_POSTED &&\n event.accountId &&\n event.fileId\n ) {\n this.addPostedFile(context, event.accountId, event.fileId);\n }\n }\n }\n }\n\n /**\n * Check if an event contains a source URL (FILE_POSTED or MESSAGE_POSTED).\n */\n private isSourceUrlEvent(event: PostEvent): boolean {\n return (\n event.eventType === PostEventType.FILE_POSTED ||\n event.eventType === PostEventType.MESSAGE_POSTED\n );\n }\n\n /**\n * Add a source URL to the context for an account.\n */\n private addSourceUrl(\n context: ResumeContext,\n accountId: AccountId,\n sourceUrl: string,\n ): void {\n const existing = context.sourceUrlsByAccount.get(accountId);\n if (existing) {\n existing.push(sourceUrl);\n } else {\n context.sourceUrlsByAccount.set(accountId, [sourceUrl]);\n }\n }\n\n /**\n * Add a posted file to the context for an account.\n */\n private addPostedFile(\n context: ResumeContext,\n accountId: AccountId,\n fileId: EntityId,\n ): void {\n const existing = context.postedFilesByAccount.get(accountId);\n if (existing) {\n existing.add(fileId);\n } else {\n context.postedFilesByAccount.set(accountId, new Set([fileId]));\n }\n }\n\n // ========================================================================\n // Resume Context Helpers\n // ========================================================================\n\n /**\n * Check if an account should be skipped based on resume context.\n * @param {ResumeContext} context - The resume context\n * @param {AccountId} accountId - The account to check\n * @returns {boolean} True if the account should be skipped\n */\n shouldSkipAccount(context: ResumeContext, accountId: AccountId): boolean {\n return context.completedAccountIds.has(accountId);\n }\n\n /**\n * Check if a file should be skipped for a specific account based on resume context.\n * @param {ResumeContext} context - The resume context\n * @param {AccountId} accountId - The account ID\n * @param {EntityId} fileId - The file ID to check\n * @returns {boolean} True if the file should be skipped\n */\n shouldSkipFile(\n context: ResumeContext,\n accountId: AccountId,\n fileId: EntityId,\n ): boolean {\n // If account is completed, all files should be skipped (handled by shouldSkipAccount)\n // This method is for checking individual files within a non-completed account\n\n if (context.resumeMode === PostRecordResumeMode.NEW) {\n return false;\n }\n\n if (context.resumeMode === PostRecordResumeMode.CONTINUE_RETRY) {\n // CONTINUE_RETRY retries all files for non-completed accounts\n return false;\n }\n\n // CONTINUE mode: skip files that were already posted\n const postedFiles = context.postedFilesByAccount.get(accountId);\n return postedFiles?.has(fileId) ?? false;\n }\n\n /**\n * Get source URLs from prior attempts for cross-website propagation.\n * @param {ResumeContext} context - The resume context\n * @param {AccountId} accountId - The account to get URLs for\n * @returns {string[]} Source URLs from the prior attempt\n */\n getSourceUrlsForAccount(\n context: ResumeContext,\n accountId: AccountId,\n ): string[] {\n return context.sourceUrlsByAccount.get(accountId) ?? [];\n }\n\n /**\n * Get all source URLs from all accounts in the resume context.\n * @param {ResumeContext} context - The resume context\n * @returns {string[]} All source URLs\n */\n getAllSourceUrls(context: ResumeContext): string[] {\n const allUrls: string[] = [];\n for (const urls of context.sourceUrlsByAccount.values()) {\n allUrls.push(...urls);\n }\n return allUrls;\n }\n\n /**\n * Get all source URLs from the resume context, excluding a specific account.\n * Used for cross-website source URL propagation to avoid self-referential URLs.\n * @param {ResumeContext} context - The resume context\n * @param {AccountId} excludeAccountId - The account ID to exclude\n * @returns {string[]} Source URLs from all accounts except the excluded one\n */\n getSourceUrlsExcludingAccount(\n context: ResumeContext,\n excludeAccountId: AccountId,\n ): string[] {\n const allUrls: string[] = [];\n for (const [accountId, urls] of context.sourceUrlsByAccount.entries()) {\n if (accountId !== excludeAccountId) {\n allUrls.push(...urls);\n }\n }\n return allUrls;\n }\n}\n" }, { "path": "apps/client-server/src/app/post-parsers/models/description-node/converters/base-converter.ts", "content": "/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { UsernameShortcut } from '@postybirb/types';\nimport { ConversionContext } from '../description-node.base';\nimport { InlineTypes, isTextNode, TipTapNode } from '../description-node.types';\n\n/**\n * Base converter for TipTap JSON → output format.\n *\n * Converters process TipTap nodes directly (no wrapper classes).\n * Block-level nodes are dispatched to `convertBlockNode`, inline shortcut\n * atoms to `convertInlineNode`, and text nodes to `convertTextNode`.\n */\nexport abstract class BaseConverter {\n /** Current depth for nested block rendering */\n protected currentDepth = 0;\n\n /** Used to prevent loop when default shortcut is insert into default section */\n private processingDefaultDescription = false;\n\n abstract convertBlockNode(\n node: TipTapNode,\n context: ConversionContext,\n ): string;\n\n abstract convertInlineNode(\n node: TipTapNode,\n context: ConversionContext,\n ): string;\n\n abstract convertTextNode(\n node: TipTapNode,\n context: ConversionContext,\n ): string;\n\n /**\n * Converts an array of top-level TipTap nodes (block nodes).\n */\n convertBlocks(nodes: TipTapNode[], context: ConversionContext): string {\n const results = nodes.map((node) => this.convertBlockNode(node, context));\n return results.join(this.getBlockSeparator());\n }\n\n /**\n * Converts raw TipTap block data. Handles default description recursion guard.\n */\n convertRawBlocks(blocks: TipTapNode[], context: ConversionContext): string {\n const isDefaultDescription = blocks === context.defaultDescription;\n if (isDefaultDescription) {\n if (this.processingDefaultDescription) {\n return '';\n }\n this.processingDefaultDescription = true;\n }\n\n try {\n return this.convertBlocks(blocks, context);\n } finally {\n if (isDefaultDescription) {\n this.processingDefaultDescription = false;\n }\n }\n }\n\n /**\n * Returns the separator to use between blocks.\n */\n protected abstract getBlockSeparator(): string;\n\n /**\n * Converts the `content` array of a block node.\n * Dispatches each child to the appropriate handler based on type.\n */\n protected convertContent(\n content: TipTapNode[] | undefined,\n context: ConversionContext,\n ): string {\n if (!content || content.length === 0) return '';\n\n return content\n .map((child) => {\n if (isTextNode(child)) {\n return this.convertTextNode(child, context);\n }\n if (InlineTypes.includes(child.type)) {\n return this.convertInlineNode(child, context);\n }\n // Nested block nodes (e.g., listItem content)\n return this.convertBlockNode(child, context);\n })\n .join('');\n }\n\n /**\n * Converts children blocks with increased depth.\n */\n protected convertChildren(\n children: TipTapNode[],\n context: ConversionContext,\n ): string {\n if (!children || children.length === 0) return '';\n\n this.currentDepth += 1;\n try {\n const results = children.map((child) =>\n this.convertBlockNode(child, context),\n );\n return results.join(this.getBlockSeparator());\n } finally {\n this.currentDepth -= 1;\n }\n }\n\n /**\n * Helper to check if a shortcut should be rendered for this website.\n */\n protected shouldRenderShortcut(\n node: TipTapNode,\n context: ConversionContext,\n ): boolean {\n const attrs = node.attrs ?? {};\n const onlyTo = (attrs.only?.split(',') ?? [])\n .map((s: string) => s.trim().toLowerCase())\n .filter((s: string) => s.length > 0);\n\n if (onlyTo.length === 0) return true;\n\n return onlyTo.includes(context.website.toLowerCase());\n }\n\n /**\n * Helper to resolve username shortcut link.\n */\n protected getUsernameShortcutLink(\n node: TipTapNode,\n context: ConversionContext,\n ):\n | {\n url: string;\n username: string;\n }\n | undefined {\n const attrs = node.attrs ?? {};\n const username = (attrs.username as string)?.trim() ?? '';\n\n let convertedUsername = username;\n let effectiveShortcutId = attrs.shortcut;\n\n const converted = context.usernameConversions?.get(username);\n if (converted && converted !== username) {\n convertedUsername = converted;\n effectiveShortcutId = context.website;\n }\n\n const shortcut: UsernameShortcut | undefined =\n context.shortcuts[effectiveShortcutId];\n const url =\n shortcut?.convert?.call(node, context.website, effectiveShortcutId) ??\n shortcut?.url;\n\n return convertedUsername && url\n ? {\n url: url.replace('$1', convertedUsername),\n username: convertedUsername,\n }\n : undefined;\n }\n}\n" }, { "path": "apps/client-server/src/app/post-parsers/models/description-node/converters/bbcode-converter.ts", "content": "/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { ConversionContext } from '../description-node.base';\nimport { TipTapNode } from '../description-node.types';\nimport { BaseConverter } from './base-converter';\n\nexport class BBCodeConverter extends BaseConverter {\n protected getBlockSeparator(): string {\n return '\\n';\n }\n\n convertBlockNode(\n node: TipTapNode,\n context: ConversionContext,\n ): string {\n const attrs = node.attrs ?? {};\n\n if (node.type === 'defaultShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n return this.convertRawBlocks(context.defaultDescription, context);\n }\n\n // For FA: More than 5 dashes in a line are replaced with a horizontal divider.\n if (node.type === 'horizontalRule') return '--------';\n\n if (node.type === 'image') return '';\n if (node.type === 'hardBreak') return '\\n';\n\n // List containers\n if (node.type === 'bulletList') {\n return (node.content ?? [])\n .map((child) => this.convertBlockNode(child, context))\n .join('\\n');\n }\n\n if (node.type === 'orderedList') {\n return (node.content ?? [])\n .map((child) => this.convertBlockNode(child, context))\n .join('\\n');\n }\n\n if (node.type === 'listItem') {\n const inner = (node.content ?? [])\n .map((child) => {\n if (child.type === 'paragraph') {\n return this.convertContent(child.content, context);\n }\n return this.convertBlockNode(child, context);\n })\n .join('');\n return `• ${inner}`;\n }\n\n if (node.type === 'blockquote') {\n const inner = (node.content ?? [])\n .map((child) => this.convertBlockNode(child, context))\n .join('\\n');\n return `[quote]${inner}[/quote]`;\n }\n\n if (node.type === 'paragraph') {\n let text = this.convertContent(node.content, context);\n\n // Apply text alignment if not default\n if (attrs.textAlign && attrs.textAlign !== 'left') {\n text = `[${attrs.textAlign}]${text}[/${attrs.textAlign}]`;\n }\n\n // Apply indentation\n if (attrs.indent && attrs.indent > 0) {\n const spaces = '\\u00A0\\u00A0\\u00A0\\u00A0'.repeat(attrs.indent);\n text = `${spaces}${text}`;\n }\n\n return text;\n }\n\n if (node.type === 'heading') {\n const level = attrs.level ?? 1;\n let text = `[h${level}]${this.convertContent(node.content, context)}[/h${level}]`;\n\n if (attrs.textAlign && attrs.textAlign !== 'left') {\n text = `[${attrs.textAlign}]${text}[/${attrs.textAlign}]`;\n }\n\n if (attrs.indent && attrs.indent > 0) {\n const spaces = '\\u00A0\\u00A0\\u00A0\\u00A0'.repeat(attrs.indent);\n text = `${spaces}${text}`;\n }\n\n return text;\n }\n\n // Fallback\n return this.convertContent(node.content, context);\n }\n\n convertInlineNode(\n node: TipTapNode,\n context: ConversionContext,\n ): string {\n const attrs = node.attrs ?? {};\n\n if (node.type === 'username') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n const sc = this.getUsernameShortcutLink(node, context);\n if (sc?.url.startsWith('http')) {\n return `[url=${sc.url}]${sc.username}[/url]`;\n }\n return sc ? `${sc.url ?? sc.username}` : '';\n }\n\n if (node.type === 'customShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n const shortcutBlocks = context.customShortcuts.get(attrs.id);\n if (shortcutBlocks) {\n return this.convertRawBlocks(shortcutBlocks, context);\n }\n return '';\n }\n\n if (node.type === 'titleShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n return context.title ?? '';\n }\n\n if (node.type === 'tagsShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n return context.tags?.map((e) => `#${e}`).join(' ') ?? '';\n }\n\n if (node.type === 'contentWarningShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n return context.contentWarningText ?? '';\n }\n\n if (node.type === 'hardBreak') return '\\n';\n\n return this.convertContent(node.content, context);\n }\n\n convertTextNode(\n node: TipTapNode,\n context: ConversionContext,\n ): string {\n const textNode = node as any;\n if (!textNode.text) return '';\n\n if (textNode.text === '\\n' || textNode.text === '\\r\\n') {\n return '\\n';\n }\n\n const marks = textNode.marks ?? [];\n\n // Check for link mark\n const linkMark = marks.find((m: any) => m.type === 'link');\n if (linkMark) {\n const href = linkMark.attrs?.href ?? '';\n const innerText = this.renderTextWithMarks(textNode.text, marks.filter((m: any) => m.type !== 'link'));\n return `[url=${href}]${innerText}[/url]`;\n }\n\n return this.renderTextWithMarks(textNode.text, marks);\n }\n\n /**\n * Renders text with BBCode formatting marks applied.\n */\n private renderTextWithMarks(text: string, marks: any[]): string {\n const segments: string[] = [];\n\n for (const mark of marks) {\n switch (mark.type) {\n case 'bold':\n segments.push('b');\n break;\n case 'italic':\n segments.push('i');\n break;\n case 'underline':\n segments.push('u');\n break;\n case 'strike':\n segments.push('s');\n break;\n default:\n break;\n }\n }\n\n if (!segments.length) {\n return text;\n }\n\n let segmentedText = `${segments.map((e) => `[${e}]`).join('')}${text}${segments\n .reverse()\n .map((e) => `[/${e}]`)\n .join('')}`;\n\n // Check for textStyle mark with color\n const textStyleMark = marks.find((m: any) => m.type === 'textStyle');\n if (textStyleMark?.attrs?.color) {\n segmentedText = `[color=${textStyleMark.attrs.color}]${segmentedText}[/color]`;\n }\n\n return segmentedText;\n }\n}\n\n" }, { "path": "apps/client-server/src/app/post-parsers/models/description-node/converters/custom-converter.ts", "content": "/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { ConversionContext } from '../description-node.base';\nimport { TipTapNode } from '../description-node.types';\nimport { BaseConverter } from './base-converter';\n\nexport type CustomNodeHandler = (\n node: TipTapNode,\n context: ConversionContext,\n) => string;\n\n/**\n * Converter that uses custom handlers for each node type.\n * This allows websites like Tumblr to inject their own conversion logic.\n */\nexport class CustomConverter extends BaseConverter {\n constructor(\n private readonly blockHandler: CustomNodeHandler,\n private readonly inlineHandler?: CustomNodeHandler,\n private readonly textHandler?: CustomNodeHandler,\n ) {\n super();\n }\n\n protected getBlockSeparator(): string {\n return '\\n';\n }\n\n convertBlockNode(\n node: TipTapNode,\n context: ConversionContext,\n ): string {\n return this.blockHandler(node, context);\n }\n\n convertInlineNode(\n node: TipTapNode,\n context: ConversionContext,\n ): string {\n if (this.inlineHandler) {\n return this.inlineHandler(node, context);\n }\n return this.convertContent(node.content, context);\n }\n\n convertTextNode(\n node: TipTapNode,\n context: ConversionContext,\n ): string {\n if (this.textHandler) {\n return this.textHandler(node, context);\n }\n return (node as any).text ?? '';\n }\n}\n\n" }, { "path": "apps/client-server/src/app/post-parsers/models/description-node/converters/html-converter.ts", "content": "/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { encode } from 'html-entities';\nimport { ConversionContext } from '../description-node.base';\nimport { TipTapNode } from '../description-node.types';\nimport { BaseConverter } from './base-converter';\n\nexport class HtmlConverter extends BaseConverter {\n protected getBlockSeparator(): string {\n return '';\n }\n\n convertBlockNode(\n node: TipTapNode,\n context: ConversionContext,\n ): string {\n const attrs = node.attrs ?? {};\n\n // Handle special block types\n if (node.type === 'defaultShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n return this.convertRawBlocks(context.defaultDescription, context);\n }\n\n if (node.type === 'horizontalRule') return '
';\n if (node.type === 'image') return this.convertImage(node);\n if (node.type === 'hardBreak') return '
';\n\n // List containers: render as
    /
      wrapping children\n if (node.type === 'bulletList') {\n const items = (node.content ?? [])\n .map((child) => this.convertBlockNode(child, context))\n .join('');\n return `
        ${items}
      `;\n }\n\n if (node.type === 'orderedList') {\n const items = (node.content ?? [])\n .map((child) => this.convertBlockNode(child, context))\n .join('');\n return `
        ${items}
      `;\n }\n\n if (node.type === 'listItem') {\n // listItem content is typically [paragraph, ...]. Render inline content of inner paragraphs.\n const inner = (node.content ?? [])\n .map((child) => {\n if (child.type === 'paragraph') {\n return this.convertContent(child.content, context);\n }\n return this.convertBlockNode(child, context);\n })\n .join('');\n return `
    1. ${inner}
      2. `;\n }\n\n if (node.type === 'blockquote') {\n const inner = (node.content ?? [])\n .map((child) => this.convertBlockNode(child, context))\n .join('');\n return `
      ${inner}
      `;\n }\n\n // Regular blocks: paragraph, heading, etc.\n const tag = this.getBlockTag(node);\n const styles = this.getBlockStyles(node);\n const content = this.convertContent(node.content, context);\n\n return `<${tag}${styles ? ` style=\"${styles}\"` : ''}>${content}`;\n }\n\n convertInlineNode(\n node: TipTapNode,\n context: ConversionContext,\n ): string {\n const attrs = node.attrs ?? {};\n\n if (node.type === 'username') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n const sc = this.getUsernameShortcutLink(node, context);\n if (!sc) return '';\n if (!sc.url.startsWith('http')) return `${sc.url}`;\n return `${sc.username}`;\n }\n\n if (node.type === 'customShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n const shortcutBlocks = context.customShortcuts.get(attrs.id);\n if (shortcutBlocks) {\n return this.convertRawBlocks(shortcutBlocks, context);\n }\n return '';\n }\n\n if (node.type === 'titleShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n return context.title\n ? `${encode(context.title, { level: 'html5' })}`\n : '';\n }\n\n if (node.type === 'tagsShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n return context.tags?.length\n ? `${context.tags.map((t) => encode(`#${t}`, { level: 'html5' })).join(' ')}`\n : '';\n }\n\n if (node.type === 'contentWarningShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n return context.contentWarningText\n ? `${encode(context.contentWarningText, { level: 'html5' })}`\n : '';\n }\n\n if (node.type === 'hardBreak') return '
      ';\n\n // Fallback: render content\n const content = this.convertContent(node.content, context);\n return content ? `${content}` : '';\n }\n\n convertTextNode(\n node: TipTapNode,\n context: ConversionContext,\n ): string {\n const textNode = node as any;\n if (!textNode.text) return '';\n\n // Handle line breaks from merged blocks\n if (textNode.text === '\\n' || textNode.text === '\\r\\n') {\n return '
      ';\n }\n\n const marks = textNode.marks ?? [];\n const segments: string[] = [];\n const styles: string[] = [];\n\n // Check for link mark — wrap entire text in \n const linkMark = marks.find((m: any) => m.type === 'link');\n if (linkMark) {\n const href = linkMark.attrs?.href ?? '';\n const innerHtml = this.renderTextWithMarks(textNode.text, marks.filter((m: any) => m.type !== 'link'));\n return `${innerHtml}`;\n }\n\n return this.renderTextWithMarks(textNode.text, marks);\n }\n\n /**\n * Renders text with formatting marks (bold, italic, etc.) applied.\n */\n private renderTextWithMarks(text: string, marks: any[]): string {\n const segments: string[] = [];\n const styles: string[] = [];\n\n for (const mark of marks) {\n switch (mark.type) {\n case 'bold':\n segments.push('b');\n break;\n case 'italic':\n segments.push('i');\n break;\n case 'underline':\n segments.push('u');\n break;\n case 'strike':\n segments.push('s');\n break;\n default:\n break;\n }\n }\n\n // Check for textStyle mark with color\n const textStyleMark = marks.find((m: any) => m.type === 'textStyle');\n if (textStyleMark?.attrs?.color) {\n styles.push(`color: ${textStyleMark.attrs.color}`);\n }\n\n const encodedText = encode(text, { level: 'html5' }).replace(/\\n/g, '
      ');\n\n if (!segments.length && !styles.length) {\n return encodedText;\n }\n\n const stylesString = styles.join(';');\n return `${segments.map((s) => `<${s}>`).join('')}${encodedText}${segments\n .reverse()\n .map((s) => ``)\n .join('')}`;\n }\n\n private getBlockTag(node: TipTapNode): string {\n const attrs = node.attrs ?? {};\n if (node.type === 'paragraph') return 'div';\n if (node.type === 'heading') return `h${attrs.level ?? 1}`;\n return 'div';\n }\n\n private getBlockStyles(node: TipTapNode): string {\n const attrs = node.attrs ?? {};\n const styles: string[] = [];\n\n if (\n attrs.textAlign &&\n attrs.textAlign !== 'left'\n ) {\n styles.push(`text-align: ${attrs.textAlign}`);\n }\n\n if (attrs.indent && attrs.indent > 0) {\n styles.push(`margin-left: ${attrs.indent * 2}em`);\n }\n\n return styles.join(';');\n }\n\n private convertImage(node: TipTapNode): string {\n const attrs = node.attrs ?? {};\n const src = attrs.src || '';\n const alt = attrs.alt || '';\n const width = attrs.width || '';\n const height = attrs.height || '';\n\n let imgTag = `\"${alt}\"`;\n${imgTag}`;\n }\n}\n\n" }, { "path": "apps/client-server/src/app/post-parsers/models/description-node/converters/npf-converter.spec.ts", "content": "import { NPFTextBlock, TipTapNode } from '@postybirb/types';\nimport { ConversionContext } from '../description-node.base';\nimport { NpfConverter } from './npf-converter';\n\ndescribe('NpfConverter', () => {\n let converter: NpfConverter;\n let context: ConversionContext;\n\n beforeEach(() => {\n converter = new NpfConverter();\n context = {\n website: 'tumblr',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n });\n\n describe('convertBlockNode', () => {\n it('should convert a simple paragraph to NPF text block', () => {\n const node: TipTapNode = {\n type: 'paragraph',\n content: [\n {\n type: 'text',\n text: 'Hello, World!',\n },\n ],\n };\n\n const resultJson = converter.convertBlockNode(node, context);\n const result = JSON.parse(resultJson);\n\n expect(result).toEqual({\n type: 'text',\n text: 'Hello, World!',\n });\n });\n\n it('should convert a paragraph with bold text to NPF with formatting', () => {\n const node: TipTapNode = {\n type: 'paragraph',\n content: [\n {\n type: 'text',\n text: 'This is ',\n },\n {\n type: 'text',\n text: 'bold',\n marks: [{ type: 'bold' }],\n },\n {\n type: 'text',\n text: ' text',\n },\n ],\n };\n\n const resultJson = converter.convertBlockNode(node, context);\n const result = JSON.parse(resultJson) as NPFTextBlock;\n\n expect(result.type).toBe('text');\n expect(result.text).toBe('This is bold text');\n expect(result.formatting).toEqual([\n {\n start: 8,\n end: 12,\n type: 'bold',\n },\n ]);\n });\n\n it('should convert a paragraph with link mark to NPF with link formatting', () => {\n const node: TipTapNode = {\n type: 'paragraph',\n content: [\n {\n type: 'text',\n text: 'Visit ',\n },\n {\n type: 'text',\n text: 'PostyBirb',\n marks: [\n {\n type: 'link',\n attrs: { href: 'https://postybirb.com' },\n },\n ],\n },\n ],\n };\n\n const resultJson = converter.convertBlockNode(node, context);\n const result = JSON.parse(resultJson) as NPFTextBlock;\n\n expect(result.type).toBe('text');\n expect(result.text).toBe('Visit PostyBirb');\n expect(result.formatting).toEqual([\n {\n start: 6,\n end: 15,\n type: 'link',\n url: 'https://postybirb.com',\n },\n ]);\n });\n\n it('should convert heading to NPF text block with subtype', () => {\n const node: TipTapNode = {\n type: 'heading',\n attrs: { level: 1 },\n content: [\n {\n type: 'text',\n text: 'My Heading',\n },\n ],\n };\n\n const resultJson = converter.convertBlockNode(node, context);\n const result = JSON.parse(resultJson);\n\n expect(result).toEqual({\n type: 'text',\n text: 'My Heading',\n subtype: 'heading1',\n });\n });\n\n it('should convert image to NPF image block', () => {\n const node: TipTapNode = {\n type: 'image',\n attrs: {\n src: 'https://example.com/image.jpg',\n alt: 'My Image',\n width: 800,\n height: 600,\n },\n };\n\n const resultJson = converter.convertBlockNode(node, context);\n const result = JSON.parse(resultJson);\n\n expect(result).toEqual({\n type: 'image',\n media: [\n {\n url: 'https://example.com/image.jpg',\n type: 'image/jpeg',\n width: 800,\n height: 600,\n },\n ],\n alt_text: 'My Image',\n });\n });\n\n it('should handle multiple formatting types on same text', () => {\n const node: TipTapNode = {\n type: 'paragraph',\n content: [\n {\n type: 'text',\n text: 'Bold and Italic',\n marks: [{ type: 'bold' }, { type: 'italic' }],\n },\n ],\n };\n\n const resultJson = converter.convertBlockNode(node, context);\n const result = JSON.parse(resultJson) as NPFTextBlock;\n\n expect(result.type).toBe('text');\n expect(result.text).toBe('Bold and Italic');\n expect(result.formatting).toEqual([\n {\n start: 0,\n end: 15,\n type: 'bold',\n },\n {\n start: 0,\n end: 15,\n type: 'italic',\n },\n ]);\n });\n\n it('should handle custom shortcuts', () => {\n const shortcutContent: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n {\n type: 'text',\n text: 'Commission Info',\n marks: [{ type: 'bold' }],\n },\n ],\n },\n ];\n\n context.customShortcuts.set('cs-1', shortcutContent);\n\n const node: TipTapNode = {\n type: 'paragraph',\n content: [\n {\n type: 'text',\n text: 'Check out my ',\n },\n {\n type: 'customShortcut',\n attrs: { id: 'cs-1' },\n },\n ],\n };\n\n const resultJson = converter.convertBlockNode(node, context);\n const result = JSON.parse(resultJson) as NPFTextBlock;\n\n expect(result.type).toBe('text');\n expect(result.text).toBe('Check out my Commission Info');\n expect(result.formatting).toEqual([\n {\n start: 13,\n end: 28,\n type: 'bold',\n },\n ]);\n });\n\n it('should handle double spaces', () => {\n const shortcutContent: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n {\n type: 'text',\n text: 'Commission Info',\n marks: [{ type: 'bold' }],\n },\n ],\n },\n ];\n\n context.customShortcuts.set('cs-1', shortcutContent);\n\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'aaa' }],\n },\n {\n type: 'defaultShortcut',\n },\n {\n type: 'paragraph',\n content: [],\n },\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'aaa' }],\n },\n ];\n\n const resultJson = converter.convertBlocks(nodes, context);\n const result = JSON.parse(resultJson) as NPFTextBlock;\n\n expect(result).toMatchInlineSnapshot(`\n [\n {\n \"text\": \"aaa\",\n \"type\": \"text\",\n },\n {\n \"text\": \"\",\n \"type\": \"text\",\n },\n {\n \"text\": \"aaa\",\n \"type\": \"text\",\n },\n ]\n `);\n });\n });\n});\n" }, { "path": "apps/client-server/src/app/post-parsers/models/description-node/converters/npf-converter.ts", "content": "/* eslint-disable @typescript-eslint/no-explicit-any */\nimport {\n NPFContentBlock,\n NPFImageBlock,\n NPFInlineFormatting,\n NPFMediaObject,\n NPFTextBlock,\n} from '@postybirb/types';\nimport { ConversionContext } from '../description-node.base';\nimport { isTextNode, TipTapMark, TipTapNode } from '../description-node.types';\nimport { BaseConverter } from './base-converter';\n\n/**\n * Converter that outputs Tumblr NPF (Nuevo Post Format) blocks.\n */\nexport class NpfConverter extends BaseConverter {\n private currentFormatting: NPFInlineFormatting[] = [];\n\n private currentPosition = 0;\n\n private blocks: NPFContentBlock[] = [];\n\n protected getBlockSeparator(): string {\n return '';\n }\n\n /**\n * Override convertBlocks to accumulate NPF blocks and return JSON.\n */\n convertBlocks(nodes: TipTapNode[], context: ConversionContext): string {\n this.blocks = [];\n for (const node of nodes) {\n this.convertBlockNodeRecursive(node, context, 0);\n }\n return JSON.stringify(this.blocks);\n }\n\n /**\n * Recursively converts a block node and its children to NPF blocks.\n */\n private convertBlockNodeRecursive(\n node: TipTapNode,\n context: ConversionContext,\n indentLevel: number,\n ): void {\n if (node.type === 'defaultShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return;\n if (context.defaultDescription && context.defaultDescription.length > 0) {\n for (const defaultBlock of context.defaultDescription) {\n this.convertBlockNodeRecursive(defaultBlock, context, indentLevel);\n }\n }\n return;\n }\n\n // Handle list containers — recurse into items\n if (node.type === 'bulletList') {\n for (const item of node.content ?? []) {\n this.convertListItemToNpf(\n item,\n context,\n 'unordered-list-item',\n indentLevel,\n );\n }\n return;\n }\n\n if (node.type === 'orderedList') {\n for (const item of node.content ?? []) {\n this.convertListItemToNpf(\n item,\n context,\n 'ordered-list-item',\n indentLevel,\n );\n }\n return;\n }\n\n if (node.type === 'blockquote') {\n for (const child of node.content ?? []) {\n const block = this.convertBlockNodeToNpf(child, context);\n if (block.type === 'text') {\n block.subtype = 'quote';\n }\n this.blocks.push(block);\n }\n return;\n }\n\n const block = this.convertBlockNodeToNpf(node, context);\n\n // Apply indent_level for nested blocks (NPF supports 0-7)\n if (indentLevel > 0 && block.type === 'text' && indentLevel <= 7) {\n block.indent_level = indentLevel;\n if (!block.subtype) {\n block.subtype = 'indented';\n }\n }\n\n this.blocks.push(block);\n }\n\n /**\n * Convert a listItem node to NPF text block with appropriate subtype.\n */\n private convertListItemToNpf(\n node: TipTapNode,\n context: ConversionContext,\n subtype: 'ordered-list-item' | 'unordered-list-item',\n indentLevel: number,\n ): void {\n // listItem typically contains [paragraph, ...nested lists]\n for (const child of node.content ?? []) {\n if (child.type === 'paragraph') {\n this.currentFormatting = [];\n this.currentPosition = 0;\n const text = this.extractText(child.content ?? [], context);\n const npfBlock: NPFTextBlock = {\n type: 'text',\n text,\n subtype,\n formatting:\n this.currentFormatting.length > 0\n ? this.currentFormatting\n : undefined,\n };\n if (indentLevel > 0 && indentLevel <= 7) {\n npfBlock.indent_level = indentLevel;\n }\n this.blocks.push(npfBlock);\n } else if (child.type === 'bulletList') {\n for (const item of child.content ?? []) {\n this.convertListItemToNpf(\n item,\n context,\n 'unordered-list-item',\n indentLevel + 1,\n );\n }\n } else if (child.type === 'orderedList') {\n for (const item of child.content ?? []) {\n this.convertListItemToNpf(\n item,\n context,\n 'ordered-list-item',\n indentLevel + 1,\n );\n }\n } else {\n this.convertBlockNodeRecursive(child, context, indentLevel);\n }\n }\n }\n\n /**\n * Stub method required by BaseConverter interface.\n */\n convertBlockNode(node: TipTapNode, context: ConversionContext): string {\n const block = this.convertBlockNodeToNpf(node, context);\n return JSON.stringify(block);\n }\n\n /**\n * Internal method that returns a single NPF block.\n */\n private convertBlockNodeToNpf(\n node: TipTapNode,\n context: ConversionContext,\n ): NPFContentBlock {\n this.currentFormatting = [];\n this.currentPosition = 0;\n\n switch (node.type) {\n case 'paragraph':\n return this.convertParagraph(node, context);\n case 'heading':\n return this.convertHeading(node, context);\n case 'image':\n return this.convertImage(node);\n case 'horizontalRule':\n return { type: 'text', text: '' };\n case 'defaultShortcut':\n return { type: 'text', text: '' };\n default:\n return this.convertParagraph(node, context);\n }\n }\n\n private convertParagraph(\n node: TipTapNode,\n context: ConversionContext,\n ): NPFTextBlock {\n const text = this.extractText(node.content ?? [], context);\n const formatting =\n this.currentFormatting.length > 0 ? this.currentFormatting : undefined;\n return { type: 'text', text, formatting };\n }\n\n private convertHeading(\n node: TipTapNode,\n context: ConversionContext,\n ): NPFTextBlock {\n const text = this.extractText(node.content ?? [], context);\n const level = parseInt(node.attrs?.level || '1', 10);\n const subtype = level === 1 ? 'heading1' : 'heading2';\n return {\n type: 'text',\n text,\n subtype,\n formatting:\n this.currentFormatting.length > 0 ? this.currentFormatting : undefined,\n };\n }\n\n private convertImage(node: TipTapNode): NPFImageBlock {\n const attrs = node.attrs ?? {};\n const url = attrs.src || '';\n const alt = attrs.alt || '';\n const width = attrs.width ? parseInt(String(attrs.width), 10) : undefined;\n const height = attrs.height\n ? parseInt(String(attrs.height), 10)\n : undefined;\n\n const media: NPFMediaObject[] = [\n {\n url,\n type: this.getMimeType(url),\n width: width || undefined,\n height: height || undefined,\n },\n ];\n\n const imageBlock: NPFImageBlock = {\n type: 'image',\n media,\n alt_text: alt || undefined,\n };\n\n return imageBlock;\n }\n\n convertInlineNode(node: TipTapNode, context: ConversionContext): string {\n const attrs = node.attrs ?? {};\n const startPos = this.currentPosition;\n let text = '';\n\n if (node.type === 'customShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n const shortcutBlocks = context.customShortcuts.get(attrs.id);\n if (shortcutBlocks) {\n for (const block of shortcutBlocks) {\n text += this.extractText(block.content ?? [], context);\n }\n }\n return text;\n }\n\n if (node.type === 'username') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n const sc = this.getUsernameShortcutLink(node, context);\n if (!sc) {\n text = attrs.username ?? '';\n } else if (!sc.url.startsWith('http')) {\n text = sc.url;\n } else {\n text = sc.username;\n if (text.length > 0) {\n this.currentFormatting.push({\n start: startPos,\n end: startPos + text.length,\n type: 'link',\n url: sc.url,\n });\n }\n }\n this.currentPosition += text.length;\n return text;\n }\n\n if (node.type === 'titleShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n text = context.title ?? '';\n this.currentPosition += text.length;\n this.addFormattingForMarks(node.marks, startPos, this.currentPosition);\n return text;\n }\n\n if (node.type === 'tagsShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n text = context.tags?.map((e) => `#${e}`).join(' ') ?? '';\n this.currentPosition += text.length;\n this.addFormattingForMarks(node.marks, startPos, this.currentPosition);\n return text;\n }\n\n if (node.type === 'contentWarningShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n text = context.contentWarningText ?? '';\n this.currentPosition += text.length;\n this.addFormattingForMarks(node.marks, startPos, this.currentPosition);\n return text;\n }\n\n if (node.type === 'hardBreak') {\n text = '\\n';\n this.currentPosition += text.length;\n return text;\n }\n\n // Generic inline: extract text from content\n text = this.extractText(node.content ?? [], context);\n return text;\n }\n\n convertTextNode(node: TipTapNode): string {\n const text = node.text ?? '';\n const startPos = this.currentPosition;\n this.currentPosition += text.length;\n\n this.addFormattingForMarks(node.marks, startPos, this.currentPosition);\n\n return text;\n }\n\n /**\n * Extracts plain text from content array and builds formatting.\n */\n private extractText(\n content: TipTapNode[],\n context: ConversionContext,\n ): string {\n let text = '';\n for (const node of content) {\n if (isTextNode(node)) {\n text += this.convertTextNode(node);\n } else {\n text += this.convertInlineNode(node, context);\n }\n }\n return text;\n }\n\n /**\n * Adds formatting entries for marks on a text node.\n */\n private addFormattingForMarks(\n marks: TipTapMark[],\n start: number,\n end: number,\n ): void {\n if (!marks || start === end) return;\n\n for (const mark of marks) {\n switch (mark.type) {\n case 'bold':\n this.currentFormatting.push({ start, end, type: 'bold' });\n break;\n case 'italic':\n this.currentFormatting.push({ start, end, type: 'italic' });\n break;\n case 'strike':\n this.currentFormatting.push({\n start,\n end,\n type: 'strikethrough',\n });\n break;\n case 'textStyle':\n if (mark.attrs?.color) {\n this.currentFormatting.push({\n start,\n end,\n type: 'color',\n hex: mark.attrs.color,\n });\n }\n break;\n case 'link':\n if (typeof mark.attrs?.href === 'string')\n this.currentFormatting.push({\n start,\n end,\n type: 'link',\n url: mark.attrs.href,\n });\n break;\n default:\n break;\n }\n }\n }\n\n /**\n * Gets MIME type from file URL.\n */\n private getMimeType(url: string): string | undefined {\n const ext = url.split('.').pop()?.toLowerCase();\n const mimeMap: Record = {\n jpg: 'image/jpeg',\n jpeg: 'image/jpeg',\n png: 'image/png',\n gif: 'image/gif',\n webp: 'image/webp',\n mp4: 'video/mp4',\n webm: 'video/webm',\n mp3: 'audio/mp3',\n ogg: 'audio/ogg',\n wav: 'audio/wav',\n };\n return ext ? mimeMap[ext] : undefined;\n }\n}\n" }, { "path": "apps/client-server/src/app/post-parsers/models/description-node/converters/plaintext-converter.ts", "content": "/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { ConversionContext } from '../description-node.base';\nimport { TipTapNode } from '../description-node.types';\nimport { BaseConverter } from './base-converter';\n\nexport class PlainTextConverter extends BaseConverter {\n protected getBlockSeparator(): string {\n return '\\r\\n';\n }\n\n convertBlockNode(node: TipTapNode, context: ConversionContext): string {\n if (node.type === 'defaultShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n return this.convertRawBlocks(context.defaultDescription, context);\n }\n\n if (node.type === 'horizontalRule') return '----------';\n if (node.type === 'image') return '';\n if (node.type === 'hardBreak') return '\\r\\n';\n\n // List containers\n if (node.type === 'bulletList' || node.type === 'orderedList') {\n return (node.content ?? [])\n .map((child) => this.convertBlockNode(child, context))\n .join('\\r\\n');\n }\n\n if (node.type === 'listItem') {\n const inner = (node.content ?? [])\n .map((child) => {\n if (child.type === 'paragraph') {\n return this.convertContent(child.content, context);\n }\n return this.convertBlockNode(child, context);\n })\n .join('');\n return `- ${inner}`;\n }\n\n if (node.type === 'blockquote') {\n return (node.content ?? [])\n .map((child) => {\n const text = this.convertBlockNode(child, context);\n return `> ${text}`;\n })\n .join('\\r\\n');\n }\n\n // Indent paragraph/heading content\n if (node.type === 'paragraph' || node.type === 'heading') {\n const attrs = node.attrs ?? {};\n let text = this.convertContent(node.content, context);\n if (attrs.indent && attrs.indent > 0) {\n const spaces = ' '.repeat(attrs.indent);\n text = `${spaces}${text}`;\n }\n return text;\n }\n\n return this.convertContent(node.content, context);\n }\n\n convertInlineNode(node: TipTapNode, context: ConversionContext): string {\n const attrs = node.attrs ?? {};\n\n if (node.type === 'username') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n const sc = this.getUsernameShortcutLink(node, context);\n return sc ? sc.url : '';\n }\n\n if (node.type === 'customShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n const shortcutBlocks = context.customShortcuts.get(attrs.id);\n if (shortcutBlocks) {\n return this.convertRawBlocks(shortcutBlocks, context);\n }\n return '';\n }\n\n if (node.type === 'titleShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n return context.title ?? '';\n }\n\n if (node.type === 'tagsShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n return context.tags?.map((e) => `#${e}`).join(' ') ?? '';\n }\n\n if (node.type === 'contentWarningShortcut') {\n if (!this.shouldRenderShortcut(node, context)) return '';\n return context.contentWarningText ?? '';\n }\n\n if (node.type === 'hardBreak') return '\\r\\n';\n\n return this.convertContent(node.content, context);\n }\n\n convertTextNode(node: TipTapNode, context: ConversionContext): string {\n const textNode = node as any;\n\n // Check for link mark — append URL\n const marks = textNode.marks ?? [];\n const linkMark = marks.find((m: any) => m.type === 'link');\n if (linkMark) {\n return `${textNode.text}: ${linkMark.attrs?.href ?? ''}`;\n }\n\n return textNode.text ?? '';\n }\n}\n" }, { "path": "apps/client-server/src/app/post-parsers/models/description-node/description-node-tree.ts", "content": "/* eslint-disable @typescript-eslint/no-explicit-any */\nimport TurndownService from 'turndown';\nimport { BaseConverter } from './converters/base-converter';\nimport { BBCodeConverter } from './converters/bbcode-converter';\nimport {\n CustomConverter,\n CustomNodeHandler,\n} from './converters/custom-converter';\nimport { HtmlConverter } from './converters/html-converter';\nimport { PlainTextConverter } from './converters/plaintext-converter';\nimport { ConversionContext } from './description-node.base';\nimport { TipTapNode } from './description-node.types';\n\nexport type InsertionOptions = {\n insertTitle?: string;\n insertTags?: string[];\n insertAd: boolean;\n};\n\nexport class DescriptionNodeTree {\n private readonly nodes: TipTapNode[];\n\n private readonly insertionOptions: InsertionOptions;\n\n private context: ConversionContext;\n\n /** Empty paragraph used as spacing before the ad */\n private readonly spacing: TipTapNode = {\n type: 'paragraph',\n content: [],\n };\n\n /** PostyBirb ad in TipTap JSON format */\n private readonly ad: TipTapNode = {\n type: 'paragraph',\n content: [\n {\n type: 'text',\n text: 'Posted using PostyBirb',\n marks: [\n {\n type: 'link',\n attrs: { href: 'https://postybirb.com', target: '_blank' },\n },\n ],\n },\n ],\n };\n\n constructor(\n context: ConversionContext,\n nodes: TipTapNode[],\n insertionOptions: InsertionOptions,\n ) {\n this.context = context;\n this.insertionOptions = insertionOptions;\n this.nodes = nodes ?? [];\n }\n\n toBBCode(): string {\n const converter = new BBCodeConverter();\n return converter.convertBlocks(this.withInsertions(), this.context);\n }\n\n toPlainText(): string {\n const converter = new PlainTextConverter();\n return converter.convertBlocks(this.withInsertions(), this.context);\n }\n\n toHtml(): string {\n const converter = new HtmlConverter();\n return converter.convertBlocks(this.withInsertions(), this.context);\n }\n\n toMarkdown(turndownService?: TurndownService): string {\n const converter = turndownService ?? new TurndownService();\n\n converter.addRule('nestedIndent', {\n filter: (node) =>\n node.nodeName === 'DIV' &&\n node.getAttribute('style')?.includes('margin-left'),\n replacement: (content) =>\n `\\n\\n> ${content.trim().replace(/\\n/g, '\\n> ')}\\n\\n`,\n });\n\n const html = this.toHtml();\n return converter.turndown(html);\n }\n\n parseCustom(blockHandler: CustomNodeHandler): string {\n const converter = new CustomConverter(blockHandler);\n return converter.convertBlocks(this.withInsertions(), this.context);\n }\n\n parseWithConverter(converter: BaseConverter): string {\n return converter.convertBlocks(this.withInsertions(), this.context);\n }\n\n public updateContext(updates: Partial): void {\n this.context = { ...this.context, ...updates };\n }\n\n /**\n * Finds all TipTap nodes of a specific type in the tree (recursively).\n */\n public findNodesByType(type: string): TipTapNode[] {\n const found: TipTapNode[] = [];\n\n const traverse = (nodes: TipTapNode[]) => {\n for (const node of nodes) {\n if (node.type === type) {\n found.push(node);\n }\n if (node.content) {\n traverse(node.content);\n }\n }\n };\n\n traverse(this.nodes);\n return found;\n }\n\n /**\n * Finds all custom shortcut IDs in the tree.\n */\n public findCustomShortcutIds(): Set {\n const ids = new Set();\n const shortcuts = this.findNodesByType('customShortcut');\n\n for (const shortcut of shortcuts) {\n const id = shortcut.attrs?.id;\n if (id) {\n ids.add(id);\n }\n }\n\n return ids;\n }\n\n /**\n * Finds all usernames in the tree.\n */\n public findUsernames(): Set {\n const usernames = new Set();\n const usernameNodes = this.findNodesByType('username');\n\n for (const node of usernameNodes) {\n const username = (node.attrs?.username as string)?.trim();\n if (username) {\n usernames.add(username);\n }\n }\n\n return usernames;\n }\n\n /**\n * Checks if a TipTap node is structurally empty\n * (no content, or content is only whitespace text nodes).\n * Only considers paragraph/heading nodes as trimmable.\n */\n private isEmptyNode(node: TipTapNode): boolean {\n if (node.type !== 'paragraph' && node.type !== 'heading') {\n return false;\n }\n\n if (!node.content || node.content.length === 0) {\n return true;\n }\n\n return node.content.every(\n (child) =>\n child.type === 'text' &&\n (!(child as any).text || (child as any).text.trim() === ''),\n );\n }\n\n /**\n * Trims structurally empty nodes from the start and end of a block array,\n * preserving empty nodes in the middle (intentional blank lines).\n */\n private trimEmptyEdgeNodes(nodes: TipTapNode[]): TipTapNode[] {\n let start = 0;\n let end = nodes.length - 1;\n\n while (start < nodes.length && this.isEmptyNode(nodes[start])) {\n start++;\n }\n\n while (end >= start && this.isEmptyNode(nodes[end])) {\n end--;\n }\n\n if (start > end) return [];\n\n return nodes.slice(start, end + 1);\n }\n\n private withInsertions(): TipTapNode[] {\n // Trim empty edge nodes before insertions so converters receive clean input\n const nodes = this.trimEmptyEdgeNodes([...this.nodes]);\n const { insertAd, insertTags, insertTitle } = this.insertionOptions;\n\n if (insertTitle) {\n nodes.unshift({\n type: 'heading',\n attrs: { level: 2 },\n content: [\n {\n type: 'text',\n text: insertTitle,\n },\n ],\n });\n }\n\n if (insertTags) {\n nodes.push({\n type: 'paragraph',\n content: [\n {\n type: 'text',\n text: insertTags.map((e) => `#${e}`).join(' '),\n },\n ],\n });\n }\n\n if (insertAd) {\n const lastNode = nodes[nodes.length - 1];\n const isLastNodeSpacing =\n lastNode?.type === 'paragraph' &&\n (!lastNode.content || lastNode.content.length === 0);\n\n // Avoid duplicated spacings\n if (!isLastNodeSpacing) {\n nodes.push(this.spacing);\n }\n\n nodes.push(this.ad);\n }\n\n return nodes;\n }\n}\n\n" }, { "path": "apps/client-server/src/app/post-parsers/models/description-node/description-node.base.ts", "content": "/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { UsernameShortcut } from '@postybirb/types';\nimport { TipTapNode } from './description-node.types';\n\n/**\n * Context provided to all converters during conversion.\n */\nexport interface ConversionContext {\n website: string;\n shortcuts: Record;\n customShortcuts: Map;\n defaultDescription: TipTapNode[];\n title?: string;\n tags?: string[];\n usernameConversions?: Map;\n contentWarningText?: string;\n}\n" }, { "path": "apps/client-server/src/app/post-parsers/models/description-node/description-node.types.ts", "content": "/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { TipTapMark, TipTapNode } from '@postybirb/types';\n\n/**\n * Re-export TipTap types from the shared types library for use in the server parser.\n */\nexport type { TipTapMark, TipTapNode };\n\n/**\n * A TipTap node used as a block-level element (paragraph, heading, list, etc.).\n * In TipTap, block nodes have `type`, optional `attrs`, and optional `content`.\n */\nexport interface ITipTapBlockNode extends TipTapNode {\n attrs?: Record;\n content?: TipTapNode[];\n}\n\n/**\n * A TipTap text node. Has `type: 'text'`, `text`, and optional `marks`.\n */\nexport interface ITipTapTextNode extends TipTapNode {\n type: 'text';\n text: string;\n marks?: TipTapMark[];\n}\n\n/**\n * Known block-level node types in TipTap.\n */\nexport const BlockTypes: string[] = [\n 'doc',\n 'paragraph',\n 'heading',\n 'blockquote',\n 'bulletList',\n 'orderedList',\n 'listItem',\n 'horizontalRule',\n 'hardBreak',\n 'image',\n 'defaultShortcut',\n];\n\n/**\n * Known inline/atom node types in TipTap (rendered inline, not block-level).\n */\nexport const InlineTypes: string[] = [\n 'username',\n 'customShortcut',\n 'titleShortcut',\n 'tagsShortcut',\n 'contentWarningShortcut',\n];\n\n/**\n * Helper to check if a TipTap node is a text node.\n */\nexport function isTextNode(node: TipTapNode): node is ITipTapTextNode {\n return node.type === 'text' && typeof (node as any).text === 'string';\n}\n\n/**\n * Helper to check if a TipTap node is an inline shortcut node.\n */\nexport function isInlineShortcut(node: TipTapNode): boolean {\n return InlineTypes.includes(node.type);\n}\n\n/**\n * Helper to check if a text node has a specific mark.\n */\nexport function hasMark(node: ITipTapTextNode, markType: string): boolean {\n return node.marks?.some((m) => m.type === markType) ?? false;\n}\n\n/**\n * Helper to get a mark's attrs from a text node.\n */\nexport function getMarkAttrs(\n node: ITipTapTextNode,\n markType: string,\n): Record | undefined {\n return node.marks?.find((m) => m.type === markType)?.attrs;\n}\n" }, { "path": "apps/client-server/src/app/post-parsers/models/description-node.spec.ts", "content": "import { TipTapNode } from '@postybirb/types';\nimport { DescriptionNodeTree } from './description-node/description-node-tree';\nimport { ConversionContext } from './description-node/description-node.base';\n\ndescribe('DescriptionNode', () => {\n it('should support username shortcuts', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Hello, ', marks: [{ type: 'bold' }] },\n {\n type: 'username',\n attrs: {\n shortcut: 'test',\n only: '',\n username: 'User',\n },\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {\n test: {\n id: 'test',\n url: 'https://test.postybirb.com/$1',\n },\n },\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe('Hello, https://test.postybirb.com/User');\n expect(tree.toHtml()).toBe(\n '
      Hello, User
      ',\n );\n expect(tree.toBBCode()).toBe(\n '[b]Hello, [/b][url=https://test.postybirb.com/User]User[/url]',\n );\n });\n\n it('should support username shortcut conversion', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Hello, ', marks: [{ type: 'bold' }] },\n {\n type: 'username',\n attrs: {\n shortcut: 'test',\n only: '',\n username: 'User',\n },\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {\n test: {\n id: 'test',\n url: 'https://test.postybirb.com/$1',\n convert: (websiteName) => {\n if (websiteName === 'test') {\n return '';\n }\n return undefined;\n },\n },\n },\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe('Hello, ');\n expect(tree.toHtml()).toBe(\n '
      Hello,
      ',\n );\n expect(tree.toBBCode()).toBe('[b]Hello, [/b]');\n });\n\n it('should handle multiple paragraphs', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'First paragraph.' }],\n },\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Second paragraph.' }],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe('First paragraph.\\r\\nSecond paragraph.');\n expect(tree.toHtml()).toBe(\n '
      First paragraph.
      Second paragraph.
      ',\n );\n expect(tree.toBBCode()).toBe('First paragraph.\\nSecond paragraph.');\n });\n\n describe('findUsernames', () => {\n it('should find all usernames in the tree', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Hello ' },\n {\n type: 'username',\n attrs: {\n shortcut: 'test',\n only: '',\n username: 'User1',\n },\n },\n { type: 'text', text: ' and ' },\n {\n type: 'username',\n attrs: {\n shortcut: 'test',\n only: '',\n username: 'User2',\n },\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n const usernames = tree.findUsernames();\n expect(usernames.size).toBe(2);\n expect(usernames.has('User1')).toBe(true);\n expect(usernames.has('User2')).toBe(true);\n });\n\n it('should find usernames across multiple paragraphs', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n {\n type: 'username',\n attrs: { shortcut: 'test', only: '', username: 'Alice' },\n },\n ],\n },\n {\n type: 'paragraph',\n content: [\n {\n type: 'username',\n attrs: { shortcut: 'test', only: '', username: 'Bob' },\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n const usernames = tree.findUsernames();\n expect(usernames.size).toBe(2);\n expect(usernames.has('Alice')).toBe(true);\n expect(usernames.has('Bob')).toBe(true);\n });\n\n it('should return empty set when no usernames exist', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Just plain text' }],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n const usernames = tree.findUsernames();\n expect(usernames.size).toBe(0);\n });\n\n it('should handle duplicate usernames', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n {\n type: 'username',\n attrs: {\n shortcut: 'test',\n only: '',\n username: 'SameUser',\n },\n },\n { type: 'text', text: ' and ' },\n {\n type: 'username',\n attrs: {\n shortcut: 'test',\n only: '',\n username: 'SameUser',\n },\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n const usernames = tree.findUsernames();\n expect(usernames.size).toBe(1);\n expect(usernames.has('SameUser')).toBe(true);\n });\n });\n\n describe('findCustomShortcutIds', () => {\n it('should find all custom shortcut IDs in the tree', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Here are some shortcuts: ' },\n {\n type: 'customShortcut',\n attrs: { id: 'shortcut-1' },\n },\n { type: 'text', text: ' and ' },\n {\n type: 'customShortcut',\n attrs: { id: 'shortcut-2' },\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n const shortcutIds = tree.findCustomShortcutIds();\n expect(shortcutIds.size).toBe(2);\n expect(shortcutIds.has('shortcut-1')).toBe(true);\n expect(shortcutIds.has('shortcut-2')).toBe(true);\n });\n\n it('should find shortcuts across multiple paragraphs', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n {\n type: 'customShortcut',\n attrs: { id: 'shortcut-a' },\n },\n ],\n },\n {\n type: 'paragraph',\n content: [\n {\n type: 'customShortcut',\n attrs: { id: 'shortcut-b' },\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n const shortcutIds = tree.findCustomShortcutIds();\n expect(shortcutIds.size).toBe(2);\n expect(shortcutIds.has('shortcut-a')).toBe(true);\n expect(shortcutIds.has('shortcut-b')).toBe(true);\n });\n\n it('should return empty set when no custom shortcuts exist', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Just plain text' }],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n const shortcutIds = tree.findCustomShortcutIds();\n expect(shortcutIds.size).toBe(0);\n });\n\n it('should handle duplicate shortcut IDs', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n {\n type: 'customShortcut',\n attrs: { id: 'same-id' },\n },\n { type: 'text', text: ' and ' },\n {\n type: 'customShortcut',\n attrs: { id: 'same-id' },\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n const shortcutIds = tree.findCustomShortcutIds();\n expect(shortcutIds.size).toBe(1);\n expect(shortcutIds.has('same-id')).toBe(true);\n });\n\n it('should handle shortcuts without IDs gracefully', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n {\n type: 'customShortcut',\n attrs: { id: '' },\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n const shortcutIds = tree.findCustomShortcutIds();\n expect(shortcutIds.size).toBe(0);\n });\n });\n\n describe('updateContext', () => {\n it('should allow updating context after tree creation', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n {\n type: 'username',\n attrs: {\n shortcut: 'test',\n only: '',\n username: 'TestUser',\n },\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {\n test: {\n id: 'test',\n url: 'https://test.postybirb.com/$1',\n },\n },\n customShortcuts: new Map(),\n defaultDescription: [],\n usernameConversions: new Map(),\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n // Before update - no conversion\n expect(tree.toPlainText()).toBe('https://test.postybirb.com/TestUser');\n\n // Update context with username conversion\n tree.updateContext({\n usernameConversions: new Map([['TestUser', 'ConvertedUser']]),\n });\n\n // After update - should use converted username\n expect(tree.toPlainText()).toBe(\n 'https://test.postybirb.com/ConvertedUser',\n );\n\n // Verify the tree still finds the original username\n const usernames = tree.findUsernames();\n expect(usernames.has('TestUser')).toBe(true);\n });\n\n it('should convert cross-platform username tags to target website', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n {\n type: 'username',\n attrs: {\n shortcut: 'twitter',\n only: '',\n username: 'abcd',\n },\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'bluesky',\n shortcuts: {\n twitter: {\n id: 'twitter',\n url: 'https://x.com/$1',\n },\n bluesky: {\n id: 'bluesky',\n url: 'https://bsky.app/profile/$1',\n },\n },\n customShortcuts: new Map(),\n defaultDescription: [],\n usernameConversions: new Map([['abcd', 'abcd.bsky.app']]),\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe('https://bsky.app/profile/abcd.bsky.app');\n expect(tree.toHtml()).toBe(\n '',\n );\n });\n\n it('should keep original username when no conversion exists', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n {\n type: 'username',\n attrs: {\n shortcut: 'twitter',\n only: '',\n username: 'someuser',\n },\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'bluesky',\n shortcuts: {\n twitter: {\n id: 'twitter',\n url: 'https://x.com/$1',\n },\n bluesky: {\n id: 'bluesky',\n url: 'https://bsky.app/profile/$1',\n },\n },\n customShortcuts: new Map(),\n defaultDescription: [],\n usernameConversions: new Map(),\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe('https://x.com/someuser');\n expect(tree.toHtml()).toBe(\n '',\n );\n });\n\n it('should convert usernames when shortcut ID matches target website', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n {\n type: 'username',\n attrs: { shortcut: 'bluesky', only: '', username: 'x' },\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'bluesky',\n shortcuts: {\n twitter: {\n id: 'twitter',\n url: 'https://x.com/$1',\n },\n bluesky: {\n id: 'bluesky',\n url: 'https://bsky.app/profile/$1',\n },\n },\n customShortcuts: new Map(),\n defaultDescription: [],\n usernameConversions: new Map([['x', 'bluesky_user']]),\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe('https://bsky.app/profile/bluesky_user');\n expect(tree.toHtml()).toBe(\n '',\n );\n });\n });\n\n describe('blockquote nesting', () => {\n it('should render blockquotes in HTML', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Para 1' }],\n },\n {\n type: 'blockquote',\n content: [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Para 1 nested' }],\n },\n ],\n },\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Para 2' }],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toHtml()).toBe(\n '
      Para 1
      Para 1 nested
      Para 2
      ',\n );\n });\n\n it('should render blockquotes in plain text with > prefix', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Para 1' }],\n },\n {\n type: 'blockquote',\n content: [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Para 1 nested' }],\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe('Para 1\\r\\n> Para 1 nested');\n });\n\n it('should render blockquotes in BBCode with [quote] tags', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Para 1' }],\n },\n {\n type: 'blockquote',\n content: [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Para 1 nested' }],\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toBBCode()).toBe('Para 1\\n[quote]Para 1 nested[/quote]');\n });\n\n it('should handle deeply nested blockquotes (multi-level)', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Level 0' }],\n },\n {\n type: 'blockquote',\n content: [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Level 1' }],\n },\n {\n type: 'blockquote',\n content: [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Level 2' }],\n },\n ],\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe('Level 0\\r\\n> Level 1\\r\\n> > Level 2');\n expect(tree.toBBCode()).toBe(\n 'Level 0\\n[quote]Level 1\\n[quote]Level 2[/quote][/quote]',\n );\n });\n\n it('should handle multiple paragraphs at same blockquote level', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Para 1' }],\n },\n {\n type: 'blockquote',\n content: [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Para 1 nested' }],\n },\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Para 1 nested 2' }],\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe(\n 'Para 1\\r\\n> Para 1 nested\\r\\n> Para 1 nested 2',\n );\n });\n\n it('should find usernames in blockquotes', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Para 1 ' }],\n },\n {\n type: 'blockquote',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Nested ' },\n {\n type: 'username',\n attrs: {\n shortcut: 'test',\n only: '',\n username: 'NestedUser',\n },\n },\n ],\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {\n test: {\n id: 'test',\n url: 'https://test.postybirb.com/$1',\n },\n },\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n const usernames = tree.findUsernames();\n expect(usernames.has('NestedUser')).toBe(true);\n });\n\n it('should render blockquotes in Markdown', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Para 1' }],\n },\n {\n type: 'blockquote',\n content: [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Para 1 nested' }],\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toMarkdown()).toBe('Para 1\\n\\n> Para 1 nested');\n });\n\n it('should render deeply nested blockquotes in Markdown', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Level 0' }],\n },\n {\n type: 'blockquote',\n content: [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Level 1' }],\n },\n {\n type: 'blockquote',\n content: [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Level 2' }],\n },\n ],\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toMarkdown()).toBe('Level 0\\n\\n> Level 1\\n> \\n> > Level 2');\n });\n });\n\n describe('system inline shortcuts', () => {\n it('should render titleShortcut with title from context', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Title: ' },\n { type: 'titleShortcut', attrs: {} },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n title: 'My Amazing Artwork',\n tags: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe('Title: My Amazing Artwork');\n expect(tree.toHtml()).toBe(\n '
      Title: My Amazing Artwork
      ',\n );\n expect(tree.toBBCode()).toBe('Title: My Amazing Artwork');\n });\n\n it('should render tagsShortcut with tags from context', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Tags: ' },\n { type: 'tagsShortcut', attrs: {} },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n title: '',\n tags: ['art', 'digital', 'fantasy'],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe('Tags: #art #digital #fantasy');\n expect(tree.toHtml()).toBe(\n '
      Tags: #art #digital #fantasy
      ',\n );\n expect(tree.toBBCode()).toBe('Tags: #art #digital #fantasy');\n });\n\n it('should render contentWarningShortcut with content warning from context', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'CW: ' },\n { type: 'contentWarningShortcut', attrs: {} },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n title: '',\n tags: [],\n contentWarningText: 'Mild Violence',\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe('CW: Mild Violence');\n expect(tree.toHtml()).toBe('
      CW: Mild Violence
      ');\n expect(tree.toBBCode()).toBe('CW: Mild Violence');\n });\n\n it('should render empty string when title is not in context', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Title: ' },\n { type: 'titleShortcut', attrs: {} },\n { type: 'text', text: ' end' },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe('Title: end');\n });\n\n it('should render empty string when tags array is empty', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Tags: ' },\n { type: 'tagsShortcut', attrs: {} },\n { type: 'text', text: ' end' },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n tags: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe('Tags: end');\n });\n\n it('should render all system shortcuts together', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n { type: 'titleShortcut', attrs: {} },\n { type: 'text', text: ' - ' },\n { type: 'contentWarningShortcut', attrs: {} },\n ],\n },\n {\n type: 'paragraph',\n content: [{ type: 'tagsShortcut', attrs: {} }],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n title: 'My Art',\n tags: ['tag1', 'tag2'],\n contentWarningText: 'NSFW',\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe('My Art - NSFW\\r\\n#tag1 #tag2');\n expect(tree.toHtml()).toBe(\n '
      My Art - NSFW
      #tag1 #tag2
      ',\n );\n });\n\n it('should HTML encode special characters in title', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [{ type: 'titleShortcut', attrs: {} }],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n title: '',\n tags: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toHtml()).toBe(\n '
      <script>alert("xss")</script>
      ',\n );\n });\n });\n\n describe('Username shortcuts', () => {\n it('should support username attrs format', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Hello, ', marks: [{ type: 'bold' }] },\n {\n type: 'username',\n attrs: {\n shortcut: 'test',\n only: '',\n username: 'TestUser',\n },\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {\n test: {\n id: 'test',\n url: 'https://test.postybirb.com/$1',\n },\n },\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe(\n 'Hello, https://test.postybirb.com/TestUser',\n );\n expect(tree.toHtml()).toBe(\n '
      Hello, TestUser
      ',\n );\n expect(tree.toBBCode()).toBe(\n '[b]Hello, [/b][url=https://test.postybirb.com/TestUser]TestUser[/url]',\n );\n });\n\n it('should find usernames from attrs format', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n {\n type: 'username',\n attrs: {\n shortcut: 'twitter',\n only: '',\n username: 'alice',\n },\n },\n { type: 'text', text: ' and ' },\n {\n type: 'username',\n attrs: {\n shortcut: 'twitter',\n only: '',\n username: 'bob',\n },\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {},\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n const usernames = tree.findUsernames();\n expect(usernames.size).toBe(2);\n expect(usernames.has('alice')).toBe(true);\n expect(usernames.has('bob')).toBe(true);\n });\n\n it('should support username conversion', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Follow me: ' },\n {\n type: 'username',\n attrs: {\n shortcut: 'twitter',\n only: '',\n username: 'myusername',\n },\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {\n twitter: {\n id: 'twitter',\n url: 'https://twitter.com/$1',\n },\n test: {\n id: 'test',\n url: 'https://test.postybirb.com/$1',\n },\n },\n customShortcuts: new Map(),\n defaultDescription: [],\n usernameConversions: new Map([['myusername', 'converted_username']]),\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe(\n 'Follow me: https://test.postybirb.com/converted_username',\n );\n expect(tree.toHtml()).toBe(\n '',\n );\n });\n\n it('should handle empty username gracefully', () => {\n const nodes: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n {\n type: 'username',\n attrs: {\n shortcut: 'test',\n only: '',\n username: '',\n },\n },\n ],\n },\n ];\n\n const context: ConversionContext = {\n website: 'test',\n shortcuts: {\n test: {\n id: 'test',\n url: 'https://test.postybirb.com/$1',\n },\n },\n customShortcuts: new Map(),\n defaultDescription: [],\n };\n\n const tree = new DescriptionNodeTree(context, nodes, {\n insertAd: false,\n });\n\n expect(tree.toPlainText()).toBe('');\n expect(tree.findUsernames().size).toBe(0);\n });\n });\n});\n" }, { "path": "apps/client-server/src/app/post-parsers/parsers/content-warning-parser.ts", "content": "import { Injectable } from '@nestjs/common';\nimport { BaseWebsiteOptions } from '../../websites/models/base-website-options';\nimport { DefaultWebsiteOptions } from '../../websites/models/default-website-options';\n\n@Injectable()\nexport class ContentWarningParser {\n public async parse(\n defaultOptions: DefaultWebsiteOptions,\n websiteOptions: BaseWebsiteOptions,\n ): Promise {\n const defaultWarningForm = defaultOptions.getFormFieldFor('contentWarning');\n const websiteWarningForm = websiteOptions.getFormFieldFor('contentWarning');\n const merged = websiteOptions.mergeDefaults(defaultOptions);\n\n const warning = merged.contentWarning ?? '';\n const field = websiteWarningForm ?? defaultWarningForm;\n const maxLength = field?.maxLength ?? Infinity;\n\n return warning.trim().slice(0, maxLength);\n }\n}\n" }, { "path": "apps/client-server/src/app/post-parsers/parsers/description-parser.service.spec.ts", "content": "/* eslint-disable max-classes-per-file */\nimport { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport { DescriptionField } from '@postybirb/form-builder';\nimport {\n Description,\n DescriptionType,\n DescriptionValue,\n IWebsiteOptions,\n TipTapNode,\n} from '@postybirb/types';\nimport { WEBSITE_IMPLEMENTATIONS } from '../../constants';\nimport { CustomShortcutsService } from '../../custom-shortcuts/custom-shortcuts.service';\nimport { SettingsService } from '../../settings/settings.service';\nimport { UserConvertersService } from '../../user-converters/user-converters.service';\nimport { BaseWebsiteOptions } from '../../websites/models/base-website-options';\nimport { DefaultWebsiteOptions } from '../../websites/models/default-website-options';\nimport { UnknownWebsite } from '../../websites/website';\nimport { DescriptionParserService } from './description-parser.service';\n\ndescribe('DescriptionParserService', () => {\n let module: TestingModule;\n let service: DescriptionParserService;\n let settingsService: SettingsService;\n let customShortcutsService: CustomShortcutsService;\n let userConvertersService: UserConvertersService;\n\n const testDescription: Description = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Hello, ', marks: [{ type: 'bold' }] },\n { type: 'text', text: 'World!' },\n ],\n },\n {\n type: 'paragraph',\n content: [\n {\n type: 'text',\n text: 'A link',\n marks: [\n {\n type: 'link',\n attrs: { href: 'https://postybirb.com' },\n },\n ],\n },\n ],\n },\n ],\n };\n\n beforeEach(async () => {\n clearDatabase();\n module = await Test.createTestingModule({\n providers: [\n DescriptionParserService,\n {\n provide: SettingsService,\n useValue: {\n getSettings: jest.fn(),\n getDefaultSettings: jest.fn(),\n },\n },\n {\n provide: CustomShortcutsService,\n useValue: {\n findById: jest.fn(),\n },\n },\n {\n provide: UserConvertersService,\n useValue: {\n convert: jest\n .fn()\n .mockImplementation((instance, username) =>\n Promise.resolve(username),\n ),\n },\n },\n {\n provide: WEBSITE_IMPLEMENTATIONS,\n useValue: [],\n },\n ],\n }).compile();\n service = module.get(DescriptionParserService);\n settingsService = module.get(SettingsService);\n customShortcutsService = module.get(CustomShortcutsService);\n userConvertersService = module.get(UserConvertersService);\n settingsService.getDefaultSettings = jest.fn().mockResolvedValue({\n settings: {\n hiddenWebsites: [],\n language: 'en',\n allowAd: false,\n },\n });\n });\n\n afterAll(async () => {\n await module.close();\n });\n\n function createWebsiteOptions(\n description: Description | undefined,\n ): IWebsiteOptions {\n return {\n data: {\n description: {\n description,\n },\n },\n } as IWebsiteOptions;\n }\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n\n it('should parse plaintext description', async () => {\n const instance = {\n decoratedProps: {\n allowAd: true,\n metadata: {\n name: 'Test',\n },\n },\n };\n\n class PlaintextBaseWebsiteOptions extends BaseWebsiteOptions {\n @DescriptionField({ descriptionType: DescriptionType.PLAINTEXT })\n description: DescriptionValue;\n }\n\n const defaultOptions = createWebsiteOptions(testDescription);\n const websiteOptions = createWebsiteOptions(undefined);\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new PlaintextBaseWebsiteOptions(websiteOptions.data),\n [],\n '',\n );\n expect(description).toMatchInlineSnapshot(`\n \"Hello, World!\n A link: https://postybirb.com\"\n `);\n });\n\n it('should parse html description', async () => {\n const instance = {\n decoratedProps: {\n allowAd: true,\n metadata: {\n name: 'Test',\n },\n },\n };\n\n const defaultOptions = createWebsiteOptions(testDescription);\n const websiteOptions = createWebsiteOptions(undefined);\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(websiteOptions.data),\n [],\n '',\n );\n expect(description).toMatchInlineSnapshot(\n `\"
      Hello, World!
      \"`,\n );\n });\n\n it('should parse markdown description', async () => {\n const instance = {\n decoratedProps: {\n allowAd: true,\n metadata: {\n name: 'Test',\n },\n },\n };\n\n class MarkdownBaseWebsiteOptions extends BaseWebsiteOptions {\n @DescriptionField({ descriptionType: DescriptionType.MARKDOWN })\n description: DescriptionValue;\n }\n\n const defaultOptions = createWebsiteOptions(testDescription);\n const websiteOptions = createWebsiteOptions(undefined);\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new MarkdownBaseWebsiteOptions(websiteOptions.data),\n [],\n '',\n );\n expect(description).toMatchInlineSnapshot(`\n \"**Hello,** World!\n\n [A link](https://postybirb.com)\"\n `);\n });\n\n it('should return empty for description type NONE', async () => {\n const instance = {\n decoratedProps: {\n allowAd: true,\n },\n };\n\n class NoneBaseWebsiteOptions extends BaseWebsiteOptions {\n @DescriptionField({ descriptionType: DescriptionType.NONE })\n description: DescriptionValue;\n }\n\n const defaultOptions = createWebsiteOptions(testDescription);\n const websiteOptions = createWebsiteOptions(undefined);\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new NoneBaseWebsiteOptions(websiteOptions.data),\n [],\n '',\n );\n expect(description).toEqual(undefined);\n });\n\n it('should insert ad if allowed in settings and website', async () => {\n settingsService.getDefaultSettings = jest.fn().mockResolvedValue({\n settings: {\n hiddenWebsites: [],\n language: 'en',\n allowAd: true,\n },\n });\n const instance = {\n decoratedProps: {\n allowAd: true,\n metadata: {\n name: 'Test',\n },\n },\n };\n\n const defaultOptions = createWebsiteOptions(testDescription);\n const websiteOptions = createWebsiteOptions(undefined);\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(websiteOptions.data),\n [],\n '',\n );\n\n expect(description).toMatchInlineSnapshot(\n `\"
      Hello, World!
      \"`,\n );\n });\n\n it('should not insert ad if allowed in settings and not website', async () => {\n settingsService.getDefaultSettings = jest.fn().mockResolvedValue({\n settings: {\n hiddenWebsites: [],\n language: 'en',\n allowAd: true,\n },\n });\n const instance = {\n decoratedProps: {\n allowAd: false,\n metadata: {\n name: 'Test',\n },\n },\n };\n\n const defaultOptions = createWebsiteOptions(testDescription);\n const websiteOptions = createWebsiteOptions(undefined);\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(websiteOptions.data),\n [],\n '',\n );\n\n expect(description).toMatchInlineSnapshot(\n `\"
      Hello, World!
      \"`,\n );\n });\n\n it('should pass blocks through without merging', () => {\n const blocks: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n {\n type: 'text',\n text: 'Test\\nIn the same block!',\n },\n ],\n },\n {\n type: 'paragraph',\n content: [\n {\n type: 'text',\n text: 'New block',\n },\n ],\n },\n {\n type: 'paragraph',\n attrs: { textAlign: 'center' },\n content: [\n {\n type: 'text',\n text: 'block',\n },\n ],\n },\n ];\n\n const result = service.mergeBlocks(blocks);\n expect(result).toBe(blocks);\n });\n\n it('should insert default when available', async () => {\n const instance = {\n decoratedProps: {\n allowAd: true,\n metadata: {\n name: 'Test',\n },\n },\n };\n\n class PlaintextBaseWebsiteOptions extends BaseWebsiteOptions {\n @DescriptionField({ descriptionType: DescriptionType.PLAINTEXT })\n description: DescriptionValue;\n }\n\n const defaultOptions = createWebsiteOptions(testDescription);\n const websiteDesc: Description = {\n type: 'doc',\n content: [\n {\n type: 'defaultShortcut',\n },\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Hello, Basic' }],\n },\n ],\n };\n const websiteOptions = createWebsiteOptions(websiteDesc);\n websiteOptions.data.description.overrideDefault = true;\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new PlaintextBaseWebsiteOptions(websiteOptions.data),\n [],\n '',\n );\n expect(description).toMatchInlineSnapshot(`\n \"Hello, World!\n A link: https://postybirb.com\n Hello, Basic\"\n `);\n });\n\n describe('Custom Shortcuts', () => {\n it('should inject single custom shortcut', async () => {\n const instance = {\n decoratedProps: {\n allowAd: false,\n metadata: {\n name: 'Test',\n },\n },\n };\n\n const shortcutContent: Description = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n {\n type: 'text',\n text: 'Commission Info',\n marks: [{ type: 'bold' }],\n },\n ],\n },\n ],\n };\n\n customShortcutsService.findById = jest.fn().mockResolvedValue({\n id: 'cs-1',\n name: 'commission',\n shortcut: shortcutContent,\n });\n\n const descriptionWithShortcut: Description = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Check out my ' },\n { type: 'customShortcut', attrs: { id: 'cs-1' } },\n ],\n },\n ],\n };\n\n const defaultOptions = createWebsiteOptions(descriptionWithShortcut);\n const websiteOptions = createWebsiteOptions(undefined);\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(websiteOptions.data),\n [],\n '',\n );\n\n expect(customShortcutsService.findById).toHaveBeenCalledWith('cs-1');\n expect(description).toMatchInlineSnapshot(\n `\"
      Check out my
      Commission Info
      \"`,\n );\n });\n\n it('should inject multiple custom shortcuts', async () => {\n const instance = {\n decoratedProps: {\n allowAd: false,\n metadata: {\n name: 'Test',\n },\n },\n };\n\n const commissionShortcut: Description = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Commissions Open!' }],\n },\n ],\n };\n\n const priceShortcut: Description = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [{ type: 'text', text: '$50 per hour' }],\n },\n ],\n };\n\n customShortcutsService.findById = jest\n .fn()\n .mockImplementation((id: string) => {\n if (id === 'cs-1')\n return Promise.resolve({\n id: 'cs-1',\n name: 'commission',\n shortcut: commissionShortcut,\n });\n if (id === 'cs-2')\n return Promise.resolve({\n id: 'cs-2',\n name: 'price',\n shortcut: priceShortcut,\n });\n return Promise.resolve(null);\n });\n\n const descriptionWithShortcuts: Description = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'customShortcut', attrs: { id: 'cs-1' } },\n { type: 'text', text: ' - ' },\n { type: 'customShortcut', attrs: { id: 'cs-2' } },\n ],\n },\n ],\n };\n\n const defaultOptions = createWebsiteOptions(descriptionWithShortcuts);\n const websiteOptions = createWebsiteOptions(undefined);\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(websiteOptions.data),\n [],\n '',\n );\n\n expect(customShortcutsService.findById).toHaveBeenCalledWith('cs-1');\n expect(customShortcutsService.findById).toHaveBeenCalledWith('cs-2');\n expect(description).toMatchInlineSnapshot(\n `\"
      Commissions Open!
      -
      $50 per hour
      \"`,\n );\n });\n\n it('should handle missing custom shortcut gracefully', async () => {\n const instance = {\n decoratedProps: {\n allowAd: false,\n metadata: {\n name: 'Test',\n },\n },\n };\n\n customShortcutsService.findById = jest.fn().mockResolvedValue(null);\n\n const descriptionWithMissing: Description = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Before ' },\n {\n type: 'customShortcut',\n attrs: { id: 'cs-missing' },\n },\n { type: 'text', text: ' After' },\n ],\n },\n ],\n };\n\n const defaultOptions = createWebsiteOptions(descriptionWithMissing);\n const websiteOptions = createWebsiteOptions(undefined);\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(websiteOptions.data),\n [],\n '',\n );\n\n expect(customShortcutsService.findById).toHaveBeenCalledWith(\n 'cs-missing',\n );\n expect(description).toMatchInlineSnapshot(`\"
      Before After
      \"`);\n });\n\n it('should resolve custom shortcuts with different output formats', async () => {\n const instance = {\n decoratedProps: {\n allowAd: false,\n metadata: {\n name: 'Test',\n },\n },\n };\n\n class PlaintextBaseWebsiteOptions extends BaseWebsiteOptions {\n @DescriptionField({ descriptionType: DescriptionType.PLAINTEXT })\n description: DescriptionValue;\n }\n\n const shortcutContent: Description = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n {\n type: 'text',\n text: 'Bold Text',\n marks: [{ type: 'bold' }],\n },\n ],\n },\n ],\n };\n\n customShortcutsService.findById = jest.fn().mockResolvedValue({\n id: 'cs-1',\n name: 'bold',\n shortcut: shortcutContent,\n });\n\n const descriptionWithShortcut: Description = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Text: ' },\n { type: 'customShortcut', attrs: { id: 'cs-1' } },\n ],\n },\n ],\n };\n\n const defaultOptions = createWebsiteOptions(descriptionWithShortcut);\n const websiteOptions = createWebsiteOptions(undefined);\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new PlaintextBaseWebsiteOptions(websiteOptions.data),\n [],\n '',\n );\n\n expect(description).toMatchInlineSnapshot(`\"Text: Bold Text\"`);\n });\n\n it('should resolve custom shortcuts with links and styling', async () => {\n const instance = {\n decoratedProps: {\n allowAd: false,\n metadata: {\n name: 'Test',\n },\n },\n };\n\n const shortcutWithLink: Description = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Visit my ' },\n {\n type: 'text',\n text: 'portfolio',\n marks: [\n { type: 'bold' },\n {\n type: 'link',\n attrs: { href: 'https://portfolio.example.com' },\n },\n ],\n },\n ],\n },\n ],\n };\n\n customShortcutsService.findById = jest.fn().mockResolvedValue({\n id: 'cs-link',\n name: 'portfolio',\n shortcut: shortcutWithLink,\n });\n\n const descriptionWithShortcut: Description = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'customShortcut', attrs: { id: 'cs-link' } },\n ],\n },\n ],\n };\n\n const defaultOptions = createWebsiteOptions(descriptionWithShortcut);\n const websiteOptions = createWebsiteOptions(undefined);\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(websiteOptions.data),\n [],\n '',\n );\n\n expect(description).toMatchInlineSnapshot(\n `\"
      Visit my portfolio
      \"`,\n );\n });\n });\n\n describe('System Inline Shortcuts', () => {\n it('should render titleShortcut with submission title', async () => {\n const instance = {\n decoratedProps: {\n allowAd: false,\n metadata: {\n name: 'Test',\n },\n },\n };\n\n const descriptionWithTitle: Description = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Artwork: ' },\n { type: 'titleShortcut', attrs: {} },\n ],\n },\n ],\n };\n\n const defaultOptions = createWebsiteOptions(descriptionWithTitle);\n const websiteOptions = createWebsiteOptions(undefined);\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(websiteOptions.data),\n [],\n 'My Amazing Art',\n );\n\n expect(description).toMatchInlineSnapshot(\n `\"
      Artwork: My Amazing Art
      \"`,\n );\n });\n\n it('should render tagsShortcut with submission tags', async () => {\n const instance = {\n decoratedProps: {\n allowAd: false,\n metadata: {\n name: 'Test',\n },\n },\n };\n\n const descriptionWithTags: Description = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Tags: ' },\n { type: 'tagsShortcut', attrs: {} },\n ],\n },\n ],\n };\n\n const defaultOptions = createWebsiteOptions(descriptionWithTags);\n const websiteOptions = createWebsiteOptions(undefined);\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(websiteOptions.data),\n ['art', 'digital', 'fantasy'],\n '',\n );\n\n expect(description).toMatchInlineSnapshot(\n `\"
      Tags: #art #digital #fantasy
      \"`,\n );\n });\n\n it('should render contentWarningShortcut with content warning', async () => {\n const instance = {\n decoratedProps: {\n allowAd: false,\n metadata: {\n name: 'Test',\n },\n },\n };\n\n const descriptionWithCW: Description = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Content Warning: ' },\n { type: 'contentWarningShortcut', attrs: {} },\n ],\n },\n ],\n };\n\n const defaultOptions = new DefaultWebsiteOptions({\n description: {\n description: descriptionWithCW,\n overrideDefault: false,\n },\n contentWarning: 'Mild Violence',\n });\n const websiteOptions = new BaseWebsiteOptions({});\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n defaultOptions,\n websiteOptions,\n [],\n '',\n );\n\n expect(description).toMatchInlineSnapshot(\n `\"
      Content Warning: Mild Violence
      \"`,\n );\n });\n\n it('should not double-insert title when titleShortcut is present and insertTitle is true', async () => {\n const instance = {\n decoratedProps: {\n allowAd: false,\n metadata: {\n name: 'Test',\n },\n },\n };\n\n const descriptionWithTitle: Description = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Title: ' },\n { type: 'titleShortcut', attrs: {} },\n ],\n },\n ],\n };\n\n const defaultOptions = createWebsiteOptions(descriptionWithTitle);\n defaultOptions.data.description.insertTitle = true;\n const websiteOptions = createWebsiteOptions(undefined);\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(websiteOptions.data),\n [],\n 'My Title',\n );\n\n expect(description).toMatchInlineSnapshot(\n `\"
      Title: My Title
      \"`,\n );\n expect((description.match(/My Title/g) || []).length).toBe(1);\n });\n\n it('should not double-insert tags when tagsShortcut is present and insertTags is true', async () => {\n const instance = {\n decoratedProps: {\n allowAd: false,\n metadata: {\n name: 'Test',\n },\n },\n };\n\n const descriptionWithTags: Description = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Tags: ' },\n { type: 'tagsShortcut', attrs: {} },\n ],\n },\n ],\n };\n\n const defaultOptions = createWebsiteOptions(descriptionWithTags);\n defaultOptions.data.description.insertTags = true;\n const websiteOptions = createWebsiteOptions(undefined);\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(websiteOptions.data),\n ['tag1', 'tag2'],\n '',\n );\n\n expect(description).toMatchInlineSnapshot(\n `\"
      Tags: #tag1 #tag2
      \"`,\n );\n expect((description.match(/#tag1 #tag2/g) || []).length).toBe(1);\n });\n\n it('should render all system shortcuts together in plaintext', async () => {\n const instance = {\n decoratedProps: {\n allowAd: false,\n metadata: {\n name: 'Test',\n },\n },\n };\n\n class PlaintextBaseWebsiteOptions extends BaseWebsiteOptions {\n @DescriptionField({ descriptionType: DescriptionType.PLAINTEXT })\n description: DescriptionValue;\n }\n\n const descriptionWithAll: Description = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n content: [\n { type: 'titleShortcut', attrs: {} },\n { type: 'text', text: ' (' },\n { type: 'contentWarningShortcut', attrs: {} },\n { type: 'text', text: ')' },\n ],\n },\n {\n type: 'paragraph',\n content: [{ type: 'tagsShortcut', attrs: {} }],\n },\n ],\n };\n\n const defaultOptions = new DefaultWebsiteOptions({\n description: {\n description: descriptionWithAll,\n overrideDefault: false,\n },\n contentWarning: 'NSFW',\n });\n const websiteOptions = new PlaintextBaseWebsiteOptions({});\n const description = await service.parse(\n instance as unknown as UnknownWebsite,\n defaultOptions,\n websiteOptions,\n ['art', 'digital'],\n 'My Art',\n );\n\n expect(description).toMatchInlineSnapshot(`\n \"My Art (NSFW)\n #art #digital\"\n `);\n });\n });\n\n // TODO: Add test for description type CUSTOM\n});\n" }, { "path": "apps/client-server/src/app/post-parsers/parsers/description-parser.service.ts", "content": "import { Inject, Injectable } from '@nestjs/common';\nimport {\n DescriptionType,\n TipTapNode,\n UsernameShortcut,\n} from '@postybirb/types';\nimport { Class } from 'type-fest';\nimport { WEBSITE_IMPLEMENTATIONS } from '../../constants';\nimport { CustomShortcutsService } from '../../custom-shortcuts/custom-shortcuts.service';\nimport { SettingsService } from '../../settings/settings.service';\nimport { UserConvertersService } from '../../user-converters/user-converters.service';\nimport { BaseWebsiteOptions } from '../../websites/models/base-website-options';\nimport { DefaultWebsiteOptions } from '../../websites/models/default-website-options';\nimport { isWithCustomDescriptionParser } from '../../websites/models/website-modifiers/with-custom-description-parser';\nimport { isWithRuntimeDescriptionParser } from '../../websites/models/website-modifiers/with-runtime-description-parser';\nimport { UnknownWebsite, Website } from '../../websites/website';\nimport {\n DescriptionNodeTree,\n InsertionOptions,\n} from '../models/description-node/description-node-tree';\nimport { ConversionContext } from '../models/description-node/description-node.base';\n\n@Injectable()\nexport class DescriptionParserService {\n private readonly websiteShortcuts: Record = {};\n\n constructor(\n private readonly settingsService: SettingsService,\n @Inject(WEBSITE_IMPLEMENTATIONS)\n private readonly websiteImplementations: Class[],\n private readonly customShortcutsService?: CustomShortcutsService,\n private readonly userConvertersService?: UserConvertersService,\n ) {\n this.websiteImplementations.forEach((website) => {\n const shortcut: UsernameShortcut | undefined =\n website.prototype.decoratedProps.usernameShortcut;\n if (shortcut) {\n this.websiteShortcuts[shortcut.id] = shortcut;\n }\n });\n }\n\n public async parse(\n instance: Website,\n defaultOptions: DefaultWebsiteOptions,\n websiteOptions: BaseWebsiteOptions,\n tags: string[],\n title: string,\n ): Promise {\n const mergedOptions = websiteOptions.mergeDefaults(defaultOptions);\n const { descriptionType, hidden } =\n mergedOptions.getFormFieldFor('description');\n\n if (descriptionType === DescriptionType.NONE || hidden) {\n return undefined;\n }\n\n const settings = await this.settingsService.getDefaultSettings();\n let { allowAd } = settings.settings;\n\n if (!instance.decoratedProps.allowAd) {\n allowAd = false;\n }\n\n const descriptionValue = mergedOptions.description;\n const descriptionBlocks: TipTapNode[] =\n descriptionValue.description?.content ?? [];\n\n const { contentWarning } = mergedOptions;\n\n // Detect presence of shortcut tags to prevent double insertion\n const hasTitleShortcut = this.hasInlineContentType(\n descriptionBlocks,\n 'titleShortcut',\n );\n const hasTagsShortcut = this.hasInlineContentType(\n descriptionBlocks,\n 'tagsShortcut',\n );\n\n const insertionOptions: InsertionOptions = {\n insertTitle:\n descriptionValue.insertTitle && !hasTitleShortcut ? title : undefined,\n insertTags:\n descriptionValue.insertTags && !hasTagsShortcut ? tags : undefined,\n insertAd: allowAd,\n };\n\n /*\n * We choose to merge blocks here to avoid confusing user expectations.\n * Most editors want you to use Shift + Enter to insert a new line. But in\n * most cases this is not something the user cares about. They just want to\n * see the description on a line-by-line basis. So we choose to merge similar\n * blocks together to avoid confusion.\n */\n const mergedDescriptionBlocks = this.mergeBlocks(descriptionBlocks);\n\n // Pre-resolve default description\n const defaultDescription = this.mergeBlocks(\n defaultOptions.description.description?.content ?? [],\n );\n\n for (let i = defaultDescription.length - 1; i >= 0; i--) {\n const element = defaultDescription[i];\n const isSpacing =\n element?.type === 'paragraph' &&\n (!element.content || element.content.length === 0);\n if (isSpacing) {\n defaultDescription.splice(i);\n } else break;\n }\n\n // Build tree once with minimal context\n const context: ConversionContext = {\n website: instance.decoratedProps.metadata.name,\n shortcuts: this.websiteShortcuts,\n customShortcuts: new Map(),\n defaultDescription,\n title,\n tags,\n usernameConversions: new Map(),\n contentWarningText: contentWarning,\n };\n\n const tree = new DescriptionNodeTree(\n context,\n mergedDescriptionBlocks,\n insertionOptions,\n );\n\n // Resolve and inject into the same tree\n const customShortcuts = await this.resolveCustomShortcutsFromTree(tree);\n const usernameConversions = await this.resolveUsernamesFromTree(\n tree,\n instance,\n );\n\n tree.updateContext({\n customShortcuts,\n usernameConversions,\n });\n\n return this.createDescription(instance, descriptionType, tree);\n }\n\n private createDescription(\n instance: Website,\n descriptionType: DescriptionType,\n tree: DescriptionNodeTree,\n ): string {\n switch (descriptionType) {\n case DescriptionType.MARKDOWN:\n return tree.toMarkdown();\n case DescriptionType.HTML:\n return tree.toHtml();\n case DescriptionType.PLAINTEXT:\n return tree.toPlainText();\n case DescriptionType.BBCODE:\n return tree.toBBCode();\n case DescriptionType.CUSTOM:\n if (isWithCustomDescriptionParser(instance)) {\n const converter = instance.getDescriptionConverter();\n return tree.parseWithConverter(converter);\n }\n throw new Error(\n `Website does not implement custom description parser: ${instance.constructor.name}`,\n );\n case DescriptionType.RUNTIME:\n if (isWithRuntimeDescriptionParser(instance)) {\n return this.createDescription(\n instance,\n instance.getRuntimeParser(),\n tree,\n );\n }\n throw new Error(\n `Website does not implement runtime description mapping: ${instance.constructor.name}`,\n );\n default:\n throw new Error(`Unsupported description type: ${descriptionType}`);\n }\n }\n\n /**\n * Pre-resolves all custom shortcuts found in the description tree.\n * Note: Does not handle nested shortcuts - users should not create circular references.\n */\n private async resolveCustomShortcutsFromTree(\n tree: DescriptionNodeTree,\n ): Promise> {\n const customShortcuts = new Map();\n const shortcutIds = tree.findCustomShortcutIds();\n\n for (const id of shortcutIds) {\n const shortcut = await this.customShortcutsService?.findById(id);\n if (shortcut) {\n const shortcutBlocks = this.mergeBlocks(\n shortcut.shortcut?.content ?? [],\n );\n customShortcuts.set(id, shortcutBlocks);\n }\n }\n\n return customShortcuts;\n }\n\n /**\n * Pre-resolves all usernames found in the description tree.\n */\n private async resolveUsernamesFromTree(\n tree: DescriptionNodeTree,\n instance: Website,\n ): Promise> {\n const usernameConversions = new Map();\n const usernames = tree.findUsernames();\n\n for (const username of usernames) {\n const converted =\n (await this.userConvertersService?.convert(instance, username)) ??\n username;\n usernameConversions.set(username, converted);\n }\n\n return usernameConversions;\n }\n\n public mergeBlocks(blocks: TipTapNode[]): TipTapNode[] {\n return blocks;\n }\n\n /**\n * Recursively checks if TipTap nodes contain a specific inline content type.\n */\n private hasInlineContentType(blocks: TipTapNode[], type: string): boolean {\n for (const block of blocks) {\n if (block?.type === type) return true;\n if (Array.isArray(block?.content)) {\n if (this.hasInlineContentType(block.content, type)) {\n return true;\n }\n }\n }\n return false;\n }\n}\n" }, { "path": "apps/client-server/src/app/post-parsers/parsers/rating-parser.spec.ts", "content": "import { IWebsiteOptions, SubmissionRating } from '@postybirb/types';\nimport { BaseWebsiteOptions } from '../../websites/models/base-website-options';\nimport { DefaultWebsiteOptions } from '../../websites/models/default-website-options';\nimport { RatingParser } from './rating-parser';\n\ndescribe('RatingParser', () => {\n let parser: RatingParser;\n\n beforeEach(() => {\n parser = new RatingParser();\n });\n\n it('should parse rating', () => {\n const options: IWebsiteOptions = {\n data: {\n rating: SubmissionRating.ADULT,\n },\n } as IWebsiteOptions;\n const defaultOptions: IWebsiteOptions = {\n data: {\n rating: SubmissionRating.GENERAL,\n },\n } as IWebsiteOptions;\n expect(\n parser.parse(\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(options.data),\n ),\n ).toEqual(SubmissionRating.ADULT);\n });\n\n it('should parse default rating', () => {\n const options: IWebsiteOptions = {\n data: {},\n } as IWebsiteOptions;\n const defaultOptions: IWebsiteOptions = {\n data: {\n rating: SubmissionRating.GENERAL,\n },\n } as IWebsiteOptions;\n expect(\n parser.parse(\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(options.data),\n ),\n ).toEqual(SubmissionRating.GENERAL);\n });\n\n it('should throw on parsing no rating', () => {\n expect(() =>\n parser.parse(\n { rating: null } as unknown as DefaultWebsiteOptions,\n { rating: null } as unknown as BaseWebsiteOptions,\n ),\n ).toThrow(Error);\n });\n});\n" }, { "path": "apps/client-server/src/app/post-parsers/parsers/rating-parser.ts", "content": "import { SubmissionRating } from '@postybirb/types';\nimport { BaseWebsiteOptions } from '../../websites/models/base-website-options';\nimport { DefaultWebsiteOptions } from '../../websites/models/default-website-options';\n\nexport class RatingParser {\n public parse(\n defaultOptions: DefaultWebsiteOptions,\n websiteOptions: BaseWebsiteOptions,\n ): SubmissionRating {\n if (websiteOptions.rating) {\n return websiteOptions.rating;\n }\n\n if (defaultOptions.rating) {\n return defaultOptions.rating;\n }\n\n throw new Error('No rating found');\n }\n}\n" }, { "path": "apps/client-server/src/app/post-parsers/parsers/tag-parser.service.spec.ts", "content": "/* eslint-disable max-classes-per-file */\nimport { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport { TagField } from '@postybirb/form-builder';\nimport { IWebsiteOptions, TagValue } from '@postybirb/types';\nimport { TagConvertersService } from '../../tag-converters/tag-converters.service';\nimport { BaseWebsiteOptions } from '../../websites/models/base-website-options';\nimport { DefaultWebsiteOptions } from '../../websites/models/default-website-options';\nimport { UnknownWebsite } from '../../websites/website';\nimport { TagParserService } from './tag-parser.service';\n\ndescribe('TagParserService', () => {\n let module: TestingModule;\n let service: TagParserService;\n\n beforeEach(async () => {\n clearDatabase();\n module = await Test.createTestingModule({\n providers: [\n TagParserService,\n {\n provide: TagConvertersService,\n useValue: {\n convert: (_, tags: string[]) => tags,\n },\n },\n ],\n }).compile();\n service = module.get(TagParserService);\n });\n\n afterAll(async () => {\n await module.close();\n });\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n\n it('should parse tags', async () => {\n const instance = {};\n const defaultOptions: IWebsiteOptions = {\n data: {\n tags: {\n tags: ['default'],\n },\n },\n } as IWebsiteOptions;\n const websiteOptions: IWebsiteOptions = {\n data: {\n tags: {\n tags: ['website'],\n },\n },\n } as IWebsiteOptions;\n const tags = [\n ...websiteOptions.data.tags.tags,\n ...defaultOptions.data.tags.tags,\n ];\n const result = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(websiteOptions.data),\n );\n\n expect(result).toEqual(tags);\n });\n\n it('should parse tags with default tags override', async () => {\n const instance = {};\n const defaultOptions: IWebsiteOptions = {\n data: {\n tags: {\n tags: ['default'],\n },\n },\n } as IWebsiteOptions;\n const websiteOptions: IWebsiteOptions = {\n data: {\n tags: {\n tags: ['website'],\n overrideDefault: true,\n },\n },\n } as IWebsiteOptions;\n const { tags } = websiteOptions.data.tags;\n\n const result = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(websiteOptions.data),\n );\n\n expect(result).toEqual(tags);\n });\n\n it('should parse tags with no website options', async () => {\n const instance = {};\n const defaultOptions: IWebsiteOptions = {\n data: {\n tags: {\n tags: ['default'],\n },\n },\n } as IWebsiteOptions;\n const websiteOptions: IWebsiteOptions = {\n data: {},\n } as IWebsiteOptions;\n const { tags } = defaultOptions.data.tags;\n\n const result = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(websiteOptions.data),\n );\n\n expect(result).toEqual(tags);\n });\n\n it('should parse tags with no website options and no default tags', async () => {\n const instance = {};\n const defaultOptions: IWebsiteOptions = {\n data: {},\n } as IWebsiteOptions;\n const websiteOptions: IWebsiteOptions = {\n data: {},\n } as IWebsiteOptions;\n const tags = [];\n\n const result = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(websiteOptions.data),\n );\n\n expect(result).toEqual(tags);\n });\n\n it('should parse tags with no tag support and return empty', async () => {\n const instance = {};\n const defaultOptions: IWebsiteOptions = {\n data: {\n tags: {\n tags: ['default'],\n },\n },\n } as IWebsiteOptions;\n const websiteOptions: IWebsiteOptions = {\n data: {\n tags: {\n tags: ['website'],\n },\n },\n } as IWebsiteOptions;\n const tags = [];\n class TagOptions extends BaseWebsiteOptions {\n @TagField({ hidden: true })\n tags: TagValue;\n }\n const result = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new TagOptions(websiteOptions.data),\n );\n\n expect(result).toEqual(tags);\n });\n\n it('should parse tags with custom instance tag parser', async () => {\n const instance = {};\n class TagOptions extends BaseWebsiteOptions {\n @TagField({})\n tags: TagValue;\n\n protected processTag(tag: string): string {\n return tag.toUpperCase();\n }\n }\n const defaultOptions: IWebsiteOptions = {\n data: {\n tags: {\n tags: ['default'],\n },\n },\n } as IWebsiteOptions;\n const websiteOptions: IWebsiteOptions = {\n data: {\n tags: {\n tags: ['website'],\n },\n },\n } as IWebsiteOptions;\n const tags = [\n ...websiteOptions.data.tags.tags,\n ...defaultOptions.data.tags.tags,\n ].map((tag) => tag.toUpperCase());\n\n const result = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new TagOptions(websiteOptions.data),\n );\n\n expect(result).toEqual(tags);\n });\n\n it('should truncate tags to maxTags', async () => {\n const instance = {};\n const defaultOptions: IWebsiteOptions = {\n data: {\n tags: {\n tags: ['default', 'default2'],\n },\n },\n } as IWebsiteOptions;\n const websiteOptions: IWebsiteOptions = {\n data: {\n tags: {\n tags: ['website', 'website2'],\n },\n },\n } as IWebsiteOptions;\n const tags = [websiteOptions.data.tags.tags[0]];\n class TagOptions extends BaseWebsiteOptions {\n @TagField({ maxTags: 1 })\n tags: TagValue;\n }\n const result = await service.parse(\n instance as unknown as UnknownWebsite,\n new DefaultWebsiteOptions(defaultOptions.data),\n new TagOptions(websiteOptions.data),\n );\n\n expect(result).toEqual(tags);\n });\n});\n" }, { "path": "apps/client-server/src/app/post-parsers/parsers/tag-parser.service.ts", "content": "import { Injectable } from '@nestjs/common';\nimport { TagConvertersService } from '../../tag-converters/tag-converters.service';\nimport { BaseWebsiteOptions } from '../../websites/models/base-website-options';\nimport { DefaultWebsiteOptions } from '../../websites/models/default-website-options';\nimport { Website } from '../../websites/website';\n\n@Injectable()\nexport class TagParserService {\n constructor(private readonly tagConvertersService: TagConvertersService) {}\n\n public async parse(\n instance: Website,\n defaultOptions: DefaultWebsiteOptions,\n websiteOptions: BaseWebsiteOptions,\n ): Promise {\n const mergedOptions = websiteOptions.mergeDefaults(defaultOptions);\n return mergedOptions.getProcessedTags((tag) =>\n this.tagConvertersService.convert(instance, tag),\n );\n }\n}\n" }, { "path": "apps/client-server/src/app/post-parsers/parsers/title-parser.spec.ts", "content": "/* eslint-disable max-classes-per-file */\nimport { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport { TitleField } from '@postybirb/form-builder';\nimport { IWebsiteOptions } from '@postybirb/types';\nimport { FormGeneratorService } from '../../form-generator/form-generator.service';\nimport { BaseWebsiteOptions } from '../../websites/models/base-website-options';\nimport { DefaultWebsiteOptions } from '../../websites/models/default-website-options';\nimport { TitleParser } from './title-parser';\n\ndescribe('TitleParserService', () => {\n let module: TestingModule;\n let service: TitleParser;\n\n beforeEach(async () => {\n clearDatabase();\n module = await Test.createTestingModule({\n providers: [\n TitleParser,\n {\n provide: FormGeneratorService,\n useValue: {\n getDefaultForm: jest.fn(),\n generateForm: jest.fn(),\n },\n },\n ],\n }).compile();\n service = module.get(TitleParser);\n });\n\n afterAll(async () => {\n await module.close();\n });\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n\n it('should parse title', async () => {\n const defaultOptions: IWebsiteOptions = {\n id: 'default',\n data: {\n title: 'default',\n },\n } as IWebsiteOptions;\n const websiteOptions: IWebsiteOptions = {\n id: 'website',\n data: {\n title: 'website',\n },\n } as IWebsiteOptions;\n\n const title = await service.parse(\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(websiteOptions.data),\n );\n\n expect(title).toBe('website');\n });\n\n it('should parse title with no website options', async () => {\n class TestWebsiteOptions extends BaseWebsiteOptions {\n @TitleField({ maxLength: 5 })\n public title: string;\n }\n\n class TestDefaultWebsiteOptions extends DefaultWebsiteOptions {\n @TitleField({ maxLength: 10 })\n public title: string;\n }\n\n const defaultOptions: IWebsiteOptions = {\n id: 'default',\n data: {\n title: 'default',\n },\n } as IWebsiteOptions;\n const websiteOptions: IWebsiteOptions = {\n id: 'website',\n data: {},\n } as IWebsiteOptions;\n\n const title = await service.parse(\n new TestDefaultWebsiteOptions(defaultOptions.data),\n new TestWebsiteOptions(websiteOptions.data),\n );\n\n // Title should be truncated\n expect(title).toBe('defau');\n });\n\n it('should parse title and use default form if website form is not available', async () => {\n const defaultOptions: IWebsiteOptions = {\n id: 'default',\n data: {\n title: 'default',\n },\n } as IWebsiteOptions;\n const websiteOptions: IWebsiteOptions = {\n id: 'website',\n data: {},\n } as IWebsiteOptions;\n\n const title = await service.parse(\n new DefaultWebsiteOptions(defaultOptions.data),\n new BaseWebsiteOptions(websiteOptions.data),\n );\n\n expect(title).toBe('default');\n });\n});\n" }, { "path": "apps/client-server/src/app/post-parsers/parsers/title-parser.ts", "content": "import { Injectable } from '@nestjs/common';\nimport { BaseWebsiteOptions } from '../../websites/models/base-website-options';\nimport { DefaultWebsiteOptions } from '../../websites/models/default-website-options';\n\n@Injectable()\nexport class TitleParser {\n public async parse(\n defaultOptions: DefaultWebsiteOptions,\n websiteOptions: BaseWebsiteOptions,\n ): Promise {\n const defaultTitleForm = defaultOptions.getFormFieldFor('title');\n const websiteTitleForm = websiteOptions.getFormFieldFor('title');\n const merged = websiteOptions.mergeDefaults(defaultOptions);\n\n const title = merged.title ?? '';\n const field = websiteTitleForm ?? defaultTitleForm;\n const maxLength = field?.maxLength ?? Infinity;\n\n return title.trim().slice(0, maxLength);\n }\n}\n" }, { "path": "apps/client-server/src/app/post-parsers/post-parsers.module.ts", "content": "import { Module, forwardRef } from '@nestjs/common';\nimport { CustomShortcutsModule } from '../custom-shortcuts/custom-shortcuts.module';\nimport { FormGeneratorModule } from '../form-generator/form-generator.module';\nimport { SettingsModule } from '../settings/settings.module';\nimport { TagConvertersModule } from '../tag-converters/tag-converters.module';\nimport { UserConvertersModule } from '../user-converters/user-converters.module';\nimport { WebsiteImplProvider } from '../websites/implementations/provider';\nimport { DescriptionParserService } from './parsers/description-parser.service';\nimport { TagParserService } from './parsers/tag-parser.service';\nimport { PostParsersService } from './post-parsers.service';\n\n@Module({\n imports: [\n TagConvertersModule,\n UserConvertersModule,\n FormGeneratorModule,\n SettingsModule,\n forwardRef(() => CustomShortcutsModule),\n ],\n providers: [\n PostParsersService,\n TagParserService,\n WebsiteImplProvider,\n DescriptionParserService,\n ],\n exports: [PostParsersService, TagParserService, DescriptionParserService],\n})\nexport class PostParsersModule {}\n" }, { "path": "apps/client-server/src/app/post-parsers/post-parsers.service.ts", "content": "import { Injectable } from '@nestjs/common';\nimport {\n ISubmission,\n IWebsiteFormFields,\n IWebsiteOptions,\n PostData,\n} from '@postybirb/types';\nimport { DefaultWebsiteOptions } from '../websites/models/default-website-options';\nimport { UnknownWebsite } from '../websites/website';\nimport { ContentWarningParser } from './parsers/content-warning-parser';\nimport { DescriptionParserService } from './parsers/description-parser.service';\nimport { RatingParser } from './parsers/rating-parser';\nimport { TagParserService } from './parsers/tag-parser.service';\nimport { TitleParser } from './parsers/title-parser';\n\n@Injectable()\nexport class PostParsersService {\n private readonly ratingParser: RatingParser = new RatingParser();\n\n private readonly titleParser: TitleParser = new TitleParser();\n\n private readonly contentWarningParser: ContentWarningParser =\n new ContentWarningParser();\n\n constructor(\n private readonly tagParser: TagParserService,\n private readonly descriptionParser: DescriptionParserService,\n ) {}\n\n public async parse(\n submission: ISubmission,\n instance: UnknownWebsite,\n websiteOptions: IWebsiteOptions,\n ): Promise> {\n const defaultOptions: IWebsiteOptions = submission.options.find(\n (o) => o.isDefault,\n );\n const defaultOpts = Object.assign(new DefaultWebsiteOptions(), {\n ...defaultOptions.data,\n });\n const websiteOpts = Object.assign(instance.getModelFor(submission.type), {\n ...websiteOptions.data,\n });\n const tags = await this.tagParser.parse(instance, defaultOpts, websiteOpts);\n const title = await this.titleParser.parse(defaultOpts, websiteOpts);\n const contentWarning = await this.contentWarningParser.parse(\n defaultOpts,\n websiteOpts,\n );\n\n return {\n submission,\n options: {\n ...defaultOptions.data,\n ...websiteOptions.data,\n tags,\n description: await this.descriptionParser.parse(\n instance,\n defaultOpts,\n websiteOpts,\n tags,\n title,\n ),\n title,\n contentWarning,\n rating: this.ratingParser.parse(defaultOpts, websiteOpts),\n },\n };\n }\n}\n" }, { "path": "apps/client-server/src/app/remote/models/update-cookies-remote.dto.ts", "content": "import { UpdateCookiesRemote } from '@postybirb/types';\nimport { IsString } from 'class-validator';\n\nexport class UpdateCookiesRemoteDto implements UpdateCookiesRemote {\n @IsString()\n accountId: string;\n\n @IsString()\n cookies: string;\n}\n" }, { "path": "apps/client-server/src/app/remote/remote.controller.ts", "content": "import { Body, Controller, Get, Param, Post } from '@nestjs/common';\nimport { UpdateCookiesRemoteDto } from './models/update-cookies-remote.dto';\nimport { RemoteService } from './remote.service';\n\n@Controller('remote')\nexport class RemoteController {\n constructor(private readonly remoteService: RemoteService) {}\n\n @Get('ping/:password')\n ping(@Param('password') password: string) {\n return this.remoteService.validate(password);\n }\n\n @Post('set-cookies')\n setCookies(@Body() updateCookies: UpdateCookiesRemoteDto) {\n return this.remoteService.setCookies(updateCookies);\n }\n}\n" }, { "path": "apps/client-server/src/app/remote/remote.middleware.ts", "content": "import {\n Injectable,\n NestMiddleware,\n UnauthorizedException,\n} from '@nestjs/common';\nimport { NextFunction, Request, Response } from 'express';\nimport { RemoteService } from './remote.service';\n\n@Injectable()\nexport class RemotePasswordMiddleware implements NestMiddleware {\n constructor(private readonly remoteService: RemoteService) {}\n\n async use(req: Request, res: Response, next: NextFunction) {\n try {\n if (req.baseUrl.startsWith('/api/file/')) {\n // Skip authentication for file API routes\n // This is mostly just to avoid nuisance password injection into query params\n next();\n return;\n }\n\n if (req.baseUrl.startsWith('/api/remote/ping')) {\n // Skip authentication for ping API routes to let user check the password\n next();\n return;\n }\n\n const remotePassword = req.headers['x-remote-password'] as string;\n\n if (!remotePassword) {\n throw new UnauthorizedException('No remote password provided');\n }\n\n const isValid = await this.remoteService.validate(remotePassword);\n if (!isValid) {\n throw new UnauthorizedException('Invalid remote password');\n }\n\n next(); // Proceed to the next middleware or route handler\n } catch (error) {\n next(error); // Pass the error to the global error handler\n }\n }\n}\n" }, { "path": "apps/client-server/src/app/remote/remote.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { SettingsModule } from '../settings/settings.module';\nimport { RemoteController } from './remote.controller';\nimport { RemoteService } from './remote.service';\n\n@Module({\n imports: [SettingsModule],\n controllers: [RemoteController],\n providers: [RemoteService],\n exports: [RemoteService],\n})\nexport class RemoteModule {}\n" }, { "path": "apps/client-server/src/app/remote/remote.service.ts", "content": "import { Injectable, UnauthorizedException } from '@nestjs/common';\nimport { Logger } from '@postybirb/logger';\nimport { getRemoteConfig } from '@postybirb/utils/electron';\nimport { session } from 'electron';\nimport { UpdateCookiesRemoteDto } from './models/update-cookies-remote.dto';\n\n@Injectable()\nexport class RemoteService {\n protected readonly logger = Logger(this.constructor.name);\n\n async validate(password: string): Promise {\n const remoteConfig = await getRemoteConfig();\n // if (!remoteConfig.enabled) {\n // this.logger.error('Remote access is not enabled');\n // throw new UnauthorizedException('Remote access is not enabled');\n // }\n\n if (remoteConfig.password !== password) {\n this.logger.error('Invalid remote access password');\n throw new UnauthorizedException('Invalid remote access password');\n }\n\n return true;\n }\n\n /**\n * Set cookies for the specified account ID.\n * To share cookies with the remote host, this method should be called.\n *\n * @param {UpdateCookiesRemoteDto} updateCookies\n */\n async setCookies(updateCookies: UpdateCookiesRemoteDto) {\n this.logger\n .withMetadata({ accountId: updateCookies.accountId })\n .info('Updating cookies from remote client');\n\n const clientCookies = Buffer.from(updateCookies.cookies, 'base64').toString(\n 'utf-8',\n );\n const cookies = JSON.parse(clientCookies);\n\n if (!Array.isArray(cookies)) {\n this.logger.error('Invalid cookies format received from remote client');\n throw new Error('Invalid cookies format received from remote client');\n }\n if (cookies.length === 0) {\n this.logger.warn('No cookies provided for account, skipping update');\n return;\n }\n const accountSession = session.fromPartition(\n `persist:${updateCookies.accountId}`,\n );\n await accountSession.clearStorageData();\n await Promise.all(\n cookies.map((cookie) =>\n accountSession.cookies.set(this.convertCookie(cookie)),\n ),\n );\n }\n\n private convertCookie(cookie: Electron.Cookie): Electron.CookiesSetDetails {\n const url = `${cookie.secure ? 'https' : 'http'}://${cookie.domain}${cookie.path || ''}`;\n const details: Electron.CookiesSetDetails = {\n domain: `.${cookie.domain}`.replace('..', '.'),\n httpOnly: cookie.httpOnly || false,\n name: cookie.name,\n secure: cookie.secure || false,\n url: url.replace('://.', '://'),\n value: cookie.value,\n };\n\n if (cookie.expirationDate) {\n details.expirationDate = cookie.expirationDate;\n }\n\n return details;\n }\n}\n" }, { "path": "apps/client-server/src/app/security-and-authentication/ssl.ts", "content": "import { Logger } from '@postybirb/logger';\nimport { app } from 'electron';\nimport { mkdir, readFile, stat, writeFile } from 'fs/promises';\nimport forge from 'node-forge';\nimport { join } from 'path';\n\nexport class SSL {\n private static cachedCerts?: { key: string; cert: string };\n\n static async getOrCreateSSL(): Promise<{ key: string; cert: string }> {\n // Return cached certs if available\n if (this.cachedCerts) {\n return this.cachedCerts;\n }\n\n const logger = Logger().withContext({ name: 'SSL' });\n const path = join(app.getPath('userData'), 'auth');\n const keyPath = join(path, 'key.pem');\n const certPath = join(path, 'cert.pem');\n\n // Check if certificates already exist\n let exists = false;\n try {\n await stat(certPath);\n exists = true;\n } catch {\n try {\n await mkdir(path, { recursive: true });\n } catch (err) {\n if (err.code !== 'EEXIST') {\n logger.error(err);\n }\n }\n }\n\n if (exists) {\n const certs = {\n key: (await readFile(keyPath)).toString(),\n cert: (await readFile(certPath)).toString(),\n };\n this.cachedCerts = certs;\n return certs;\n }\n\n logger.trace('Creating SSL certs...');\n const { pki } = forge;\n\n // Generate RSA key pair - will use native crypto if available\n const keys = pki.rsa.generateKeyPair(2048);\n const cert = pki.createCertificate();\n\n cert.publicKey = keys.publicKey;\n cert.serialNumber = '01';\n cert.validity.notBefore = new Date();\n cert.validity.notAfter = new Date();\n cert.validity.notAfter.setFullYear(\n cert.validity.notBefore.getFullYear() + 99,\n );\n\n const attrs = [\n { name: 'commonName', value: 'postybirb.com' },\n { name: 'countryName', value: 'US' },\n { shortName: 'ST', value: 'Virginia' },\n { name: 'localityName', value: 'Arlington' },\n { name: 'organizationName', value: 'PostyBirb' },\n { shortName: 'OU', value: 'PostyBirb' },\n ];\n cert.setSubject(attrs);\n cert.setIssuer(attrs);\n cert.sign(keys.privateKey);\n\n const pkey = pki.privateKeyToPem(keys.privateKey);\n const pcert = pki.certificateToPem(cert);\n\n await Promise.all([writeFile(keyPath, pkey), writeFile(certPath, pcert)]);\n\n logger.info('SSL Certs created');\n\n const certs = { cert: pcert, key: pkey };\n this.cachedCerts = certs;\n return certs;\n }\n}\n" }, { "path": "apps/client-server/src/app/settings/dtos/update-settings.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { ISettingsOptions, IUpdateSettingsDto } from '@postybirb/types';\nimport { IsObject } from 'class-validator';\n\n/**\n * Settings update request object.\n */\nexport class UpdateSettingsDto implements IUpdateSettingsDto {\n @ApiProperty()\n @IsObject()\n settings: ISettingsOptions;\n}\n" }, { "path": "apps/client-server/src/app/settings/dtos/update-startup-settings.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { StartupOptions } from '@postybirb/utils/electron';\nimport { IsBoolean, IsOptional, IsString } from 'class-validator';\n\nexport class UpdateStartupSettingsDto implements StartupOptions {\n @ApiProperty()\n @IsOptional()\n @IsString()\n appDataPath: string;\n\n @ApiProperty()\n @IsOptional()\n @IsString()\n port: string;\n\n @ApiProperty()\n @IsOptional()\n @IsBoolean()\n startAppOnSystemStartup: boolean;\n\n @ApiProperty()\n @IsOptional()\n @IsBoolean()\n spellchecker: boolean;\n}\n" }, { "path": "apps/client-server/src/app/settings/settings.controller.ts", "content": "import { Body, Controller, Get, Param, Patch } from '@nestjs/common';\nimport { ApiNotFoundResponse, ApiOkResponse, ApiTags } from '@nestjs/swagger';\nimport { EntityId } from '@postybirb/types';\nimport { PostyBirbController } from '../common/controller/postybirb-controller';\nimport { UpdateSettingsDto } from './dtos/update-settings.dto';\nimport { UpdateStartupSettingsDto } from './dtos/update-startup-settings.dto';\nimport { SettingsService } from './settings.service';\n\n/**\n * CRUD operations for settings.\n * @class SettingsController\n */\n@ApiTags('settings')\n@Controller('settings')\nexport class SettingsController extends PostyBirbController<'SettingsSchema'> {\n constructor(readonly service: SettingsService) {\n super(service);\n }\n\n @Patch(':id')\n @ApiOkResponse({ description: 'Update successful.' })\n @ApiNotFoundResponse({ description: 'Settings profile not found.' })\n update(\n @Body() updateSettingsDto: UpdateSettingsDto,\n @Param('id') id: EntityId,\n ) {\n return this.service\n .update(id, updateSettingsDto)\n .then((entity) => entity.toDTO());\n }\n\n @Get('startup')\n getStartupSettings() {\n return this.service.getStartupSettings();\n }\n\n @Patch('startup/system-startup')\n updateStartupSettings(@Body() startupOptions: UpdateStartupSettingsDto) {\n return this.service.updateStartupSettings(startupOptions);\n }\n}\n" }, { "path": "apps/client-server/src/app/settings/settings.events.ts", "content": "import { SETTINGS_UPDATES } from '@postybirb/socket-events';\nimport { SettingsDto } from '@postybirb/types';\nimport { WebsocketEvent } from '../web-socket/models/web-socket-event';\n\nexport type SettingsEventTypes = SettingsUpdateEvent;\n\nclass SettingsUpdateEvent implements WebsocketEvent {\n event: string = SETTINGS_UPDATES;\n\n data: SettingsDto[];\n}\n" }, { "path": "apps/client-server/src/app/settings/settings.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { SettingsController } from './settings.controller';\nimport { SettingsService } from './settings.service';\n\n@Module({\n providers: [SettingsService],\n controllers: [SettingsController],\n exports: [SettingsService],\n})\nexport class SettingsModule {}\n" }, { "path": "apps/client-server/src/app/settings/settings.service.spec.ts", "content": "import { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport { UpdateSettingsDto } from './dtos/update-settings.dto';\nimport { SettingsService } from './settings.service';\n\ndescribe('SettingsService', () => {\n let service: SettingsService;\n let module: TestingModule;\n\n beforeEach(async () => {\n clearDatabase();\n module = await Test.createTestingModule({\n providers: [SettingsService],\n }).compile();\n\n service = module.get(SettingsService);\n await service.onModuleInit();\n });\n\n afterAll(async () => {\n await module.close();\n });\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n\n it('should update entities', async () => {\n const groups = await service.findAll();\n expect(groups).toHaveLength(1);\n\n const record = groups[0];\n\n const updateDto = new UpdateSettingsDto();\n updateDto.settings = {\n desktopNotifications: {\n enabled: true,\n showOnDirectoryWatcherError: true,\n showOnDirectoryWatcherSuccess: true,\n showOnPostError: true,\n showOnPostSuccess: true,\n },\n tagSearchProvider: {\n id: undefined,\n showWikiInHelpOnHover: false,\n },\n hiddenWebsites: ['test'],\n language: 'en',\n allowAd: true,\n queuePaused: false,\n };\n await service.update(record.id, updateDto);\n const updatedRec = await service.findById(record.id);\n expect(updatedRec.settings).toEqual(updateDto.settings);\n });\n});\n" }, { "path": "apps/client-server/src/app/settings/settings.service.ts", "content": "import {\n BadRequestException,\n Injectable,\n OnModuleInit,\n Optional,\n} from '@nestjs/common';\nimport { SETTINGS_UPDATES } from '@postybirb/socket-events';\nimport { EntityId, SettingsConstants } from '@postybirb/types';\nimport {\n StartupOptions,\n getStartupOptions,\n setStartupOptions,\n} from '@postybirb/utils/electron';\nimport { eq } from 'drizzle-orm';\nimport { PostyBirbService } from '../common/service/postybirb-service';\nimport { Settings } from '../drizzle/models';\nimport { WSGateway } from '../web-socket/web-socket-gateway';\nimport { UpdateSettingsDto } from './dtos/update-settings.dto';\n\n@Injectable()\nexport class SettingsService\n extends PostyBirbService<'SettingsSchema'>\n implements OnModuleInit\n{\n constructor(@Optional() webSocket: WSGateway) {\n super('SettingsSchema', webSocket);\n this.repository.subscribe('SettingsSchema', () => this.emit());\n }\n\n /**\n * Initializes default settings if required.\n * Also updates existing settings with any new default fields that might be missing.\n * Heavy merge operations are deferred to avoid blocking application startup.\n */\n async onModuleInit() {\n const defaultSettingsCount = await this.repository.count(\n eq(this.schema.profile, SettingsConstants.DEFAULT_PROFILE_NAME),\n );\n\n if (!defaultSettingsCount) {\n this.createDefaultSettings();\n } else {\n // Defer the settings merge check to avoid blocking startup\n setImmediate(async () => {\n // Get existing default settings\n const existingSettings = await this.getDefaultSettings();\n if (existingSettings) {\n // Check if there are any missing fields compared to the current default settings\n const currentDefaults = SettingsConstants.DEFAULT_SETTINGS;\n let hasChanges = false;\n const updatedSettings = { ...existingSettings.settings };\n\n // Recursively merge missing fields\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n const mergeObjects = (target: any, source: any, path = ''): boolean => {\n let changed = false;\n\n Object.keys(source).forEach((key) => {\n const fullPath = path ? `${path}.${key}` : key;\n\n // If key doesn't exist in target, add it\n if (!(key in target)) {\n // eslint-disable-next-line no-param-reassign\n target[key] = source[key];\n this.logger.debug(`Added missing setting: ${fullPath}`);\n changed = true;\n }\n // If both are objects, recursively merge\n else if (\n typeof source[key] === 'object' &&\n source[key] !== null &&\n typeof target[key] === 'object' &&\n target[key] !== null &&\n !Array.isArray(source[key]) &&\n !Array.isArray(target[key])\n ) {\n const nestedChanged = mergeObjects(\n target[key],\n source[key],\n fullPath,\n );\n if (nestedChanged) changed = true;\n }\n });\n\n return changed;\n };\n\n hasChanges = mergeObjects(updatedSettings, currentDefaults);\n\n // Update database if there were changes\n if (hasChanges) {\n this.logger.debug('Updating default settings with missing fields');\n await this.repository.update(existingSettings.id, {\n settings: updatedSettings,\n });\n }\n }\n });\n }\n }\n\n // Not sure if we'll ever need this\n // eslint-disable-next-line @typescript-eslint/no-unused-vars\n create(createDto: unknown): Promise {\n throw new Error('Method not implemented.');\n }\n\n /**\n * Creates the default settings record.\n */\n private createDefaultSettings() {\n this.repository\n .insert({\n profile: SettingsConstants.DEFAULT_PROFILE_NAME,\n settings: SettingsConstants.DEFAULT_SETTINGS,\n })\n .then((entity) => {\n this.logger.withMetadata(entity).debug('Default settings created');\n })\n .catch((err: Error) => {\n this.logger.withError(err).error('Unable to create default settings');\n });\n }\n\n /**\n * Emits settings.\n */\n async emit() {\n super.emit({\n event: SETTINGS_UPDATES,\n data: (await this.findAll()).map((entity) => entity.toDTO()),\n });\n }\n\n /**\n * Gets the startup settings.\n */\n public getStartupSettings() {\n return getStartupOptions();\n }\n\n /**\n * Gets the default settings.\n */\n public getDefaultSettings() {\n return this.repository.findOne({\n where: (setting, { eq: equals }) =>\n equals(setting.profile, SettingsConstants.DEFAULT_PROFILE_NAME),\n });\n }\n\n /**\n * Updates app startup settings.\n */\n public updateStartupSettings(startUpOptions: Partial) {\n if (startUpOptions.appDataPath) {\n // eslint-disable-next-line no-param-reassign\n startUpOptions.appDataPath = startUpOptions.appDataPath.trim();\n }\n\n if (startUpOptions.port) {\n // eslint-disable-next-line no-param-reassign\n startUpOptions.port = startUpOptions.port.trim();\n const port = parseInt(startUpOptions.port, 10);\n if (Number.isNaN(port) || port < 1024 || port > 65535) {\n throw new BadRequestException('Invalid port');\n }\n }\n\n setStartupOptions({ ...startUpOptions });\n }\n\n /**\n * Updates settings.\n *\n * @param {string} id\n * @param {UpdateSettingsDto} updateSettingsDto\n * @return {*}\n */\n async update(id: EntityId, updateSettingsDto: UpdateSettingsDto) {\n this.logger\n .withMetadata(updateSettingsDto)\n .info(`Updating Settings '${id}'`);\n\n return this.repository.update(id, updateSettingsDto);\n }\n\n /**\n * Tests remote connection to a PostyBirb host.\n *\n * @param {string} hostUrl\n * @param {string} password\n * @return {Promise<{ success: boolean; message: string }>\n */\n async testRemoteConnection(\n hostUrl: string,\n password: string,\n ): Promise<{ success: boolean; message: string }> {\n try {\n if (!hostUrl || !password) {\n return {\n success: false,\n message: 'Host URL and password are required',\n };\n }\n\n // Clean up the URL\n const cleanUrl = hostUrl.trim().replace(/\\/$/, '');\n const testUrl = `${cleanUrl}/api/remote/ping/${encodeURIComponent(password)}`;\n\n this.logger.debug(`Testing remote connection to: ${cleanUrl}`);\n\n const response = await fetch(testUrl, {\n method: 'GET',\n headers: {\n 'Content-Type': 'application/json',\n },\n // Set a reasonable timeout\n signal: AbortSignal.timeout(10000), // 10 seconds\n });\n\n if (response.ok) {\n const result = await response.json();\n if (result === true) {\n return {\n success: true,\n message: 'Connection successful! Host is reachable and password is correct.',\n };\n }\n }\n\n // Handle different HTTP status codes\n switch (response.status) {\n case 401:\n return {\n success: false,\n message: 'Authentication failed. Please check your password.',\n };\n case 404:\n return {\n success: false,\n message: 'Host not found. Please check the URL.',\n };\n case 500:\n return {\n success: false,\n message: 'Host server error. The remote host may not be configured properly.',\n };\n default:\n return {\n success: false,\n message: `Connection failed with status ${response.status}`,\n };\n }\n } catch (error) {\n this.logger.withError(error).error('Remote connection test failed');\n\n if (error instanceof TypeError && error.message.includes('fetch')) {\n return {\n success: false,\n message: 'Network error. Please check the host URL and ensure the host is running.',\n };\n }\n\n if (error.name === 'AbortError') {\n return {\n success: false,\n message: 'Connection timeout. The host may be unreachable.',\n };\n }\n\n return {\n success: false,\n message: `Connection test failed: ${error.message}`,\n };\n }\n }\n}\n" }, { "path": "apps/client-server/src/app/submission/dtos/apply-multi-submission.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { IApplyMultiSubmissionDto, SubmissionId } from '@postybirb/types';\nimport { IsArray, IsBoolean, IsNotEmpty, IsString } from 'class-validator';\n\nexport class ApplyMultiSubmissionDto implements IApplyMultiSubmissionDto {\n @ApiProperty()\n @IsString()\n @IsNotEmpty()\n submissionToApply: SubmissionId;\n\n @ApiProperty()\n @IsArray()\n @IsString({ each: true })\n @IsNotEmpty()\n submissionIds: SubmissionId[];\n\n @ApiProperty()\n @IsBoolean()\n merge: boolean;\n}\n" }, { "path": "apps/client-server/src/app/submission/dtos/apply-template-options.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { SubmissionId } from '@postybirb/types';\nimport { Type } from 'class-transformer';\nimport {\n IsArray,\n IsBoolean,\n IsNotEmpty,\n IsString,\n ValidateNested,\n} from 'class-validator';\nimport { TemplateOptionDto } from './template-option.dto';\n\n/**\n * DTO for applying selected template options to multiple submissions.\n */\nexport class ApplyTemplateOptionsDto {\n @ApiProperty({ description: 'Submission IDs to apply template options to' })\n @IsArray()\n @IsString({ each: true })\n @IsNotEmpty()\n targetSubmissionIds: SubmissionId[];\n\n @ApiProperty({\n description: 'Template options to apply',\n type: [TemplateOptionDto],\n })\n @IsArray()\n @ValidateNested({ each: true })\n @Type(() => TemplateOptionDto)\n options: TemplateOptionDto[];\n\n @ApiProperty({ description: 'Whether to replace title with template title' })\n @IsBoolean()\n overrideTitle: boolean;\n\n @ApiProperty({\n description: 'Whether to replace description with template description',\n })\n @IsBoolean()\n overrideDescription: boolean;\n}\n" }, { "path": "apps/client-server/src/app/submission/dtos/create-submission.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport {\n ICreateSubmissionDefaultOptions,\n ICreateSubmissionDto,\n IFileMetadata,\n SubmissionType,\n} from '@postybirb/types';\nimport { Transform } from 'class-transformer';\nimport {\n IsArray,\n IsBoolean,\n IsEnum,\n IsObject,\n IsOptional,\n IsString,\n} from 'class-validator';\n\n/**\n * Helper to parse JSON strings from FormData.\n * Returns the parsed object or the original value if already an object.\n */\nfunction parseJsonField(value: unknown): T | undefined {\n if (value === undefined || value === null || value === '') {\n return undefined;\n }\n if (typeof value === 'string') {\n try {\n return JSON.parse(value) as T;\n } catch {\n return undefined;\n }\n }\n return value as T;\n}\n\nexport class CreateSubmissionDto implements ICreateSubmissionDto {\n @ApiProperty()\n @IsOptional()\n @IsString()\n name: string;\n\n @ApiProperty({ enum: SubmissionType })\n @IsOptional()\n @IsEnum(SubmissionType)\n type: SubmissionType;\n\n @ApiProperty()\n @IsOptional()\n @IsBoolean()\n @Transform(({ value }) => value === 'true' || value === true)\n isTemplate?: boolean;\n\n @ApiProperty()\n @IsOptional()\n @IsBoolean()\n @Transform(({ value }) => value === 'true' || value === true)\n isMultiSubmission?: boolean;\n\n @ApiProperty({ description: 'Default options to apply to all created submissions' })\n @IsOptional()\n @IsObject()\n @Transform(({ value }) => parseJsonField(value))\n defaultOptions?: ICreateSubmissionDefaultOptions;\n\n @ApiProperty({ description: 'Per-file metadata for batch uploads' })\n @IsOptional()\n @IsArray()\n @Transform(({ value }) => parseJsonField(value))\n fileMetadata?: IFileMetadata[];\n}\n" }, { "path": "apps/client-server/src/app/submission/dtos/reorder-submission-files.dto.ts", "content": "import { IReorderSubmissionFilesDto } from '@postybirb/types';\nimport { IsObject } from 'class-validator';\n\nexport class ReorderSubmissionFilesDto implements IReorderSubmissionFilesDto {\n @IsObject()\n order: Record;\n}\n" }, { "path": "apps/client-server/src/app/submission/dtos/reorder-submission.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { SubmissionId } from '@postybirb/types';\nimport { IsIn, IsString } from 'class-validator';\n\nexport class ReorderSubmissionDto {\n @ApiProperty({ description: 'The ID of the submission to move' })\n @IsString()\n id: SubmissionId;\n\n @ApiProperty({\n description: 'The ID of the target submission to position relative to',\n })\n @IsString()\n targetId: SubmissionId;\n\n @ApiProperty({\n description: 'Whether to place before or after the target',\n enum: ['before', 'after'],\n })\n @IsIn(['before', 'after'])\n position: 'before' | 'after';\n}\n" }, { "path": "apps/client-server/src/app/submission/dtos/template-option.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { AccountId, IWebsiteFormFields } from '@postybirb/types';\nimport { IsNotEmpty, IsObject, IsString } from 'class-validator';\n\n/**\n * Single template option to apply.\n */\nexport class TemplateOptionDto {\n @ApiProperty()\n @IsString()\n @IsNotEmpty()\n accountId: AccountId;\n\n @ApiProperty({ type: Object })\n @IsObject()\n data: IWebsiteFormFields;\n}\n" }, { "path": "apps/client-server/src/app/submission/dtos/update-alt-file.dto.ts", "content": "import { IUpdateAltFileDto } from '@postybirb/types';\nimport { IsString } from 'class-validator';\n\nexport class UpdateAltFileDto implements IUpdateAltFileDto {\n @IsString()\n text: string;\n}\n" }, { "path": "apps/client-server/src/app/submission/dtos/update-submission-template-name.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { IUpdateSubmissionTemplateNameDto } from '@postybirb/types';\nimport { IsString } from 'class-validator';\n\nexport class UpdateSubmissionTemplateNameDto\n implements IUpdateSubmissionTemplateNameDto\n{\n @ApiProperty()\n @IsString()\n name: string;\n}\n" }, { "path": "apps/client-server/src/app/submission/dtos/update-submission.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport {\n ISubmissionMetadata,\n IUpdateSubmissionDto,\n IWebsiteFormFields,\n ScheduleType,\n WebsiteOptionsDto\n} from '@postybirb/types';\nimport {\n IsArray,\n IsBoolean,\n IsEnum,\n IsISO8601,\n IsObject,\n IsOptional,\n IsString,\n} from 'class-validator';\n\nexport class UpdateSubmissionDto implements IUpdateSubmissionDto {\n @ApiProperty()\n @IsOptional()\n @IsBoolean()\n isArchived?: boolean;\n\n @ApiProperty()\n @IsOptional()\n @IsBoolean()\n isScheduled?: boolean;\n\n @ApiProperty()\n @IsOptional()\n @IsString()\n @IsISO8601()\n scheduledFor?: string | null | undefined;\n\n @ApiProperty({ enum: ScheduleType })\n @IsOptional()\n @IsEnum(ScheduleType)\n scheduleType?: ScheduleType;\n\n @ApiProperty()\n @IsOptional()\n @IsString()\n cron?: string | null | undefined;\n\n @ApiProperty()\n @IsOptional()\n @IsArray()\n deletedWebsiteOptions?: string[];\n\n @ApiProperty()\n @IsOptional()\n @IsArray()\n newOrUpdatedOptions?: WebsiteOptionsDto[];\n\n @ApiProperty()\n @IsOptional()\n @IsObject()\n metadata?: ISubmissionMetadata;\n}\n" }, { "path": "apps/client-server/src/app/submission/file-submission.controller.ts", "content": "import {\n BadRequestException,\n Body,\n Controller,\n Delete,\n Get,\n Param,\n Patch,\n Post,\n UploadedFile,\n UploadedFiles,\n UseInterceptors,\n} from '@nestjs/common';\nimport { FileInterceptor, FilesInterceptor } from '@nestjs/platform-express';\nimport {\n ApiBadRequestResponse,\n ApiConsumes,\n ApiOkResponse,\n ApiTags,\n} from '@nestjs/swagger';\nimport {\n EntityId,\n SubmissionFileMetadata,\n SubmissionId,\n} from '@postybirb/types';\nimport { MulterFileInfo } from '../file/models/multer-file-info';\nimport { ReorderSubmissionFilesDto } from './dtos/reorder-submission-files.dto';\nimport { UpdateAltFileDto } from './dtos/update-alt-file.dto';\nimport { FileSubmissionService } from './services/file-submission.service';\nimport { SubmissionService } from './services/submission.service';\n\ntype Target = 'file' | 'thumbnail';\n\n/**\n * Specific REST operations for File Submissions.\n * i.e. as thumbnail changes.\n * @class FileSubmissionController\n */\n@ApiTags('file-submission')\n@Controller('file-submission')\nexport class FileSubmissionController {\n constructor(\n private service: FileSubmissionService,\n private submissionService: SubmissionService,\n ) {}\n\n private findOne(id: SubmissionId) {\n return this.submissionService.findById(id).then((record) => record.toDTO());\n }\n\n @Post('add/:target/:id')\n @ApiConsumes('multipart/form-data')\n @ApiOkResponse({ description: 'File appended.' })\n @ApiBadRequestResponse({ description: 'Bad request made.' })\n @UseInterceptors(FilesInterceptor('files', undefined, { preservePath: true }))\n async appendFile(\n @Param('target') target: Target,\n @Param('id') id: SubmissionId,\n @UploadedFiles() files: MulterFileInfo[],\n ) {\n switch (target) {\n case 'file':\n await Promise.all(\n files.map((file) => this.service.appendFile(id, file)),\n );\n break;\n case 'thumbnail':\n default:\n throw new BadRequestException(`Unsupported add target '${target}'`);\n }\n\n return this.findOne(id);\n }\n\n @Post('replace/:target/:id/:fileId')\n @ApiConsumes('multipart/form-data')\n @ApiOkResponse({ description: 'File replaced.' })\n @ApiBadRequestResponse({ description: 'Bad request made.' })\n @UseInterceptors(FileInterceptor('file', { preservePath: true }))\n async replaceFile(\n @Param('target') target: Target,\n @Param('id') id: SubmissionId,\n @Param('fileId') fileId: EntityId,\n @UploadedFile() file: MulterFileInfo,\n ) {\n switch (target) {\n case 'file':\n await this.service.replaceFile(id, fileId, file);\n break;\n case 'thumbnail':\n await this.service.replaceThumbnail(id, fileId, file);\n break;\n default:\n throw new BadRequestException(`Unsupported replace target '${target}'`);\n }\n\n return this.findOne(id);\n }\n\n @Delete('remove/:target/:id/:fileId')\n @ApiOkResponse({ description: 'File removed.' })\n @ApiBadRequestResponse({ description: 'Bad request made.' })\n async removeFile(\n @Param('target') target: Target,\n @Param('id') id: SubmissionId,\n @Param('fileId') fileId: EntityId,\n ) {\n switch (target) {\n case 'file':\n await this.service.removeFile(id, fileId);\n break;\n case 'thumbnail':\n default:\n throw new BadRequestException(`Unsupported remove target '${target}'`);\n }\n\n return this.findOne(id);\n }\n\n @Get('alt/:id')\n @ApiOkResponse({ description: 'Alt File Text.' })\n async getAltFileText(@Param('id') id: EntityId) {\n return this.service.getAltFileText(id);\n }\n\n @Patch('alt/:id')\n @ApiOkResponse({ description: 'Updated Alt File Text.' })\n async updateAltFileText(\n @Param('id') id: EntityId,\n @Body() update: UpdateAltFileDto,\n ) {\n return this.service.updateAltFileText(id, update);\n }\n\n @Patch('metadata/:id')\n @ApiOkResponse({ description: 'Updated Metadata.' })\n async updateMetadata(\n @Param('id') id: EntityId,\n @Body() update: SubmissionFileMetadata,\n ) {\n return this.service.updateMetadata(id, update);\n }\n\n @Patch('reorder')\n @ApiOkResponse({ description: 'Files reordered.' })\n async reorderFiles(@Body() update: ReorderSubmissionFilesDto) {\n return this.service.reorderFiles(update);\n }\n}\n" }, { "path": "apps/client-server/src/app/submission/services/file-submission.service.ts", "content": "import {\n BadRequestException,\n forwardRef,\n Inject,\n Injectable,\n} from '@nestjs/common';\nimport {\n EntityId,\n FileSubmission,\n FileType,\n isFileSubmission,\n ISubmission,\n SubmissionFileMetadata,\n SubmissionId,\n SubmissionType,\n} from '@postybirb/types';\nimport { getFileType } from '@postybirb/utils/file-type';\nimport { PostyBirbService } from '../../common/service/postybirb-service';\nimport { PostyBirbDatabase } from '../../drizzle/postybirb-database/postybirb-database';\nimport { FileService } from '../../file/file.service';\nimport { MulterFileInfo } from '../../file/models/multer-file-info';\nimport { CreateSubmissionDto } from '../dtos/create-submission.dto';\nimport { ReorderSubmissionFilesDto } from '../dtos/reorder-submission-files.dto';\nimport { UpdateAltFileDto } from '../dtos/update-alt-file.dto';\nimport { ISubmissionService } from './submission-service.interface';\nimport { SubmissionService } from './submission.service';\n\n/**\n * Service that implements logic for manipulating a FileSubmission.\n * All actions perform mutations on the original object.\n *\n * @class FileSubmissionService\n * @implements {ISubmissionService}\n */\n@Injectable()\nexport class FileSubmissionService\n extends PostyBirbService<'SubmissionSchema'>\n implements ISubmissionService\n{\n constructor(\n private readonly fileService: FileService,\n @Inject(forwardRef(() => SubmissionService))\n private readonly submissionService: SubmissionService,\n ) {\n super(\n new PostyBirbDatabase('SubmissionSchema', {\n files: true,\n }),\n );\n }\n\n async populate(\n submission: FileSubmission,\n createSubmissionDto: CreateSubmissionDto,\n file: MulterFileInfo,\n ): Promise {\n // eslint-disable-next-line no-param-reassign\n submission.metadata = {\n ...submission.metadata,\n };\n\n await this.appendFile(submission, file);\n }\n\n private guardIsFileSubmission(submission: ISubmission) {\n if (!isFileSubmission(submission)) {\n throw new BadRequestException(\n `Submission '${(submission as ISubmission).id}' is not a ${SubmissionType.FILE} submission.`,\n );\n }\n\n if (submission.metadata.template) {\n throw new BadRequestException(\n `Submission '${submission.id}' is a template and cannot have files.`,\n );\n }\n }\n\n /**\n * Guards against mixing different file types in the same submission.\n * For example, prevents adding an IMAGE file to a submission that already contains a TEXT (PDF) file.\n *\n * @param {FileSubmission} submission - The submission to check\n * @param {MulterFileInfo} file - The new file being added\n * @throws {BadRequestException} if file types are incompatible\n */\n private guardFileTypeCompatibility(\n submission: FileSubmission,\n file: MulterFileInfo,\n ) {\n if (!submission.files || submission.files.length === 0) {\n return; // No existing files, any type is allowed\n }\n\n const newFileType = getFileType(file.originalname);\n const existingFileType = getFileType(submission.files[0].fileName);\n\n if (newFileType !== existingFileType) {\n const fileTypeLabels: Record = {\n [FileType.IMAGE]: 'IMAGE',\n [FileType.VIDEO]: 'VIDEO',\n [FileType.AUDIO]: 'AUDIO',\n [FileType.TEXT]: 'TEXT',\n [FileType.UNKNOWN]: 'UNKNOWN',\n };\n throw new BadRequestException(\n `Cannot add ${fileTypeLabels[newFileType]} file to a submission containing ${fileTypeLabels[existingFileType]} files. All files in a submission must be of the same type.`,\n );\n }\n }\n\n /**\n * Adds a file to a submission.\n *\n * @param {string} id\n * @param {MulterFileInfo} file\n */\n async appendFile(id: EntityId | FileSubmission, file: MulterFileInfo) {\n const submission = (\n typeof id === 'string'\n ? await this.repository.findById(id, {\n failOnMissing: true,\n })\n : id\n ) as FileSubmission;\n\n this.guardIsFileSubmission(submission);\n this.guardFileTypeCompatibility(submission, file);\n\n const createdFile = await this.fileService.create(file, submission);\n this.logger\n .withMetadata(submission)\n .info(`Created file ${createdFile.id} = ${submission.id}`);\n\n await this.repository.update(submission.id, {\n metadata: submission.metadata,\n });\n\n return submission;\n }\n\n async replaceFile(id: EntityId, fileId: EntityId, file: MulterFileInfo) {\n const submission = (await this.repository.findById(\n id,\n )) as unknown as FileSubmission;\n this.guardIsFileSubmission(submission);\n\n await this.fileService.update(file, fileId, false);\n }\n\n /**\n * Replaces a thumbnail file.\n *\n * @param {SubmissionId} id\n * @param {EntityId} fileId\n * @param {MulterFileInfo} file\n */\n async replaceThumbnail(\n id: SubmissionId,\n fileId: EntityId,\n file: MulterFileInfo,\n ) {\n const submission = (await this.repository.findById(\n id,\n )) as unknown as FileSubmission;\n this.guardIsFileSubmission(submission);\n\n await this.fileService.update(file, fileId, true);\n }\n\n /**\n * Removes a file of thumbnail that matches file id.\n *\n * @param {SubmissionId} id\n * @param {EntityId} fileId\n */\n async removeFile(id: SubmissionId, fileId: EntityId) {\n const submission = (await this.repository.findById(\n id,\n )) as unknown as FileSubmission;\n this.guardIsFileSubmission(submission);\n\n await this.fileService.remove(fileId);\n await this.repository.update(submission.id, {\n metadata: submission.metadata,\n });\n }\n\n getAltFileText(id: EntityId) {\n return this.fileService.getAltText(id);\n }\n\n updateAltFileText(id: EntityId, update: UpdateAltFileDto) {\n return this.fileService.updateAltText(id, update);\n }\n\n updateMetadata(id: EntityId, update: SubmissionFileMetadata) {\n return this.fileService.updateMetadata(id, update);\n }\n\n reorderFiles(update: ReorderSubmissionFilesDto) {\n return this.fileService.reorderFiles(update);\n }\n}\n" }, { "path": "apps/client-server/src/app/submission/services/message-submission.service.ts", "content": "import { Injectable } from '@nestjs/common';\nimport { MessageSubmission } from '@postybirb/types';\nimport { CreateSubmissionDto } from '../dtos/create-submission.dto';\nimport { ISubmissionService } from './submission-service.interface';\n\n@Injectable()\nexport class MessageSubmissionService\n implements ISubmissionService\n{\n async populate(\n submission: MessageSubmission,\n createSubmissionDto: CreateSubmissionDto,\n ): Promise {\n // Do nothing for now\n }\n}\n" }, { "path": "apps/client-server/src/app/submission/services/submission-service.interface.ts", "content": "import { ISubmission, SubmissionMetadataType } from '@postybirb/types';\nimport { MulterFileInfo } from '../../file/models/multer-file-info';\nimport { CreateSubmissionDto } from '../dtos/create-submission.dto';\n\nexport interface ISubmissionService<\n T extends ISubmission,\n> {\n populate(\n submission: T,\n createSubmissionDto: CreateSubmissionDto,\n file?: MulterFileInfo,\n ): Promise;\n}\n" }, { "path": "apps/client-server/src/app/submission/services/submission.service.spec.ts", "content": "import { BadRequestException, NotFoundException } from '@nestjs/common';\nimport { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport { PostyBirbDirectories, writeSync } from '@postybirb/fs';\nimport {\n FileSubmissionMetadata,\n IWebsiteFormFields,\n ScheduleType,\n SubmissionRating,\n SubmissionType,\n WebsiteOptionsDto,\n} from '@postybirb/types';\nimport { readFileSync } from 'fs';\nimport { join } from 'path';\nimport { AccountModule } from '../../account/account.module';\nimport { AccountService } from '../../account/account.service';\nimport { CreateAccountDto } from '../../account/dtos/create-account.dto';\nimport { FileConverterService } from '../../file-converter/file-converter.service';\nimport { FileService } from '../../file/file.service';\nimport { MulterFileInfo } from '../../file/models/multer-file-info';\nimport { CreateFileService } from '../../file/services/create-file.service';\nimport { UpdateFileService } from '../../file/services/update-file.service';\nimport { FormGeneratorModule } from '../../form-generator/form-generator.module';\nimport { SharpInstanceManager } from '../../image-processing/sharp-instance-manager';\nimport { PostParsersModule } from '../../post-parsers/post-parsers.module';\nimport { UserSpecifiedWebsiteOptionsModule } from '../../user-specified-website-options/user-specified-website-options.module';\nimport { UserSpecifiedWebsiteOptionsService } from '../../user-specified-website-options/user-specified-website-options.service';\nimport { waitUntilPromised } from '../../utils/wait.util';\nimport { ValidationService } from '../../validation/validation.service';\nimport { WebsiteOptionsService } from '../../website-options/website-options.service';\nimport { WebsiteImplProvider } from '../../websites/implementations/provider';\nimport { WebsiteRegistryService } from '../../websites/website-registry.service';\nimport { WebsitesModule } from '../../websites/websites.module';\nimport { CreateSubmissionDto } from '../dtos/create-submission.dto';\nimport { UpdateSubmissionDto } from '../dtos/update-submission.dto';\nimport { FileSubmissionService } from './file-submission.service';\nimport { MessageSubmissionService } from './message-submission.service';\nimport { SubmissionService } from './submission.service';\n\ndescribe('SubmissionService', () => {\n let testFile: Buffer | null = null;\n let service: SubmissionService;\n let websiteOptionsService: WebsiteOptionsService;\n let accountService: AccountService;\n let module: TestingModule;\n\n beforeAll(() => {\n testFile = readFileSync(\n join(__dirname, '../../../test-files/small_image.jpg'),\n );\n });\n\n beforeEach(async () => {\n clearDatabase();\n try {\n module = await Test.createTestingModule({\n imports: [\n AccountModule,\n WebsitesModule,\n UserSpecifiedWebsiteOptionsModule,\n PostParsersModule,\n FormGeneratorModule,\n ],\n providers: [\n SubmissionService,\n CreateFileService,\n UpdateFileService,\n SharpInstanceManager,\n FileService,\n FileSubmissionService,\n MessageSubmissionService,\n AccountService,\n WebsiteRegistryService,\n UserSpecifiedWebsiteOptionsService,\n ValidationService,\n WebsiteOptionsService,\n WebsiteImplProvider,\n FileConverterService,\n ],\n }).compile();\n\n service = module.get(SubmissionService);\n websiteOptionsService = module.get(\n WebsiteOptionsService,\n );\n accountService = module.get(AccountService);\n await accountService.onModuleInit();\n } catch (e) {\n console.error(e);\n }\n });\n\n afterAll(async () => {\n await module.close();\n });\n\n function setup(): string {\n const path = `${PostyBirbDirectories.DATA_DIRECTORY}/${Date.now()}.jpg`;\n writeSync(path, testFile);\n return path;\n }\n\n async function createAccount() {\n const dto = new CreateAccountDto();\n dto.groups = ['test'];\n dto.name = 'test';\n dto.website = 'test';\n\n const record = await accountService.create(dto);\n return record;\n }\n\n function createSubmissionDto(): CreateSubmissionDto {\n const dto = new CreateSubmissionDto();\n dto.name = 'Test';\n dto.type = SubmissionType.MESSAGE;\n return dto;\n }\n\n function createMulterData(path: string): MulterFileInfo {\n return {\n fieldname: 'file',\n originalname: 'small_image.jpg',\n encoding: '',\n mimetype: 'image/jpeg',\n size: testFile.length,\n destination: '',\n filename: 'small_image.jpg',\n path,\n origin: undefined,\n };\n }\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n\n it('should create message entities', async () => {\n const createDto = createSubmissionDto();\n const record = await service.create(createDto);\n\n const records = await service.findAll();\n expect(records).toHaveLength(1);\n expect(records[0].type).toEqual(createDto.type);\n expect(records[0].options).toHaveLength(1);\n expect(record.toDTO()).toEqual({\n id: record.id,\n createdAt: record.createdAt,\n updatedAt: record.updatedAt,\n type: record.type,\n isScheduled: false,\n isTemplate: false,\n isArchived: false,\n isInitialized: true,\n isMultiSubmission: false,\n schedule: {\n scheduleType: ScheduleType.NONE,\n },\n metadata: {},\n files: [],\n order: 1,\n posts: [],\n postQueueRecord: undefined,\n options: [record.options[0].toDTO()],\n validations: [],\n });\n });\n\n it('should delete entity', async () => {\n const createDto = createSubmissionDto();\n const record = await service.create(createDto);\n\n const records = await service.findAll();\n expect(records).toHaveLength(1);\n expect(records[0].type).toEqual(createDto.type);\n\n await service.remove(record.id);\n expect(await service.findAll()).toHaveLength(0);\n });\n\n it('should throw exception on message submission with provided file', async () => {\n const createDto = createSubmissionDto();\n createDto.type = SubmissionType.MESSAGE;\n await expect(\n service.create(createDto, {} as MulterFileInfo),\n ).rejects.toThrow(BadRequestException);\n });\n\n it('should create file entities', async () => {\n const createDto = createSubmissionDto();\n delete createDto.name; // To ensure file name check\n createDto.type = SubmissionType.FILE;\n const path = setup();\n const fileInfo = createMulterData(path);\n\n const record = await service.create(createDto, fileInfo);\n const defaultOptions = record.options[0];\n const file = record.files[0];\n\n const records = await service.findAll();\n expect(records).toHaveLength(1);\n expect(records[0].type).toEqual(createDto.type);\n expect(records[0].options).toHaveLength(1);\n expect(records[0].files).toHaveLength(1);\n expect(record.toDTO()).toEqual({\n id: record.id,\n createdAt: record.createdAt,\n updatedAt: record.updatedAt,\n type: record.type,\n isScheduled: false,\n postQueueRecord: undefined,\n isTemplate: false,\n isMultiSubmission: false,\n isArchived: false,\n isInitialized: true,\n schedule: {\n scheduleType: ScheduleType.NONE,\n },\n metadata: {},\n files: [\n {\n createdAt: file.createdAt,\n primaryFileId: file.primaryFileId,\n fileName: fileInfo.originalname,\n hasThumbnail: true,\n hasCustomThumbnail: false,\n hasAltFile: false,\n hash: file.hash,\n height: 202,\n width: 138,\n id: file.id,\n mimeType: fileInfo.mimetype,\n size: testFile.length,\n submissionId: record.id,\n altFileId: null,\n thumbnailId: file.thumbnailId,\n updatedAt: file.updatedAt,\n metadata: {\n altText: '',\n dimensions: {\n default: {\n height: 202,\n width: 138,\n },\n },\n ignoredWebsites: [],\n sourceUrls: [],\n spoilerText: '',\n },\n order: file.order,\n },\n ],\n posts: [],\n order: 1,\n options: [defaultOptions.toObject()],\n validations: [],\n });\n });\n\n it('should throw on missing file on file submission', async () => {\n const createDto = createSubmissionDto();\n createDto.type = SubmissionType.FILE;\n await expect(service.create(createDto)).rejects.toThrow(\n BadRequestException,\n );\n });\n\n it('should remove entities', async () => {\n const fileService = module.get(FileService);\n const optionsService = module.get(\n WebsiteOptionsService,\n );\n\n const createDto = createSubmissionDto();\n createDto.type = SubmissionType.FILE;\n const path = setup();\n const fileInfo = createMulterData(path);\n\n const record = await service.create(createDto, fileInfo);\n const fileId = record.files[0].id;\n\n expect(await service.findAll()).toHaveLength(1);\n expect(await optionsService.findAll()).toHaveLength(1);\n expect(await fileService.findFile(fileId)).toBeDefined();\n\n await service.remove(record.id);\n expect(await service.findAll()).toHaveLength(0);\n expect(await optionsService.findAll()).toHaveLength(0);\n await expect(fileService.findFile(fileId)).rejects.toThrow(\n NotFoundException,\n );\n });\n\n it('should update entity props', async () => {\n const createDto = createSubmissionDto();\n const record = await service.create(createDto);\n\n const updateDto = new UpdateSubmissionDto();\n updateDto.isScheduled = true;\n updateDto.scheduleType = ScheduleType.RECURRING;\n updateDto.scheduledFor = '*';\n updateDto.metadata = {\n test: 'test',\n } as unknown;\n\n const updatedRecord = await service.update(record.id, updateDto);\n expect(updatedRecord.isScheduled).toEqual(updateDto.isScheduled);\n expect(updatedRecord.schedule.scheduleType).toEqual(updateDto.scheduleType);\n expect(updatedRecord.schedule.scheduledFor).toEqual(updateDto.scheduledFor);\n expect(updatedRecord.metadata).toEqual(updateDto.metadata);\n });\n\n it('should remove entity options', async () => {\n const createDto = createSubmissionDto();\n const record = await service.create(createDto);\n\n const updateDto = new UpdateSubmissionDto();\n updateDto.deletedWebsiteOptions = [record.options[0].id];\n\n const updatedRecord = await service.update(record.id, updateDto);\n expect(updatedRecord.options).toHaveLength(0);\n });\n\n it('should update entity options', async () => {\n const createDto = createSubmissionDto();\n const record = await service.create(createDto);\n\n const updateDto = new UpdateSubmissionDto();\n updateDto.newOrUpdatedOptions = [\n {\n ...record.options[0],\n data: {\n rating: SubmissionRating.GENERAL,\n title: 'Updated',\n },\n } as unknown as WebsiteOptionsDto,\n ];\n\n const updatedRecord = await service.update(record.id, updateDto);\n expect(updatedRecord.options[0].data.title).toEqual('Updated');\n });\n\n it('should serialize entity', async () => {\n const createDto = createSubmissionDto();\n const record = await service.create(createDto);\n\n const serialized = JSON.stringify(record.toDTO());\n expect(serialized).toBeDefined();\n });\n\n it('should reorder entities', async () => {\n const createDto = createSubmissionDto();\n const record1 = await service.create(createDto);\n const record2 = await service.create(createDto);\n const record3 = await service.create(createDto);\n\n // Move record1 after record2 (from position 0 to after position 1)\n await service.reorder(record1.id, record2.id, 'after');\n const records = (await service.findAll()).sort((a, b) => a.order - b.order);\n expect(records[0].id).toEqual(record2.id);\n expect(records[1].id).toEqual(record1.id);\n expect(records[2].id).toEqual(record3.id);\n });\n\n it('should create multi submissions onModuleInit', async () => {\n service.onModuleInit();\n await waitUntilPromised(async () => {\n const records = await service.findAll();\n return records.length === 2;\n }, 50);\n const records = await service.findAll();\n expect(records).toHaveLength(2);\n expect(records[0].isMultiSubmission).toBeTruthy();\n expect(records[1].isMultiSubmission).toBeTruthy();\n });\n\n it('should apply template', async () => {\n const account = await createAccount();\n const createDto = createSubmissionDto();\n const record = await service.create(createDto);\n\n createDto.isTemplate = true;\n createDto.name = 'Template';\n const template = await service.create(createDto);\n await websiteOptionsService.create({\n submissionId: template.id,\n accountId: account.id,\n data: {\n title: 'Template Test',\n rating: SubmissionRating.MATURE,\n },\n });\n\n const updatedTemplate = await service.updateTemplateName(template.id, {\n name: 'Updated',\n });\n\n expect(updatedTemplate.metadata.template.name).toEqual('Updated');\n\n const updatedRecord = await service.applyOverridingTemplate(\n record.id,\n template.id,\n );\n const defaultOptions = updatedRecord.options[0];\n // The default title should not be updated\n expect(defaultOptions.data.title).not.toEqual('Template Test');\n const nonDefault = updatedRecord.options.find((o) => !o.isDefault);\n expect(nonDefault).toBeDefined();\n expect(nonDefault?.data.title).toEqual('Template Test');\n expect(nonDefault?.data.rating).toEqual(SubmissionRating.MATURE);\n });\n\n it('should apply multi submission merge', async () => {\n const account = await createAccount();\n const createDto = createSubmissionDto();\n const record = await service.create(createDto);\n\n createDto.isMultiSubmission = true;\n createDto.name = 'Multi';\n const multi = await service.create(createDto);\n await websiteOptionsService.create({\n submissionId: multi.id,\n accountId: account.id,\n data: {\n title: 'Multi Test',\n rating: SubmissionRating.MATURE,\n },\n });\n\n await service.applyMultiSubmission({\n submissionToApply: multi.id,\n submissionIds: [record.id],\n merge: true,\n });\n\n const updatedRecord = await service.findById(record.id);\n const defaultOptions = updatedRecord.options[0];\n const multiDefaultOptions = multi.options.find((o) => o.isDefault);\n // The default title should not be updated\n expect(defaultOptions.data.title).not.toEqual(\n multiDefaultOptions?.data.title,\n );\n const nonDefault = updatedRecord.options.find((o) => !o.isDefault);\n expect(nonDefault).toBeDefined();\n expect(nonDefault?.data.title).toEqual('Multi Test');\n expect(nonDefault?.data.rating).toEqual(SubmissionRating.MATURE);\n });\n\n it('should apply multi submission without merge', async () => {\n const account = await createAccount();\n const createDto = createSubmissionDto();\n const record = await service.create(createDto);\n\n createDto.isMultiSubmission = true;\n createDto.name = 'Multi';\n const multi = await service.create(createDto);\n await websiteOptionsService.create({\n submissionId: multi.id,\n accountId: account.id,\n data: {\n title: 'Multi Test',\n rating: SubmissionRating.MATURE,\n },\n });\n\n await service.applyMultiSubmission({\n submissionToApply: multi.id,\n submissionIds: [record.id],\n merge: false,\n });\n\n const updatedRecord = await service.findById(record.id);\n const multiSubmission = await service.findById(multi.id);\n expect(updatedRecord.options).toHaveLength(multiSubmission.options.length);\n const defaultOptions = updatedRecord.options.find((o) => o.isDefault);\n const nonDefault = updatedRecord.options.find((o) => !o.isDefault);\n expect(nonDefault).toBeDefined();\n expect(nonDefault.data).toEqual(multiSubmission.options[1].data);\n expect(defaultOptions).toBeDefined();\n expect(defaultOptions.data).toEqual({\n ...multiSubmission.options[0].data,\n title: defaultOptions.data.title,\n });\n });\n\n it('should duplicate submission', async () => {\n const account = await createAccount();\n const createDto = createSubmissionDto();\n createDto.type = SubmissionType.FILE;\n const path = setup();\n const fileInfo = createMulterData(path);\n\n const record = await service.create(createDto, fileInfo);\n await websiteOptionsService.create({\n submissionId: record.id,\n accountId: account.id,\n data: {\n title: 'Duplicate Test',\n rating: SubmissionRating.MATURE,\n },\n });\n\n await service.duplicate(record.id);\n\n const records = await service.findAll();\n const duplicated = records.find((r) => r.id !== record.id);\n expect(duplicated).toBeDefined();\n expect(duplicated?.type).toEqual(record.type);\n expect(duplicated?.options).toHaveLength(2);\n expect(duplicated?.files).toHaveLength(1);\n expect(duplicated.order).toEqual(record.order);\n\n // Check that the metadata references the new file IDs\n const duplicatedFileId = duplicated?.files[0].id;\n const duplicatedMetadata = duplicated?.metadata as FileSubmissionMetadata;\n expect(duplicatedFileId).toBeDefined();\n\n for (const file of duplicated.files) {\n expect(record.files.find((f) => f.id === file.id)).toBeUndefined();\n }\n\n // Check that the original metadata is preserved\n const originalMetadata = record?.metadata as FileSubmissionMetadata;\n for (const file of record.files) {\n expect(duplicated.files.find((f) => f.id === file.id)).toBeUndefined();\n }\n });\n});\n" }, { "path": "apps/client-server/src/app/submission/services/submission.service.ts", "content": "/* eslint-disable no-param-reassign */\nimport {\n BadRequestException,\n forwardRef,\n Inject,\n Injectable,\n NotFoundException,\n OnModuleInit,\n Optional,\n} from '@nestjs/common';\nimport {\n FileBufferSchema,\n Insert,\n SubmissionFileSchema,\n SubmissionSchema,\n WebsiteOptionsSchema,\n} from '@postybirb/database';\nimport { SUBMISSION_UPDATES } from '@postybirb/socket-events';\nimport {\n FileSubmission,\n FileSubmissionMetadata,\n ISubmissionDto,\n ISubmissionMetadata,\n MessageSubmission,\n NULL_ACCOUNT_ID,\n ScheduleType,\n SubmissionId,\n SubmissionMetadataType,\n SubmissionType,\n} from '@postybirb/types';\nimport { IsTestEnvironment } from '@postybirb/utils/electron';\nimport { eq } from 'drizzle-orm';\nimport * as path from 'path';\nimport { PostyBirbService } from '../../common/service/postybirb-service';\nimport { FileBuffer, Submission, WebsiteOptions } from '../../drizzle/models';\nimport { PostyBirbDatabase } from '../../drizzle/postybirb-database/postybirb-database';\nimport { withTransactionContext } from '../../drizzle/transaction-context';\nimport { MulterFileInfo } from '../../file/models/multer-file-info';\nimport { WSGateway } from '../../web-socket/web-socket-gateway';\nimport { WebsiteOptionsService } from '../../website-options/website-options.service';\nimport { ApplyMultiSubmissionDto } from '../dtos/apply-multi-submission.dto';\nimport { ApplyTemplateOptionsDto } from '../dtos/apply-template-options.dto';\nimport { CreateSubmissionDto } from '../dtos/create-submission.dto';\nimport { UpdateSubmissionTemplateNameDto } from '../dtos/update-submission-template-name.dto';\nimport { UpdateSubmissionDto } from '../dtos/update-submission.dto';\nimport { FileSubmissionService } from './file-submission.service';\nimport { MessageSubmissionService } from './message-submission.service';\n\ntype SubmissionEntity = Submission;\n\n/**\n * Service that handles the vast majority of submission management logic.\n * @class SubmissionService\n */\n@Injectable()\nexport class SubmissionService\n extends PostyBirbService<'SubmissionSchema'>\n implements OnModuleInit\n{\n private emitDebounceTimer: ReturnType | null = null;\n\n constructor(\n @Inject(forwardRef(() => WebsiteOptionsService))\n private readonly websiteOptionsService: WebsiteOptionsService,\n @Inject(forwardRef(() => FileSubmissionService))\n private readonly fileSubmissionService: FileSubmissionService,\n private readonly messageSubmissionService: MessageSubmissionService,\n @Optional() webSocket: WSGateway,\n ) {\n super(\n new PostyBirbDatabase('SubmissionSchema', {\n options: {\n with: {\n account: true,\n },\n },\n posts: {\n with: {\n events: {\n account: true,\n },\n },\n },\n postQueueRecord: true,\n files: true,\n }),\n webSocket,\n );\n this.repository.subscribe(\n [\n 'PostRecordSchema',\n 'PostQueueRecordSchema',\n 'SubmissionFileSchema',\n 'FileBufferSchema',\n ],\n () => {\n this.emit();\n },\n );\n\n this.repository.subscribe(['WebsiteOptionsSchema'], (_, action) => {\n if (action === 'delete') {\n this.emit();\n }\n });\n }\n\n async onModuleInit() {\n await this.cleanupUninitializedSubmissions();\n await this.normalizeOrders();\n for (const type of Object.values(SubmissionType)) {\n // eslint-disable-next-line no-await-in-loop\n await this.populateMultiSubmission(type);\n }\n }\n\n /**\n * Normalizes order values to sequential integers on startup.\n * This cleans up fractional values that accumulate from reordering operations.\n */\n private async normalizeOrders() {\n this.logger.info('Normalizing submission orders');\n\n for (const type of [SubmissionType.FILE, SubmissionType.MESSAGE]) {\n const submissions = (await this.repository.findAll())\n .filter((s) => s.type === type && !s.isTemplate && !s.isMultiSubmission)\n .sort((a, b) => a.order - b.order);\n\n for (let i = 0; i < submissions.length; i++) {\n if (submissions[i].order !== i) {\n // eslint-disable-next-line no-await-in-loop\n await this.repository.update(submissions[i].id, { order: i });\n }\n }\n }\n\n this.logger.info('Order normalization complete');\n }\n\n /**\n * Cleans up any submissions that were left in an uninitialized state\n * (e.g., from a crash during creation).\n */\n private async cleanupUninitializedSubmissions() {\n const all = await super.findAll();\n const uninitialized = all.filter((s) => !s.isInitialized);\n if (uninitialized.length > 0) {\n const ids = uninitialized.map((s) => s.id);\n this.logger\n .withMetadata({ submissionIds: ids })\n .info(\n `Cleaning up ${uninitialized.length} uninitialized submission(s) from previous session`,\n );\n await this.repository.deleteById(ids);\n }\n }\n\n /**\n * Emits submissions onto websocket.\n * Debounced by 50ms to avoid rapid consecutive emits.\n * Overrides base class emit to provide submission-specific behavior.\n */\n public async emit() {\n if (IsTestEnvironment()) {\n return;\n }\n\n if (this.emitDebounceTimer) {\n clearTimeout(this.emitDebounceTimer);\n }\n\n this.emitDebounceTimer = setTimeout(() => {\n this.emitDebounceTimer = null;\n this.performEmit();\n }, 50);\n }\n\n private async performEmit() {\n const now = Date.now();\n super.emit({\n event: SUBMISSION_UPDATES,\n data: await this.findAllAsDto(),\n });\n this.logger.info(`Emitted submission updates in ${Date.now() - now}ms`);\n }\n\n public async findAllAsDto(): Promise[]> {\n const all = (await super.findAll()).filter((s) => s.isInitialized);\n\n // Separate archived from non-archived for efficient processing\n const archived = all.filter((s) => s.isArchived);\n const nonArchived = all.filter((s) => !s.isArchived);\n\n // Validate non-archived submissions in parallel batches to avoid overwhelming the system\n const BATCH_SIZE = 10;\n const validatedNonArchived: ISubmissionDto[] = [];\n\n for (let i = 0; i < nonArchived.length; i += BATCH_SIZE) {\n const batch = nonArchived.slice(i, i + BATCH_SIZE);\n const batchResults = await Promise.all(\n batch.map(\n async (s) =>\n ({\n ...s.toDTO(),\n validations:\n await this.websiteOptionsService.validateSubmission(s),\n }) as ISubmissionDto,\n ),\n );\n validatedNonArchived.push(...batchResults);\n }\n\n // Archived submissions don't need validation\n const archivedDtos = archived.map(\n (s) =>\n ({\n ...s.toDTO(),\n validations: [],\n }) as ISubmissionDto,\n );\n\n return [...validatedNonArchived, ...archivedDtos];\n }\n\n /**\n * Returns all initialized submissions.\n * Overrides base class to filter out submissions still being created.\n */\n public async findAll() {\n const all = await super.findAll();\n return all.filter((s) => s.isInitialized);\n }\n\n private async populateMultiSubmission(type: SubmissionType) {\n const existing = await this.repository.findOne({\n where: (submission, { eq: equals, and }) =>\n and(\n eq(submission.type, type),\n equals(submission.isMultiSubmission, true),\n ),\n });\n if (existing) {\n return;\n }\n\n await this.create({ name: type, type, isMultiSubmission: true });\n }\n\n /**\n * Creates a submission.\n *\n * @param {CreateSubmissionDto} createSubmissionDto\n * @param {MulterFileInfo} [file]\n * @return {*} {Promise>}\n */\n async create(\n createSubmissionDto: CreateSubmissionDto,\n file?: MulterFileInfo,\n ): Promise {\n this.logger.withMetadata(createSubmissionDto).info('Creating Submission');\n\n // Templates and multi-submissions are immediately initialized since they don't need file population\n const isImmediatelyInitialized =\n !!createSubmissionDto.isMultiSubmission ||\n !!createSubmissionDto.isTemplate;\n\n let submission = new Submission({\n isScheduled: false,\n isMultiSubmission: !!createSubmissionDto.isMultiSubmission,\n isTemplate: !!createSubmissionDto.isTemplate,\n isInitialized: isImmediatelyInitialized,\n ...createSubmissionDto,\n schedule: {\n scheduledFor: undefined,\n scheduleType: ScheduleType.NONE,\n cron: undefined,\n },\n metadata: {\n template: createSubmissionDto.isTemplate\n ? { name: createSubmissionDto.name.trim() }\n : undefined,\n },\n order: (await this.repository.count()) + 1,\n });\n\n submission = await this.repository.insert(submission);\n\n // Determine the submission name/title\n let name = 'New submission';\n if (createSubmissionDto.name) {\n name = createSubmissionDto.name;\n } else if (file) {\n // Check for per-file title override from fileMetadata\n const fileMetadata = createSubmissionDto.fileMetadata?.find(\n (meta) => meta.filename === file.originalname,\n );\n if (fileMetadata?.title) {\n name = fileMetadata.title;\n } else {\n name = path.parse(file.filename).name;\n }\n }\n\n // Convert defaultOptions from DTO format to IWebsiteFormFields format\n const defaultOptions = createSubmissionDto.defaultOptions\n ? {\n tags: createSubmissionDto.defaultOptions.tags\n ? {\n overrideDefault: false,\n tags: createSubmissionDto.defaultOptions.tags,\n }\n : undefined,\n description: createSubmissionDto.defaultOptions.description,\n rating: createSubmissionDto.defaultOptions.rating,\n }\n : undefined;\n\n try {\n await this.websiteOptionsService.createDefaultSubmissionOptions(\n submission,\n name,\n defaultOptions,\n );\n\n switch (createSubmissionDto.type) {\n case SubmissionType.MESSAGE: {\n if (file) {\n throw new BadRequestException(\n 'A file was provided for SubmissionType Message.',\n );\n }\n\n await this.messageSubmissionService.populate(\n submission as unknown as MessageSubmission,\n createSubmissionDto,\n );\n break;\n }\n\n case SubmissionType.FILE: {\n if (\n createSubmissionDto.isTemplate ||\n createSubmissionDto.isMultiSubmission\n ) {\n // Don't need to populate on a template\n break;\n }\n\n if (!file) {\n throw new BadRequestException(\n 'No file provided for SubmissionType FILE.',\n );\n }\n\n // This currently mutates the submission object metadata\n await this.fileSubmissionService.populate(\n submission as unknown as FileSubmission,\n createSubmissionDto,\n file,\n );\n break;\n }\n\n default: {\n throw new BadRequestException(\n `Unknown SubmissionType: ${createSubmissionDto.type}.`,\n );\n }\n }\n\n // Re-save to capture any mutations during population and mark as initialized\n await this.repository.update(submission.id, {\n ...submission.toObject(),\n isInitialized: true,\n });\n this.emit();\n return await this.findById(submission.id);\n } catch (err) {\n // Clean up on error, tx is too much work\n this.logger.error(err, 'Error creating submission');\n await this.repository.deleteById([submission.id]);\n throw err;\n }\n }\n\n /**\n * Applies a template to a submission.\n * Primarily used when a submission is created from a template.\n * Or when applying overriding multi-submission options.\n *\n * @param {SubmissionId} id\n * @param {SubmissionId} templateId\n */\n async applyOverridingTemplate(id: SubmissionId, templateId: SubmissionId) {\n this.logger\n .withMetadata({ id, templateId })\n .info('Applying template to submission');\n const submission = await this.findById(id, { failOnMissing: true });\n const template: Submission = await this.findById(templateId, {\n failOnMissing: true,\n });\n\n if (!template.metadata.template) {\n throw new BadRequestException('Template Id provided is not a template.');\n }\n\n const defaultOption: WebsiteOptions = submission.options.find(\n (option: WebsiteOptions) => option.accountId === NULL_ACCOUNT_ID,\n );\n const defaultTitle = defaultOption?.data?.title;\n\n // Prepare all option insertions before the transaction\n const newOptionInsertions: Insert<'WebsiteOptionsSchema'>[] =\n await Promise.all(\n template.options.map((option) =>\n this.websiteOptionsService.createOptionInsertObject(\n submission,\n option.accountId,\n option.data,\n (option.isDefault ? defaultTitle : option?.data?.title) ?? '',\n ),\n ),\n );\n\n await withTransactionContext(this.repository.db, async (ctx) => {\n // clear all existing options\n await ctx\n .getDb()\n .delete(WebsiteOptionsSchema)\n .where(eq(WebsiteOptionsSchema.submissionId, id));\n\n await ctx\n .getDb()\n .insert(WebsiteOptionsSchema)\n .values(newOptionInsertions);\n\n // Track all created options for cleanup if needed\n newOptionInsertions.forEach((option) => {\n if (option.id) {\n ctx.track('WebsiteOptionsSchema', option.id);\n }\n });\n });\n\n try {\n return await this.findById(id);\n } catch (err) {\n throw new BadRequestException(err);\n }\n }\n\n /**\n * Updates a submission.\n *\n * @param {SubmissionId} id\n * @param {UpdateSubmissionDto} update\n */\n async update(id: SubmissionId, update: UpdateSubmissionDto) {\n this.logger.withMetadata(update).info(`Updating Submission '${id}'`);\n const submission = await this.findById(id, { failOnMissing: true });\n\n const scheduleType =\n update.scheduleType ?? submission.schedule.scheduleType;\n const updates: Pick<\n SubmissionEntity,\n 'metadata' | 'isArchived' | 'isScheduled' | 'schedule'\n > = {\n metadata: {\n ...submission.metadata,\n ...(update.metadata ?? {}),\n },\n isArchived: update.isArchived ?? submission.isArchived,\n isScheduled:\n scheduleType === ScheduleType.NONE\n ? false\n : (update.isScheduled ?? submission.isScheduled),\n schedule:\n scheduleType === ScheduleType.NONE\n ? {\n scheduleType: ScheduleType.NONE,\n scheduledFor: undefined,\n cron: undefined,\n }\n : {\n scheduledFor:\n scheduleType === ScheduleType.SINGLE && update.scheduledFor\n ? new Date(update.scheduledFor).toISOString()\n : (update.scheduledFor ?? submission.schedule.scheduledFor),\n scheduleType:\n update.scheduleType ?? submission.schedule.scheduleType,\n cron: update.cron ?? submission.schedule.cron,\n },\n };\n\n const optionChanges: Promise[] = [];\n\n // Removes unused website options\n if (update.deletedWebsiteOptions?.length) {\n update.deletedWebsiteOptions.forEach((deletedOptionId) => {\n optionChanges.push(this.websiteOptionsService.remove(deletedOptionId));\n });\n }\n\n // Creates or updates new website options\n if (update.newOrUpdatedOptions?.length) {\n update.newOrUpdatedOptions.forEach((option) => {\n if (option.createdAt) {\n optionChanges.push(\n this.websiteOptionsService.update(option.id, {\n data: option.data,\n }),\n );\n } else {\n optionChanges.push(\n this.websiteOptionsService.create({\n accountId: option.accountId,\n data: option.data,\n submissionId: submission.id,\n }),\n );\n }\n });\n }\n\n await Promise.allSettled(optionChanges);\n\n try {\n // Update Here\n await this.repository.update(id, updates);\n this.emit();\n return await this.findById(id);\n } catch (err) {\n throw new BadRequestException(err);\n }\n }\n\n public async remove(id: SubmissionId) {\n const result = await super.remove(id);\n this.emit();\n return result;\n }\n\n async applyMultiSubmission(applyMultiSubmissionDto: ApplyMultiSubmissionDto) {\n const { submissionToApply, submissionIds, merge } = applyMultiSubmissionDto;\n const origin = await this.repository.findById(submissionToApply, {\n failOnMissing: true,\n });\n const submissions = await this.repository.find({\n where: (submission, { inArray }) => inArray(submission.id, submissionIds),\n });\n if (merge) {\n // Keeps unique options, overwrites overlapping options\n for (const submission of submissions) {\n for (const option of origin.options) {\n const existingOption = submission.options.find(\n (o) => o.accountId === option.accountId,\n );\n if (existingOption) {\n // Don't overwrite set title\n const opts = { ...option.data, title: existingOption.data.title };\n await this.websiteOptionsService.update(existingOption.id, {\n data: opts,\n });\n } else {\n await this.websiteOptionsService.createOption(\n submission,\n option.accountId,\n option.data,\n option.isDefault ? undefined : option.data.title,\n );\n }\n }\n }\n } else {\n // Removes all options not included in the origin submission\n for (const submission of submissions) {\n const { options } = submission;\n const defaultOptions = options.find((option) => option.isDefault);\n const defaultTitle = defaultOptions?.data.title;\n await Promise.all(\n options.map((option) => this.websiteOptionsService.remove(option.id)),\n );\n // eslint-disable-next-line no-restricted-syntax\n for (const option of origin.options) {\n const opts = { ...option.data };\n if (option.isDefault) {\n opts.title = defaultTitle;\n }\n await this.websiteOptionsService.createOption(\n submission,\n option.accountId,\n opts,\n option.isDefault ? defaultTitle : option.data.title,\n );\n }\n }\n }\n\n this.emit();\n }\n\n /**\n * Applies selected template options to multiple submissions.\n * Upserts options (update if exists, create if new) with merge behavior.\n *\n * @param dto - The apply template options DTO\n * @returns Object with success/failure counts\n */\n async applyTemplateOptions(dto: ApplyTemplateOptionsDto): Promise<{\n success: number;\n failed: number;\n errors: Array<{ submissionId: SubmissionId; error: string }>;\n }> {\n const { targetSubmissionIds, options, overrideTitle, overrideDescription } =\n dto;\n\n this.logger\n .withMetadata({\n targetCount: targetSubmissionIds.length,\n optionCount: options.length,\n })\n .info('Applying template options to submissions');\n\n const results = {\n success: 0,\n failed: 0,\n errors: [] as Array<{ submissionId: SubmissionId; error: string }>,\n };\n\n for (const submissionId of targetSubmissionIds) {\n try {\n const submission = await this.findById(submissionId, {\n failOnMissing: true,\n });\n\n for (const templateOption of options) {\n // Find existing option for this account\n const existingOption = submission.options.find(\n (o) => o.accountId === templateOption.accountId,\n );\n\n // Prepare the data to apply\n const dataToApply = { ...templateOption.data };\n\n // Handle title override: only replace if overrideTitle is true AND template has non-empty title\n if (!overrideTitle || !dataToApply.title?.trim()) {\n delete dataToApply.title;\n }\n\n // Handle description override: only replace if overrideDescription is true AND template has non-empty description\n if (\n !overrideDescription ||\n !dataToApply.description?.description?.content?.length\n ) {\n delete dataToApply.description;\n }\n\n if (existingOption) {\n // Upsert: merge existing data with template data\n const mergedData = {\n ...existingOption.data,\n ...dataToApply,\n };\n await this.websiteOptionsService.update(existingOption.id, {\n data: mergedData,\n });\n } else {\n // Create new option - only pass title if it exists in dataToApply\n await this.websiteOptionsService.createOption(\n submission,\n templateOption.accountId,\n dataToApply,\n 'title' in dataToApply ? dataToApply.title : undefined,\n );\n }\n }\n\n results.success++;\n } catch (error) {\n results.failed++;\n results.errors.push({\n submissionId,\n error: error instanceof Error ? error.message : String(error),\n });\n this.logger\n .withMetadata({ submissionId, error })\n .error('Failed to apply template options to submission');\n }\n }\n\n this.emit();\n return results;\n }\n\n /**\n * Duplicates a submission.\n * @param {string} id\n */\n public async duplicate(id: SubmissionId) {\n this.logger.info(`Duplicating Submission '${id}'`);\n const entityToDuplicate = await this.repository.findOne({\n where: (submission, { eq: equals }) => equals(submission.id, id),\n with: {\n options: {\n with: {\n account: true,\n },\n },\n files: true,\n },\n });\n await withTransactionContext(this.repository.db, async (ctx) => {\n const newSubmission = (\n await ctx\n .getDb()\n .insert(SubmissionSchema)\n .values({\n metadata: entityToDuplicate.metadata,\n type: entityToDuplicate.type,\n isScheduled: entityToDuplicate.isScheduled,\n schedule: entityToDuplicate.schedule,\n isMultiSubmission: entityToDuplicate.isMultiSubmission,\n isTemplate: entityToDuplicate.isTemplate,\n isInitialized: false, // Will be set to true at the end of the transaction\n order: entityToDuplicate.order,\n })\n .returning()\n )[0];\n ctx.track('SubmissionSchema', newSubmission.id);\n\n const optionValues = entityToDuplicate.options.map((option) => ({\n ...option,\n id: undefined,\n submissionId: newSubmission.id,\n }));\n await ctx.getDb().insert(WebsiteOptionsSchema).values(optionValues);\n\n for (const file of entityToDuplicate.files) {\n await file.load();\n const newFile = (\n await ctx\n .getDb()\n .insert(SubmissionFileSchema)\n .values({\n submissionId: newSubmission.id,\n fileName: file.fileName,\n mimeType: file.mimeType,\n hash: file.hash,\n size: file.size,\n height: file.height,\n width: file.width,\n hasThumbnail: file.hasThumbnail,\n hasCustomThumbnail: file.hasCustomThumbnail,\n hasAltFile: file.hasAltFile,\n metadata: file.metadata,\n order: file.order,\n })\n .returning()\n )[0];\n ctx.track('SubmissionFileSchema', newFile.id);\n\n const primaryFile = (\n await ctx\n .getDb()\n .insert(FileBufferSchema)\n .values({\n ...file.file,\n id: undefined,\n submissionFileId: newFile.id,\n })\n .returning()\n )[0];\n ctx.track('FileBufferSchema', primaryFile.id);\n\n const thumbnail: FileBuffer | undefined = file.thumbnail\n ? (\n await ctx\n .getDb()\n .insert(FileBufferSchema)\n .values({\n ...file.thumbnail,\n id: undefined,\n submissionFileId: newFile.id,\n })\n .returning()\n )[0]\n : undefined;\n if (thumbnail) {\n ctx.track('FileBufferSchema', thumbnail.id);\n }\n\n const altFile: FileBuffer | undefined = file.altFile\n ? (\n await ctx\n .getDb()\n .insert(FileBufferSchema)\n .values({\n ...file.altFile,\n id: undefined,\n submissionFileId: newFile.id,\n })\n .returning()\n )[0]\n : undefined;\n if (altFile) {\n ctx.track('FileBufferSchema', altFile.id);\n }\n\n await ctx\n .getDb()\n .update(SubmissionFileSchema)\n .set({\n primaryFileId: primaryFile.id,\n thumbnailId: thumbnail?.id,\n altFileId: altFile?.id,\n })\n .where(eq(SubmissionFileSchema.id, newFile.id));\n\n const oldId = file.id;\n // eslint-disable-next-line prefer-destructuring\n const metadata: FileSubmissionMetadata =\n newSubmission.metadata as FileSubmissionMetadata;\n }\n\n // Save updated metadata and mark as initialized\n await ctx\n .getDb()\n .update(SubmissionSchema)\n .set({ metadata: newSubmission.metadata, isInitialized: true })\n .where(eq(SubmissionSchema.id, newSubmission.id));\n });\n this.emit();\n }\n\n async updateTemplateName(\n id: SubmissionId,\n updateSubmissionDto: UpdateSubmissionTemplateNameDto,\n ) {\n const entity = await this.findById(id, { failOnMissing: true });\n\n if (!entity.isTemplate) {\n throw new BadRequestException(`Submission '${id}' is not a template`);\n }\n\n const name = updateSubmissionDto.name.trim();\n if (!updateSubmissionDto.name) {\n throw new BadRequestException(\n 'Template name cannot be empty or whitespace',\n );\n }\n\n if (entity.metadata.template) {\n entity.metadata.template.name = name;\n }\n const result = await this.repository.update(id, {\n metadata: entity.metadata,\n });\n this.emit();\n return result;\n }\n\n async reorder(\n id: SubmissionId,\n targetId: SubmissionId,\n position: 'before' | 'after',\n ) {\n const moving = await this.findById(id, { failOnMissing: true });\n const target = await this.findById(targetId, { failOnMissing: true });\n\n // Ensure same type (FILE or MESSAGE)\n if (moving.type !== target.type) {\n throw new BadRequestException(\n 'Cannot reorder across different submission types',\n );\n }\n\n // Get all submissions of the same type, sorted by order\n // Exclude templates and multi-submissions from ordering\n const allOfType = (await this.repository.findAll())\n .filter(\n (s) =>\n s.type === moving.type && !s.isTemplate && !s.isMultiSubmission,\n )\n .sort((a, b) => a.order - b.order);\n\n const targetIndex = allOfType.findIndex((s) => s.id === targetId);\n if (targetIndex === -1) {\n throw new NotFoundException(`Target submission '${targetId}' not found`);\n }\n\n let newOrder: number;\n\n if (position === 'before') {\n if (targetIndex === 0) {\n // Insert at the very beginning\n newOrder = target.order - 1;\n } else {\n // Insert between previous and target\n const prevOrder = allOfType[targetIndex - 1].order;\n newOrder = (prevOrder + target.order) / 2;\n }\n } else if (targetIndex === allOfType.length - 1) {\n // position === 'after', Insert at the very end\n newOrder = target.order + 1;\n } else {\n // position === 'after', Insert between target and next\n const nextOrder = allOfType[targetIndex + 1].order;\n newOrder = (target.order + nextOrder) / 2;\n }\n\n await this.repository.update(id, { order: newOrder });\n this.emit();\n }\n\n async unarchive(id: SubmissionId) {\n const submission = await this.findById(id, { failOnMissing: true });\n if (!submission.isArchived) {\n throw new BadRequestException(`Submission '${id}' is not archived`);\n }\n await this.repository.update(id, {\n isArchived: false,\n });\n this.emit();\n }\n\n async archive(id: SubmissionId) {\n const submission = await this.findById(id, { failOnMissing: true });\n if (submission.isArchived) {\n throw new BadRequestException(`Submission '${id}' is already archived`);\n }\n await this.repository.update(id, {\n isArchived: true,\n isScheduled: false,\n schedule: {\n scheduledFor: undefined,\n scheduleType: ScheduleType.NONE,\n cron: undefined,\n },\n });\n this.emit();\n }\n}\n" }, { "path": "apps/client-server/src/app/submission/submission.controller.ts", "content": "import {\n Body,\n Controller,\n Get,\n Param,\n Patch,\n Post,\n UploadedFiles,\n UseInterceptors,\n} from '@nestjs/common';\nimport { FilesInterceptor } from '@nestjs/platform-express';\nimport {\n ApiBadRequestResponse,\n ApiConsumes,\n ApiNotFoundResponse,\n ApiOkResponse,\n ApiTags,\n} from '@nestjs/swagger';\nimport { ISubmissionDto, SubmissionId, SubmissionType } from '@postybirb/types';\nimport { parse } from 'path';\nimport { PostyBirbController } from '../common/controller/postybirb-controller';\nimport { MulterFileInfo } from '../file/models/multer-file-info';\nimport { ApplyMultiSubmissionDto } from './dtos/apply-multi-submission.dto';\nimport { ApplyTemplateOptionsDto } from './dtos/apply-template-options.dto';\nimport { CreateSubmissionDto } from './dtos/create-submission.dto';\nimport { ReorderSubmissionDto } from './dtos/reorder-submission.dto';\nimport { UpdateSubmissionTemplateNameDto } from './dtos/update-submission-template-name.dto';\nimport { UpdateSubmissionDto } from './dtos/update-submission.dto';\nimport { SubmissionService } from './services/submission.service';\n\n/**\n * CRUD operations on Submission data.\n *\n * @class SubmissionController\n */\n@ApiTags('submissions')\n@Controller('submissions')\nexport class SubmissionController extends PostyBirbController<'SubmissionSchema'> {\n constructor(readonly service: SubmissionService) {\n super(service);\n }\n\n @Get()\n async findAll(): Promise {\n return this.service.findAllAsDto();\n }\n\n @Post()\n @ApiConsumes('multipart/form-data')\n @ApiOkResponse({ description: 'Submission created.' })\n @ApiBadRequestResponse({ description: 'Bad request made.' })\n @UseInterceptors(FilesInterceptor('files', undefined, { preservePath: true }))\n async create(\n @Body() createSubmissionDto: CreateSubmissionDto,\n @UploadedFiles() files: MulterFileInfo[],\n ) {\n const mapper = (res) => res.toDTO();\n if ((files || []).length) {\n const results = [];\n // !NOTE: Currently this shouldn't be able to happen with the current UI, but may need to be addressed in the future.\n // Efforts have been made to prevent this from happening, with the removal of using entity.create({}) but it may still be possible.\n // There appears to be an issue where if trying to create many submissions in parallel\n // the database will attempt to create them all at once and fail on a race condition.\n // not sure if this is a database issue or a typeorm issue.\n for (const file of files) {\n const createFileSubmission = new CreateSubmissionDto();\n Object.assign(createFileSubmission, createSubmissionDto);\n if (!createSubmissionDto.name) {\n createFileSubmission.name = parse(file.originalname).name;\n }\n\n createFileSubmission.type = SubmissionType.FILE;\n results.push(await this.service.create(createFileSubmission, file));\n }\n\n return results.map(mapper);\n }\n return (\n await Promise.all([await this.service.create(createSubmissionDto)])\n ).map(mapper);\n }\n\n @Post('duplicate/:id')\n @ApiOkResponse({ description: 'Submission duplicated.' })\n @ApiNotFoundResponse({ description: 'Submission Id not found.' })\n async duplicate(@Param('id') id: SubmissionId) {\n this.service.duplicate(id);\n }\n\n @Post('unarchive/:id')\n @ApiOkResponse({ description: 'Submission unarchived.' })\n @ApiNotFoundResponse({ description: 'Submission Id not found.' })\n async unarchive(@Param('id') id: SubmissionId) {\n return this.service.unarchive(id);\n }\n\n @Post('archive/:id')\n @ApiOkResponse({ description: 'Submission archived.' })\n @ApiNotFoundResponse({ description: 'Submission Id not found.' })\n async archive(@Param('id') id: SubmissionId) {\n return this.service.archive(id);\n }\n\n // IMPORTANT: This route MUST be defined BEFORE @Patch(':id') to prevent\n // Express from matching 'reorder' as an :id parameter\n @Patch('reorder')\n @ApiOkResponse({ description: 'Submission reordered.' })\n @ApiNotFoundResponse({ description: 'Submission Id not found.' })\n async reorder(@Body() reorderDto: ReorderSubmissionDto) {\n return this.service.reorder(\n reorderDto.id,\n reorderDto.targetId,\n reorderDto.position,\n );\n }\n\n @Patch(':id')\n @ApiOkResponse({ description: 'Submission updated.' })\n @ApiNotFoundResponse({ description: 'Submission Id not found.' })\n async update(\n @Param('id') id: SubmissionId,\n @Body() updateSubmissionDto: UpdateSubmissionDto,\n ) {\n return this.service\n .update(id, updateSubmissionDto)\n .then((entity) => entity.toDTO());\n }\n\n @Patch('template/:id')\n @ApiOkResponse({ description: 'Submission updated.' })\n @ApiNotFoundResponse({ description: 'Submission Id not found.' })\n async updateTemplateName(\n @Param('id') id: SubmissionId,\n @Body() updateSubmissionDto: UpdateSubmissionTemplateNameDto,\n ) {\n return this.service\n .updateTemplateName(id, updateSubmissionDto)\n .then((entity) => entity.toDTO());\n }\n\n @Patch('apply/multi')\n @ApiOkResponse({ description: 'Submission applied to multiple submissions.' })\n @ApiNotFoundResponse({ description: 'Submission Id not found.' })\n async applyMulti(@Body() applyMultiSubmissionDto: ApplyMultiSubmissionDto) {\n return this.service.applyMultiSubmission(applyMultiSubmissionDto);\n }\n\n @Patch('apply/template/options')\n @ApiOkResponse({ description: 'Template options applied to submissions.' })\n @ApiBadRequestResponse({ description: 'Invalid request.' })\n async applyTemplateOptions(\n @Body() applyTemplateOptionsDto: ApplyTemplateOptionsDto,\n ) {\n return this.service.applyTemplateOptions(applyTemplateOptionsDto);\n }\n\n @Patch('apply/template/:id/:templateId')\n @ApiOkResponse({ description: 'Template applied to submission.' })\n @ApiNotFoundResponse({ description: 'Submission Id or Template Id not found.' })\n async applyTemplate(\n @Param('id') id: SubmissionId,\n @Param('templateId') templateId: SubmissionId,\n ) {\n return this.service\n .applyOverridingTemplate(id, templateId)\n .then((entity) => entity.toDTO());\n }\n}\n" }, { "path": "apps/client-server/src/app/submission/submission.events.ts", "content": "import { SUBMISSION_UPDATES } from '@postybirb/socket-events';\nimport { ISubmissionDto, ISubmissionMetadata } from '@postybirb/types';\nimport { WebsocketEvent } from '../web-socket/models/web-socket-event';\n\nexport type SubmissionEventTypes = SubmissionUpdateEvent;\n\nclass SubmissionUpdateEvent\n implements WebsocketEvent[]>\n{\n event: string = SUBMISSION_UPDATES;\n\n data: ISubmissionDto[];\n}\n" }, { "path": "apps/client-server/src/app/submission/submission.module.ts", "content": "import { forwardRef, Module } from '@nestjs/common';\nimport { MulterModule } from '@nestjs/platform-express';\nimport { PostyBirbDirectories } from '@postybirb/fs';\nimport { diskStorage } from 'multer';\nimport { extname } from 'path';\nimport { v4 } from 'uuid';\nimport { AccountModule } from '../account/account.module';\nimport { FileModule } from '../file/file.module';\nimport { WebsiteOptionsModule } from '../website-options/website-options.module';\nimport { WebsitesModule } from '../websites/websites.module';\nimport { FileSubmissionController } from './file-submission.controller';\nimport { FileSubmissionService } from './services/file-submission.service';\nimport { MessageSubmissionService } from './services/message-submission.service';\nimport { SubmissionService } from './services/submission.service';\nimport { SubmissionController } from './submission.controller';\n\n@Module({\n imports: [\n WebsitesModule,\n AccountModule,\n FileModule,\n forwardRef(() => WebsiteOptionsModule),\n MulterModule.register({\n limits: {\n fileSize: 3e8, // Max 300MB\n },\n storage: diskStorage({\n destination(req, file, cb) {\n cb(null, PostyBirbDirectories.TEMP_DIRECTORY);\n },\n filename(req, file, cb) {\n cb(null, v4() + extname(file.originalname)); // Appending extension\n },\n }),\n }),\n ],\n providers: [\n SubmissionService,\n MessageSubmissionService,\n FileSubmissionService,\n ],\n controllers: [SubmissionController, FileSubmissionController],\n exports: [SubmissionService, FileSubmissionService],\n})\nexport class SubmissionModule {}\n" }, { "path": "apps/client-server/src/app/tag-converters/dtos/create-tag-converter.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { ICreateTagConverterDto } from '@postybirb/types';\nimport { IsObject, IsString } from 'class-validator';\n\nexport class CreateTagConverterDto implements ICreateTagConverterDto {\n @ApiProperty()\n @IsString()\n tag: string;\n\n @ApiProperty()\n @IsObject()\n convertTo: Record;\n}\n" }, { "path": "apps/client-server/src/app/tag-converters/dtos/update-tag-converter.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { IUpdateTagConverterDto } from '@postybirb/types';\nimport { IsObject, IsString } from 'class-validator';\n\nexport class UpdateTagConverterDto implements IUpdateTagConverterDto {\n @ApiProperty()\n @IsString()\n tag: string;\n\n @ApiProperty()\n @IsObject()\n convertTo: Record;\n}\n" }, { "path": "apps/client-server/src/app/tag-converters/tag-converter.events.ts", "content": "import { TAG_CONVERTER_UPDATES } from '@postybirb/socket-events';\nimport { TagConverterDto } from '@postybirb/types';\nimport { WebsocketEvent } from '../web-socket/models/web-socket-event';\n\nexport type TagConverterEventTypes = TagConverterUpdateEvent;\n\nclass TagConverterUpdateEvent implements WebsocketEvent {\n event: string = TAG_CONVERTER_UPDATES;\n\n data: TagConverterDto[];\n}\n" }, { "path": "apps/client-server/src/app/tag-converters/tag-converters.controller.ts", "content": "import { Body, Controller, Param, Patch, Post } from '@nestjs/common';\nimport {\n ApiBadRequestResponse,\n ApiNotFoundResponse,\n ApiOkResponse,\n ApiTags,\n} from '@nestjs/swagger';\nimport { EntityId } from '@postybirb/types';\nimport { PostyBirbController } from '../common/controller/postybirb-controller';\nimport { CreateTagConverterDto } from './dtos/create-tag-converter.dto';\nimport { UpdateTagConverterDto } from './dtos/update-tag-converter.dto';\nimport { TagConvertersService } from './tag-converters.service';\n\n/**\n * CRUD operations on TagConverters\n * @class TagConvertersController\n */\n@ApiTags('tag-converters')\n@Controller('tag-converters')\nexport class TagConvertersController extends PostyBirbController<'TagConverterSchema'> {\n constructor(readonly service: TagConvertersService) {\n super(service);\n }\n\n @Post()\n @ApiOkResponse({ description: 'Tag converter created.' })\n @ApiBadRequestResponse({ description: 'Bad request made.' })\n create(@Body() createTagConverterDto: CreateTagConverterDto) {\n return this.service\n .create(createTagConverterDto)\n .then((entity) => entity.toDTO());\n }\n\n @Patch(':id')\n @ApiOkResponse({ description: 'Tag converter updated.' })\n @ApiNotFoundResponse({ description: 'Tag converter not found.' })\n update(@Body() updateDto: UpdateTagConverterDto, @Param('id') id: EntityId) {\n return this.service.update(id, updateDto).then((entity) => entity.toDTO());\n }\n}\n" }, { "path": "apps/client-server/src/app/tag-converters/tag-converters.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { TagConvertersController } from './tag-converters.controller';\nimport { TagConvertersService } from './tag-converters.service';\n\n@Module({\n controllers: [TagConvertersController],\n providers: [TagConvertersService],\n exports: [TagConvertersService],\n})\nexport class TagConvertersModule {}\n" }, { "path": "apps/client-server/src/app/tag-converters/tag-converters.service.spec.ts", "content": "import { BadRequestException } from '@nestjs/common';\nimport { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport { CreateTagConverterDto } from './dtos/create-tag-converter.dto';\nimport { UpdateTagConverterDto } from './dtos/update-tag-converter.dto';\nimport { TagConvertersService } from './tag-converters.service';\n\ndescribe('TagConvertersService', () => {\n let service: TagConvertersService;\n let module: TestingModule;\n\n function createTagConverterDto(\n tag: string,\n convertTo: Record,\n ) {\n const dto = new CreateTagConverterDto();\n dto.tag = tag;\n dto.convertTo = convertTo;\n return dto;\n }\n\n beforeEach(async () => {\n clearDatabase();\n try {\n module = await Test.createTestingModule({\n imports: [],\n providers: [TagConvertersService],\n }).compile();\n\n service = module.get(TagConvertersService);\n } catch (e) {\n console.log(e);\n }\n });\n\n afterAll(async () => {\n await module.close();\n });\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n\n it('should create entities', async () => {\n const dto = createTagConverterDto('test', { default: 'converted' });\n\n const record = await service.create(dto);\n const groups = await service.findAll();\n expect(groups).toHaveLength(1);\n expect(groups[0].tag).toEqual(dto.tag);\n expect(groups[0].convertTo).toEqual(dto.convertTo);\n expect(record.toObject()).toEqual({\n tag: dto.tag,\n convertTo: dto.convertTo,\n id: record.id,\n createdAt: record.createdAt,\n updatedAt: record.updatedAt,\n });\n });\n\n it('should fail to create duplicate tag converters', async () => {\n const dto = createTagConverterDto('test', { default: 'converted' });\n const dto2 = createTagConverterDto('test', { default: 'converted' });\n\n await service.create(dto);\n\n let expectedException = null;\n try {\n await service.create(dto2);\n } catch (err) {\n expectedException = err;\n }\n expect(expectedException).toBeInstanceOf(BadRequestException);\n });\n\n it('should update entities', async () => {\n const dto = createTagConverterDto('test', { default: 'converted' });\n\n const record = await service.create(dto);\n const groups = await service.findAll();\n expect(groups).toHaveLength(1);\n\n const updateDto = new UpdateTagConverterDto();\n updateDto.tag = 'test';\n updateDto.convertTo = { default: 'converted', test: 'converted2' };\n await service.update(record.id, updateDto);\n const updatedRec = await service.findById(record.id);\n expect(updatedRec.tag).toBe(updateDto.tag);\n expect(updatedRec.convertTo).toEqual(updateDto.convertTo);\n });\n\n it('should delete entities', async () => {\n const dto = createTagConverterDto('test', { default: 'converted' });\n\n const record = await service.create(dto);\n expect(await service.findAll()).toHaveLength(1);\n\n await service.remove(record.id);\n expect(await service.findAll()).toHaveLength(0);\n });\n\n it('should convert tags', async () => {\n const dto = createTagConverterDto('test', { default: 'converted' });\n\n await service.create(dto);\n const result = await service.convert(\n { decoratedProps: { metadata: { name: 'default' } } } as any,\n ['test', 'test2'],\n );\n expect(result).toEqual(['converted', 'test2']);\n });\n});\n" }, { "path": "apps/client-server/src/app/tag-converters/tag-converters.service.ts", "content": "import { Injectable, Optional } from '@nestjs/common';\nimport { TAG_CONVERTER_UPDATES } from '@postybirb/socket-events';\nimport { EntityId } from '@postybirb/types';\nimport { eq } from 'drizzle-orm';\nimport { PostyBirbService } from '../common/service/postybirb-service';\nimport { TagConverter } from '../drizzle/models';\nimport { WSGateway } from '../web-socket/web-socket-gateway';\nimport { Website } from '../websites/website';\nimport { CreateTagConverterDto } from './dtos/create-tag-converter.dto';\nimport { UpdateTagConverterDto } from './dtos/update-tag-converter.dto';\n\n@Injectable()\nexport class TagConvertersService extends PostyBirbService<'TagConverterSchema'> {\n constructor(@Optional() webSocket?: WSGateway) {\n super('TagConverterSchema', webSocket);\n this.repository.subscribe('TagConverterSchema', () => {\n this.emit();\n });\n }\n\n async create(createDto: CreateTagConverterDto): Promise {\n this.logger\n .withMetadata(createDto)\n .info(`Creating TagConverter '${createDto.tag}'`);\n await this.throwIfExists(eq(this.schema.tag, createDto.tag));\n return this.repository.insert(createDto);\n }\n\n update(id: EntityId, update: UpdateTagConverterDto) {\n this.logger.withMetadata(update).info(`Updating TagConverter '${id}'`);\n return this.repository.update(id, update);\n }\n\n /**\n * Converts a list of tags using user defined conversion table.\n *\n * @param {Website} instance\n * @param {string[]} tags\n * @return {*} {Promise}\n */\n async convert(instance: Website, tags: string): Promise;\n async convert(instance: Website, tags: string[]): Promise;\n async convert(\n instance: Website,\n tags: string[] | string,\n ): Promise {\n if (typeof tags === 'string') {\n return (await this.convert(instance, [tags]))[0];\n }\n\n // { tag: { $in: tags } }\n const converters = await this.repository.find({\n where: (converter, { inArray }) => inArray(converter.tag, tags),\n });\n return tags.map((tag) => {\n const converter = converters.find((c) => c.tag === tag);\n if (!converter) {\n return tag;\n }\n return (\n converter.convertTo[instance.decoratedProps.metadata.name] ??\n converter.convertTo.default ?? // NOTE: This is not currently used, but it's here for future proofing\n tag\n );\n });\n }\n\n protected async emit() {\n super.emit({\n event: TAG_CONVERTER_UPDATES,\n data: (await this.repository.findAll()).map((entity) => entity.toDTO()),\n });\n }\n}\n" }, { "path": "apps/client-server/src/app/tag-groups/dtos/create-tag-group.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { ICreateTagGroupDto } from '@postybirb/types';\nimport { IsArray, IsString } from 'class-validator';\n\nexport class CreateTagGroupDto implements ICreateTagGroupDto {\n @ApiProperty()\n @IsString()\n name: string;\n\n @ApiProperty()\n @IsArray()\n tags: string[];\n}\n" }, { "path": "apps/client-server/src/app/tag-groups/dtos/update-tag-group.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { IUpdateTagGroupDto } from '@postybirb/types';\nimport { IsArray, IsString } from 'class-validator';\n\nexport class UpdateTagGroupDto implements IUpdateTagGroupDto {\n @ApiProperty()\n @IsString()\n name: string;\n\n @ApiProperty()\n @IsArray()\n tags: string[];\n}\n" }, { "path": "apps/client-server/src/app/tag-groups/tag-group.events.ts", "content": "import { TAG_GROUP_UPDATES } from '@postybirb/socket-events';\nimport { TagGroupDto } from '@postybirb/types';\nimport { WebsocketEvent } from '../web-socket/models/web-socket-event';\n\nexport type TagGroupEventTypes = TagGroupUpdateEvent;\n\nclass TagGroupUpdateEvent implements WebsocketEvent {\n event: string = TAG_GROUP_UPDATES;\n\n data: TagGroupDto[];\n}\n" }, { "path": "apps/client-server/src/app/tag-groups/tag-groups.controller.ts", "content": "import { Body, Controller, Param, Patch, Post } from '@nestjs/common';\nimport {\n ApiBadRequestResponse,\n ApiNotFoundResponse,\n ApiOkResponse,\n ApiTags,\n} from '@nestjs/swagger';\nimport { EntityId } from '@postybirb/types';\nimport { PostyBirbController } from '../common/controller/postybirb-controller';\nimport { CreateTagGroupDto } from './dtos/create-tag-group.dto';\nimport { UpdateTagGroupDto } from './dtos/update-tag-group.dto';\nimport { TagGroupsService } from './tag-groups.service';\n\n/**\n * CRUD operations for TagGroups.\n * @class TagGroupsController\n */\n@ApiTags('tag-groups')\n@Controller('tag-groups')\nexport class TagGroupsController extends PostyBirbController<'TagGroupSchema'> {\n constructor(readonly service: TagGroupsService) {\n super(service);\n }\n\n @Post()\n @ApiOkResponse({ description: 'Tag group created.' })\n @ApiBadRequestResponse({ description: 'Bad request made.' })\n create(@Body() createDto: CreateTagGroupDto) {\n return this.service.create(createDto).then((entity) => entity.toDTO());\n }\n\n @Patch(':id')\n @ApiOkResponse({ description: 'Tag group updated.', type: Boolean })\n @ApiNotFoundResponse({ description: 'Tag group not found.' })\n update(@Param('id') id: EntityId, @Body() updateDto: UpdateTagGroupDto) {\n return this.service.update(id, updateDto).then((entity) => entity.toDTO());\n }\n}\n" }, { "path": "apps/client-server/src/app/tag-groups/tag-groups.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { TagGroupsController } from './tag-groups.controller';\nimport { TagGroupsService } from './tag-groups.service';\n\n@Module({\n providers: [TagGroupsService],\n controllers: [TagGroupsController],\n})\nexport class TagGroupsModule {}\n" }, { "path": "apps/client-server/src/app/tag-groups/tag-groups.service.spec.ts", "content": "import { BadRequestException } from '@nestjs/common';\nimport { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport { CreateTagGroupDto } from './dtos/create-tag-group.dto';\nimport { TagGroupsService } from './tag-groups.service';\n\ndescribe('TagGroupsService', () => {\n let service: TagGroupsService;\n let module: TestingModule;\n\n function createTagGroupDto(name: string, tags: string[]) {\n const dto = new CreateTagGroupDto();\n dto.name = name;\n dto.tags = tags;\n return dto;\n }\n\n beforeEach(async () => {\n clearDatabase();\n module = await Test.createTestingModule({\n providers: [TagGroupsService],\n }).compile();\n\n service = module.get(TagGroupsService);\n });\n\n afterAll(async () => {\n await module.close();\n });\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n\n it('should create entities', async () => {\n const dto = createTagGroupDto('test', ['test', 'tag group']);\n\n const record = await service.create(dto);\n const groups = await service.findAll();\n expect(groups).toHaveLength(1);\n expect(groups[0].name).toEqual(dto.name);\n expect(groups[0].tags).toEqual(dto.tags);\n expect(record.toDTO()).toEqual({\n name: dto.name,\n tags: dto.tags,\n id: record.id,\n createdAt: record.createdAt,\n updatedAt: record.updatedAt,\n });\n });\n\n it('should fail to create duplicate named groups', async () => {\n const dto = createTagGroupDto('test', ['test']);\n const dto2 = createTagGroupDto('test', ['test', 'test-dupe']);\n\n await service.create(dto);\n const groups = await service.findAll();\n expect(groups).toHaveLength(1);\n expect(groups[0].name).toEqual(dto.name);\n expect(groups[0].tags).toEqual(dto.tags);\n\n let expectedException = null;\n try {\n await service.create(dto2);\n } catch (err) {\n expectedException = err;\n }\n expect(expectedException).toBeInstanceOf(BadRequestException);\n });\n\n it('should update entities', async () => {\n const dto = createTagGroupDto('test', ['test', 'tag group']);\n\n const record = await service.create(dto);\n const groups = await service.findAll();\n expect(groups).toHaveLength(1);\n\n const updateDto = new CreateTagGroupDto();\n updateDto.name = 'test';\n updateDto.tags = ['test', 'updated'];\n await service.update(record.id, updateDto);\n const updatedRec = await service.findById(record.id);\n expect(updatedRec.name).toBe(updateDto.name);\n expect(updatedRec.tags).toEqual(updateDto.tags);\n });\n\n it('should delete entities', async () => {\n const dto = createTagGroupDto('test', ['test', 'tag group']);\n\n const record = await service.create(dto);\n expect(await service.findAll()).toHaveLength(1);\n\n await service.remove(record.id);\n expect(await service.findAll()).toHaveLength(0);\n });\n});\n" }, { "path": "apps/client-server/src/app/tag-groups/tag-groups.service.ts", "content": "import { Injectable, Optional } from '@nestjs/common';\nimport { TAG_GROUP_UPDATES } from '@postybirb/socket-events';\nimport { EntityId } from '@postybirb/types';\nimport { eq } from 'drizzle-orm';\nimport { PostyBirbService } from '../common/service/postybirb-service';\nimport { TagGroup } from '../drizzle/models';\nimport { WSGateway } from '../web-socket/web-socket-gateway';\nimport { CreateTagGroupDto } from './dtos/create-tag-group.dto';\nimport { UpdateTagGroupDto } from './dtos/update-tag-group.dto';\n\n@Injectable()\nexport class TagGroupsService extends PostyBirbService<'TagGroupSchema'> {\n constructor(@Optional() webSocket?: WSGateway) {\n super('TagGroupSchema', webSocket);\n this.repository.subscribe('TagGroupSchema', () => this.emit());\n }\n\n async create(createDto: CreateTagGroupDto): Promise {\n this.logger\n .withMetadata(createDto)\n .info(`Creating TagGroup '${createDto.name}'`);\n await this.throwIfExists(eq(this.schema.name, createDto.name));\n return this.repository.insert(createDto);\n }\n\n update(id: EntityId, update: UpdateTagGroupDto) {\n this.logger.withMetadata(update).info(`Updating TagGroup '${id}'`);\n return this.repository.update(id, update);\n }\n\n protected async emit() {\n super.emit({\n event: TAG_GROUP_UPDATES,\n data: (await this.repository.findAll()).map((entity) => entity.toDTO()),\n });\n }\n}\n" }, { "path": "apps/client-server/src/app/update/update.controller.ts", "content": "import { Controller, Get, Post } from '@nestjs/common';\nimport { UpdateState } from '@postybirb/types';\nimport { UpdateService } from './update.service';\n\n@Controller('update')\nexport class UpdateController {\n constructor(private readonly service: UpdateService) {}\n\n /**\n * Checks for updates.\n * @returns For some reason, to fix error while\n * build we need to directly import and specify type\n */\n @Get()\n checkForUpdates(): UpdateState {\n return this.service.getUpdateState();\n }\n\n @Post('start')\n update() {\n return this.service.update();\n }\n\n /**\n * Quit and install the downloaded update.\n */\n @Post('install')\n install() {\n return this.service.install();\n }\n}\n" }, { "path": "apps/client-server/src/app/update/update.events.ts", "content": "import { UPDATE_UPDATES } from '@postybirb/socket-events';\nimport { UpdateState } from '@postybirb/types';\nimport { WebsocketEvent } from '../web-socket/models/web-socket-event';\n\nexport type UpdateEventTypes = UpdateUpdateEvent;\n\nclass UpdateUpdateEvent implements WebsocketEvent {\n event: string = UPDATE_UPDATES;\n\n data: UpdateState;\n}\n" }, { "path": "apps/client-server/src/app/update/update.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { WebSocketModule } from '../web-socket/web-socket.module';\nimport { UpdateController } from './update.controller';\nimport { UpdateService } from './update.service';\n\n@Module({\n imports: [WebSocketModule],\n providers: [UpdateService],\n controllers: [UpdateController],\n})\nexport class UpdateModule {}\n" }, { "path": "apps/client-server/src/app/update/update.service.ts", "content": "import { Injectable, Optional } from '@nestjs/common';\nimport { Logger } from '@postybirb/logger';\nimport { UPDATE_UPDATES } from '@postybirb/socket-events';\nimport { ReleaseNoteInfo, UpdateState } from '@postybirb/types';\nimport { ProgressInfo, UpdateInfo, autoUpdater } from 'electron-updater';\nimport isDocker from 'is-docker';\nimport { WSGateway } from '../web-socket/web-socket-gateway';\n\n/**\n * Handles updates for the application.\n *\n * @class UpdateService\n */\n@Injectable()\nexport class UpdateService {\n private readonly logger = Logger('Updates');\n\n private updateState: UpdateState = {\n updateAvailable: false,\n updateDownloaded: false,\n updateDownloading: false,\n updateError: undefined,\n updateProgress: undefined,\n updateNotes: undefined,\n };\n\n constructor(@Optional() private readonly webSocket?: WSGateway) {\n autoUpdater.logger = this.logger;\n autoUpdater.autoDownload = false;\n autoUpdater.fullChangelog = true;\n autoUpdater.allowPrerelease = true;\n\n this.registerListeners();\n if (!isDocker()) setTimeout(() => this.checkForUpdates(), 5_000);\n }\n\n /**\n * Emit update state changes via WebSocket.\n */\n private emit() {\n if (this.webSocket) {\n this.webSocket.emit({\n event: UPDATE_UPDATES,\n data: this.getUpdateState(),\n });\n }\n }\n\n private registerListeners() {\n autoUpdater.on('update-available', (update) => {\n this.onUpdateAvailable(update);\n });\n autoUpdater.on('download-progress', (progress) => {\n this.onDownloadProgress(progress);\n });\n autoUpdater.on('error', (error) => {\n this.onUpdateError(error);\n });\n autoUpdater.on('update-downloaded', () => {\n this.onUpdateDownloaded();\n });\n }\n\n private onUpdateDownloaded() {\n this.updateState = {\n ...this.updateState,\n updateDownloaded: true,\n updateDownloading: false,\n updateProgress: 100,\n };\n this.emit();\n }\n\n private onUpdateAvailable(update: UpdateInfo) {\n this.updateState = {\n ...this.updateState,\n updateAvailable: true,\n updateNotes: (update.releaseNotes as ReleaseNoteInfo[]) ?? [],\n };\n this.emit();\n }\n\n private onUpdateError(error: Error) {\n this.logger.withError(error).error();\n this.updateState = {\n ...this.updateState,\n updateError: error.message,\n updateDownloading: false,\n };\n this.emit();\n }\n\n private onDownloadProgress(progress: ProgressInfo) {\n this.updateState = {\n ...this.updateState,\n updateProgress: progress.percent,\n };\n this.emit();\n }\n\n public checkForUpdates() {\n if (\n this.updateState.updateDownloading ||\n this.updateState.updateDownloaded\n ) {\n return;\n }\n\n autoUpdater.checkForUpdates();\n }\n\n public getUpdateState() {\n return { ...this.updateState };\n }\n\n public update() {\n if (\n !this.updateState.updateAvailable ||\n this.updateState.updateDownloaded ||\n this.updateState.updateDownloading\n ) {\n return;\n }\n\n this.updateState = {\n ...this.updateState,\n updateDownloading: true,\n };\n this.emit();\n\n autoUpdater.downloadUpdate();\n }\n\n /**\n * Quit the application and install the downloaded update.\n * Only works if an update has been downloaded.\n */\n public install() {\n if (!this.updateState.updateDownloaded) {\n return;\n }\n\n autoUpdater.quitAndInstall(false, true);\n }\n}\n" }, { "path": "apps/client-server/src/app/user-converters/dtos/create-user-converter.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { ICreateUserConverterDto } from '@postybirb/types';\nimport { IsObject, IsString } from 'class-validator';\n\nexport class CreateUserConverterDto implements ICreateUserConverterDto {\n @ApiProperty()\n @IsString()\n username: string;\n\n @ApiProperty()\n @IsObject()\n convertTo: Record;\n}\n" }, { "path": "apps/client-server/src/app/user-converters/dtos/update-user-converter.dto.ts", "content": "import { ApiProperty, PartialType } from '@nestjs/swagger';\nimport { IUpdateUserConverterDto } from '@postybirb/types';\nimport { CreateUserConverterDto } from './create-user-converter.dto';\n\nexport class UpdateUserConverterDto\n extends PartialType(CreateUserConverterDto)\n implements IUpdateUserConverterDto\n{\n @ApiProperty()\n username?: string;\n\n @ApiProperty()\n convertTo?: Record;\n}\n" }, { "path": "apps/client-server/src/app/user-converters/user-converter.events.ts", "content": "import { USER_CONVERTER_UPDATES } from '@postybirb/socket-events';\nimport { UserConverterDto } from '@postybirb/types';\nimport { WebsocketEvent } from '../web-socket/models/web-socket-event';\n\nexport type UserConverterEventTypes = UserConverterUpdateEvent;\n\nclass UserConverterUpdateEvent implements WebsocketEvent {\n event: string = USER_CONVERTER_UPDATES;\n\n data: UserConverterDto[];\n}\n" }, { "path": "apps/client-server/src/app/user-converters/user-converters.controller.ts", "content": "import { Body, Controller, Param, Patch, Post } from '@nestjs/common';\nimport {\n ApiBadRequestResponse,\n ApiNotFoundResponse,\n ApiOkResponse,\n ApiTags,\n} from '@nestjs/swagger';\nimport { EntityId } from '@postybirb/types';\nimport { PostyBirbController } from '../common/controller/postybirb-controller';\nimport { CreateUserConverterDto } from './dtos/create-user-converter.dto';\nimport { UpdateUserConverterDto } from './dtos/update-user-converter.dto';\nimport { UserConvertersService } from './user-converters.service';\n\n/**\n * CRUD operations on UserConverters\n * @class UserConvertersController\n */\n@ApiTags('user-converters')\n@Controller('user-converters')\nexport class UserConvertersController extends PostyBirbController<'UserConverterSchema'> {\n constructor(readonly service: UserConvertersService) {\n super(service);\n }\n\n @Post()\n @ApiOkResponse({ description: 'User converter created.' })\n @ApiBadRequestResponse({ description: 'Bad request made.' })\n create(@Body() createUserConverterDto: CreateUserConverterDto) {\n return this.service\n .create(createUserConverterDto)\n .then((entity) => entity.toDTO());\n }\n\n @Patch(':id')\n @ApiOkResponse({ description: 'User converter updated.' })\n @ApiNotFoundResponse({ description: 'User converter not found.' })\n update(@Body() updateDto: UpdateUserConverterDto, @Param('id') id: EntityId) {\n return this.service.update(id, updateDto).then((entity) => entity.toDTO());\n }\n}\n" }, { "path": "apps/client-server/src/app/user-converters/user-converters.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { UserConvertersController } from './user-converters.controller';\nimport { UserConvertersService } from './user-converters.service';\n\n@Module({\n controllers: [UserConvertersController],\n providers: [UserConvertersService],\n exports: [UserConvertersService],\n})\nexport class UserConvertersModule {}\n" }, { "path": "apps/client-server/src/app/user-converters/user-converters.service.spec.ts", "content": "import { BadRequestException } from '@nestjs/common';\nimport { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport { CreateUserConverterDto } from './dtos/create-user-converter.dto';\nimport { UpdateUserConverterDto } from './dtos/update-user-converter.dto';\nimport { UserConvertersService } from './user-converters.service';\n\ndescribe('UserConvertersService', () => {\n let service: UserConvertersService;\n let module: TestingModule;\n\n function createUserConverterDto(\n username: string,\n convertTo: Record,\n ) {\n const dto = new CreateUserConverterDto();\n dto.username = username;\n dto.convertTo = convertTo;\n return dto;\n }\n\n beforeEach(async () => {\n clearDatabase();\n try {\n module = await Test.createTestingModule({\n imports: [],\n providers: [UserConvertersService],\n }).compile();\n\n service = module.get(UserConvertersService);\n } catch (e) {\n console.log(e);\n }\n });\n\n afterAll(async () => {\n await module.close();\n });\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n\n it('should create entities', async () => {\n const dto = createUserConverterDto('my_friend', {\n default: 'converted_friend',\n });\n\n const record = await service.create(dto);\n const converters = await service.findAll();\n expect(converters).toHaveLength(1);\n expect(converters[0].username).toEqual(dto.username);\n expect(converters[0].convertTo).toEqual(dto.convertTo);\n expect(record.toObject()).toEqual({\n username: dto.username,\n convertTo: dto.convertTo,\n id: record.id,\n createdAt: record.createdAt,\n updatedAt: record.updatedAt,\n });\n });\n\n it('should fail to create duplicate user converters', async () => {\n const dto = createUserConverterDto('my_friend', {\n default: 'converted_friend',\n });\n const dto2 = createUserConverterDto('my_friend', {\n default: 'converted_friend2',\n });\n\n await service.create(dto);\n\n let expectedException = null;\n try {\n await service.create(dto2);\n } catch (err) {\n expectedException = err;\n }\n expect(expectedException).toBeInstanceOf(BadRequestException);\n });\n\n it('should update entities', async () => {\n const dto = createUserConverterDto('my_friend', {\n default: 'converted_friend',\n });\n\n const record = await service.create(dto);\n const converters = await service.findAll();\n expect(converters).toHaveLength(1);\n\n const updateDto = new UpdateUserConverterDto();\n updateDto.username = 'my_friend';\n updateDto.convertTo = {\n default: 'converted_friend',\n bluesky: 'converted_friend2',\n };\n await service.update(record.id, updateDto);\n const updatedRec = await service.findById(record.id);\n expect(updatedRec.username).toBe(updateDto.username);\n expect(updatedRec.convertTo).toEqual(updateDto.convertTo);\n });\n\n it('should delete entities', async () => {\n const dto = createUserConverterDto('my_friend', {\n default: 'converted_friend',\n });\n\n const record = await service.create(dto);\n expect(await service.findAll()).toHaveLength(1);\n\n await service.remove(record.id);\n expect(await service.findAll()).toHaveLength(0);\n });\n\n it('should convert usernames', async () => {\n const dto = createUserConverterDto('my_friend', {\n default: 'default_friend',\n bluesky: 'friend.bsky.social',\n });\n\n await service.create(dto);\n\n // Test conversion for bluesky\n const resultBluesky = await service.convert(\n { decoratedProps: { metadata: { name: 'bluesky' } } } as any,\n 'my_friend',\n );\n expect(resultBluesky).toEqual('friend.bsky.social');\n\n // Test conversion for unknown website (should use default)\n const resultDefault = await service.convert(\n { decoratedProps: { metadata: { name: 'unknown' } } } as any,\n 'my_friend',\n );\n expect(resultDefault).toEqual('default_friend');\n\n // Test conversion for username not in converter (should return original)\n const resultNotFound = await service.convert(\n { decoratedProps: { metadata: { name: 'bluesky' } } } as any,\n 'unknown_user',\n );\n expect(resultNotFound).toEqual('unknown_user');\n });\n});\n" }, { "path": "apps/client-server/src/app/user-converters/user-converters.service.ts", "content": "import { Injectable, Optional } from '@nestjs/common';\nimport { USER_CONVERTER_UPDATES } from '@postybirb/socket-events';\nimport { EntityId } from '@postybirb/types';\nimport { eq } from 'drizzle-orm';\nimport { PostyBirbService } from '../common/service/postybirb-service';\nimport { UserConverter } from '../drizzle/models';\nimport { WSGateway } from '../web-socket/web-socket-gateway';\nimport { Website } from '../websites/website';\nimport { CreateUserConverterDto } from './dtos/create-user-converter.dto';\nimport { UpdateUserConverterDto } from './dtos/update-user-converter.dto';\n\n@Injectable()\nexport class UserConvertersService extends PostyBirbService<'UserConverterSchema'> {\n constructor(@Optional() webSocket?: WSGateway) {\n super('UserConverterSchema', webSocket);\n this.repository.subscribe('UserConverterSchema', () => {\n this.emit();\n });\n }\n\n async create(createDto: CreateUserConverterDto): Promise {\n this.logger\n .withMetadata(createDto)\n .info(`Creating UserConverter '${createDto.username}'`);\n await this.throwIfExists(eq(this.schema.username, createDto.username));\n return this.repository.insert(createDto);\n }\n\n update(id: EntityId, update: UpdateUserConverterDto) {\n this.logger.withMetadata(update).info(`Updating UserConverter '${id}'`);\n return this.repository.update(id, update);\n }\n\n /**\n * Converts a username using user defined conversion table.\n *\n * @param {Website} instance\n * @param {string} username\n * @return {*} {Promise}\n */\n async convert(instance: Website, username: string): Promise {\n const converter = await this.repository.findOne({\n where: (c, { eq: eqFn }) => eqFn(c.username, username),\n });\n\n if (!converter) {\n return username;\n }\n\n return (\n converter.convertTo[instance.decoratedProps.metadata.name] ??\n converter.convertTo.default ??\n username\n );\n }\n\n protected async emit() {\n super.emit({\n event: USER_CONVERTER_UPDATES,\n data: (await this.repository.findAll()).map((entity) => entity.toDTO()),\n });\n }\n}\n" }, { "path": "apps/client-server/src/app/user-specified-website-options/dtos/create-user-specified-website-options.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport {\n DynamicObject,\n EntityId,\n ICreateUserSpecifiedWebsiteOptionsDto,\n SubmissionType,\n} from '@postybirb/types';\nimport { IsEnum, IsObject, IsString } from 'class-validator';\n\nexport class CreateUserSpecifiedWebsiteOptionsDto\n implements ICreateUserSpecifiedWebsiteOptionsDto\n{\n @ApiProperty()\n @IsObject()\n options: DynamicObject;\n\n @ApiProperty({ enum: SubmissionType })\n @IsEnum(SubmissionType)\n type: SubmissionType;\n\n @IsString()\n accountId: EntityId;\n}\n" }, { "path": "apps/client-server/src/app/user-specified-website-options/dtos/update-user-specified-website-options.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport {\n DynamicObject,\n IUpdateUserSpecifiedWebsiteOptionsDto,\n SubmissionType,\n} from '@postybirb/types';\nimport { IsEnum, IsObject } from 'class-validator';\n\nexport class UpdateUserSpecifiedWebsiteOptionsDto\n implements IUpdateUserSpecifiedWebsiteOptionsDto\n{\n @ApiProperty({ enum: SubmissionType })\n @IsEnum(SubmissionType)\n type: SubmissionType;\n\n @ApiProperty()\n @IsObject()\n options: DynamicObject;\n}\n" }, { "path": "apps/client-server/src/app/user-specified-website-options/user-specified-website-options.controller.ts", "content": "import { Body, Controller, Param, Patch, Post } from '@nestjs/common';\nimport {\n ApiBadRequestResponse,\n ApiNotFoundResponse,\n ApiOkResponse,\n ApiTags,\n} from '@nestjs/swagger';\nimport { EntityId } from '@postybirb/types';\nimport { PostyBirbController } from '../common/controller/postybirb-controller';\nimport { CreateUserSpecifiedWebsiteOptionsDto } from './dtos/create-user-specified-website-options.dto';\nimport { UpdateUserSpecifiedWebsiteOptionsDto } from './dtos/update-user-specified-website-options.dto';\nimport { UserSpecifiedWebsiteOptionsService } from './user-specified-website-options.service';\n\n/**\n * CRUD operations for UserSpecifiedWebsiteOptions\n * @class UserSpecifiedWebsiteOptionsController\n */\n@ApiTags('user-specified-website-options')\n@Controller('user-specified-website-options')\nexport class UserSpecifiedWebsiteOptionsController extends PostyBirbController<'UserSpecifiedWebsiteOptionsSchema'> {\n constructor(readonly service: UserSpecifiedWebsiteOptionsService) {\n super(service);\n }\n\n @Post()\n @ApiOkResponse({ description: 'Entity created or updated.' })\n @ApiBadRequestResponse({ description: 'Bad request made.' })\n async create(@Body() createDto: CreateUserSpecifiedWebsiteOptionsDto) {\n // Use upsert to handle both create and update based on accountId+type\n return this.service.upsert(createDto).then((entity) => entity.toDTO());\n }\n\n @Patch(':id')\n @ApiOkResponse({ description: 'Entity updated.', type: Boolean })\n @ApiNotFoundResponse()\n update(\n @Param('id') id: EntityId,\n @Body() updateDto: UpdateUserSpecifiedWebsiteOptionsDto,\n ) {\n return this.service.update(id, updateDto).then((entity) => entity.toDTO());\n }\n}\n" }, { "path": "apps/client-server/src/app/user-specified-website-options/user-specified-website-options.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { UserSpecifiedWebsiteOptionsController } from './user-specified-website-options.controller';\nimport { UserSpecifiedWebsiteOptionsService } from './user-specified-website-options.service';\n\n@Module({\n controllers: [UserSpecifiedWebsiteOptionsController],\n providers: [UserSpecifiedWebsiteOptionsService],\n exports: [UserSpecifiedWebsiteOptionsService],\n})\nexport class UserSpecifiedWebsiteOptionsModule {}\n" }, { "path": "apps/client-server/src/app/user-specified-website-options/user-specified-website-options.service.spec.ts", "content": "import { BadRequestException } from '@nestjs/common';\nimport { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport { NULL_ACCOUNT_ID, SubmissionType } from '@postybirb/types';\nimport { AccountModule } from '../account/account.module';\nimport { AccountService } from '../account/account.service';\nimport { CreateUserSpecifiedWebsiteOptionsDto } from './dtos/create-user-specified-website-options.dto';\nimport { UpdateUserSpecifiedWebsiteOptionsDto } from './dtos/update-user-specified-website-options.dto';\nimport { UserSpecifiedWebsiteOptionsService } from './user-specified-website-options.service';\n\ndescribe('UserSpecifiedWebsiteOptionsService', () => {\n let service: UserSpecifiedWebsiteOptionsService;\n let module: TestingModule;\n\n beforeEach(async () => {\n clearDatabase();\n module = await Test.createTestingModule({\n imports: [AccountModule],\n providers: [UserSpecifiedWebsiteOptionsService],\n }).compile();\n\n service = module.get(\n UserSpecifiedWebsiteOptionsService,\n );\n\n const accountService = module.get(AccountService);\n await accountService.onModuleInit();\n });\n\n afterAll(async () => {\n await module.close();\n });\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n\n it('should create entities', async () => {\n const dto = new CreateUserSpecifiedWebsiteOptionsDto();\n dto.accountId = NULL_ACCOUNT_ID;\n dto.options = { test: 'test' };\n dto.type = SubmissionType.MESSAGE;\n\n const record = await service.create(dto);\n expect(await service.findAll()).toHaveLength(1);\n expect(record.options).toEqual(dto.options);\n expect(record.type).toEqual(dto.type);\n expect(record.toDTO()).toEqual({\n accountId: NULL_ACCOUNT_ID,\n createdAt: record.createdAt,\n id: record.id,\n options: dto.options,\n type: dto.type,\n updatedAt: record.updatedAt,\n });\n });\n\n it('should fail to create a duplicate entity', async () => {\n const dto = new CreateUserSpecifiedWebsiteOptionsDto();\n dto.accountId = NULL_ACCOUNT_ID;\n dto.options = { test: 'test' };\n dto.type = SubmissionType.MESSAGE;\n\n await service.create(dto);\n await expect(service.create(dto)).rejects.toThrow(BadRequestException);\n });\n\n it('should upsert - create when not exists', async () => {\n const dto = new CreateUserSpecifiedWebsiteOptionsDto();\n dto.accountId = NULL_ACCOUNT_ID;\n dto.options = { test: 'test' };\n dto.type = SubmissionType.MESSAGE;\n\n const record = await service.upsert(dto);\n expect(await service.findAll()).toHaveLength(1);\n expect(record.options).toEqual(dto.options);\n });\n\n it('should upsert - update when exists', async () => {\n const dto = new CreateUserSpecifiedWebsiteOptionsDto();\n dto.accountId = NULL_ACCOUNT_ID;\n dto.options = { test: 'original' };\n dto.type = SubmissionType.MESSAGE;\n\n // First create\n const created = await service.upsert(dto);\n expect(created.options).toEqual({ test: 'original' });\n\n // Second upsert should update, not throw\n dto.options = { test: 'updated' };\n const updated = await service.upsert(dto);\n\n expect(await service.findAll()).toHaveLength(1); // Still only one record\n expect(updated.id).toEqual(created.id); // Same record\n expect(updated.options).toEqual({ test: 'updated' }); // Updated options\n });\n\n it('should upsert different account+type combinations independently', async () => {\n const dto1 = new CreateUserSpecifiedWebsiteOptionsDto();\n dto1.accountId = NULL_ACCOUNT_ID;\n dto1.options = { test: 'message' };\n dto1.type = SubmissionType.MESSAGE;\n\n const dto2 = new CreateUserSpecifiedWebsiteOptionsDto();\n dto2.accountId = NULL_ACCOUNT_ID;\n dto2.options = { test: 'file' };\n dto2.type = SubmissionType.FILE;\n\n await service.upsert(dto1);\n await service.upsert(dto2);\n\n expect(await service.findAll()).toHaveLength(2);\n });\n\n it('should update entities', async () => {\n const dto = new CreateUserSpecifiedWebsiteOptionsDto();\n dto.accountId = NULL_ACCOUNT_ID;\n dto.options = { test: 'test' };\n dto.type = SubmissionType.MESSAGE;\n\n const record = await service.create(dto);\n const options = { ...record.options };\n\n const updateDto = new UpdateUserSpecifiedWebsiteOptionsDto();\n updateDto.type = SubmissionType.MESSAGE;\n updateDto.options = { test: 'updated' };\n const updateRecord = await service.update(record.id, updateDto);\n expect(record.id).toEqual(updateRecord.id);\n expect(options).not.toEqual(updateRecord.options);\n });\n});\n" }, { "path": "apps/client-server/src/app/user-specified-website-options/user-specified-website-options.service.ts", "content": "import { Injectable } from '@nestjs/common';\nimport { AccountId, EntityId, SubmissionType } from '@postybirb/types';\nimport { and, eq } from 'drizzle-orm';\nimport { PostyBirbService } from '../common/service/postybirb-service';\nimport { UserSpecifiedWebsiteOptions } from '../drizzle/models';\nimport { CreateUserSpecifiedWebsiteOptionsDto } from './dtos/create-user-specified-website-options.dto';\nimport { UpdateUserSpecifiedWebsiteOptionsDto } from './dtos/update-user-specified-website-options.dto';\n\n@Injectable()\nexport class UserSpecifiedWebsiteOptionsService extends PostyBirbService<'UserSpecifiedWebsiteOptionsSchema'> {\n constructor() {\n super('UserSpecifiedWebsiteOptionsSchema');\n }\n\n async create(\n createDto: CreateUserSpecifiedWebsiteOptionsDto,\n ): Promise {\n this.logger\n .withMetadata(createDto)\n .info(`Creating UserSpecifiedWebsiteOptions '${createDto.accountId}'`);\n await this.throwIfExists(\n and(\n eq(this.schema.accountId, createDto.accountId),\n eq(this.schema.type, createDto.type),\n ),\n );\n return this.repository.insert({\n accountId: createDto.accountId,\n ...createDto,\n });\n }\n\n update(id: EntityId, update: UpdateUserSpecifiedWebsiteOptionsDto) {\n this.logger\n .withMetadata(update)\n .info(`Updating UserSpecifiedWebsiteOptions '${id}'`);\n return this.repository.update(id, { options: update.options });\n }\n\n /**\n * Creates or updates user-specified website options.\n * If options already exist for this account+type combination, updates them.\n * Otherwise, creates new options.\n */\n async upsert(\n dto: CreateUserSpecifiedWebsiteOptionsDto,\n ): Promise {\n const existing = await this.findByAccountAndSubmissionType(\n dto.accountId,\n dto.type,\n );\n\n if (existing) {\n this.logger\n .withMetadata(dto)\n .info(\n `Updating existing UserSpecifiedWebsiteOptions for '${dto.accountId}'`,\n );\n return this.repository.update(existing.id, { options: dto.options });\n }\n\n return this.create(dto);\n }\n\n public findByAccountAndSubmissionType(\n accountId: AccountId,\n type: SubmissionType,\n ) {\n return this.repository.findOne({\n // eslint-disable-next-line @typescript-eslint/no-shadow\n where: (options, { and, eq }) =>\n and(eq(options.accountId, accountId), eq(options.type, type)),\n });\n }\n}\n" }, { "path": "apps/client-server/src/app/utils/blocknote-to-tiptap.ts", "content": "/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { Logger } from '@postybirb/logger';\nimport { Description, TipTapMark, TipTapNode } from '@postybirb/types';\n\nconst logger = Logger();\n\n/**\n * BlockNote block shape (old format):\n * {\n * id: string,\n * type: string,\n * props: Record,\n * content: Array<{ type: 'text', text: string, styles: Record }\n * | { type: 'link', href: string, content: [...textNodes] }\n * | { type: string, props: Record }>,\n * children: Block[]\n * }\n */\n\ninterface BNTextNode {\n type: 'text';\n text: string;\n styles?: Record;\n}\n\ninterface BNLinkNode {\n type: 'link';\n href: string;\n content: BNTextNode[];\n}\n\ninterface BNInlineNode {\n type: string;\n props?: Record;\n}\n\ntype BNInlineContent = BNTextNode | BNLinkNode | BNInlineNode;\n\ninterface BNBlock {\n id?: string;\n type: string;\n props?: Record;\n content?: BNInlineContent[] | 'none' | string;\n children?: BNBlock[];\n}\n\n// Default props that BlockNote includes but are meaningless — strip them\nconst DEFAULT_PROP_VALUES: Record = {\n textColor: 'default',\n backgroundColor: 'default',\n textAlignment: 'left',\n level: 1,\n};\n\n/**\n * Returns true if the value is a BlockNote-format description (old format).\n * BlockNote descriptions are stored as an array of blocks.\n * TipTap descriptions are stored as { type: 'doc', content: [...] }.\n */\nexport function isBlockNoteFormat(desc: unknown): desc is BNBlock[] {\n return Array.isArray(desc);\n}\n\n/**\n * Convert BlockNote styles object to TipTap marks array.\n * e.g. { bold: true, italic: true, textColor: '#ff0000' }\n * → [{ type: 'bold' }, { type: 'italic' }, { type: 'textColor', attrs: { color: '#ff0000' } }]\n */\nfunction convertStyles(styles: Record): TipTapMark[] {\n const marks: TipTapMark[] = [];\n for (const [key, value] of Object.entries(styles)) {\n if (value === false || value === undefined || value === null) {\n continue;\n }\n switch (key) {\n case 'bold':\n case 'italic':\n case 'strike':\n case 'underline':\n case 'code':\n if (value === true) {\n marks.push({ type: key });\n }\n break;\n case 'textColor':\n if (value && value !== 'default') {\n marks.push({ type: 'textStyle', attrs: { color: value } });\n }\n break;\n case 'backgroundColor':\n if (value && value !== 'default') {\n marks.push({ type: 'highlight', attrs: { color: value } });\n }\n break;\n default:\n // Unknown style — preserve as-is\n if (value === true) {\n marks.push({ type: key });\n } else {\n marks.push({ type: key, attrs: { value } });\n }\n break;\n }\n }\n return marks;\n}\n\n/**\n * Convert a BlockNote inline content node to TipTap inline nodes.\n * - text → { type: 'text', text, marks }\n * - link → text nodes with link mark added\n * - custom inline shortcuts → { type, attrs }\n */\nfunction convertInlineContent(\n inlineNode: BNInlineContent,\n): TipTapNode[] {\n if (inlineNode.type === 'text') {\n const bn = inlineNode as BNTextNode;\n const node: TipTapNode = { type: 'text', text: bn.text };\n if (bn.styles && Object.keys(bn.styles).length > 0) {\n const marks = convertStyles(bn.styles);\n if (marks.length > 0) {\n node.marks = marks;\n }\n }\n return [node];\n }\n\n if (inlineNode.type === 'link') {\n const bn = inlineNode as BNLinkNode;\n const linkMark: TipTapMark = {\n type: 'link',\n attrs: { href: bn.href, target: '_blank', rel: 'noopener noreferrer nofollow' },\n };\n // Each text node inside the link gets the link mark added\n return (bn.content || []).flatMap((textNode) => {\n const converted = convertInlineContent(textNode);\n return converted.map((n) => {\n if (n.type === 'text') {\n const existing = n.marks || [];\n return { ...n, marks: [...existing, linkMark] };\n }\n return n;\n });\n });\n }\n\n // Custom inline nodes: customShortcut, titleShortcut, tagsShortcut,\n // contentWarningShortcut, username\n const bn = inlineNode as BNInlineNode;\n const attrs: Record = {};\n if (bn.props) {\n for (const [key, value] of Object.entries(bn.props)) {\n if (value !== undefined && value !== null && value !== '') {\n attrs[key] = value;\n }\n }\n }\n const node: TipTapNode = { type: bn.type };\n if (Object.keys(attrs).length > 0) {\n node.attrs = attrs;\n }\n return [node];\n}\n\n/**\n * Convert a single BlockNote block to a TipTap node.\n */\nfunction convertBlock(block: BNBlock): TipTapNode[] {\n const {type} = block;\n const props = block.props || {};\n\n // Map BlockNote block types to TipTap node types\n switch (type) {\n case 'paragraph': {\n const node: TipTapNode = { type: 'paragraph' };\n const attrs = extractBlockAttrs(props);\n if (Object.keys(attrs).length > 0) {\n node.attrs = attrs;\n }\n if (\n block.content &&\n Array.isArray(block.content) &&\n block.content.length > 0\n ) {\n node.content = block.content.flatMap(convertInlineContent);\n }\n const result: TipTapNode[] = [node];\n // BlockNote children are nested blocks (e.g. indented content)\n if (block.children && block.children.length > 0) {\n result.push(...block.children.flatMap(convertBlock));\n }\n return result;\n }\n\n case 'heading': {\n const level = props.level || 1;\n const node: TipTapNode = {\n type: 'heading',\n attrs: { level, ...extractBlockAttrs(props, ['level']) },\n };\n if (\n block.content &&\n Array.isArray(block.content) &&\n block.content.length > 0\n ) {\n node.content = block.content.flatMap(convertInlineContent);\n }\n const result: TipTapNode[] = [node];\n if (block.children && block.children.length > 0) {\n result.push(...block.children.flatMap(convertBlock));\n }\n return result;\n }\n\n case 'bulletListItem': {\n // BlockNote uses flat blocks with type 'bulletListItem'\n // TipTap uses bulletList > listItem > paragraph\n const paragraph: TipTapNode = { type: 'paragraph' };\n if (\n block.content &&\n Array.isArray(block.content) &&\n block.content.length > 0\n ) {\n paragraph.content = block.content.flatMap(convertInlineContent);\n }\n const listItem: TipTapNode = {\n type: 'listItem',\n content: [paragraph],\n };\n // Children of a list item become nested list items inside a sub-list\n if (block.children && block.children.length > 0) {\n const childItems = block.children.flatMap((child) =>\n convertBlock(child),\n );\n const subList: TipTapNode = {\n type: 'bulletList',\n content: childItems.filter((n) => n.type === 'listItem'),\n };\n if (subList.content && subList.content.length > 0) {\n listItem.content = [paragraph, subList];\n }\n }\n // Wrap in bulletList — caller will merge adjacent ones\n return [{ type: 'bulletList', content: [listItem] }];\n }\n\n case 'numberedListItem': {\n const paragraph: TipTapNode = { type: 'paragraph' };\n if (\n block.content &&\n Array.isArray(block.content) &&\n block.content.length > 0\n ) {\n paragraph.content = block.content.flatMap(convertInlineContent);\n }\n const listItem: TipTapNode = {\n type: 'listItem',\n content: [paragraph],\n };\n if (block.children && block.children.length > 0) {\n const childItems = block.children.flatMap((child) =>\n convertBlock(child),\n );\n const subList: TipTapNode = {\n type: 'orderedList',\n content: childItems.filter((n) => n.type === 'listItem'),\n };\n if (subList.content && subList.content.length > 0) {\n listItem.content = [paragraph, subList];\n }\n }\n return [{ type: 'orderedList', content: [listItem] }];\n }\n\n case 'blockquote': {\n // BlockNote stores blockquote content as inline content in a single block\n const paragraph: TipTapNode = { type: 'paragraph' };\n if (\n block.content &&\n Array.isArray(block.content) &&\n block.content.length > 0\n ) {\n paragraph.content = block.content.flatMap(convertInlineContent);\n }\n const result: TipTapNode[] = [\n { type: 'blockquote', content: [paragraph] },\n ];\n if (block.children && block.children.length > 0) {\n result.push(...block.children.flatMap(convertBlock));\n }\n return result;\n }\n\n case 'codeBlock': {\n const node: TipTapNode = { type: 'codeBlock' };\n if (props.language) {\n node.attrs = { language: props.language };\n }\n // Code blocks in BlockNote have text content\n if (\n block.content &&\n Array.isArray(block.content) &&\n block.content.length > 0\n ) {\n node.content = block.content\n .filter(\n (c): c is BNTextNode => c.type === 'text' && 'text' in c,\n )\n .map((c) => ({ type: 'text', text: c.text }));\n }\n return [node];\n }\n\n // Custom block nodes — map directly\n case 'defaultShortcut': {\n const node: TipTapNode = { type: 'defaultShortcut' };\n const attrs: Record = {};\n if (props.only) attrs.only = props.only;\n if (Object.keys(attrs).length > 0) {\n node.attrs = attrs;\n }\n return [node];\n }\n\n // Fallback for any other block type (e.g. horizontal rule, images, etc.)\n default: {\n // Map known aliases\n if (type === 'horizontalRule' || type === 'rule') {\n return [{ type: 'horizontalRule' }];\n }\n\n // Generic passthrough — preserve type and convert content\n const node: TipTapNode = { type };\n const attrs = extractBlockAttrs(props);\n if (Object.keys(attrs).length > 0) {\n node.attrs = attrs;\n }\n if (\n block.content &&\n Array.isArray(block.content) &&\n block.content.length > 0\n ) {\n node.content = block.content.flatMap(convertInlineContent);\n }\n return [node];\n }\n }\n}\n\n/**\n * Extract meaningful attrs from BlockNote props, stripping defaults.\n */\nfunction extractBlockAttrs(\n props: Record,\n skipKeys: string[] = [],\n): Record {\n const attrs: Record = {};\n for (const [key, value] of Object.entries(props)) {\n if (skipKeys.includes(key)) continue;\n if (value === undefined || value === null) continue;\n if (DEFAULT_PROP_VALUES[key] === value) continue;\n if (key === 'textAlignment' && value !== 'left') {\n attrs.textAlign = value;\n } else if (\n key !== 'textColor' &&\n key !== 'backgroundColor' &&\n key !== 'textAlignment'\n ) {\n attrs[key] = value;\n }\n }\n return attrs;\n}\n\n/**\n * Merge adjacent list nodes of the same type.\n * BlockNote creates one bulletList/orderedList per list item,\n * but TipTap expects all items in a single list node.\n */\nfunction mergeAdjacentLists(nodes: TipTapNode[]): TipTapNode[] {\n const result: TipTapNode[] = [];\n for (const node of nodes) {\n const prev = result[result.length - 1];\n if (\n prev &&\n prev.type === node.type &&\n (node.type === 'bulletList' || node.type === 'orderedList') &&\n Array.isArray(prev.content) &&\n Array.isArray(node.content)\n ) {\n prev.content = [...prev.content, ...node.content];\n } else {\n result.push(node);\n }\n }\n return result;\n}\n\n/**\n * Convert an entire BlockNote description (array of blocks) to TipTap JSON.\n */\nexport function convertBlockNoteToTipTap(blocks: BNBlock[]): Description {\n if (!Array.isArray(blocks) || blocks.length === 0) {\n return { type: 'doc', content: [] };\n }\n\n const converted = blocks.flatMap(convertBlock);\n const merged = mergeAdjacentLists(converted);\n\n return {\n type: 'doc',\n content: merged,\n };\n}\n\n/**\n * Migrate a Description field if it is in BlockNote format.\n * Returns the migrated TipTap doc, or the original if already in TipTap format.\n */\nexport function migrateDescription(desc: unknown): Description {\n if (!desc) {\n return { type: 'doc', content: [] };\n }\n\n // Already TipTap format\n if (\n typeof desc === 'object' &&\n !Array.isArray(desc) &&\n (desc as any).type === 'doc'\n ) {\n return desc as Description;\n }\n\n // BlockNote format (array of blocks)\n if (Array.isArray(desc)) {\n try {\n return convertBlockNoteToTipTap(desc as BNBlock[]);\n } catch (err) {\n logger.error(\n `Failed to migrate BlockNote description: ${(err as Error).message}`,\n (err as Error).stack,\n );\n return { type: 'doc', content: [] };\n }\n }\n\n // Unknown format\n logger.warn(`Unknown description format, resetting to empty`);\n return { type: 'doc', content: [] };\n}\n" }, { "path": "apps/client-server/src/app/utils/coerce.util.ts", "content": "export class Coerce {\n static boolean(value: string | boolean): boolean {\n return !!value.toString().match(/^(true|[1-9][0-9]*|[0-9]*[1-9]+|yes)$/i);\n }\n}\n" }, { "path": "apps/client-server/src/app/utils/filesize.util.ts", "content": "/**\n * Defines simple file size conversion methods\n * @class FileSize\n */\nexport default class FileSize {\n static megabytes(size: number): number {\n return size * 1000000;\n }\n\n static bytesToMB(size: number): number {\n return size / 1000000;\n }\n}\n" }, { "path": "apps/client-server/src/app/utils/html-parser.util.ts", "content": "import { NotFoundException } from '@nestjs/common';\n\nexport default class HtmlParserUtil {\n public static getInputValue(html: string, name: string, index = 0): string {\n // eslint-disable-next-line no-param-reassign\n index = index || 0;\n const inputs = (html.match(//gim) || [])\n .filter((input?: string) => input?.includes(`name=\"${name}\"`))\n .map((input) => input.match(/value=\"(.*?)\"/)[1]);\n\n const picked = inputs[index];\n if (!picked) {\n throw new NotFoundException(`Could not find form key: ${name}[${index}]`);\n }\n\n return picked;\n }\n}\n" }, { "path": "apps/client-server/src/app/utils/select-option.util.ts", "content": "import { SelectOption } from '@postybirb/form-builder';\n\nexport class SelectOptionUtil {\n static findOptionById(\n options: SelectOption[],\n id: string,\n ): SelectOption | undefined {\n for (const option of options) {\n if (option.value === id) {\n return option;\n }\n if ('items' in option && option.items) {\n const found = this.findOptionById(option.items, id);\n if (found) {\n return found;\n }\n }\n }\n return undefined;\n }\n}\n" }, { "path": "apps/client-server/src/app/utils/wait.util.ts", "content": "import { setInterval } from 'timers/promises';\n\nexport function wait(milliseconds: number): Promise {\n return new Promise((resolve) => {\n setTimeout(resolve, milliseconds);\n });\n}\n\nexport async function waitUntil(\n fn: () => boolean,\n milliseconds: number,\n): Promise {\n if (fn()) {\n return;\n }\n\n const interval = setInterval(milliseconds);\n\n // eslint-disable-next-line no-restricted-syntax, @typescript-eslint/no-unused-vars\n for await (const i of interval) {\n if (fn()) {\n break;\n }\n }\n}\n\nexport async function waitUntilPromised(\n fn: () => Promise,\n milliseconds: number,\n): Promise {\n if (await fn()) {\n return;\n }\n\n const interval = setInterval(milliseconds);\n\n // eslint-disable-next-line no-restricted-syntax, @typescript-eslint/no-unused-vars\n for await (const i of interval) {\n if (await fn()) {\n break;\n }\n }\n}\n" }, { "path": "apps/client-server/src/app/validation/validation.module.ts", "content": "import { Module } from '@nestjs/common';\nimport { FileConverterModule } from '../file-converter/file-converter.module';\nimport { FileModule } from '../file/file.module';\nimport { PostParsersModule } from '../post-parsers/post-parsers.module';\nimport { WebsitesModule } from '../websites/websites.module';\nimport { ValidationService } from './validation.service';\n\n@Module({\n imports: [WebsitesModule, PostParsersModule, FileConverterModule, FileModule],\n providers: [ValidationService],\n exports: [ValidationService],\n})\nexport class ValidationModule {}\n" }, { "path": "apps/client-server/src/app/validation/validation.service.spec.ts", "content": "import { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport { FileConverterService } from '../file-converter/file-converter.service';\nimport { FileModule } from '../file/file.module';\nimport { FileService } from '../file/file.service';\nimport { CreateFileService } from '../file/services/create-file.service';\nimport { UpdateFileService } from '../file/services/update-file.service';\nimport { SharpInstanceManager } from '../image-processing/sharp-instance-manager';\nimport { PostParsersModule } from '../post-parsers/post-parsers.module';\nimport { PostParsersService } from '../post-parsers/post-parsers.service';\nimport { WebsiteImplProvider } from '../websites/implementations/provider';\nimport { WebsiteRegistryService } from '../websites/website-registry.service';\nimport { WebsitesModule } from '../websites/websites.module';\nimport { ValidationService } from './validation.service';\n\ndescribe('ValidationService', () => {\n let service: ValidationService;\n\n beforeEach(async () => {\n clearDatabase();\n const module: TestingModule = await Test.createTestingModule({\n imports: [WebsitesModule, PostParsersModule, FileModule],\n providers: [\n WebsiteImplProvider,\n ValidationService,\n WebsiteRegistryService,\n PostParsersService,\n FileConverterService,\n FileService,\n CreateFileService,\n UpdateFileService,\n SharpInstanceManager,\n ],\n }).compile();\n\n service = module.get(ValidationService);\n });\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n});\n" }, { "path": "apps/client-server/src/app/validation/validation.service.ts", "content": "import { Injectable } from '@nestjs/common';\nimport { Logger } from '@postybirb/logger';\nimport {\n EntityId,\n ISubmission,\n IWebsiteOptions,\n PostData,\n SimpleValidationResult,\n SubmissionId,\n SubmissionType,\n ValidationResult,\n} from '@postybirb/types';\nimport { Account, Submission, WebsiteOptions } from '../drizzle/models';\nimport { FileConverterService } from '../file-converter/file-converter.service';\nimport { FileService } from '../file/file.service';\nimport { PostParsersService } from '../post-parsers/post-parsers.service';\nimport DefaultWebsite from '../websites/implementations/default/default.website';\nimport { DefaultWebsiteOptions } from '../websites/models/default-website-options';\nimport { isFileWebsite } from '../websites/models/website-modifiers/file-website';\nimport { isMessageWebsite } from '../websites/models/website-modifiers/message-website';\nimport { UnknownWebsite } from '../websites/website';\nimport { WebsiteRegistryService } from '../websites/website-registry.service';\nimport { validators } from './validators';\nimport {\n FieldValidator,\n Validator,\n ValidatorParams,\n} from './validators/validator.type';\n\ntype ValidationCacheRecord = {\n submissionLastUpdatedTimestamp: string;\n results: Record<\n EntityId, // WebsiteOptionId\n {\n validationResult: ValidationResult;\n websiteOptionLastUpdatedTimestamp: string;\n }\n >;\n};\n\n@Injectable()\nexport class ValidationService {\n private readonly logger = Logger(this.constructor.name);\n\n private readonly validations: Validator[] = validators;\n\n private readonly validationCache = new Map<\n SubmissionId,\n ValidationCacheRecord\n >();\n\n constructor(\n private readonly websiteRegistry: WebsiteRegistryService,\n private readonly postParserService: PostParsersService,\n private readonly fileConverterService: FileConverterService,\n private readonly fileService: FileService,\n ) {}\n\n private getCachedValidation(\n submissionId: SubmissionId,\n ): ValidationCacheRecord | undefined {\n return this.validationCache.get(submissionId);\n }\n\n private clearCachedValidation(submissionId: SubmissionId) {\n this.validationCache.delete(submissionId);\n }\n\n private setCachedValidation(\n submission: Submission,\n websiteOption: WebsiteOptions,\n validationResult: ValidationResult,\n ) {\n const cachedValidation = this.getCachedValidation(submission.id);\n if (!cachedValidation) {\n this.validationCache.set(submission.id, {\n submissionLastUpdatedTimestamp: submission.updatedAt,\n results: {\n [websiteOption.id]: {\n validationResult,\n websiteOptionLastUpdatedTimestamp: websiteOption.updatedAt,\n },\n },\n });\n } else {\n cachedValidation.results[websiteOption.id] = {\n validationResult,\n websiteOptionLastUpdatedTimestamp: websiteOption.updatedAt,\n };\n }\n }\n\n /**\n * Validate a submission for all website options.\n *\n * @param {ISubmission} submission\n * @return {*} {Promise}\n */\n public async validateSubmission(\n submission: Submission,\n ): Promise {\n if (this.isStale(submission)) {\n this.clearCachedValidation(submission.id);\n }\n return Promise.all(\n submission.options.map((website) => this.validate(submission, website)),\n );\n }\n\n /**\n * Check if a submission is stale by comparing the last updated timestamps of\n * the submission and the website options.\n *\n * @param {Submission} submission\n * @return {*} {boolean}\n */\n private isStale(submission: Submission): boolean {\n const cachedValidation = this.getCachedValidation(submission.id);\n if (!cachedValidation) {\n return false;\n }\n if (\n cachedValidation.submissionLastUpdatedTimestamp !== submission.updatedAt\n ) {\n return true;\n }\n\n return submission.options.some(\n (website) =>\n cachedValidation.results[website.id] &&\n cachedValidation.results[website.id]\n .websiteOptionLastUpdatedTimestamp !== website.updatedAt,\n );\n }\n\n /**\n * Validate an individual website option.\n *\n * @param {ISubmission} submission\n * @param {IWebsiteOptions} websiteOption\n * @return {*} {Promise}\n */\n public async validate(\n submission: Submission,\n websiteOption: WebsiteOptions,\n ): Promise {\n try {\n const cachedValidation = this.getCachedValidation(submission.id);\n if (\n cachedValidation &&\n cachedValidation.results[websiteOption.id] &&\n cachedValidation.results[websiteOption.id]\n .websiteOptionLastUpdatedTimestamp === websiteOption.updatedAt\n ) {\n return cachedValidation.results[websiteOption.id].validationResult;\n }\n\n const website = websiteOption.isDefault\n ? new DefaultWebsite(new Account(websiteOption.account))\n : this.websiteRegistry.findInstance(websiteOption.account);\n\n if (!website) {\n this.logger.error(\n `Failed to find website instance for account ${websiteOption.accountId}`,\n );\n throw new Error(\n `Failed to find website instance for account ${websiteOption.accountId}`,\n );\n }\n // All sub-validations mutate the result object\n const result: ValidationResult = {\n id: websiteOption.id,\n account: website.account.toDTO(),\n warnings: [],\n errors: [],\n };\n\n const data = await this.postParserService.parse(\n submission,\n website,\n websiteOption,\n );\n\n const defaultOptions: IWebsiteOptions = submission.options.find(\n (o) => o.isDefault,\n );\n const defaultOpts = Object.assign(new DefaultWebsiteOptions(), {\n ...defaultOptions.data,\n });\n const mergedWebsiteOptions = Object.assign(\n website.getModelFor(submission.type),\n websiteOption.data,\n ).mergeDefaults(defaultOpts);\n\n const params: ValidatorParams = {\n result,\n validator: new FieldValidator(result.errors, result.warnings),\n websiteInstance: website,\n data,\n submission,\n fileConverterService: this.fileConverterService,\n fileService: this.fileService,\n mergedWebsiteOptions,\n };\n\n // eslint-disable-next-line no-restricted-syntax\n for (const validation of this.validations) {\n await validation(params);\n }\n\n const instanceResult = await this.validateWebsiteInstance(\n websiteOption.id,\n submission,\n website,\n data,\n );\n\n result.warnings.push(...(instanceResult.warnings ?? []));\n result.errors.push(...(instanceResult.errors ?? []));\n\n this.setCachedValidation(submission, websiteOption, result);\n return result;\n } catch (error) {\n this.logger.warn(\n `Failed to validate website options ${websiteOption.id}`,\n error,\n );\n return {\n id: websiteOption.id,\n account: new Account(websiteOption.account).toDTO(),\n warnings: [\n {\n id: 'validation.failed',\n values: {\n message: error.message,\n },\n },\n ],\n };\n }\n }\n\n private async validateWebsiteInstance(\n websiteId: EntityId,\n submission: ISubmission,\n website: UnknownWebsite,\n postData: PostData,\n ): Promise {\n let result: SimpleValidationResult;\n try {\n if (submission.type === SubmissionType.FILE && isFileWebsite(website)) {\n result = await website.onValidateFileSubmission(postData);\n }\n\n if (\n submission.type === SubmissionType.MESSAGE &&\n isMessageWebsite(website)\n ) {\n result = await website.onValidateMessageSubmission(postData);\n }\n\n return {\n id: websiteId,\n account: website.account.toDTO(),\n warnings: result?.warnings,\n errors: result?.errors,\n };\n } catch (error) {\n this.logger.warn(\n `Failed to validate website instance for submission ${submission.id}, website ${websiteId}, type ${submission.type}, instance ${website.constructor.name}`,\n error,\n );\n return {\n id: websiteId,\n account: website.account.toDTO(),\n warnings: [\n {\n id: 'validation.failed',\n values: {\n message: error.message,\n },\n },\n ],\n };\n }\n }\n}\n" }, { "path": "apps/client-server/src/app/validation/validators/common-field-validators.ts", "content": "import { ValidatorParams } from './validator.type';\n\n/**\n * Validates that a required text field (input/textarea) is not empty.\n */\nexport async function validateRequiredTextField({\n result,\n data,\n mergedWebsiteOptions,\n}: ValidatorParams) {\n const fields = mergedWebsiteOptions.getFormFields();\n\n for (const [fieldName, field] of Object.entries(fields)) {\n // Skip if field is not required or hidden\n if (!field.required || field.hidden) continue;\n\n // Only check text field types (includes title, content warning etc.)\n if (field.formField !== 'input' && field.formField !== 'textarea') continue;\n\n // Check if the field is a title field\n // Check if the field is a content warning field\n if (\n (field.formField === 'input' && field.label === 'title') ||\n (field.formField === 'input' && field.label === 'contentWarning')\n ) {\n continue;\n }\n\n const value = data.options[fieldName] as string;\n\n // Check if the value is empty\n if (!value || value.trim() === '') {\n result.errors.push({\n id: 'validation.field.required',\n field: fieldName,\n values: {},\n });\n }\n }\n}\n\n/**\n * Validates that a required select field has a value selected.\n */\nexport async function validateRequiredSelectField({\n result,\n data,\n mergedWebsiteOptions,\n}: ValidatorParams) {\n const fields = mergedWebsiteOptions.getFormFields();\n\n for (const [fieldName, field] of Object.entries(fields)) {\n // Skip if field is not required or hidden\n if (!field.required || field.hidden) continue;\n\n // Only check select fields\n if (field.formField !== 'select') continue;\n\n // Skip fields with min selected, they gets handled by selectFieldValidator\n if ('minSelected' in field && typeof field.minSelected === 'number')\n continue;\n\n const value = data.options[fieldName];\n const isEmpty = Array.isArray(value) ? value.length === 0 : !value;\n\n if (isEmpty) {\n result.errors.push({\n id: 'validation.field.required',\n field: fieldName,\n values: {},\n });\n }\n }\n}\n\n/**\n * Validates that a required radio field has a value selected.\n */\nexport async function validateRequiredRadioField({\n result,\n data,\n mergedWebsiteOptions,\n}: ValidatorParams) {\n const fields = mergedWebsiteOptions.getFormFields();\n\n for (const [fieldName, field] of Object.entries(fields)) {\n // Skip if field is not required or hidden\n if (!field.required || field.hidden) continue;\n\n // Only check radio fields (includes rating fields)\n if (field.formField !== 'radio' && field.formField !== 'rating') continue;\n\n const value = data.options[fieldName];\n\n if (!value) {\n result.errors.push({\n id: 'validation.field.required',\n field: fieldName,\n values: {},\n });\n }\n }\n}\n\n/**\n * Validates that a required boolean field (checkbox) is checked.\n */\nexport async function validateRequiredBooleanField({\n result,\n data,\n mergedWebsiteOptions,\n}: ValidatorParams) {\n const fields = mergedWebsiteOptions.getFormFields();\n\n for (const [fieldName, field] of Object.entries(fields)) {\n // Skip if field is not required or hidden\n if (!field.required || field.hidden) continue;\n\n // Only check checkbox fields\n if (field.formField !== 'checkbox') continue;\n\n const value = data.options[fieldName] as boolean;\n\n if (typeof value !== 'boolean') {\n result.errors.push({\n id: 'validation.field.required',\n field: fieldName,\n values: {},\n });\n }\n }\n}\n\n/**\n * Validates that a required description field has content.\n */\nexport async function validateRequiredDescriptionField({\n result,\n data,\n mergedWebsiteOptions,\n}: ValidatorParams) {\n const fields = mergedWebsiteOptions.getFormFields();\n\n for (const [fieldName, field] of Object.entries(fields)) {\n // Skip if field is not required or hidden\n if (!field.required || field.hidden) continue;\n\n // Only check description fields\n if (field.formField !== 'description') continue;\n\n // Description field value structure\n let value: string = data.options[fieldName] || '';\n value = value.replaceAll('
      ', '').trim();\n\n if (!value || value.length === 0) {\n result.errors.push({\n id: 'validation.field.required',\n field: fieldName,\n values: {},\n });\n }\n }\n}\n" }, { "path": "apps/client-server/src/app/validation/validators/datetime-field-validators.ts", "content": "import { ValidatorParams } from './validator.type';\n\n/**\n * Validates that a datetime field value is a valid ISO date string.\n */\nexport async function validateDateTimeFormat({\n result,\n data,\n mergedWebsiteOptions,\n}: ValidatorParams) {\n const fields = mergedWebsiteOptions.getFormFields();\n\n for (const [fieldName, field] of Object.entries(fields)) {\n // Skip if field is hidden\n if (field.hidden) continue;\n\n // Only check datetime fields\n if (field.formField !== 'datetime') continue;\n\n const value = data.options[fieldName] as string;\n\n // Skip if no value (empty is handled by required validator)\n if (!value || value.trim() === '') continue;\n\n // Try to parse as ISO date string\n const date = new Date(value);\n if (Number.isNaN(date.getTime())) {\n result.errors.push({\n id: 'validation.datetime.invalid-format',\n field: fieldName,\n values: {\n value,\n },\n });\n }\n }\n}\n\n/**\n * Validates that a datetime field value is not before the minimum allowed date.\n */\nexport async function validateDateTimeMinimum({\n result,\n data,\n mergedWebsiteOptions,\n}: ValidatorParams) {\n const fields = mergedWebsiteOptions.getFormFields();\n\n for (const [fieldName, field] of Object.entries(fields)) {\n // Skip if field is hidden\n if (field.hidden) continue;\n\n // Only check datetime fields\n if (field.formField !== 'datetime') continue;\n\n // Skip if no min constraint\n if (!('min' in field) || !field.min) continue;\n\n const value = data.options[fieldName] as string;\n\n // Skip if no value\n if (!value || value.trim() === '') continue;\n\n const date = new Date(value);\n const minDate = new Date(field.min);\n\n // Skip if date is invalid (handled by format validator)\n if (Number.isNaN(date.getTime())) continue;\n\n if (date < minDate) {\n result.errors.push({\n id: 'validation.datetime.min',\n field: fieldName,\n values: {\n currentDate: value,\n minDate: field.min,\n },\n });\n }\n }\n}\n\n/**\n * Validates that a datetime field value is not after the maximum allowed date.\n */\nexport async function validateDateTimeMaximum({\n result,\n data,\n mergedWebsiteOptions,\n}: ValidatorParams) {\n const fields = mergedWebsiteOptions.getFormFields();\n\n for (const [fieldName, field] of Object.entries(fields)) {\n // Skip if field is hidden\n if (field.hidden) continue;\n\n // Only check datetime fields\n if (field.formField !== 'datetime') continue;\n\n // Skip if no max constraint\n if (!('max' in field) || !field.max) continue;\n\n const value = data.options[fieldName] as string;\n\n // Skip if no value\n if (!value || value.trim() === '') continue;\n\n const date = new Date(value);\n const maxDate = new Date(field.max);\n\n // Skip if date is invalid (handled by format validator)\n if (Number.isNaN(date.getTime())) continue;\n\n if (date > maxDate) {\n result.errors.push({\n id: 'validation.datetime.max',\n field: fieldName,\n values: {\n currentDate: value,\n maxDate: field.max,\n },\n });\n }\n }\n}\n\n/**\n * Validates that a datetime field value is within the allowed range (min and max).\n * This is a convenience validator that checks both min and max constraints.\n */\nexport async function validateDateTimeRange({\n result,\n data,\n mergedWebsiteOptions,\n}: ValidatorParams) {\n const fields = mergedWebsiteOptions.getFormFields();\n\n for (const [fieldName, field] of Object.entries(fields)) {\n // Skip if field is hidden\n if (field.hidden) continue;\n\n // Only check datetime fields\n if (field.formField !== 'datetime') continue;\n\n // Skip if no range constraints\n const hasMin = 'min' in field && field.min;\n const hasMax = 'max' in field && field.max;\n if (!hasMin || !hasMax) continue;\n\n const value = data.options[fieldName] as string;\n\n // Skip if no value\n if (!value || value.trim() === '') continue;\n\n const date = new Date(value);\n const minDate = new Date(field.min);\n const maxDate = new Date(field.max);\n\n // Skip if date is invalid (handled by format validator)\n if (Number.isNaN(date.getTime())) continue;\n\n if (date < minDate || date > maxDate) {\n result.errors.push({\n id: 'validation.datetime.range',\n field: fieldName,\n values: {\n currentDate: value,\n minDate: field.min,\n maxDate: field.max,\n },\n });\n }\n }\n}\n" }, { "path": "apps/client-server/src/app/validation/validators/description-validators.ts", "content": "import { DescriptionType } from '@postybirb/types';\nimport DefaultWebsite from '../../websites/implementations/default/default.website';\nimport { ValidatorParams } from './validator.type';\n\nexport async function validateDescriptionMaxLength({\n data,\n mergedWebsiteOptions,\n validator,\n}: ValidatorParams) {\n const { hidden, descriptionType, maxDescriptionLength } =\n mergedWebsiteOptions.getFormFieldFor('description');\n\n if (\n descriptionType === undefined ||\n descriptionType === DescriptionType.NONE ||\n hidden\n ) {\n return;\n }\n\n const { description } = data.options;\n const maxLength = maxDescriptionLength ?? Number.MAX_SAFE_INTEGER;\n if (description.length > maxLength) {\n validator.warning(\n 'validation.description.max-length',\n { currentLength: description.length, maxLength },\n 'description',\n );\n }\n}\n\nexport async function validateDescriptionMinLength({\n data,\n mergedWebsiteOptions,\n validator,\n}: ValidatorParams) {\n const { hidden, descriptionType, minDescriptionLength } =\n mergedWebsiteOptions.getFormFieldFor('description');\n\n if (\n descriptionType === undefined ||\n descriptionType === DescriptionType.NONE ||\n hidden\n ) {\n return;\n }\n\n const { description } = data.options;\n const minLength = minDescriptionLength ?? -1;\n if (description.length < minLength) {\n validator.error(\n 'validation.description.min-length',\n { minLength, currentLength: description.length },\n 'description',\n );\n }\n}\n\nexport async function validateTagsPresence({\n data,\n mergedWebsiteOptions,\n validator,\n websiteInstance,\n}: ValidatorParams) {\n if (websiteInstance instanceof DefaultWebsite) return;\n\n const tagsField = mergedWebsiteOptions.getFormFieldFor('tags');\n const descriptionField = mergedWebsiteOptions.getFormFieldFor('description');\n const { tags, description } = data.options;\n\n if (tagsField.hidden || descriptionField.hidden) return;\n if (!description || !tags.length) return;\n\n const presentTags = tags.filter((e) => description.includes(`#${e}`));\n\n if (descriptionField.expectsInlineTags) {\n if (presentTags.length === 0) {\n // Tags are missing in the description\n validator.warning(\n 'validation.description.missing-tags',\n {},\n 'description',\n );\n }\n } else if (presentTags.length === tags.length) {\n // All tags are in description\n validator.warning(\n 'validation.description.unexpected-tags',\n {},\n 'description',\n );\n }\n}\n\nexport async function validateTitlePresence({\n data,\n mergedWebsiteOptions,\n validator,\n websiteInstance,\n}: ValidatorParams) {\n if (websiteInstance instanceof DefaultWebsite) return;\n\n const titleField = mergedWebsiteOptions.getFormFieldFor('tags');\n const descriptionField = mergedWebsiteOptions.getFormFieldFor('description');\n const { title, description } = data.options;\n if (titleField.hidden || descriptionField.hidden) return;\n if (!description || !title) return;\n\n const hasTitleText = description.includes(title);\n\n if (descriptionField.expectsInlineTitle) {\n if (!hasTitleText) {\n // Title is missing in the description\n validator.warning(\n 'validation.description.missing-title',\n {},\n 'description',\n );\n }\n } else if (hasTitleText) {\n // Title is in the description\n validator.warning(\n 'validation.description.unexpected-title',\n {},\n 'description',\n );\n }\n}\n" }, { "path": "apps/client-server/src/app/validation/validators/file-submission-validators.ts", "content": "import {\n FileSubmission,\n FileType,\n ISubmission,\n ISubmissionFile,\n SubmissionType,\n ValidationMessage,\n} from '@postybirb/types';\nimport { getFileType } from '@postybirb/utils/file-type';\nimport { parse } from 'path';\nimport { getSupportedFileSize } from '../../websites/decorators/supports-files.decorator';\nimport DefaultWebsite from '../../websites/implementations/default/default.website';\nimport {\n ImplementedFileWebsite,\n isFileWebsite,\n} from '../../websites/models/website-modifiers/file-website';\nimport { UnknownWebsite } from '../../websites/website';\nimport { ValidatorParams } from './validator.type';\n\nfunction isFileHandlingWebsite(\n websiteInstance: UnknownWebsite,\n): websiteInstance is ImplementedFileWebsite {\n return isFileWebsite(websiteInstance);\n}\n\nfunction isFileSubmission(\n submission: ISubmission,\n): submission is FileSubmission {\n return submission.type === SubmissionType.FILE;\n}\n\nfunction isFileFiltered(\n file: ISubmissionFile,\n submission: FileSubmission,\n websiteInstance: UnknownWebsite,\n): boolean {\n if (file.metadata?.ignoredWebsites?.includes(websiteInstance.accountId)) {\n return true;\n }\n return false;\n}\n\nasync function validateTextFileRequiresFallback({\n result,\n websiteInstance,\n submission,\n fileService,\n}: ValidatorParams & { file: ISubmissionFile }) {\n if (\n !isFileHandlingWebsite(websiteInstance) ||\n !isFileSubmission(submission) ||\n websiteInstance instanceof DefaultWebsite\n ) {\n return;\n }\n\n for (const file of submission.files) {\n if (isFileFiltered(file, submission, websiteInstance)) {\n continue;\n }\n if (getFileType(file.fileName) === FileType.TEXT) {\n const supportedMimeTypes =\n websiteInstance.decoratedProps.fileOptions?.acceptedMimeTypes ?? [];\n if (supportedMimeTypes.length === 0) {\n // Assume empty to accept all file types if no accepted mime types are specified\n continue;\n }\n // Check if the alt file has content by querying its size\n let altFileHasContent = false;\n if (file.altFileId) {\n const altFileSize = await fileService.getAltFileSize(file.altFileId);\n altFileHasContent = altFileSize > 0;\n }\n // Fail validation if the file is not supported and alt file is empty or missing\n if (!supportedMimeTypes.includes(file.mimeType) && !altFileHasContent) {\n result.errors.push({\n id: 'validation.file.text-file-no-fallback',\n field: 'files',\n values: {\n fileName: file.fileName,\n fileExtension: parse(file.fileName).ext,\n fileId: file.id,\n },\n });\n }\n }\n }\n}\n\nexport async function validateNotAllFilesIgnored({\n result,\n websiteInstance,\n submission,\n}: ValidatorParams) {\n if (\n !isFileHandlingWebsite(websiteInstance) ||\n !isFileSubmission(submission) ||\n websiteInstance instanceof DefaultWebsite\n ) {\n return;\n }\n\n const numFiles = submission.files.filter(\n (file) => !isFileFiltered(file, submission, websiteInstance),\n ).length;\n if (numFiles === 0) {\n result.warnings.push({\n id: 'validation.file.all-ignored',\n field: 'files',\n values: {},\n });\n }\n}\n\nexport async function validateAcceptedFiles({\n result,\n websiteInstance,\n submission,\n data,\n fileConverterService,\n ...rest\n}: ValidatorParams) {\n if (\n !isFileHandlingWebsite(websiteInstance) ||\n !isFileSubmission(submission) ||\n websiteInstance instanceof DefaultWebsite\n ) {\n return;\n }\n\n const acceptedMimeTypes =\n websiteInstance.decoratedProps.fileOptions?.acceptedMimeTypes ?? [];\n const supportedFileTypes =\n websiteInstance.decoratedProps.fileOptions?.supportedFileTypes ?? [];\n\n if (!acceptedMimeTypes.length && !supportedFileTypes.length) {\n return;\n }\n\n submission.files.forEach((file) => {\n if (isFileFiltered(file, submission, websiteInstance)) {\n return;\n }\n if (!acceptedMimeTypes.includes(file.mimeType)) {\n const fileType = getFileType(file.fileName);\n if (!supportedFileTypes.includes(fileType)) {\n result.errors.push({\n id: 'validation.file.unsupported-file-type',\n field: 'files',\n values: {\n fileName: file.fileName,\n fileType: getFileType(file.fileName),\n fileId: file.id,\n },\n });\n }\n\n if (fileType === FileType.TEXT) {\n validateTextFileRequiresFallback({\n result,\n websiteInstance,\n submission,\n file,\n data,\n fileConverterService,\n ...rest,\n });\n return;\n }\n\n if (!fileConverterService.canConvert(file.mimeType, acceptedMimeTypes)) {\n result.errors.push({\n id: 'validation.file.invalid-mime-type',\n field: 'files',\n values: {\n mimeType: file.mimeType,\n acceptedMimeTypes,\n fileId: file.id,\n },\n });\n }\n }\n });\n}\n\nexport async function validateFileBatchSize({\n result,\n websiteInstance,\n submission,\n}: ValidatorParams) {\n if (\n !isFileHandlingWebsite(websiteInstance) ||\n !isFileSubmission(submission) ||\n websiteInstance instanceof DefaultWebsite\n ) {\n return;\n }\n\n const maxBatchSize =\n websiteInstance.decoratedProps.fileOptions?.fileBatchSize ?? 0;\n const numFiles = submission.files.filter(\n (file) => !isFileFiltered(file, submission, websiteInstance),\n ).length;\n if (numFiles > maxBatchSize) {\n const expectedBatchesToCreate = Math.ceil(numFiles / maxBatchSize);\n result.warnings.push({\n id: 'validation.file.file-batch-size',\n field: 'files',\n values: {\n maxBatchSize,\n expectedBatchesToCreate,\n },\n });\n }\n}\n\nexport async function validateFileSize({\n result,\n websiteInstance,\n submission,\n}: ValidatorParams) {\n if (\n !isFileHandlingWebsite(websiteInstance) ||\n !isFileSubmission(submission) ||\n websiteInstance instanceof DefaultWebsite\n ) {\n return;\n }\n\n submission.files.forEach((file) => {\n if (isFileFiltered(file, submission, websiteInstance)) {\n return;\n }\n\n const maxFileSize = getSupportedFileSize(websiteInstance, file);\n if (maxFileSize && file.size > maxFileSize) {\n const issue: ValidationMessage = {\n id: 'validation.file.file-size',\n field: 'files',\n values: {\n maxFileSize,\n fileSize: file.size,\n fileName: file.fileName,\n fileId: file.id,\n },\n };\n if (getFileType(file.fileName) === FileType.IMAGE) {\n result.warnings.push(issue);\n } else {\n result.errors.push(issue);\n }\n }\n });\n}\n\nexport async function validateImageFileDimensions({\n result,\n websiteInstance,\n submission,\n}: ValidatorParams) {\n if (\n !isFileHandlingWebsite(websiteInstance) ||\n !isFileSubmission(submission) ||\n websiteInstance instanceof DefaultWebsite\n ) {\n return;\n }\n\n submission.files.forEach((file) => {\n if (isFileFiltered(file, submission, websiteInstance)) {\n return;\n }\n if (getFileType(file.fileName) === FileType.IMAGE) {\n const resizeProps = websiteInstance.calculateImageResize(file);\n if (resizeProps) {\n result.warnings.push({\n id: 'validation.file.image-resize',\n field: 'files',\n values: {\n fileName: file.fileName,\n resizeProps,\n fileId: file.id,\n },\n });\n }\n }\n });\n}\n" }, { "path": "apps/client-server/src/app/validation/validators/index.ts", "content": "import * as commonFieldValidators from './common-field-validators';\nimport * as dateTimeValidators from './datetime-field-validators';\nimport * as descriptionValidators from './description-validators';\nimport * as fileValidators from './file-submission-validators';\nimport * as selectFieldValidators from './select-field-validators';\nimport * as tagValidators from './tag-validators';\nimport * as titleValidators from './title-validators';\nimport { Validator } from './validator.type';\n\nexport const validators: Validator[] = [\n ...Object.values(titleValidators),\n ...Object.values(descriptionValidators),\n ...Object.values(tagValidators),\n ...Object.values(fileValidators),\n ...Object.values(commonFieldValidators),\n ...Object.values(selectFieldValidators),\n ...Object.values(dateTimeValidators),\n];\n" }, { "path": "apps/client-server/src/app/validation/validators/select-field-validators.ts", "content": "import {\n FieldAggregateType,\n SelectFieldType,\n SelectOption,\n} from '@postybirb/form-builder';\nimport { ValidatorParams } from './validator.type';\n\nfunction isSelectField(field: FieldAggregateType): field is SelectFieldType {\n return field.formField === 'select';\n}\n\nexport async function validateSelectFieldMinSelected({\n result,\n data,\n mergedWebsiteOptions,\n}: ValidatorParams) {\n const fields = mergedWebsiteOptions.getFormFields();\n for (const [fieldName, field] of Object.entries(fields)) {\n if (!isSelectField(field)) continue;\n\n const options = data.options[fieldName];\n const { minSelected } = field;\n if (!minSelected) continue;\n\n const selected = options?.length ?? 0;\n if (selected < minSelected) {\n result.errors.push({\n id: 'validation.select-field.min-selected',\n field: fieldName,\n values: {\n currentSelected: selected,\n minSelected,\n },\n });\n }\n }\n}\n\n/**\n * Validates that the selected value(s) for a select field are among the valid options.\n */\nexport async function validateSelectFieldValidOptions({\n result,\n data,\n mergedWebsiteOptions,\n}: ValidatorParams) {\n const fields = mergedWebsiteOptions.getFormFields();\n\n for (const [fieldName, field] of Object.entries(fields)) {\n if (!isSelectField(field)) continue;\n if (!field.options) continue;\n\n // Skip if field has no value\n if (\n data.options[fieldName] === undefined ||\n data.options[fieldName] === null\n ) {\n continue;\n }\n\n // Get the current value(s)\n const currentValue = data.options[fieldName];\n\n // Skip if no value\n if (\n currentValue === undefined ||\n currentValue === null ||\n currentValue === ''\n )\n continue;\n\n // Handle discriminator-based options differently (like overallFileType)\n if ('discriminator' in field.options) {\n // Skip validation for discriminator-based options as they're dynamically determined\n continue;\n }\n\n // Flatten all available options to a simple array of values\n const availableOptions = flattenSelectOptions(field.options);\n\n if (field.allowMultiple) {\n // For multi-select, validate each selected value\n if (Array.isArray(currentValue)) {\n const invalidOptions = currentValue.filter(\n (value) => !availableOptions.includes(value),\n );\n\n if (invalidOptions.length > 0) {\n result.errors.push({\n id: 'validation.select-field.invalid-option',\n field: fieldName,\n values: {\n invalidOptions,\n fieldName,\n fieldLabel: field.label,\n },\n });\n }\n }\n } else if (currentValue && !availableOptions.includes(currentValue)) {\n // For single-select, validate the selected value\n result.errors.push({\n id: 'validation.select-field.invalid-option',\n field: fieldName,\n values: {\n invalidOptions: [currentValue],\n },\n });\n }\n }\n}\n\n/**\n * Helper function to flatten nested select options into a single array of values\n */\nfunction flattenSelectOptions(options: SelectOption[]): string[] {\n const result: string[] = [];\n\n for (const option of options) {\n if ('items' in option && Array.isArray(option.items)) {\n // This is a group of options\n for (const item of option.items) {\n if ('value' in item) {\n result.push(String(item.value));\n }\n }\n } else if ('value' in option) {\n // This is a single option\n result.push(String(option.value));\n }\n }\n\n return result;\n}\n" }, { "path": "apps/client-server/src/app/validation/validators/tag-validators.ts", "content": "import { ValidatorParams } from './validator.type';\n\nexport async function validateMaxTags({\n data,\n mergedWebsiteOptions,\n validator,\n}: ValidatorParams) {\n const { hidden, maxTags } = mergedWebsiteOptions.getFormFieldFor('tags');\n if (hidden) return;\n\n const { tags } = data.options;\n const maxLength = maxTags ?? Number.MAX_SAFE_INTEGER;\n\n if (tags.length > maxLength) {\n validator.warning(\n 'validation.tags.max-tags',\n { maxLength, currentLength: tags.length },\n 'tags',\n );\n }\n}\n\nexport async function validateMinTags({\n data,\n validator,\n mergedWebsiteOptions,\n}: ValidatorParams) {\n const tagField = mergedWebsiteOptions.getFormFieldFor('tags');\n if (tagField.hidden) return;\n\n const { tags } = data.options;\n const minLength = tagField.minTags ?? -1;\n\n if (tags.length < minLength) {\n validator.error(\n 'validation.tags.min-tags',\n { currentLength: tags.length, minLength },\n 'tags',\n );\n }\n}\n\nexport async function validateMaxTagLength({\n data,\n mergedWebsiteOptions,\n validator,\n}: ValidatorParams) {\n const { hidden, maxTagLength } = mergedWebsiteOptions.getFormFieldFor('tags');\n if (hidden) return;\n\n const { tags } = data.options;\n const maxLength = maxTagLength ?? Number.MAX_SAFE_INTEGER;\n const invalidTags = tags.filter((tag) => tag.length > maxLength);\n\n if (invalidTags.length > 0) {\n validator.warning(\n 'validation.tags.max-tag-length',\n { tags: invalidTags, maxLength },\n 'tags',\n );\n }\n}\n\nexport async function validateTagHashtag({\n data,\n mergedWebsiteOptions,\n validator,\n}: ValidatorParams) {\n const { hidden } = mergedWebsiteOptions.getFormFieldFor('tags');\n if (hidden) return;\n\n const { tags } = data.options;\n const invalidTags = tags.filter((tag) => tag.startsWith('#'));\n\n if (invalidTags.length > 0) {\n validator.error(\n 'validation.tags.double-hashtag',\n { tags: invalidTags },\n 'tags',\n );\n }\n}\n" }, { "path": "apps/client-server/src/app/validation/validators/title-validators.ts", "content": "import { ValidatorParams } from './validator.type';\n\nexport async function validateTitleMaxLength({\n data,\n mergedWebsiteOptions,\n validator,\n}: ValidatorParams) {\n const { hidden, maxLength } = mergedWebsiteOptions.getFormFieldFor('title');\n if (hidden) return;\n\n const { title } = data.options;\n const maxTitleLength = maxLength ?? Number.MAX_SAFE_INTEGER;\n\n if (title.length > maxLength) {\n validator.warning(\n 'validation.title.max-length',\n { currentLength: title.length, maxLength: maxTitleLength },\n 'title',\n );\n }\n}\n\nexport async function validateTitleMinLength({\n data,\n mergedWebsiteOptions,\n validator,\n}: ValidatorParams) {\n const { hidden, minLength } = mergedWebsiteOptions.getFormFieldFor('title');\n if (hidden) return;\n\n const { title } = data.options;\n const minTitleLength = minLength ?? -1;\n\n if (title.length < minTitleLength) {\n validator.error(\n 'validation.title.min-length',\n { currentLength: title.length, minLength: minTitleLength },\n 'title',\n );\n }\n}\n" }, { "path": "apps/client-server/src/app/validation/validators/validator.type.ts", "content": "import {\n ISubmission,\n PostData,\n ValidationMessage,\n ValidationResult,\n} from '@postybirb/types';\nimport { FileConverterService } from '../../file-converter/file-converter.service';\nimport { FileService } from '../../file/file.service';\nimport { SubmissionValidator } from '../../websites/commons/validator';\nimport { BaseWebsiteOptions } from '../../websites/models/base-website-options';\nimport { UnknownWebsite } from '../../websites/website';\n\nexport type ValidatorParams = {\n result: ValidationResult;\n validator: FieldValidator;\n websiteInstance: UnknownWebsite;\n data: PostData;\n submission: ISubmission;\n fileConverterService: FileConverterService;\n fileService: FileService;\n mergedWebsiteOptions: BaseWebsiteOptions;\n};\n\nexport type Validator = (props: ValidatorParams) => Promise;\n\nexport class FieldValidator extends SubmissionValidator {\n constructor(\n override errors: ValidationMessage[],\n override warnings: ValidationMessage[],\n ) {\n super();\n }\n}\n" }, { "path": "apps/client-server/src/app/web-socket/models/web-socket-event.ts", "content": "export abstract class WebsocketEvent {\n event: string;\n\n data: D;\n\n constructor(data: D) {\n this.data = data;\n }\n}\n" }, { "path": "apps/client-server/src/app/web-socket/web-socket-adapter.ts", "content": "import { IoAdapter } from '@nestjs/platform-socket.io';\nimport { ServerOptions } from 'socket.io';\n\nexport class WebSocketAdapter extends IoAdapter {\n createIOServer(\n port: number,\n options?: ServerOptions & {\n namespace?: string;\n server?: unknown;\n },\n ) {\n const server = super.createIOServer(port, { ...options, cors: true });\n return server;\n }\n}\n" }, { "path": "apps/client-server/src/app/web-socket/web-socket-gateway.ts", "content": "import {\n OnGatewayInit,\n WebSocketGateway,\n WebSocketServer,\n} from '@nestjs/websockets';\nimport { getRemoteConfig } from '@postybirb/utils/electron';\nimport { Server } from 'socket.io';\nimport { WebSocketEvents } from './web-socket.events';\n\n@WebSocketGateway({ cors: true })\nexport class WSGateway implements OnGatewayInit {\n @WebSocketServer()\n private server: Server;\n\n afterInit(server: Server) {\n server.use(async (socket, next) => {\n const remoteConfig = await getRemoteConfig();\n if (socket.handshake.headers.authorization === remoteConfig.password) {\n return next();\n }\n return next(new Error('Authentication Error'));\n });\n }\n\n public emit(socketEvent: WebSocketEvents) {\n const { event, data } = socketEvent;\n this.server.emit(event, data);\n }\n}\n" }, { "path": "apps/client-server/src/app/web-socket/web-socket.events.ts", "content": "import { AccountEventTypes } from '../account/account.events';\nimport { CustomShortcutEventTypes } from '../custom-shortcuts/custom-shortcut.events';\nimport { DirectoryWatcherEventTypes } from '../directory-watchers/directory-watcher.events';\nimport { NotificationEventTypes } from '../notifications/notification.events';\nimport { SettingsEventTypes } from '../settings/settings.events';\nimport { SubmissionEventTypes } from '../submission/submission.events';\nimport { TagConverterEventTypes } from '../tag-converters/tag-converter.events';\nimport { TagGroupEventTypes } from '../tag-groups/tag-group.events';\nimport { UpdateEventTypes } from '../update/update.events';\nimport { UserConverterEventTypes } from '../user-converters/user-converter.events';\nimport { WebsiteEventTypes } from '../websites/website.events';\n\nexport type WebSocketEvents =\n | AccountEventTypes\n | DirectoryWatcherEventTypes\n | SettingsEventTypes\n | SubmissionEventTypes\n | TagGroupEventTypes\n | TagConverterEventTypes\n | UpdateEventTypes\n | UserConverterEventTypes\n | WebsiteEventTypes\n | NotificationEventTypes\n | CustomShortcutEventTypes;\n" }, { "path": "apps/client-server/src/app/web-socket/web-socket.module.ts", "content": "import { Global, Module } from '@nestjs/common';\nimport { WSGateway } from './web-socket-gateway';\n\n@Global()\n@Module({\n providers: [WSGateway],\n exports: [WSGateway],\n})\nexport class WebSocketModule {}\n" }, { "path": "apps/client-server/src/app/website-options/dtos/create-website-options.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport {\n AccountId,\n ICreateWebsiteOptionsDto,\n IWebsiteFormFields,\n SubmissionId,\n} from '@postybirb/types';\nimport { IsObject, IsOptional, IsString } from 'class-validator';\n\nexport class CreateWebsiteOptionsDto implements ICreateWebsiteOptionsDto {\n @ApiProperty()\n @IsString()\n accountId: AccountId;\n\n @ApiProperty({ type: Object })\n @IsOptional()\n @IsObject()\n data: IWebsiteFormFields;\n\n @ApiProperty()\n @IsString()\n submissionId: SubmissionId;\n\n isDefault?: boolean = false;\n}\n" }, { "path": "apps/client-server/src/app/website-options/dtos/preview-description.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport {\n EntityId,\n IPreviewDescriptionDto,\n SubmissionId,\n} from '@postybirb/types';\nimport { IsString } from 'class-validator';\n\nexport class PreviewDescriptionDto implements IPreviewDescriptionDto {\n @ApiProperty()\n @IsString()\n submissionId: SubmissionId;\n\n @ApiProperty()\n @IsString()\n websiteOptionId: EntityId;\n}\n" }, { "path": "apps/client-server/src/app/website-options/dtos/update-submission-website-options.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport {\n EntityId,\n ICreateWebsiteOptionsDto,\n IUpdateSubmissionWebsiteOptionsDto,\n} from '@postybirb/types';\nimport { IsArray, IsOptional } from 'class-validator';\n\nexport class UpdateSubmissionWebsiteOptionsDto\n implements IUpdateSubmissionWebsiteOptionsDto\n{\n @ApiProperty()\n @IsOptional()\n @IsArray()\n remove?: EntityId[];\n\n @ApiProperty()\n @IsOptional()\n @IsArray()\n add?: ICreateWebsiteOptionsDto[];\n}\n" }, { "path": "apps/client-server/src/app/website-options/dtos/update-website-options.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { IUpdateWebsiteOptionsDto, IWebsiteFormFields } from '@postybirb/types';\nimport { IsObject } from 'class-validator';\n\nexport class UpdateWebsiteOptionsDto implements IUpdateWebsiteOptionsDto {\n @ApiProperty({ type: Object })\n @IsObject()\n data: IWebsiteFormFields;\n}\n" }, { "path": "apps/client-server/src/app/website-options/dtos/validate-website-options.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport {\n EntityId,\n IValidateWebsiteOptionsDto,\n SubmissionId,\n} from '@postybirb/types';\nimport { IsString } from 'class-validator';\n\nexport class ValidateWebsiteOptionsDto implements IValidateWebsiteOptionsDto {\n @ApiProperty()\n @IsString()\n submissionId: SubmissionId;\n\n @ApiProperty()\n @IsString()\n websiteOptionId: EntityId;\n}\n" }, { "path": "apps/client-server/src/app/website-options/website-options.controller.ts", "content": "import { Body, Controller, Get, Param, Patch, Post } from '@nestjs/common';\nimport {\n ApiBadRequestResponse,\n ApiNotFoundResponse,\n ApiOkResponse,\n ApiTags,\n} from '@nestjs/swagger';\nimport { EntityId, SubmissionId } from '@postybirb/types';\nimport { PostyBirbController } from '../common/controller/postybirb-controller';\nimport { CreateWebsiteOptionsDto } from './dtos/create-website-options.dto';\nimport { PreviewDescriptionDto } from './dtos/preview-description.dto';\nimport { UpdateSubmissionWebsiteOptionsDto } from './dtos/update-submission-website-options.dto';\nimport { UpdateWebsiteOptionsDto } from './dtos/update-website-options.dto';\nimport { ValidateWebsiteOptionsDto } from './dtos/validate-website-options.dto';\nimport { WebsiteOptionsService } from './website-options.service';\n\n/**\n * CRUD operation on WebsiteOptions.\n *\n * @class WebsiteOptionsController\n */\n@ApiTags('website-option')\n@Controller('website-option')\nexport class WebsiteOptionsController extends PostyBirbController<'WebsiteOptionsSchema'> {\n constructor(readonly service: WebsiteOptionsService) {\n super(service);\n }\n\n @Post()\n @ApiOkResponse({ description: 'Website option created.' })\n @ApiBadRequestResponse({\n description: 'Bad request.',\n })\n @ApiNotFoundResponse({\n description: 'Account or website instance not found.',\n })\n create(\n @Body()\n createDto: CreateWebsiteOptionsDto,\n ) {\n return this.service.create(createDto).then((entity) => entity.toDTO());\n }\n\n @Patch(':id')\n @ApiOkResponse({ description: 'Submission option updated.', type: Boolean })\n @ApiNotFoundResponse({ description: 'Submission option Id not found.' })\n update(\n @Body()\n updateDto: UpdateWebsiteOptionsDto,\n @Param('id') id: EntityId,\n ) {\n return this.service.update(id, updateDto).then((entity) => entity.toDTO());\n }\n\n @Patch('submission/:id')\n @ApiOkResponse({ description: 'Submission updated.', type: Boolean })\n @ApiNotFoundResponse({ description: 'Submission Id not found.' })\n updateSubmission(\n @Body()\n updateDto: UpdateSubmissionWebsiteOptionsDto,\n @Param('id') submissionId: SubmissionId,\n ) {\n return this.service\n .updateSubmissionOptions(submissionId, updateDto)\n .then((entity) => entity.toDTO());\n }\n\n @Post('validate')\n @ApiOkResponse({ description: 'Submission validation completed.' })\n @ApiBadRequestResponse()\n @ApiNotFoundResponse({ description: 'Submission not found.' })\n validate(@Body() validateOptionsDto: ValidateWebsiteOptionsDto) {\n return this.service.validateWebsiteOption(validateOptionsDto);\n }\n\n @Get('validate/:submissionId')\n @ApiOkResponse({ description: 'Submission validation completed.' })\n @ApiBadRequestResponse()\n @ApiNotFoundResponse({ description: 'Submission not found.' })\n validateSubmission(@Param('submissionId') submissionId: SubmissionId) {\n return this.service.validateSubmission(submissionId);\n }\n\n @Post('preview-description')\n @ApiOkResponse({ description: 'Description preview generated.' })\n @ApiBadRequestResponse()\n @ApiNotFoundResponse({ description: 'Submission or option not found.' })\n previewDescription(@Body() dto: PreviewDescriptionDto) {\n return this.service.previewDescription(dto);\n }\n}\n" }, { "path": "apps/client-server/src/app/website-options/website-options.module.ts", "content": "import { forwardRef, Module } from '@nestjs/common';\nimport { AccountModule } from '../account/account.module';\nimport { FormGeneratorModule } from '../form-generator/form-generator.module';\nimport { PostParsersModule } from '../post-parsers/post-parsers.module';\nimport { SubmissionModule } from '../submission/submission.module';\nimport { UserSpecifiedWebsiteOptionsModule } from '../user-specified-website-options/user-specified-website-options.module';\nimport { ValidationModule } from '../validation/validation.module';\nimport { WebsitesModule } from '../websites/websites.module';\nimport { WebsiteOptionsController } from './website-options.controller';\nimport { WebsiteOptionsService } from './website-options.service';\n\n@Module({\n imports: [\n forwardRef(() => SubmissionModule),\n WebsitesModule,\n AccountModule,\n UserSpecifiedWebsiteOptionsModule,\n FormGeneratorModule,\n ValidationModule,\n PostParsersModule,\n ],\n providers: [WebsiteOptionsService],\n controllers: [WebsiteOptionsController],\n exports: [WebsiteOptionsService],\n})\nexport class WebsiteOptionsModule {}\n" }, { "path": "apps/client-server/src/app/website-options/website-options.service.spec.ts", "content": "import { Test, TestingModule } from '@nestjs/testing';\nimport { clearDatabase } from '@postybirb/database';\nimport {\n DefaultDescriptionValue,\n DefaultTagValue,\n SubmissionRating,\n SubmissionType,\n TipTapNode,\n} from '@postybirb/types';\nimport { AccountModule } from '../account/account.module';\nimport { AccountService } from '../account/account.service';\nimport { CreateAccountDto } from '../account/dtos/create-account.dto';\nimport { FileConverterService } from '../file-converter/file-converter.service';\nimport { FileService } from '../file/file.service';\nimport { CreateFileService } from '../file/services/create-file.service';\nimport { UpdateFileService } from '../file/services/update-file.service';\nimport { SharpInstanceManager } from '../image-processing/sharp-instance-manager';\nimport { FormGeneratorModule } from '../form-generator/form-generator.module';\nimport { PostParsersModule } from '../post-parsers/post-parsers.module';\nimport { CreateSubmissionDto } from '../submission/dtos/create-submission.dto';\nimport { FileSubmissionService } from '../submission/services/file-submission.service';\nimport { MessageSubmissionService } from '../submission/services/message-submission.service';\nimport { SubmissionService } from '../submission/services/submission.service';\nimport { UserSpecifiedWebsiteOptionsModule } from '../user-specified-website-options/user-specified-website-options.module';\nimport { UserSpecifiedWebsiteOptionsService } from '../user-specified-website-options/user-specified-website-options.service';\nimport { ValidationService } from '../validation/validation.service';\nimport { WebsiteImplProvider } from '../websites/implementations/provider';\nimport { WebsiteRegistryService } from '../websites/website-registry.service';\nimport { WebsitesModule } from '../websites/websites.module';\nimport { CreateWebsiteOptionsDto } from './dtos/create-website-options.dto';\nimport { WebsiteOptionsService } from './website-options.service';\n\ndescribe('WebsiteOptionsService', () => {\n let service: WebsiteOptionsService;\n let submissionService: SubmissionService;\n let accountService: AccountService;\n let module: TestingModule;\n\n async function createAccount() {\n const dto = new CreateAccountDto();\n dto.groups = ['test'];\n dto.name = 'test';\n dto.website = 'test';\n\n const record = await accountService.create(dto);\n return record;\n }\n\n async function createSubmission() {\n const dto = new CreateSubmissionDto();\n dto.name = 'test';\n dto.type = SubmissionType.MESSAGE;\n\n const record = await submissionService.create(dto);\n return record;\n }\n\n beforeEach(async () => {\n clearDatabase();\n try {\n module = await Test.createTestingModule({\n imports: [\n WebsitesModule,\n AccountModule,\n UserSpecifiedWebsiteOptionsModule,\n PostParsersModule,\n FormGeneratorModule,\n ],\n providers: [\n SubmissionService,\n CreateFileService,\n UpdateFileService,\n SharpInstanceManager,\n FileService,\n SubmissionService,\n FileSubmissionService,\n MessageSubmissionService,\n AccountService,\n WebsiteRegistryService,\n ValidationService,\n WebsiteOptionsService,\n WebsiteImplProvider,\n UserSpecifiedWebsiteOptionsService,\n FileConverterService,\n ],\n }).compile();\n\n service = module.get(WebsiteOptionsService);\n submissionService = module.get(SubmissionService);\n accountService = module.get(AccountService);\n await accountService.onModuleInit();\n } catch (e) {\n console.error(e);\n }\n });\n\n afterAll(async () => {\n await module.close();\n });\n\n it('should be defined', () => {\n expect(service).toBeDefined();\n });\n\n it('should create entity', async () => {\n const account = await createAccount();\n const submission = await createSubmission();\n\n const dto = new CreateWebsiteOptionsDto();\n dto.data = {\n contentWarning: '',\n title: 'title',\n tags: DefaultTagValue(),\n description: DefaultDescriptionValue(),\n rating: SubmissionRating.GENERAL,\n };\n dto.accountId = account.id;\n dto.submissionId = submission.id;\n\n const record = await service.create(dto);\n const groups = await service.findAll();\n expect(groups).toHaveLength(2); // 2 because default\n\n expect(groups[1].accountId).toEqual(dto.accountId);\n expect(groups[1].isDefault).toEqual(false);\n expect(groups[1].data).toEqual(dto.data);\n expect(groups[1].submission.id).toEqual(dto.submissionId);\n\n expect(record.toDTO()).toEqual({\n data: record.data,\n isDefault: false,\n id: record.id,\n accountId: account.id,\n createdAt: record.createdAt,\n updatedAt: record.updatedAt,\n account: record.account.toObject(),\n submissionId: submission.id,\n submission: record.submission.toDTO(),\n });\n });\n\n it('should remove entity', async () => {\n const account = await createAccount();\n const submission = await createSubmission();\n\n const dto = new CreateWebsiteOptionsDto();\n dto.data = {\n title: 'title',\n tags: DefaultTagValue(),\n description: DefaultDescriptionValue(),\n rating: SubmissionRating.GENERAL,\n };\n dto.accountId = account.id;\n dto.submissionId = submission.id;\n\n const record = await service.create(dto);\n expect(await service.findAll()).toHaveLength(2); // 2 because default\n\n await service.remove(record.id);\n expect(await service.findAll()).toHaveLength(1);\n });\n\n it('should remove entity when parent is removed', async () => {\n const account = await createAccount();\n const submission = await createSubmission();\n\n const dto = new CreateWebsiteOptionsDto();\n dto.data = {\n title: 'title',\n tags: DefaultTagValue(),\n description: DefaultDescriptionValue(),\n rating: SubmissionRating.GENERAL,\n };\n dto.accountId = account.id;\n dto.submissionId = submission.id;\n\n await service.create(dto);\n expect(await service.findAll()).toHaveLength(2); // 2 because default\n\n await submissionService.remove(submission.id);\n expect(await service.findAll()).toHaveLength(0);\n });\n\n it('should update entity', async () => {\n const account = await createAccount();\n const submission = await createSubmission();\n\n const dto = new CreateWebsiteOptionsDto();\n dto.data = {\n contentWarning: '',\n title: 'title',\n tags: DefaultTagValue(),\n description: DefaultDescriptionValue(),\n rating: SubmissionRating.GENERAL,\n };\n dto.accountId = account.id;\n dto.submissionId = submission.id;\n\n const record = await service.create(dto);\n expect(record.accountId).toEqual(dto.accountId);\n expect(record.isDefault).toEqual(false);\n expect(record.data).toEqual(dto.data);\n expect(record.submission.id).toEqual(dto.submissionId);\n\n const update = await service.update(record.id, {\n data: {\n title: 'title updated',\n tags: DefaultTagValue(),\n description: DefaultDescriptionValue(),\n rating: SubmissionRating.GENERAL,\n },\n });\n\n expect(update.data.title).toEqual('title updated');\n });\n\n it('filters nested inline content', async () => {\n const blocks: TipTapNode[] = [\n {\n type: 'paragraph',\n content: [\n { type: 'text', text: 'Hello, ', marks: [{ type: 'bold' }] },\n {\n type: 'customShortcut',\n attrs: {\n id: 'to-delete',\n },\n content: [\n {\n type: 'text',\n text: 'User',\n },\n ],\n },\n ],\n },\n ];\n\n const { changed, filtered } = service.filterCustomShortcut(\n blocks,\n 'to-delete',\n );\n expect(changed).toBeTruthy();\n expect(filtered).toEqual([\n {\n type: 'paragraph',\n content: [{ type: 'text', text: 'Hello, ', marks: [{ type: 'bold' }] }],\n },\n ]);\n });\n});\n" }, { "path": "apps/client-server/src/app/website-options/website-options.service.ts", "content": "import {\n forwardRef,\n Inject,\n Injectable,\n NotFoundException,\n OnModuleInit,\n} from '@nestjs/common';\nimport { Insert } from '@postybirb/database';\nimport {\n AccountId,\n Description,\n DescriptionType,\n DescriptionValue,\n DynamicObject,\n EntityId,\n IDescriptionPreviewResult,\n ISubmission,\n ISubmissionMetadata,\n IWebsiteFormFields,\n NULL_ACCOUNT_ID,\n SubmissionId,\n SubmissionMetadataType,\n SubmissionType,\n TipTapNode,\n ValidationResult,\n} from '@postybirb/types';\nimport { AccountService } from '../account/account.service';\nimport { PostyBirbService } from '../common/service/postybirb-service';\nimport { Account, Submission, WebsiteOptions } from '../drizzle/models';\nimport { PostyBirbDatabase } from '../drizzle/postybirb-database/postybirb-database';\nimport { FormGeneratorService } from '../form-generator/form-generator.service';\nimport { PostParsersService } from '../post-parsers/post-parsers.service';\nimport { SubmissionService } from '../submission/services/submission.service';\nimport { UserSpecifiedWebsiteOptionsService } from '../user-specified-website-options/user-specified-website-options.service';\nimport {\n isBlockNoteFormat,\n migrateDescription,\n} from '../utils/blocknote-to-tiptap';\nimport { ValidationService } from '../validation/validation.service';\nimport DefaultWebsite from '../websites/implementations/default/default.website';\nimport { DefaultWebsiteOptions } from '../websites/models/default-website-options';\nimport { WebsiteRegistryService } from '../websites/website-registry.service';\nimport { CreateWebsiteOptionsDto } from './dtos/create-website-options.dto';\nimport { PreviewDescriptionDto } from './dtos/preview-description.dto';\nimport { UpdateSubmissionWebsiteOptionsDto } from './dtos/update-submission-website-options.dto';\nimport { UpdateWebsiteOptionsDto } from './dtos/update-website-options.dto';\nimport { ValidateWebsiteOptionsDto } from './dtos/validate-website-options.dto';\n\n@Injectable()\nexport class WebsiteOptionsService\n extends PostyBirbService<'WebsiteOptionsSchema'>\n implements OnModuleInit\n{\n private readonly submissionRepository = new PostyBirbDatabase(\n 'SubmissionSchema',\n );\n\n constructor(\n @Inject(forwardRef(() => SubmissionService))\n private readonly submissionService: SubmissionService,\n private readonly accountService: AccountService,\n private readonly userSpecifiedOptionsService: UserSpecifiedWebsiteOptionsService,\n private readonly formGeneratorService: FormGeneratorService,\n private readonly validationService: ValidationService,\n private readonly postParsersService: PostParsersService,\n private readonly websiteRegistry: WebsiteRegistryService,\n ) {\n super(\n new PostyBirbDatabase('WebsiteOptionsSchema', {\n account: true,\n submission: true,\n }),\n );\n\n this.repository.subscribe('CustomShortcutSchema', (ids, action) => {\n if (action === 'delete') {\n for (const id of ids) {\n this.onCustomShortcutDelete(id).catch((err) =>\n this.logger.error(\n `Error handling custom shortcut delete for id '${id}': ${err.message}`,\n err.stack,\n ),\n );\n }\n }\n });\n }\n\n async onModuleInit() {\n await this.migrateBlockNoteDescriptions();\n }\n\n /**\n * One-time migration: convert any BlockNote-format descriptions\n * (stored as arrays) to TipTap format ({ type: 'doc', content: [] }).\n * Covers website options, custom shortcuts, and user-specified website options.\n */\n private async migrateBlockNoteDescriptions() {\n let migrated = 0;\n\n // 1. Migrate website options\n const options = await this.findAll();\n for (const option of options) {\n const descValue = option.data?.description as\n | DescriptionValue\n | undefined;\n const desc = descValue?.description;\n if (desc && isBlockNoteFormat(desc)) {\n const converted = migrateDescription(desc);\n await this.repository.update(option.id, {\n data: {\n ...option.data,\n description: {\n ...descValue,\n description: converted,\n },\n },\n });\n migrated++;\n }\n }\n\n // 2. Migrate custom shortcuts\n const customShortcutRepo = new PostyBirbDatabase('CustomShortcutSchema');\n const shortcuts = await customShortcutRepo.findAll();\n for (const shortcut of shortcuts) {\n const desc = (shortcut as DynamicObject).shortcut;\n if (desc && isBlockNoteFormat(desc)) {\n const converted = migrateDescription(desc);\n await customShortcutRepo.update(shortcut.id, {\n shortcut: converted,\n } as unknown);\n migrated++;\n }\n }\n\n // 3. Migrate user-specified website options\n const userOptsRepo = new PostyBirbDatabase(\n 'UserSpecifiedWebsiteOptionsSchema',\n );\n const userOpts = await userOptsRepo.findAll();\n for (const userOpt of userOpts) {\n const opts = (userOpt as DynamicObject).options as DynamicObject;\n if (opts?.description) {\n const descValue = opts.description as DescriptionValue | undefined;\n const desc = descValue?.description;\n if (desc && isBlockNoteFormat(desc)) {\n const converted = migrateDescription(desc);\n await userOptsRepo.update(userOpt.id, {\n options: {\n ...opts,\n description: {\n ...descValue,\n description: converted,\n },\n },\n } as unknown);\n migrated++;\n }\n }\n }\n\n if (migrated > 0) {\n this.logger.info(\n `Migrated ${migrated} BlockNote description(s) to TipTap format`,\n );\n }\n }\n\n /**\n * Creates a submission option for a submission.\n * No longer remember why this is a separate method from create.\n *\n * @param {ISubmission} submission\n * @param {AccountId} accountId\n * @param {DynamicObject} data\n * @param {string} [title]\n * @return {*} {Promise}\n */\n async createOption(\n submission: ISubmission,\n accountId: AccountId,\n data: DynamicObject,\n title?: string,\n ): Promise {\n const option = await this.createOptionInsertObject(\n submission,\n accountId,\n data,\n title,\n );\n return this.repository.insert(option);\n }\n\n /**\n * Creates a submission option for a submission.\n *\n * @param {Submission} submission\n * @param {AccountId} accountId\n * @param {DynamicObject} data\n * @param {string} [title]\n */\n async createOptionInsertObject(\n submission: ISubmission,\n accountId: AccountId,\n data: DynamicObject,\n title?: string,\n ): Promise> {\n const account = await this.accountService.findById(accountId, {\n failOnMissing: true,\n });\n const isDefault = accountId === NULL_ACCOUNT_ID;\n\n const userDefinedDefaultOptions =\n await this.userSpecifiedOptionsService.findByAccountAndSubmissionType(\n accountId,\n submission.type,\n );\n\n const formFields = isDefault\n ? await this.formGeneratorService.getDefaultForm(submission.type)\n : await this.formGeneratorService.generateForm({\n accountId: account.id,\n type: submission.type,\n });\n\n // Populate with the form fields to get the default values\n const websiteData: IWebsiteFormFields = {\n ...Object.entries(formFields).reduce(\n (acc, [key, field]) => ({\n ...acc,\n [key]: field.defaultValue,\n }),\n {} as IWebsiteFormFields,\n ),\n };\n\n const mergedData: IWebsiteFormFields = {\n ...(isDefault ? new DefaultWebsiteOptions() : {}), // Only merge default options if this is the default option\n ...websiteData, // Merge default form fields\n ...(userDefinedDefaultOptions?.options ?? {}), // Merge user defined options\n ...data, // Merge user defined data\n title, // Override title (optional)\n };\n\n // For non-default options, keep rating as undefined to represent\n // \"inherit from default\" mode unless explicitly provided in data.\n if (!isDefault && !data.rating) {\n mergedData.rating = undefined as unknown as typeof mergedData.rating;\n }\n\n const option: Insert<'WebsiteOptionsSchema'> = {\n submissionId: submission.id,\n accountId: account.id,\n data: mergedData,\n isDefault,\n };\n\n return option;\n }\n\n /**\n * The default create method for WebsiteOptions.\n * Performs user saved options and other merging operations.\n * Performs and update if it already exists.\n *\n * @param {CreateWebsiteOptionsDto} createDto\n * @return {*}\n */\n async create(createDto: CreateWebsiteOptionsDto) {\n const account = await this.accountService.findById(createDto.accountId, {\n failOnMissing: true,\n });\n\n let submission: ISubmission;\n try {\n submission = await this.submissionRepository.findById(\n createDto.submissionId,\n { failOnMissing: true },\n );\n } catch (err) {\n throw new NotFoundException(\n `Submission ${createDto.submissionId} not found.`,\n );\n }\n\n const exists = await this.repository.findOne({\n where: (wo, { and, eq }) =>\n and(eq(wo.submissionId, submission.id), eq(wo.accountId, account.id)),\n });\n if (exists) {\n // Opt to just update the existing option\n return this.update(exists.id, { data: createDto.data });\n }\n\n const formFields =\n account.id === NULL_ACCOUNT_ID\n ? await this.formGeneratorService.getDefaultForm(submission.type)\n : await this.formGeneratorService.generateForm({\n accountId: account.id,\n type: submission.type,\n });\n // Populate with the form fields to get the default values\n const websiteData: IWebsiteFormFields = {\n ...Object.entries(formFields).reduce(\n (acc, [key, field]) => ({\n ...acc,\n [key]:\n createDto.data?.[key as keyof IWebsiteFormFields] === undefined\n ? field.defaultValue\n : createDto.data?.[key as keyof IWebsiteFormFields],\n }),\n {} as IWebsiteFormFields,\n ),\n };\n\n const isDefault = account.id === NULL_ACCOUNT_ID;\n\n // For non-default options, keep rating as undefined to represent\n // \"inherit from default\" mode unless explicitly provided in the DTO.\n if (!isDefault && createDto.data?.rating === undefined) {\n websiteData.rating = undefined as unknown as typeof websiteData.rating;\n }\n\n const record = await this.repository.insert({\n submissionId: submission.id,\n data: websiteData,\n accountId: account.id,\n isDefault,\n });\n this.submissionService.emit();\n return record;\n }\n\n async update(id: EntityId, update: UpdateWebsiteOptionsDto) {\n this.logger.withMetadata(update).info(`Updating WebsiteOptions '${id}'`);\n const result = await this.repository.update(id, update);\n this.submissionService.emit();\n return result;\n }\n\n /**\n * Creates the default submission option that stores shared data\n * across multiple submission options.\n *\n * @param {Submission} submission\n * @param {string} title\n * @param {Partial} [defaultOptions] - Optional default options to merge\n * @return {*} {Promise}\n */\n async createDefaultSubmissionOptions(\n submission: ISubmission,\n title: string,\n defaultOptions?: Partial,\n ): Promise {\n this.logger\n .withMetadata({ id: submission.id })\n .info('Creating Default Website Options');\n\n const options: Insert<'WebsiteOptionsSchema'> = {\n isDefault: true,\n submissionId: submission.id,\n accountId: NULL_ACCOUNT_ID,\n data: await this.populateDefaultWebsiteOptions(\n NULL_ACCOUNT_ID,\n submission.type,\n title,\n defaultOptions,\n ),\n };\n\n return this.repository.insert(options);\n }\n\n private async populateDefaultWebsiteOptions(\n accountId: AccountId,\n type: SubmissionType,\n title?: string,\n defaultOptions?: Partial,\n ): Promise {\n const userSpecifiedOptions =\n (\n await this.userSpecifiedOptionsService.findByAccountAndSubmissionType(\n NULL_ACCOUNT_ID,\n type,\n )\n )?.options ?? {};\n\n const websiteFormFields: IWebsiteFormFields = {\n ...new DefaultWebsiteOptions(),\n ...userSpecifiedOptions,\n title,\n };\n\n // Merge provided default options (tags, description, rating)\n if (defaultOptions) {\n if (defaultOptions.tags) {\n websiteFormFields.tags = {\n overrideDefault: false,\n tags: defaultOptions.tags.tags ?? defaultOptions.tags.tags ?? [],\n };\n }\n if (defaultOptions.description) {\n websiteFormFields.description = defaultOptions.description;\n }\n if (defaultOptions.rating) {\n websiteFormFields.rating = defaultOptions.rating;\n }\n }\n\n return websiteFormFields;\n }\n\n /**\n * Validates a submission option against a website instance.\n * @param {ValidateWebsiteOptionsDto} validate\n * @return {Promise}\n */\n async validateWebsiteOption(\n validate: ValidateWebsiteOptionsDto,\n ): Promise {\n const { websiteOptionId, submissionId } = validate;\n const submission = await this.submissionService.findById(submissionId, {\n failOnMissing: true,\n });\n const websiteOption = submission.options.find(\n (option) => option.id === websiteOptionId,\n );\n return this.validationService.validate(submission, websiteOption);\n }\n\n /**\n * Validates all submission options for a submission.\n * Accepts either a submission ID (will fetch from DB) or a Submission object directly.\n * When a Submission object is provided, it avoids a redundant database query.\n * @param {SubmissionId | Submission} submissionOrId\n * @return {*} {Promise}\n */\n async validateSubmission(\n submissionOrId: SubmissionId | Submission,\n ): Promise {\n const submission =\n typeof submissionOrId === 'string'\n ? await this.submissionService.findById(submissionOrId)\n : submissionOrId;\n return this.validationService.validateSubmission(submission);\n }\n\n /**\n * Previews the parsed description for a specific website option.\n * Parses the description the same way it would be parsed during posting,\n * and returns both the output format type and the rendered string.\n * @param {PreviewDescriptionDto} dto\n * @return {Promise}\n */\n async previewDescription(\n dto: PreviewDescriptionDto,\n ): Promise {\n const { websiteOptionId, submissionId } = dto;\n const submission = await this.submissionService.findById(submissionId, {\n failOnMissing: true,\n });\n const websiteOption = submission.options.find(\n (option) => option.id === websiteOptionId,\n );\n if (!websiteOption) {\n throw new NotFoundException(\n `Website option ${websiteOptionId} not found`,\n );\n }\n\n const website = websiteOption.isDefault\n ? new DefaultWebsite(new Account(websiteOption.account))\n : this.websiteRegistry.findInstance(websiteOption.account);\n\n if (!website) {\n throw new NotFoundException(\n `Website instance for account ${websiteOption.accountId} not found`,\n );\n }\n\n const data = await this.postParsersService.parse(\n submission,\n website,\n websiteOption,\n );\n\n // Determine the description output type using the same logic as the parser\n const defaultOptions = submission.options.find((o) => o.isDefault);\n const defaultOpts = Object.assign(new DefaultWebsiteOptions(), {\n ...defaultOptions.data,\n });\n const websiteOpts = Object.assign(website.getModelFor(submission.type), {\n ...websiteOption.data,\n });\n const mergedOptions = websiteOpts.mergeDefaults(defaultOpts);\n const { descriptionType } = mergedOptions.getFormFieldFor('description');\n\n return {\n descriptionType: descriptionType as DescriptionType,\n description: data.options.description ?? '',\n };\n }\n\n async updateSubmissionOptions(\n submissionId: SubmissionId,\n updateDto: UpdateSubmissionWebsiteOptionsDto,\n ) {\n const submission = await this.submissionService.findById(submissionId, {\n failOnMissing: true,\n });\n\n const { remove, add } = updateDto;\n if (remove?.length) {\n const items = submission.options;\n const removableIds = [];\n for (const id of remove) {\n const option = items.find((opt) => opt.id === id);\n if (option) {\n removableIds.push(id);\n }\n }\n this.logger.debug(\n `Removing option(s) [${removableIds.join(', ')}] from submission ${submissionId}`,\n );\n await this.repository.deleteById(removableIds);\n }\n\n if (add?.length) {\n const options = await Promise.all(\n add.map((dto) =>\n this.createOptionInsertObject(submission, dto.accountId, dto.data),\n ),\n );\n await this.repository.insert(options);\n }\n\n this.submissionService.emit();\n return this.submissionService.findById(submissionId);\n }\n\n private async onCustomShortcutDelete(id: EntityId) {\n const websiteOptions = await this.findAll();\n for (const option of websiteOptions) {\n const { data } = option;\n const descValue: DescriptionValue | undefined = data?.description;\n const doc: Description | undefined = descValue?.description;\n const blocks = doc?.content;\n\n if (!blocks || !Array.isArray(blocks) || blocks.length === 0) {\n continue;\n }\n\n const { changed, filtered } = this.filterCustomShortcut(\n blocks,\n String(id),\n );\n if (changed) {\n const updatedDescription: DescriptionValue = {\n ...(descValue as DescriptionValue),\n description: { type: 'doc', content: filtered },\n };\n\n await this.repository.update(option.id, {\n data: {\n ...data,\n description: updatedDescription,\n },\n });\n this.submissionService.emit();\n }\n }\n }\n\n /**\n * Removes inline customShortcut items matching the given id from a TipTap content array.\n * Simple recursive filter without whitespace normalization.\n */\n public filterCustomShortcut(\n blocks: TipTapNode[],\n deleteId: string,\n ): {\n changed: boolean;\n filtered: TipTapNode[];\n } {\n let changed = false;\n\n const isObject = (v: unknown): v is Record =>\n typeof v === 'object' && v !== null;\n\n const filterInline = (content: unknown[]): unknown[] => {\n const out: unknown[] = [];\n for (const node of content) {\n if (!isObject(node)) {\n out.push(node);\n continue;\n }\n\n const {\n type,\n attrs,\n content: nodeContent,\n } = node as {\n type?: string;\n attrs?: Record;\n content?: unknown[];\n };\n\n if (type === 'customShortcut' && String(attrs?.id ?? '') === deleteId) {\n changed = true;\n continue; // drop this inline\n }\n\n // Recurse if this inline node has its own content\n if (Array.isArray(nodeContent)) {\n const clone = { ...node } as Record & {\n content?: unknown[];\n };\n clone.content = filterInline(nodeContent);\n out.push(clone);\n } else {\n out.push(node);\n }\n }\n return out;\n };\n\n const filterBlocks = (arr: TipTapNode[]): TipTapNode[] =>\n arr.map((blk) => {\n const clone: TipTapNode = { ...blk };\n if (Array.isArray(clone.content)) {\n clone.content = filterInline(clone.content) as TipTapNode[];\n }\n return clone;\n });\n\n const filtered = filterBlocks(blocks);\n return { changed, filtered };\n }\n}\n" }, { "path": "apps/client-server/src/app/websites/commons/post-builder.spec.ts", "content": "import { IFileBuffer } from '@postybirb/types';\nimport { FormFile } from '../../../../../../libs/http/src/lib/form-file'; // Direct import to avoid electron loading\nimport { CancellableToken } from '../../post/models/cancellable-token';\nimport { PostingFile } from '../../post/models/posting-file';\nimport { PostBuilder } from './post-builder';\n\n// Mocks\nconst mockWebsite = {\n account: { id: 'test-account' },\n constructor: { name: 'MockWebsite' },\n};\n\njest.mock('@postybirb/logger', () => ({\n Logger: () => ({\n withMetadata: () => ({ debug: jest.fn() }),\n debug: jest.fn(),\n }),\n}));\njest.mock('@postybirb/http', () => ({\n Http: {\n post: jest.fn().mockResolvedValue({ statusCode: 200, body: { id: '123' } }),\n },\n FormFile: FormFile,\n}));\n\nfunction createPostingFile(overrides = {}) {\n // Minimal IFileBuffer\n const fileBuffer: IFileBuffer = {\n id: 'file1',\n buffer: Buffer.from('test'),\n mimeType: 'image/png',\n width: 100,\n height: 100,\n fileName: 'file1.png',\n ...overrides,\n submissionFileId: '',\n size: 0,\n createdAt: '',\n updatedAt: '',\n };\n return new PostingFile('file1', fileBuffer);\n}\n\ndescribe('PostBuilder', () => {\n let builder: PostBuilder;\n let token: CancellableToken;\n\n beforeEach(() => {\n token = new CancellableToken();\n builder = new PostBuilder(mockWebsite as any, token);\n });\n\n it('should set headers', () => {\n builder.withHeader('X-Test', 'abc');\n expect((builder as any).headers['X-Test']).toBe('abc');\n });\n\n it('should set post type', () => {\n builder.asMultipart();\n expect((builder as any).postType).toBe('multipart');\n builder.asJson();\n expect((builder as any).postType).toBe('json');\n builder.asUrlEncoded();\n expect((builder as any).postType).toBe('urlencoded');\n });\n\n it('should merge data with withData', () => {\n builder.withData({ a: 1, b: 2 });\n builder.withData({ b: 3, c: 4 });\n expect((builder as any).data).toEqual({ a: 1, b: 3, c: 4 });\n });\n\n it('should set, get, and remove fields', () => {\n builder.setField('foo', 'bar');\n expect(builder.getField('foo')).toBe('bar');\n builder.removeField('foo');\n expect(builder.getField('foo')).toBeUndefined();\n });\n\n it('should set fields conditionally', () => {\n builder.setConditional('x', true, 1, 2);\n expect((builder as any).data['x']).toBe(1);\n builder.setConditional('y', false, 1, 2);\n expect((builder as any).data['y']).toBe(2);\n builder.setConditional('z', false, 1);\n expect((builder as any).data['z']).toBeUndefined();\n });\n\n it('should iterate with forEach', () => {\n builder.forEach(['a', 'b'], (item, idx, b) => {\n b.setField(`item${idx}`, item);\n });\n expect((builder as any).data['item0']).toBe('a');\n expect((builder as any).data['item1']).toBe('b');\n });\n\n it('should add files and thumbnails', () => {\n const file = createPostingFile();\n builder.addFile('file', file);\n expect((builder as any).data['file']).toBeInstanceOf(Object); // FormFile\n builder.addFiles('files', [file, file]);\n expect(Array.isArray((builder as any).data['files'])).toBe(true);\n builder.addThumbnail('thumb', file);\n expect((builder as any).data['thumb']).toBeInstanceOf(Object); // FormFile\n });\n\n it('should add image as thumbnail if no thumbnail', () => {\n const file = createPostingFile();\n builder.addThumbnail('thumb', file);\n expect((builder as any).data['thumb']).toBeInstanceOf(Object); // FormFile\n });\n\n it('should set empty string as thumbnail for non-image', () => {\n const file = createPostingFile({\n fileName: 'file1.txt',\n mimeType: 'text/plain',\n });\n builder.addThumbnail('thumb', file);\n expect((builder as any).data['thumb']).toBe('');\n });\n\n it('should call whenTrue only if predicate is true', () => {\n const cb = jest.fn();\n builder.whenTrue(true, cb);\n expect(cb).toHaveBeenCalled();\n cb.mockClear();\n builder.whenTrue(false, cb);\n expect(cb).not.toHaveBeenCalled();\n });\n\n it('should build data for json and multipart', () => {\n builder.setField('a', true).setField('b', [true, false, undefined]);\n expect(builder.build()).toEqual({ a: true, b: [true, false, undefined] });\n builder.asMultipart();\n expect(builder.build()).toEqual({ a: 'true', b: ['true', 'false'] });\n });\n\n it('should sanitize file fields for logging', () => {\n const file = createPostingFile();\n builder.addFile('file', file);\n const data = { file: builder.getField('file'), other: 123 };\n const sanitized = (builder as any).sanitizeDataForLogging(data);\n expect(typeof sanitized.file).toBe('string');\n expect(sanitized.other).toBe(123);\n });\n\n it('should throw if cancelled before send', async () => {\n token.cancel();\n await expect(builder.send('http://test')).rejects.toThrow(\n 'Task was cancelled.',\n );\n });\n\n it('should call Http.post and return value on send', async () => {\n const result = await builder.send<{ id: string }>('http://test');\n expect(result.body.id).toBe('123');\n });\n\n it('should convert PostingFile to FormFile', () => {\n const file = createPostingFile();\n const builder = new PostBuilder(mockWebsite as any, token);\n builder.setField('file', file);\n expect((builder as any).data['file']).toBeInstanceOf(FormFile);\n\n builder.addFile('file2', file);\n expect((builder as any).data['file2']).toBeInstanceOf(FormFile);\n });\n});\n" }, { "path": "apps/client-server/src/app/websites/commons/post-builder.ts", "content": "import {\n FormFile,\n Http,\n HttpRequestOptions,\n PostOptions,\n} from '@postybirb/http';\nimport { Logger } from '@postybirb/logger';\nimport { FileType, PostResponse } from '@postybirb/types';\nimport { CancellableToken } from '../../post/models/cancellable-token';\nimport { PostingFile } from '../../post/models/posting-file';\nimport { UnknownWebsite } from '../website';\n\n/**\n * Represents a field value that can be stored in the post data.\n */\ntype FieldValue = string | number | boolean | null | undefined | object;\n\n/**\n * Represents a value that can be either a single field value or an array of field values.\n */\ntype Value = FieldValue | FieldValue[];\n\n/**\n * A builder class for constructing HTTP POST requests to various websites.\n * Uses the builder pattern to allow fluent method chaining for configuring\n * request data, headers, and content type.\n *\n * @example\n * ```typescript\n * const response = await new PostBuilder(website, cancellationToken)\n * .asMultipart()\n * .withHeader('Authorization', 'Bearer token')\n * .setField('title', 'My Post')\n * .addFile('image', postingFile)\n * .send('https://api.example.com/posts');\n * ```\n */\nexport class PostBuilder {\n private readonly logger = Logger('PostBuilder');\n\n /**\n * The type of POST request to send (json, multipart, or urlencoded).\n * @private\n */\n private postType: PostOptions['type'] = 'json';\n\n /**\n * The data payload that will be sent in the POST request.\n * @private\n */\n private data: Record = {};\n\n /**\n * Custom HTTP request options to be send with the request.\n * @private\n * @type {HttpRequestOptions}\n */\n private readonly httpRequestOptions: HttpRequestOptions = {};\n\n /**\n * HTTP headers to include with the request.\n * @private\n */\n private readonly headers: Record = {};\n\n /**\n * Set of field names that are expected to contain file data based on input.\n * Used to enhance logging and debugging by identifying which fields\n * are intended for file uploads.\n * @private\n */\n private readonly fileFields = new Set();\n\n /**\n * When true, the request will be sent via Electron's BrowserWindow.loadURL\n * with raw data bytes instead of using net.request ClientRequest.\n * @private\n */\n private rawData = false;\n\n /**\n * Creates a new PostBuilder instance.\n *\n * @param website - The website instance for which the post is being built\n * @param cancellationToken - Token used to cancel the request if needed\n */\n constructor(\n private readonly website: UnknownWebsite,\n private readonly cancellationToken: CancellableToken,\n ) {}\n\n /**\n * Adds an HTTP header to the request.\n *\n * @param key - The header name\n * @param value - The header value\n * @returns The PostBuilder instance for method chaining\n *\n * @example\n * ```typescript\n * builder.withHeader('Content-Type', 'application/json')\n * .withHeader('Authorization', 'Bearer token');\n * ```\n */\n withHeader(key: string, value: string) {\n this.headers[key] = value;\n return this;\n }\n\n /**\n * Adds multiple headers to the request.\n * Merges the provided headers with existing ones.\n *\n * @param headers - Object containing key-value pairs of headers\n * @returns The PostBuilder instance for method chaining\n *\n * @example\n * ```typescript\n * builder.withHeaders({\n * 'Content-Type': 'application/json',\n * 'Authorization': 'Bearer token'\n * });\n * ```\n */\n withHeaders(headers: Record) {\n Object.entries(headers).forEach(([key, value]) => {\n this.headers[key] = value;\n });\n return this;\n }\n\n /**\n * Configures the request to use multipart/form-data encoding.\n * This is typically used when uploading files or sending binary data.\n *\n * @returns The PostBuilder instance for method chaining\n *\n * @example\n * ```typescript\n * builder.asMultipart().addFile('image', file);\n * ```\n */\n asMultipart() {\n this.postType = 'multipart';\n return this;\n }\n\n /**\n * Configures the request to use JSON encoding (default).\n * The request body will be sent as JSON with appropriate Content-Type header.\n *\n * @returns The PostBuilder instance for method chaining\n */\n asJson() {\n this.postType = 'json';\n return this;\n }\n\n /**\n * Configures the request to use URL-encoded form data.\n * The request body will be sent as application/x-www-form-urlencoded.\n *\n * @returns The PostBuilder instance for method chaining\n */\n asUrlEncoded(skipIndex = false) {\n this.postType = 'urlencoded';\n this.httpRequestOptions.skipUrlEncodedIndexing = skipIndex;\n return this;\n }\n\n /**\n * Configures the request to be sent via Electron's BrowserWindow.loadURL\n * with raw data bytes instead of the standard net.request flow.\n * Can be combined with any content type (multipart, json, urlencoded).\n *\n * @returns The PostBuilder instance for method chaining\n *\n * @example\n * ```typescript\n * builder.asMultipart().asRawData().addFile('image', file).send(url);\n * ```\n */\n asRawData() {\n this.rawData = true;\n return this;\n }\n\n /**\n * Merges the provided data object with the existing request data.\n * Existing keys will be overwritten with new values.\n *\n * @param data - Object containing key-value pairs to add to the request\n * @returns The PostBuilder instance for method chaining\n *\n * @example\n * ```typescript\n * builder.withData({\n * title: 'My Post',\n * description: 'Post description',\n * tags: ['tag1', 'tag2']\n * });\n * ```\n */\n withData(data: Record) {\n this.data = { ...this.data, ...data };\n return this;\n }\n\n getField(key: string): T | undefined {\n return this.data[key] as T | undefined;\n }\n\n removeField(key: string) {\n delete this.data[key];\n if (this.fileFields.has(key)) {\n this.fileFields.delete(key);\n }\n return this;\n }\n\n /**\n * Sets a single field in the request data.\n * Handles null values by converting them to undefined.\n *\n * @param key - The field name\n * @param value - The field value (can be a single value or array)\n * @returns The PostBuilder instance for method chaining\n *\n * @example\n * ```typescript\n * builder.setField('title', 'My Post Title')\n * .setField('tags', ['art', 'digital', 'illustration']);\n * ```\n */\n setField(key: string, value: Value) {\n this.insert(key, value);\n return this;\n }\n\n /**\n * Conditionally sets a field based on a predicate.\n * Useful for setting fields based on user preferences or feature flags.\n *\n * @param key - The field name\n * @param predicate - Boolean condition to evaluate\n * @param truthy - Value to set if predicate is true\n * @param falsy - Value to set if predicate is false (optional)\n * @returns The PostBuilder instance for method chaining\n *\n * @example\n * ```typescript\n * builder.setConditional('nsfw', post.isNsfw, true, false)\n * .setConditional('rating', post.rating > 0, post.rating);\n * ```\n */\n setConditional(\n key: string,\n predicate: boolean,\n truthy: Value,\n falsy?: Value,\n ) {\n this.insert(key, predicate ? truthy : falsy);\n return this;\n }\n\n /**\n * Iterates over an array and executes a callback for each item.\n *\n * @param items - Array of items to iterate over\n * @param callback - Function to execute for each item\n * @returns The PostBuilder instance for method chaining\n *\n * @example\n * ```typescript\n * builder.forEach(options.matureContent, (item, index, b) => {\n * b.setField(`attributes[${item}]`, 'true');\n * });\n * ```\n */\n forEach(\n items: T[] | undefined | null,\n callback: (item: T, index: number, builder: PostBuilder) => void,\n ) {\n if (items) {\n items.forEach((item, index) => callback(item, index, this));\n }\n return this;\n }\n\n /**\n * Adds a file to the request data using the specified field name.\n * The file is converted to the appropriate post format.\n *\n * @param key - The field name for the file\n * @param file - The PostingFile instance to add\n * @returns The PostBuilder instance for method chaining\n *\n * @example\n * ```typescript\n * builder.addFile('artwork', postingFile)\n * .addFile('reference', referenceFile);\n * ```\n */\n addFile(key: string, file: PostingFile | FormFile) {\n this.insert(key, file);\n return this;\n }\n\n /**\n * Adds multiple files to the request data under the specified field name.\n * Each file is converted to the appropriate post format.\n *\n * @param key - The field name for the files\n * @param files - Array of PostingFile instances to add\n * @returns The PostBuilder instance for method chaining\n *\n * @example\n * ```typescript\n * builder.addFiles('images', [file1, file2, file3]);\n * ```\n */\n addFiles(key: string, files: PostingFile[]) {\n this.data[key] = files.map((file) => file.toPostFormat());\n this.fileFields.add(key);\n return this;\n }\n\n /**\n * Adds a thumbnail to the request data.\n * If the file has a thumbnail, it uses that; otherwise, for image files,\n * it uses the original file as the thumbnail.\n *\n * @param key - The field name for the thumbnail\n * @param file - The PostingFile instance from which to extract the thumbnail\n * @returns The PostBuilder instance for method chaining\n *\n * @example\n * ```typescript\n * builder.addFile('video', videoFile)\n * .addThumbnail('thumbnail', videoFile);\n * ```\n */\n addThumbnail(key: string, file: PostingFile) {\n if (file.thumbnail) {\n this.data[key] = file.thumbnailToPostFormat();\n this.fileFields.add(key);\n } else if (file.fileType === FileType.IMAGE) {\n this.data[key] = file.toPostFormat();\n this.fileFields.add(key);\n } else {\n this.data[key] = '';\n }\n return this;\n }\n\n /**\n * Conditionally executes a callback based on a predicate.\n *\n * @param predicate - Boolean condition to evaluate\n * @param callback - Function to execute if predicate is true\n * @returns The PostBuilder instance for method chaining\n *\n * @example\n * ```typescript\n * builder.whenTrue(rating !== 'general', (b) => {\n * b.removeField('explicit');\n * });\n * ```\n */\n whenTrue(predicate: boolean, callback: (builder: PostBuilder) => void) {\n if (predicate) {\n callback(this);\n }\n return this;\n }\n\n /**\n * Sends the constructed POST request to the specified URL.\n * Validates the response and handles cancellation.\n *\n * @template ReturnValue - The expected type of the response body\n * @param url - The URL to send the POST request to\n * @returns Promise resolving to the response body\n * @throws {Error} If the request is cancelled or the response is invalid\n *\n * @example\n * ```typescript\n * interface ApiResponse {\n * id: string;\n * status: 'success' | 'error';\n * }\n *\n * const response = await builder.send('https://api.example.com/posts');\n * console.log(response.body.id);\n * ```\n */\n async send(url: string) {\n this.cancellationToken.throwIfCancelled();\n const data = this.build();\n this.logger\n .withMetadata({\n website: this.website.constructor.name,\n postType: this.postType,\n url,\n headers: Object.keys(this.headers),\n data: this.sanitizeDataForLogging(data),\n httpRequestOptions: this.httpRequestOptions,\n })\n .debug(`Sending ${this.postType} request to ${url} with data:`);\n\n const maxRetries = 2;\n let attempt = 0;\n let lastError: unknown;\n\n while (attempt <= maxRetries) {\n try {\n const value = await Http.post(url, {\n partition: this.website.account.id,\n type: this.postType,\n data,\n headers: this.headers,\n options: this.httpRequestOptions,\n uploadAsRawData: this.rawData,\n });\n this.logger.debug(`Received response from ${url}:`, value.statusCode);\n PostResponse.validateBody(this.website, value, undefined, url);\n return value;\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n } catch (error: any) {\n const knownErrors = ['ECONNRESET', 'ERR_CONNECTION_RESET'];\n let isKnownError = false;\n for (const knownError of knownErrors) {\n const isKnown =\n error &&\n (error.code === knownError ||\n (typeof error.message === 'string' &&\n error.message.includes(knownError)));\n if (isKnown) {\n attempt++;\n lastError = error;\n if (attempt > maxRetries) break;\n this.logger.debug(\n `Retrying request to ${url} due to ${knownError} (attempt ${attempt})`,\n );\n isKnownError = true;\n break;\n }\n }\n\n // If the error is not a known retryable error, log it and throw\n if (!isKnownError) {\n this.logger.error(\n `Failed to send request to ${url} after ${attempt} attempts:`,\n error,\n );\n throw error;\n }\n // If known error, continue loop (unless maxRetries exceeded)\n await new Promise((resolve) => {\n // Wait for 1 second before retrying\n setTimeout(resolve, 1_000);\n });\n }\n }\n throw lastError;\n }\n\n /**\n * Builds and returns the final data object that will be sent in the request.\n * Handles special formatting for multipart requests:\n * - Removes undefined values\n * - Converts boolean values to strings\n * - Filters undefined values from arrays\n * - Converts boolean values in arrays to strings\n *\n * @returns The processed data object ready for transmission\n *\n * @example\n * ```typescript\n * const data = builder.build();\n * // For multipart: { \"nsfw\": \"true\", \"tags\": [\"art\", \"digital\"] }\n * // For JSON: { \"nsfw\": true, \"tags\": [\"art\", \"digital\"] }\n * ```\n */\n public build(): Record {\n const data = { ...this.data };\n if (this.postType === 'multipart') {\n Object.keys(data).forEach((key) => {\n // If the value is undefined and we don't allow it, delete the key\n // This is necessary for multipart/form-data where undefined values are not allowed\n if (data[key] === undefined) {\n delete data[key];\n }\n\n if (typeof data[key] === 'boolean') {\n // Convert boolean values to string for multipart/form-data\n data[key] = data[key] ? 'true' : 'false';\n }\n\n if (Array.isArray(data[key])) {\n // If the value is an array, filter out undefined values\n data[key] = (data[key] as FieldValue[])\n .filter((v) => v !== undefined)\n .map((v) => {\n // Convert boolean values to string for multipart/form-data\n if (typeof v === 'boolean') {\n return v ? 'true' : 'false';\n }\n return v;\n });\n }\n });\n }\n\n return data;\n }\n\n /**\n * Converts a PostingFile or FieldValue to the appropriate format for posting.\n * If the value is a PostingFile, it converts it to a FormFile.\n *\n * @param value - The value to convert\n * @returns The converted value in the appropriate format\n */\n private convert(value: FieldValue | PostingFile): FieldValue | FormFile {\n if (value instanceof PostingFile) {\n return value.toPostFormat();\n }\n return value;\n }\n\n /**\n * Inserts a key-value pair into the data object.\n * If the value is a PostingFile, it converts it to the appropriate format.\n * If the value is a FormFile, it adds the key to the fileFields set.\n *\n * @param key - The field name\n * @param value - The field value (can be a PostingFile or FieldValue)\n */\n private insert(\n key: string,\n value: FieldValue | PostingFile | FormFile,\n ): void {\n const v = this.convert(value);\n this.data[key] = v;\n if (v instanceof FormFile) {\n this.fileFields.add(key);\n }\n }\n\n /**\n * Exposes ability to retrieve data for logging outside of the builder\n * if needed.\n */\n public getSanitizedData(): Record {\n return this.sanitizeDataForLogging(this.data);\n }\n\n private sanitizeDataForLogging(\n data: Record,\n ): Record {\n const sanitizedData: Record = {};\n for (const [key, value] of Object.entries(data)) {\n if (this.fileFields.has(key)) {\n // For file fields, we don't log the actual file content\n if (Array.isArray(value)) {\n sanitizedData[key] = value.map((v) =>\n v instanceof FormFile ? v.toString() : v,\n );\n } else {\n sanitizedData[key] = value.toString();\n }\n } else {\n sanitizedData[key] = value;\n }\n }\n return sanitizedData;\n }\n}\n" }, { "path": "apps/client-server/src/app/websites/commons/validator-passthru.ts", "content": "import { IWebsiteFormFields, SimpleValidationResult } from '@postybirb/types';\n\nexport async function validatorPassthru(): Promise<\n SimpleValidationResult\n> {\n return {};\n}\n" }, { "path": "apps/client-server/src/app/websites/commons/validator.ts", "content": "import {\n IWebsiteFormFields,\n SimpleValidationResult,\n ValidationMessage,\n ValidationMessages,\n} from '@postybirb/types';\n\ntype KeysToOmit =\n | 'mergeDefaults'\n | 'getFormFieldFor'\n | 'getFormFields'\n | 'getProcessedTags';\n\ntype ValidationArray = ValidationMessage<\n Fields,\n keyof ValidationMessages\n>[];\n\nexport class SubmissionValidator {\n protected readonly warnings: ValidationArray = [];\n\n protected readonly errors: ValidationArray = [];\n\n /**\n * Adds error to the validation result\n *\n * @param id - Error localization message id. {@link ValidationMessages}\n * @param values - Values to fill in the message\n * @param field - Associates the error to a input field\n */\n error(\n id: T,\n values: ValidationMessages[T],\n field?: keyof Omit,\n ) {\n this.errors.push({ id, values, field });\n }\n\n /**\n * Adds warning to the validation result\n *\n * @param id - Warning localization message id. {@link ValidationMessages}\n * @param values - Values to fill in the message\n * @param field - Associates the warning to a input field\n */\n warning(\n id: T,\n values: ValidationMessages[T],\n field?: keyof Omit,\n ) {\n this.warnings.push({ id, values, field });\n }\n\n /**\n * Returns validation result. Should be used in the onValidateFileSubmission or onValidateMessageSubmission\n */\n get result(): SimpleValidationResult {\n return { errors: this.errors, warnings: this.warnings };\n }\n}\n" }, { "path": "apps/client-server/src/app/websites/decorators/disable-ads.decorator.ts", "content": "import { injectWebsiteDecoratorProps } from './website-decorator-props';\n\nexport function DisableAds() {\n return function website(constructor) {\n injectWebsiteDecoratorProps(constructor, {\n allowAd: false,\n });\n };\n}\n" }, { "path": "apps/client-server/src/app/websites/decorators/login-flow.decorator.ts", "content": "import { CustomLoginType, UserLoginType } from '@postybirb/types';\nimport { Class } from 'type-fest';\nimport { UnknownWebsite } from '../website';\nimport { injectWebsiteDecoratorProps } from './website-decorator-props';\n\n/**\n * Identifies the website as having a user login flow.\n * Meaning that they will login to the website using the website url provided.\n * @param {string} url\n */\nexport function UserLoginFlow(url: string) {\n return function website(constructor: Class) {\n const loginFlow: UserLoginType = {\n type: 'user',\n url,\n };\n\n injectWebsiteDecoratorProps(constructor, { loginFlow });\n };\n}\n\n/**\n * Identifies the website as having a custom login flow.\n * Meaning that they will login through a custom provided form / component.\n * Defaults the name of the class if no name is provided.\n * @param {string} loginComponentName\n */\nexport function CustomLoginFlow(loginComponentName?: string) {\n return function website(constructor: Class) {\n const loginFlow: CustomLoginType = {\n type: 'custom',\n loginComponentName: loginComponentName ?? constructor.name,\n };\n\n injectWebsiteDecoratorProps(constructor, { loginFlow });\n };\n}\n" }, { "path": "apps/client-server/src/app/websites/decorators/supports-files.decorator.ts", "content": "import { ISubmissionFile, WebsiteFileOptions } from '@postybirb/types';\nimport {\n getFileType,\n getFileTypeFromMimeType,\n} from '@postybirb/utils/file-type';\nimport { parse } from 'path';\nimport { Class } from 'type-fest';\nimport { getDynamicFileSizeLimits } from '../models/website-modifiers/with-dynamic-file-size-limits';\nimport { UnknownWebsite } from '../website';\nimport { injectWebsiteDecoratorProps } from './website-decorator-props';\n\nexport function SupportsFiles(\n websiteFileOptions: Omit,\n);\nexport function SupportsFiles(acceptedMimeTypes: string[]);\nexport function SupportsFiles(\n websiteFileOptionsOrMimeTypes:\n | Omit\n | string[],\n) {\n return function website(constructor: Class) {\n let websiteFileOptions: WebsiteFileOptions = Array.isArray(\n websiteFileOptionsOrMimeTypes,\n )\n ? {\n acceptedMimeTypes: websiteFileOptionsOrMimeTypes,\n supportedFileTypes: [],\n }\n : { ...websiteFileOptionsOrMimeTypes, supportedFileTypes: [] };\n\n websiteFileOptions = {\n acceptedFileSizes: {},\n acceptedMimeTypes: [],\n acceptsExternalSourceUrls: false,\n fileBatchSize: 1,\n ...websiteFileOptions,\n };\n\n websiteFileOptions.acceptedMimeTypes.forEach((mimeType) => {\n const fileType = getFileTypeFromMimeType(mimeType);\n if (!websiteFileOptions.supportedFileTypes.includes(fileType)) {\n websiteFileOptions.supportedFileTypes.push(fileType);\n }\n });\n\n injectWebsiteDecoratorProps(constructor, {\n fileOptions: websiteFileOptions,\n });\n };\n}\n\nexport function getSupportedFileSize(\n instance: UnknownWebsite,\n file: ISubmissionFile,\n) {\n const acceptedFileSizes =\n instance.decoratedProps.fileOptions?.acceptedFileSizes;\n\n const dynamicFileSizeLimits = getDynamicFileSizeLimits(instance);\n\n if (!acceptedFileSizes && !dynamicFileSizeLimits) {\n return undefined;\n }\n\n const limits = { ...acceptedFileSizes, ...dynamicFileSizeLimits };\n\n return (\n limits[file.mimeType] ??\n limits[`${file.mimeType.split('/')[0]}/*`] ??\n limits[parse(file.fileName).ext] ??\n limits[getFileType(file.fileName)] ??\n limits['*'] ??\n Number.MAX_SAFE_INTEGER\n );\n}\n" }, { "path": "apps/client-server/src/app/websites/decorators/supports-username-shortcut.decorator.ts", "content": "import { UsernameShortcut } from '@postybirb/types';\nimport { Class } from 'type-fest';\nimport { UnknownWebsite } from '../website';\nimport { injectWebsiteDecoratorProps } from './website-decorator-props';\n\n/**\n * Sets a username shortcut for a website.\n * @param {UsernameShortcut} usernameShortcut\n */\nexport function SupportsUsernameShortcut(usernameShortcut: UsernameShortcut) {\n return function website(constructor: Class) {\n injectWebsiteDecoratorProps(constructor, { usernameShortcut });\n };\n}\n" }, { "path": "apps/client-server/src/app/websites/decorators/website-decorator-props.ts", "content": "import {\n CustomLoginType,\n IWebsiteMetadata,\n UserLoginType,\n UsernameShortcut,\n WebsiteFileOptions,\n} from '@postybirb/types';\nimport { LogLayer } from 'loglayer';\nimport { Class } from 'type-fest';\nimport { UnknownWebsite } from '../website';\n\nexport type WebsiteDecoratorProps = {\n /**\n * Set by {@link SupportsFiles}\n *\n * Defines the file options for a website.\n * @type {WebsiteFileOptions}\n */\n fileOptions?: WebsiteFileOptions;\n\n /**\n * Set by {@link UserLoginFlow} or {@link CustomLoginFlow}\n *\n * Defines login flow properties for a website.\n * This is used to determine how a user will login to a website.\n * @type {UserLoginType} - User will login through a webview using the provided url.\n * @type {CustomLoginType} - User will login through a custom login flow created by the implementer.\n * @type {(UserLoginType | CustomLoginType)}\n */\n loginFlow: UserLoginType | CustomLoginType;\n\n /**\n * Set by {@link WebsiteMetadata}\n *\n * Defines the metadata for a website.\n * This is usually for display or internal Ids.\n * @type {IWebsiteMetadata}\n */\n metadata: IWebsiteMetadata;\n\n /**\n * Set by {@SupportsUsernameShortcut}\n *\n * Defines the username shortcut for a website.\n * This is used to modify links to users for websites that support it.\n * @type {UsernameShortcut}\n */\n usernameShortcut?: UsernameShortcut;\n\n /**\n * Disable Ads in description by using {@link DisableAdSupport}\n *\n * @type {boolean}\n */\n allowAd: boolean;\n};\n\nexport function defaultWebsiteDecoratorProps(): WebsiteDecoratorProps {\n return {\n fileOptions: undefined,\n loginFlow: undefined,\n metadata: undefined,\n allowAd: true,\n };\n}\n\n/**\n * Injects basic website decorator properties into a website instance.\n *\n * @param {Class} constructor\n * @param {WebsiteDecoratorProps} props\n */\nexport function injectWebsiteDecoratorProps(\n constructor: Class,\n props: Partial,\n): void {\n if (!constructor.prototype.decoratedProps) {\n Object.assign(constructor.prototype, {\n decoratedProps: defaultWebsiteDecoratorProps(),\n });\n }\n\n Object.entries(props).forEach(([key, value]) => {\n if (value !== undefined) {\n // eslint-disable-next-line no-param-reassign\n constructor.prototype.decoratedProps[key] = value;\n }\n });\n}\n\nexport function validateWebsiteDecoratorProps(\n logger: LogLayer,\n websiteName: string,\n props: WebsiteDecoratorProps,\n): boolean {\n if (!props.loginFlow) {\n logger\n .withContext({ websiteName })\n .error(\n 'Website is missing login flow. Please set a login flow using UserLoginFlow or CustomLoginFlow decorators.',\n );\n return false;\n }\n\n if (!props.metadata) {\n logger\n .withContext({ websiteName })\n .error(\n 'Website is missing metadata. Please set metadata using WebsiteMetadata decorator.',\n );\n return false;\n }\n\n if (!props.metadata.name) {\n logger.withContext({ websiteName }).error(`Missing metadata field 'name'`);\n }\n\n return true;\n}\n" }, { "path": "apps/client-server/src/app/websites/decorators/website-metadata.decorator.ts", "content": "import { IWebsiteMetadata } from '@postybirb/types';\nimport { Class } from 'type-fest';\nimport { UnknownWebsite } from '../website';\nimport { injectWebsiteDecoratorProps } from './website-decorator-props';\n\nexport function WebsiteMetadata(metadata: IWebsiteMetadata) {\n return function website(constructor: Class) {\n const m = { ...metadata };\n // Determine default login refresh\n if (!metadata.refreshInterval) {\n // Default (1 hour)\n m.refreshInterval = 60 * 60_000;\n }\n\n injectWebsiteDecoratorProps(constructor, { metadata: m });\n };\n}\n" }, { "path": "apps/client-server/src/app/websites/dtos/oauth-website-request.dto.ts", "content": "import { ApiProperty } from '@nestjs/swagger';\nimport { DynamicObject, IOAuthWebsiteRequestDto } from '@postybirb/types';\nimport { IsObject, IsString } from 'class-validator';\n\nexport class OAuthWebsiteRequestDto\n implements IOAuthWebsiteRequestDto\n{\n @ApiProperty()\n @IsString()\n id: string;\n\n @ApiProperty({\n type: Object,\n })\n @IsObject()\n data: T;\n\n @ApiProperty()\n @IsString()\n route: string;\n}\n" }, { "path": "apps/client-server/src/app/websites/implementations/artconomy/artconomy.website.ts", "content": "import { Http } from '@postybirb/http';\nimport {\n ILoginState,\n ImageResizeProps,\n ISubmissionFile,\n PostData,\n PostResponse,\n SubmissionRating,\n} from '@postybirb/types';\nimport { CancellableToken } from '../../../post/models/cancellable-token';\nimport { PostingFile } from '../../../post/models/posting-file';\nimport FileSize from '../../../utils/filesize.util';\nimport { PostBuilder } from '../../commons/post-builder';\nimport { validatorPassthru } from '../../commons/validator-passthru';\nimport { UserLoginFlow } from '../../decorators/login-flow.decorator';\nimport { SupportsFiles } from '../../decorators/supports-files.decorator';\nimport { SupportsUsernameShortcut } from '../../decorators/supports-username-shortcut.decorator';\nimport { WebsiteMetadata } from '../../decorators/website-metadata.decorator';\nimport { DataPropertyAccessibility } from '../../models/data-property-accessibility';\nimport { FileWebsite } from '../../models/website-modifiers/file-website';\nimport { MessageWebsite } from '../../models/website-modifiers/message-website';\nimport { Website } from '../../website';\nimport { ArtconomyAccountData } from './models/artconomy-account-data';\nimport { ArtconomyFileSubmission } from './models/artconomy-file-submission';\nimport { ArtconomyMessageSubmission } from './models/artconomy-message-submission';\n\n@WebsiteMetadata({\n name: 'artconomy',\n displayName: 'Artconomy',\n})\n@UserLoginFlow('https://artconomy.com/auth/login')\n@SupportsUsernameShortcut({\n id: 'artconomy',\n url: 'https://artconomy.com/profile/$1/about',\n})\n@SupportsFiles({\n acceptedMimeTypes: [\n 'image/png',\n 'image/jpeg',\n 'image/jpg',\n 'image/gif',\n 'video/mp4',\n 'application/msword',\n 'application/rtf',\n 'text/plain',\n 'audio/mp3',\n ],\n acceptedFileSizes: {\n '*': FileSize.megabytes(49),\n },\n})\nexport default class Artconomy\n extends Website\n implements\n FileWebsite,\n MessageWebsite\n{\n protected BASE_URL = 'https://artconomy.com';\n\n public externallyAccessibleWebsiteDataProperties: DataPropertyAccessibility =\n {};\n\n public async onLogin(): Promise {\n try {\n const authCheck = await Http.get<{ username: string; id: number }>(\n `${this.BASE_URL}/api/profiles/data/requester/`,\n {\n partition: this.accountId,\n headers: {\n 'Content-Type': 'application/json',\n },\n },\n );\n\n if (authCheck.statusCode === 200 && authCheck.body.username !== '_') {\n // Get CSRF token from cookies\n const cookies = await Http.getWebsiteCookies(\n this.accountId,\n this.BASE_URL,\n );\n const csrfCookie = cookies.find((c) => c.name === 'csrftoken');\n await this.setWebsiteData({\n id: authCheck.body.id,\n username: authCheck.body.username,\n csrfToken: csrfCookie?.value || '',\n });\n return this.loginState.setLogin(true, authCheck.body.username);\n }\n\n return this.loginState.setLogin(false, null);\n } catch (error) {\n return this.loginState.setLogin(false, null);\n }\n }\n\n createFileModel(): ArtconomyFileSubmission {\n return new ArtconomyFileSubmission();\n }\n\n calculateImageResize(file: ISubmissionFile): ImageResizeProps {\n return undefined;\n }\n\n async onPostFileSubmission(\n postData: PostData,\n files: PostingFile[],\n cancellationToken: CancellableToken,\n ): Promise {\n const { id, username, csrfToken } = this.getWebsiteData();\n\n if (!id || !username || !csrfToken) {\n return PostResponse.fromWebsite(this).withException(\n new Error('Not properly logged in to Artconomy'),\n );\n }\n\n // Upload primary asset using PostBuilder\n const primaryUpload = await new PostBuilder(this, cancellationToken)\n .asMultipart()\n .addFile('files[]', files[0])\n .withHeader('X-CSRFTOKEN', csrfToken)\n .withHeader('Referer', this.BASE_URL)\n .send<{ id: string }>(`${this.BASE_URL}/api/lib/asset/`);\n\n if (!primaryUpload.body?.id) {\n return PostResponse.fromWebsite(this)\n .withException(\n new Error(\n `Asset upload failed: ${primaryUpload.statusMessage || 'Unknown error'}`,\n ),\n )\n .withAdditionalInfo(primaryUpload.body);\n }\n\n const primaryAsset = primaryUpload.body.id;\n let thumbnailAsset: string | null = null;\n\n // Upload thumbnail if available\n if (files[0].thumbnail) {\n const thumbnailUpload = await new PostBuilder(this, cancellationToken)\n .asMultipart()\n .addThumbnail('files[]', files[0])\n .withHeader('X-CSRFTOKEN', csrfToken)\n .withHeader('Referer', this.BASE_URL)\n .send<{ id: string }>(`${this.BASE_URL}/api/lib/asset/`);\n\n if (!thumbnailUpload.body?.id) {\n return PostResponse.fromWebsite(this)\n .withException(\n new Error(\n `Thumbnail upload failed: ${thumbnailUpload.statusMessage || 'Unknown error'}`,\n ),\n )\n .withAdditionalInfo(thumbnailUpload.body);\n }\n thumbnailAsset = thumbnailUpload.body.id;\n }\n\n cancellationToken.throwIfCancelled();\n\n // Create submission using PostBuilder\n const postResponse = await new PostBuilder(this, cancellationToken)\n .asJson()\n .setField('file', primaryAsset)\n .setField('preview', thumbnailAsset)\n .setField('title', postData.options.title)\n .setField('caption', postData.options.description)\n .setField('tags', postData.options.tags)\n .setField('rating', this.getRating(postData.options.rating))\n .setField('private', postData.options.isPrivate)\n .setField('comments_disabled', postData.options.commentsDisabled)\n .setConditional('artists', postData.options.isArtist, [id], [])\n .withHeader('X-CSRFTOKEN', csrfToken)\n .withHeader('Referer', this.BASE_URL)\n .send<{\n id: string;\n }>(`${this.BASE_URL}/api/profiles/account/${username}/submissions/`);\n\n if (!postResponse.body?.id) {\n return PostResponse.fromWebsite(this)\n .withException(\n new Error(\n `Submission creation failed: ${postResponse.statusMessage || 'Unknown error'}`,\n ),\n )\n .withAdditionalInfo(postResponse.body);\n }\n\n return PostResponse.fromWebsite(this)\n .withSourceUrl(`${this.BASE_URL}/submissions/${postResponse.body.id}`)\n .withMessage('File posted successfully');\n }\n\n onValidateFileSubmission = validatorPassthru;\n\n createMessageModel(): ArtconomyMessageSubmission {\n return new ArtconomyMessageSubmission();\n }\n\n async onPostMessageSubmission(\n postData: PostData,\n cancellationToken: CancellableToken,\n ): Promise {\n cancellationToken.throwIfCancelled();\n\n const { username, csrfToken } = this.getWebsiteData();\n\n if (!username || !csrfToken) {\n return PostResponse.fromWebsite(this).withException(\n new Error('Not properly logged in to Artconomy'),\n );\n }\n\n const postResponse = await new PostBuilder(this, cancellationToken)\n .asJson()\n .setField('subject', postData.options.title)\n .setField('body', postData.options.description)\n .withHeader('X-CSRFTOKEN', csrfToken)\n .withHeader('Referer', this.BASE_URL)\n .send<{\n id: number;\n }>(`${this.BASE_URL}/api/profiles/account/${username}/journals/`);\n\n if (!postResponse.body?.id) {\n return PostResponse.fromWebsite(this)\n .withException(\n new Error(\n `Journal creation failed: ${postResponse.statusMessage || 'Unknown error'}`,\n ),\n )\n .withAdditionalInfo(postResponse.body);\n }\n\n return PostResponse.fromWebsite(this)\n .withSourceUrl(\n `${this.BASE_URL}/profile/${username}/journals/${postResponse.body.id}`,\n )\n .withMessage('Journal posted successfully');\n }\n\n onValidateMessageSubmission = validatorPassthru;\n\n private getRating(rating: SubmissionRating): number {\n switch (rating) {\n case SubmissionRating.GENERAL:\n return 0;\n case SubmissionRating.MATURE:\n return 1;\n case SubmissionRating.ADULT:\n return 2;\n case SubmissionRating.EXTREME:\n return 3;\n default:\n // Safest assumption\n return 2;\n }\n }\n}\n" }, { "path": "apps/client-server/src/app/websites/implementations/artconomy/models/artconomy-account-data.ts", "content": "import { SelectOption } from '@postybirb/form-builder'; \n\nexport type ArtconomyAccountData = { \n id?: number;\n username?: string;\n csrfToken?: string;\n}\n" }, { "path": "apps/client-server/src/app/websites/implementations/artconomy/models/artconomy-file-submission.ts", "content": "import {\n BooleanField,\n DescriptionField,\n RatingField,\n TagField,\n} from '@postybirb/form-builder';\nimport {\n DescriptionType,\n DescriptionValue,\n SubmissionRating,\n TagValue,\n} from '@postybirb/types';\nimport { BaseWebsiteOptions } from '../../../models/base-website-options';\n\nexport class ArtconomyFileSubmission extends BaseWebsiteOptions {\n @DescriptionField({\n descriptionType: DescriptionType.MARKDOWN,\n maxDescriptionLength: 2000,\n })\n description: DescriptionValue;\n\n @TagField({\n minTags: 5,\n })\n tags: TagValue;\n\n @RatingField({\n options: [\n { value: SubmissionRating.GENERAL, label: 'Clean/Safe' },\n { value: SubmissionRating.MATURE, label: 'Risque' },\n { value: SubmissionRating.ADULT, label: 'Adult' },\n { value: SubmissionRating.EXTREME, label: 'Offensive/Disturbing' },\n ],\n })\n rating: SubmissionRating;\n\n @BooleanField({\n label: 'private',\n section: 'website',\n span: 6,\n defaultValue: false,\n })\n isPrivate: boolean;\n\n @BooleanField({\n label: 'disableComments',\n section: 'website',\n span: 6,\n defaultValue: false,\n })\n commentsDisabled: boolean;\n\n @BooleanField({\n label: 'originalWork',\n section: 'website',\n span: 6,\n defaultValue: true,\n })\n isArtist: boolean;\n}\n" }, { "path": "apps/client-server/src/app/websites/implementations/artconomy/models/artconomy-message-submission.ts", "content": "import { DescriptionField } from '@postybirb/form-builder';\nimport { DescriptionType, DescriptionValue } from '@postybirb/types';\nimport { BaseWebsiteOptions } from '../../../models/base-website-options';\n\nexport class ArtconomyMessageSubmission extends BaseWebsiteOptions {\n @DescriptionField({\n descriptionType: DescriptionType.HTML\n })\n description: DescriptionValue;\n}\n" }, { "path": "apps/client-server/src/app/websites/implementations/aryion/aryion.website.ts", "content": "import { SelectOption } from '@postybirb/form-builder';\nimport { Http } from '@postybirb/http';\nimport {\n FileType,\n ILoginState,\n ImageResizeProps,\n PostData,\n PostResponse,\n SimpleValidationResult,\n} from '@postybirb/types';\nimport { HTMLElement, parse } from 'node-html-parser';\nimport { CancellableToken } from '../../../post/models/cancellable-token';\nimport { PostingFile } from '../../../post/models/posting-file';\nimport FileSize from '../../../utils/filesize.util';\nimport { SelectOptionUtil } from '../../../utils/select-option.util';\nimport { PostBuilder } from '../../commons/post-builder';\nimport { UserLoginFlow } from '../../decorators/login-flow.decorator';\nimport { SupportsFiles } from '../../decorators/supports-files.decorator';\nimport { WebsiteMetadata } from '../../decorators/website-metadata.decorator';\nimport { DataPropertyAccessibility } from '../../models/data-property-accessibility';\nimport { FileWebsite } from '../../models/website-modifiers/file-website';\nimport { Website } from '../../website';\nimport { AryionAccountData } from './models/aryion-account-data';\nimport { AryionFileSubmission } from './models/aryion-file-submission';\n\n@WebsiteMetadata({\n name: 'aryion',\n displayName: 'Aryion',\n})\n@UserLoginFlow('https://aryion.com/forum/ucp.php?mode=login')\n@SupportsFiles({\n acceptedMimeTypes: [\n 'image/jpeg',\n 'image/jpg',\n 'image/gif',\n 'image/png',\n 'application/msword',\n 'application/vnd.openxmlformats-officedocument.wordprocessingml.document',\n 'application/vnd.ms-excel',\n 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',\n 'application/x-shockwave-flash',\n 'application/vnd.visio',\n 'text/plain',\n 'application/rtf',\n 'video/x-msvideo',\n 'video/mpeg',\n 'video/x-flv',\n 'video/mp4',\n 'application/pdf',\n ],\n acceptedFileSizes: {\n [FileType.IMAGE]: FileSize.megabytes(20),\n [FileType.VIDEO]: FileSize.megabytes(100),\n [FileType.TEXT]: FileSize.megabytes(100),\n 'application/pdf': FileSize.megabytes(100),\n },\n})\nexport default class Aryion\n extends Website\n implements FileWebsite\n{\n protected BASE_URL = 'https://aryion.com';\n\n public externallyAccessibleWebsiteDataProperties: DataPropertyAccessibility =\n {\n folders: true,\n };\n\n public async onLogin(): Promise {\n const res = await Http.get(`${this.BASE_URL}/g4/treeview.php`, {\n partition: this.accountId,\n });\n\n if (\n res.body.includes('user-link') &&\n !res.body.includes('Login to read messages')\n ) {\n const $ = parse(res.body);\n const userLink = $.querySelector('.user-link');\n const username = userLink ? userLink.text : 'Unknown User';\n this.loginState.setLogin(true, username);\n await this.getFolders($);\n } else {\n this.loginState.logout();\n }\n\n return this.loginState.getState();\n }\n\n private async getFolders($: HTMLElement): Promise {\n const folders: SelectOption[] = [];\n const treeviews = $.querySelectorAll('.treeview');\n\n treeviews.forEach((treeview) => {\n // Process each top-level
    2. element\n const topLevelItems = treeview.querySelectorAll(':scope > li');\n topLevelItems.forEach((li) => {\n this.parseFolderItem(li, folders);\n });\n });\n\n this.websiteDataStore.setData({\n ...this.websiteDataStore.getData(),\n folders,\n });\n }\n\n private parseFolderItem(li: HTMLElement, parent: SelectOption[]): void {\n // Find the span element that contains the folder info\n const folderSpan = li.querySelector(':scope > span');\n if (!folderSpan) return;\n\n const dataTid = folderSpan.getAttribute('data-tid');\n const folderName = folderSpan.text.trim();\n\n if (!dataTid || !folderName) return;\n\n // Check if this folder has children (look for a
        sibling)\n const childrenUl = li.querySelector(':scope > ul');\n\n if (childrenUl) {\n // This is a parent folder with children\n const childItems: SelectOption[] = [];\n\n // Process each child
      • element\n const childLis = childrenUl.querySelectorAll(':scope > li');\n childLis.forEach((childLi) => {\n this.parseFolderItem(childLi, childItems);\n });\n\n // Create a group entry for this folder\n parent.push({\n label: folderName,\n items: childItems,\n value: dataTid,\n });\n } else {\n // This is a leaf folder (no children)\n parent.push({\n value: dataTid,\n label: folderName,\n });\n }\n }\n\n createFileModel(): AryionFileSubmission {\n return new AryionFileSubmission();\n }\n\n calculateImageResize(): ImageResizeProps {\n return undefined;\n }\n\n async onPostFileSubmission(\n postData: PostData,\n files: PostingFile[],\n cancellationToken: CancellableToken,\n ): Promise {\n cancellationToken.throwIfCancelled();\n\n const { options } = postData;\n const file = files[0];\n\n // Filter out 'vore' and 'non-vore' tags from the tags list\n const filteredTags = options.tags\n .filter((tag) => !tag.toLowerCase().match(/^vore$/i))\n .filter((tag) => !tag.toLowerCase().match(/^non-vore$/i));\n\n const builder = new PostBuilder(this, cancellationToken)\n .asMultipart()\n .addFile('file', file)\n .addFile('thumb', file)\n .setField('desc', options.description)\n .setField('title', options.title)\n .setField('tags', filteredTags.join('\\n'))\n .setField('reqtag[]', options.requiredTag === '1' ? 'Non-Vore' : '')\n .setField('view_perm', options.viewPermissions)\n .setField('comment_perm', options.commentPermissions)\n .setField('tag_perm', options.tagPermissions)\n .setField('scrap', options.scraps ? 'on' : '')\n .setField('parentid', options.folder)\n .setField('action', 'new-item')\n .setField('MAX_FILE_SIZE', '104857600');\n\n const result = await builder.send(\n `${this.BASE_URL}/g4/itemaction.php`,\n );\n\n try {\n // Split errors/warnings if they exist and handle them separately\n const responses = result.body\n .trim()\n .split('\\n')\n .map((r) => r?.trim());\n\n if (responses.length > 1 && responses[0].indexOf('Warning:') === -1) {\n return PostResponse.fromWebsite(this)\n .withAdditionalInfo(result.body)\n .withException(new Error('Server returned warnings or errors'));\n }\n\n // Parse the JSON response\n const jsonResponse = responses[responses.length - 1].replace(\n /(