mirror of
https://github.com/outline/outline.git
synced 2026-05-07 18:41:12 -05:00
2086 lines
54 KiB
TypeScript
2086 lines
54 KiB
TypeScript
import path from "path";
|
||
import fractionalIndex from "fractional-index";
|
||
import fs from "fs-extra";
|
||
import invariant from "invariant";
|
||
import JSZip from "jszip";
|
||
import Router from "koa-router";
|
||
import escapeRegExp from "lodash/escapeRegExp";
|
||
import has from "lodash/has";
|
||
import isNil from "lodash/isNil";
|
||
import remove from "lodash/remove";
|
||
import uniq from "lodash/uniq";
|
||
import mime from "mime-types";
|
||
import { Op, ScopeOptions, Sequelize, WhereOptions } from "sequelize";
|
||
import { v4 as uuidv4 } from "uuid";
|
||
import { StatusFilter, TeamPreference, UserRole } from "@shared/types";
|
||
import { subtractDate } from "@shared/utils/date";
|
||
import slugify from "@shared/utils/slugify";
|
||
import documentCreator from "@server/commands/documentCreator";
|
||
import documentDuplicator from "@server/commands/documentDuplicator";
|
||
import documentLoader from "@server/commands/documentLoader";
|
||
import documentMover from "@server/commands/documentMover";
|
||
import documentPermanentDeleter from "@server/commands/documentPermanentDeleter";
|
||
import documentUpdater from "@server/commands/documentUpdater";
|
||
import env from "@server/env";
|
||
import {
|
||
InvalidRequestError,
|
||
AuthenticationError,
|
||
ValidationError,
|
||
IncorrectEditionError,
|
||
} from "@server/errors";
|
||
import Logger from "@server/logging/Logger";
|
||
import auth from "@server/middlewares/authentication";
|
||
import multipart from "@server/middlewares/multipart";
|
||
import { rateLimiter } from "@server/middlewares/rateLimiter";
|
||
import { transaction } from "@server/middlewares/transaction";
|
||
import validate from "@server/middlewares/validate";
|
||
import {
|
||
Attachment,
|
||
Backlink,
|
||
Collection,
|
||
Document,
|
||
Event,
|
||
Revision,
|
||
SearchQuery,
|
||
User,
|
||
View,
|
||
UserMembership,
|
||
Group,
|
||
GroupUser,
|
||
GroupMembership,
|
||
} from "@server/models";
|
||
import AttachmentHelper from "@server/models/helpers/AttachmentHelper";
|
||
import { DocumentHelper } from "@server/models/helpers/DocumentHelper";
|
||
import { ProsemirrorHelper } from "@server/models/helpers/ProsemirrorHelper";
|
||
import SearchHelper from "@server/models/helpers/SearchHelper";
|
||
import { TextHelper } from "@server/models/helpers/TextHelper";
|
||
import { authorize, can, cannot } from "@server/policies";
|
||
import {
|
||
presentDocument,
|
||
presentPolicies,
|
||
presentMembership,
|
||
presentPublicTeam,
|
||
presentUser,
|
||
presentGroupMembership,
|
||
presentGroup,
|
||
} from "@server/presenters";
|
||
import DocumentImportTask, {
|
||
DocumentImportTaskResponse,
|
||
} from "@server/queues/tasks/DocumentImportTask";
|
||
import EmptyTrashTask from "@server/queues/tasks/EmptyTrashTask";
|
||
import FileStorage from "@server/storage/files";
|
||
import { APIContext } from "@server/types";
|
||
import { RateLimiterStrategy } from "@server/utils/RateLimiter";
|
||
import ZipHelper from "@server/utils/ZipHelper";
|
||
import { getTeamFromContext } from "@server/utils/passport";
|
||
import { assertPresent } from "@server/validation";
|
||
import pagination from "../middlewares/pagination";
|
||
import * as T from "./schema";
|
||
|
||
const router = new Router();
|
||
|
||
router.post(
|
||
"documents.list",
|
||
auth(),
|
||
pagination(),
|
||
validate(T.DocumentsListSchema),
|
||
async (ctx: APIContext<T.DocumentsListReq>) => {
|
||
const {
|
||
sort,
|
||
direction,
|
||
template,
|
||
collectionId,
|
||
backlinkDocumentId,
|
||
parentDocumentId,
|
||
userId: createdById,
|
||
statusFilter,
|
||
} = ctx.input.body;
|
||
|
||
// always filter by the current team
|
||
const { user } = ctx.state.auth;
|
||
const where: WhereOptions<Document> & {
|
||
[Op.and]: WhereOptions<Document>[];
|
||
} = {
|
||
teamId: user.teamId,
|
||
[Op.and]: [
|
||
{
|
||
deletedAt: {
|
||
[Op.eq]: null,
|
||
},
|
||
},
|
||
],
|
||
};
|
||
|
||
// Exclude archived docs by default
|
||
if (!statusFilter) {
|
||
where[Op.and].push({ archivedAt: { [Op.eq]: null } });
|
||
}
|
||
|
||
if (template) {
|
||
where[Op.and].push({
|
||
template: true,
|
||
});
|
||
}
|
||
|
||
// if a specific user is passed then add to filters. If the user doesn't
|
||
// exist in the team then nothing will be returned, so no need to check auth
|
||
if (createdById) {
|
||
where[Op.and].push({ createdById });
|
||
}
|
||
|
||
let documentIds: string[] = [];
|
||
|
||
// if a specific collection is passed then we need to check auth to view it
|
||
if (collectionId) {
|
||
where[Op.and].push({ collectionId: [collectionId] });
|
||
const collection = await Collection.scope({
|
||
method: ["withMembership", user.id],
|
||
}).findByPk(collectionId);
|
||
authorize(user, "readDocument", collection);
|
||
|
||
// index sort is special because it uses the order of the documents in the
|
||
// collection.documentStructure rather than a database column
|
||
if (sort === "index") {
|
||
documentIds = (collection?.documentStructure || [])
|
||
.map((node) => node.id)
|
||
.slice(ctx.state.pagination.offset, ctx.state.pagination.limit);
|
||
where[Op.and].push({ id: documentIds });
|
||
} // if it's not a backlink request, filter by all collections the user has access to
|
||
} else if (!backlinkDocumentId) {
|
||
const collectionIds = await user.collectionIds();
|
||
where[Op.and].push({
|
||
collectionId:
|
||
template && can(user, "readTemplate", user.team)
|
||
? {
|
||
[Op.or]: [{ [Op.in]: collectionIds }, { [Op.is]: null }],
|
||
}
|
||
: collectionIds,
|
||
});
|
||
}
|
||
|
||
if (parentDocumentId) {
|
||
const [groupMembership, membership] = await Promise.all([
|
||
GroupMembership.findOne({
|
||
where: {
|
||
documentId: parentDocumentId,
|
||
},
|
||
include: [
|
||
{
|
||
model: Group,
|
||
required: true,
|
||
include: [
|
||
{
|
||
model: GroupUser,
|
||
required: true,
|
||
where: {
|
||
userId: user.id,
|
||
},
|
||
},
|
||
],
|
||
},
|
||
],
|
||
}),
|
||
UserMembership.findOne({
|
||
where: {
|
||
userId: user.id,
|
||
documentId: parentDocumentId,
|
||
},
|
||
}),
|
||
]);
|
||
|
||
if (groupMembership || membership) {
|
||
remove(where[Op.and], (cond) => has(cond, "collectionId"));
|
||
}
|
||
|
||
where[Op.and].push({ parentDocumentId });
|
||
}
|
||
|
||
// Explicitly passing 'null' as the parentDocumentId allows listing documents
|
||
// that have no parent document (aka they are at the root of the collection)
|
||
if (parentDocumentId === null) {
|
||
where[Op.and].push({
|
||
parentDocumentId: {
|
||
[Op.is]: null,
|
||
},
|
||
});
|
||
}
|
||
|
||
if (backlinkDocumentId) {
|
||
const sourceDocumentIds = await Backlink.findSourceDocumentIdsForUser(
|
||
backlinkDocumentId,
|
||
user
|
||
);
|
||
|
||
where[Op.and].push({ id: sourceDocumentIds });
|
||
|
||
// For safety, ensure the collectionId is not set in the query.
|
||
remove(where[Op.and], (cond) => has(cond, "collectionId"));
|
||
}
|
||
|
||
const statusQuery = [];
|
||
if (statusFilter?.includes(StatusFilter.Published)) {
|
||
statusQuery.push({
|
||
[Op.and]: [
|
||
{
|
||
publishedAt: {
|
||
[Op.ne]: null,
|
||
},
|
||
archivedAt: {
|
||
[Op.eq]: null,
|
||
},
|
||
},
|
||
],
|
||
});
|
||
}
|
||
|
||
if (statusFilter?.includes(StatusFilter.Draft)) {
|
||
statusQuery.push({
|
||
[Op.and]: [
|
||
{
|
||
publishedAt: {
|
||
[Op.eq]: null,
|
||
},
|
||
archivedAt: {
|
||
[Op.eq]: null,
|
||
},
|
||
[Op.or]: [
|
||
// Only ever include draft results for the user's own documents
|
||
{ createdById: user.id },
|
||
{ "$memberships.id$": { [Op.ne]: null } },
|
||
],
|
||
},
|
||
],
|
||
});
|
||
}
|
||
|
||
if (statusFilter?.includes(StatusFilter.Archived)) {
|
||
statusQuery.push({
|
||
archivedAt: {
|
||
[Op.ne]: null,
|
||
},
|
||
});
|
||
}
|
||
|
||
if (statusQuery.length) {
|
||
where[Op.and].push({
|
||
[Op.or]: statusQuery,
|
||
});
|
||
}
|
||
|
||
const [documents, total] = await Promise.all([
|
||
Document.withMembershipScope(user.id).findAll({
|
||
where,
|
||
order: [
|
||
[
|
||
// this needs to be done otherwise findAll will throw citing
|
||
// that the column "document"."index" doesn't exist – value of sort
|
||
// is required to be a column name
|
||
sort === "index" ? "updatedAt" : sort,
|
||
direction,
|
||
],
|
||
],
|
||
offset: ctx.state.pagination.offset,
|
||
limit: ctx.state.pagination.limit,
|
||
}),
|
||
Document.count({ where }),
|
||
]);
|
||
|
||
if (sort === "index") {
|
||
// sort again so as to retain the order of documents as in collection.documentStructure
|
||
documents.sort(
|
||
(a, b) => documentIds.indexOf(a.id) - documentIds.indexOf(b.id)
|
||
);
|
||
}
|
||
|
||
const data = await Promise.all(
|
||
documents.map((document) => presentDocument(ctx, document))
|
||
);
|
||
const policies = presentPolicies(user, documents);
|
||
|
||
ctx.body = {
|
||
pagination: { ...ctx.state.pagination, total },
|
||
data,
|
||
policies,
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.archived",
|
||
auth({ role: UserRole.Member }),
|
||
pagination(),
|
||
validate(T.DocumentsArchivedSchema),
|
||
async (ctx: APIContext<T.DocumentsArchivedReq>) => {
|
||
const { sort, direction, collectionId } = ctx.input.body;
|
||
|
||
const { user } = ctx.state.auth;
|
||
|
||
let where: WhereOptions<Document> = {
|
||
teamId: user.teamId,
|
||
archivedAt: {
|
||
[Op.ne]: null,
|
||
},
|
||
};
|
||
|
||
let documentIds: string[] = [];
|
||
|
||
// if a specific collection is passed then we need to check auth to view it
|
||
if (collectionId) {
|
||
where = { ...where, collectionId };
|
||
const collection = await Collection.scope({
|
||
method: ["withMembership", user.id],
|
||
}).findByPk(collectionId);
|
||
authorize(user, "readDocument", collection);
|
||
|
||
// index sort is special because it uses the order of the documents in the
|
||
// collection.documentStructure rather than a database column
|
||
if (sort === "index") {
|
||
documentIds = (collection?.documentStructure || [])
|
||
.map((node) => node.id)
|
||
.slice(ctx.state.pagination.offset, ctx.state.pagination.limit);
|
||
where = { ...where, id: documentIds };
|
||
} // otherwise, filter by all collections the user has access to
|
||
} else {
|
||
const collectionIds = await user.collectionIds();
|
||
where = {
|
||
...where,
|
||
collectionId: collectionIds,
|
||
};
|
||
}
|
||
|
||
const documents = await Document.withMembershipScope(user.id).findAll({
|
||
where,
|
||
order: [
|
||
[
|
||
// this needs to be done otherwise findAll will throw citing
|
||
// that the column "document"."index" doesn't exist – value of sort
|
||
// is required to be a column name
|
||
sort === "index" ? "updatedAt" : sort,
|
||
direction,
|
||
],
|
||
],
|
||
offset: ctx.state.pagination.offset,
|
||
limit: ctx.state.pagination.limit,
|
||
});
|
||
|
||
if (sort === "index") {
|
||
// sort again so as to retain the order of documents as in collection.documentStructure
|
||
documents.sort(
|
||
(a, b) => documentIds.indexOf(a.id) - documentIds.indexOf(b.id)
|
||
);
|
||
}
|
||
|
||
const data = await Promise.all(
|
||
documents.map((document) => presentDocument(ctx, document))
|
||
);
|
||
const policies = presentPolicies(user, documents);
|
||
|
||
ctx.body = {
|
||
pagination: ctx.state.pagination,
|
||
data,
|
||
policies,
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.deleted",
|
||
auth({ role: UserRole.Member }),
|
||
pagination(),
|
||
validate(T.DocumentsDeletedSchema),
|
||
async (ctx: APIContext<T.DocumentsDeletedReq>) => {
|
||
const { sort, direction } = ctx.input.body;
|
||
const { user } = ctx.state.auth;
|
||
const collectionIds = await user.collectionIds({
|
||
paranoid: false,
|
||
});
|
||
const membershipScope: Readonly<ScopeOptions> = {
|
||
method: ["withMembership", user.id],
|
||
};
|
||
const viewScope: Readonly<ScopeOptions> = {
|
||
method: ["withViews", user.id],
|
||
};
|
||
const documents = await Document.scope([
|
||
membershipScope,
|
||
viewScope,
|
||
"withDrafts",
|
||
]).findAll({
|
||
where: {
|
||
teamId: user.teamId,
|
||
deletedAt: {
|
||
[Op.ne]: null,
|
||
},
|
||
[Op.or]: [
|
||
{
|
||
collectionId: {
|
||
[Op.in]: collectionIds,
|
||
},
|
||
},
|
||
{
|
||
createdById: user.id,
|
||
collectionId: {
|
||
[Op.is]: null,
|
||
},
|
||
},
|
||
],
|
||
},
|
||
paranoid: false,
|
||
order: [[sort, direction]],
|
||
offset: ctx.state.pagination.offset,
|
||
limit: ctx.state.pagination.limit,
|
||
});
|
||
const data = await Promise.all(
|
||
documents.map((document) => presentDocument(ctx, document))
|
||
);
|
||
const policies = presentPolicies(user, documents);
|
||
|
||
ctx.body = {
|
||
pagination: ctx.state.pagination,
|
||
data,
|
||
policies,
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.viewed",
|
||
auth(),
|
||
pagination(),
|
||
validate(T.DocumentsViewedSchema),
|
||
async (ctx: APIContext<T.DocumentsViewedReq>) => {
|
||
const { sort, direction } = ctx.input.body;
|
||
const { user } = ctx.state.auth;
|
||
const collectionIds = await user.collectionIds();
|
||
const userId = user.id;
|
||
const views = await View.findAll({
|
||
where: {
|
||
userId,
|
||
},
|
||
order: [[sort, direction]],
|
||
include: [
|
||
{
|
||
model: Document.scope([
|
||
"withDrafts",
|
||
{ method: ["withMembership", userId] },
|
||
]),
|
||
required: true,
|
||
where: {
|
||
collectionId: collectionIds,
|
||
},
|
||
include: [
|
||
{
|
||
model: Collection.scope({
|
||
method: ["withMembership", userId],
|
||
}),
|
||
as: "collection",
|
||
},
|
||
],
|
||
},
|
||
],
|
||
offset: ctx.state.pagination.offset,
|
||
limit: ctx.state.pagination.limit,
|
||
});
|
||
const documents = views.map((view) => {
|
||
const document = view.document;
|
||
document.views = [view];
|
||
return document;
|
||
});
|
||
const data = await Promise.all(
|
||
documents.map((document) => presentDocument(ctx, document))
|
||
);
|
||
const policies = presentPolicies(user, documents);
|
||
|
||
ctx.body = {
|
||
pagination: ctx.state.pagination,
|
||
data,
|
||
policies,
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.drafts",
|
||
auth(),
|
||
pagination(),
|
||
validate(T.DocumentsDraftsSchema),
|
||
async (ctx: APIContext<T.DocumentsDraftsReq>) => {
|
||
const { collectionId, dateFilter, direction, sort } = ctx.input.body;
|
||
const { user } = ctx.state.auth;
|
||
|
||
if (collectionId) {
|
||
const collection = await Collection.scope({
|
||
method: ["withMembership", user.id],
|
||
}).findByPk(collectionId);
|
||
authorize(user, "readDocument", collection);
|
||
}
|
||
|
||
const collectionIds = collectionId
|
||
? [collectionId]
|
||
: await user.collectionIds();
|
||
const where: WhereOptions = {
|
||
createdById: user.id,
|
||
collectionId: {
|
||
[Op.or]: [{ [Op.in]: collectionIds }, { [Op.is]: null }],
|
||
},
|
||
publishedAt: {
|
||
[Op.is]: null,
|
||
},
|
||
};
|
||
|
||
if (dateFilter) {
|
||
where.updatedAt = {
|
||
[Op.gte]: subtractDate(new Date(), dateFilter),
|
||
};
|
||
} else {
|
||
delete where.updatedAt;
|
||
}
|
||
|
||
const documents = await Document.withMembershipScope(user.id, {
|
||
includeDrafts: true,
|
||
}).findAll({
|
||
where,
|
||
order: [[sort, direction]],
|
||
offset: ctx.state.pagination.offset,
|
||
limit: ctx.state.pagination.limit,
|
||
});
|
||
const data = await Promise.all(
|
||
documents.map((document) => presentDocument(ctx, document))
|
||
);
|
||
const policies = presentPolicies(user, documents);
|
||
|
||
ctx.body = {
|
||
pagination: ctx.state.pagination,
|
||
data,
|
||
policies,
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.info",
|
||
auth({ optional: true }),
|
||
validate(T.DocumentsInfoSchema),
|
||
async (ctx: APIContext<T.DocumentsInfoReq>) => {
|
||
const { id, shareId } = ctx.input.body;
|
||
const { user } = ctx.state.auth;
|
||
const apiVersion = getAPIVersion(ctx);
|
||
const teamFromCtx = await getTeamFromContext(ctx);
|
||
const { document, share, collection } = await documentLoader({
|
||
id,
|
||
shareId,
|
||
user,
|
||
teamId: teamFromCtx?.id,
|
||
});
|
||
const isPublic = cannot(user, "read", document);
|
||
const [serializedDocument, team] = await Promise.all([
|
||
presentDocument(ctx, document, {
|
||
isPublic,
|
||
shareId,
|
||
}),
|
||
teamFromCtx?.id === document.teamId ? teamFromCtx : document.$get("team"),
|
||
]);
|
||
|
||
// Passing apiVersion=2 has a single effect, to change the response payload to
|
||
// include top level keys for document, sharedTree, and team.
|
||
const data =
|
||
apiVersion >= 2
|
||
? {
|
||
document: serializedDocument,
|
||
team: team
|
||
? presentPublicTeam(
|
||
team,
|
||
!!team?.getPreference(TeamPreference.PublicBranding)
|
||
)
|
||
: undefined,
|
||
sharedTree:
|
||
share && share.includeChildDocuments
|
||
? collection?.getDocumentTree(share.documentId)
|
||
: null,
|
||
}
|
||
: serializedDocument;
|
||
ctx.body = {
|
||
data,
|
||
policies: isPublic ? undefined : presentPolicies(user, [document]),
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.users",
|
||
auth(),
|
||
pagination(),
|
||
validate(T.DocumentsUsersSchema),
|
||
async (ctx: APIContext<T.DocumentsUsersReq>) => {
|
||
const { id, userId, query } = ctx.input.body;
|
||
const actor = ctx.state.auth.user;
|
||
const { offset, limit } = ctx.state.pagination;
|
||
const document = await Document.findByPk(id, {
|
||
userId: actor.id,
|
||
});
|
||
authorize(actor, "read", document);
|
||
|
||
let where: WhereOptions<User> = {
|
||
teamId: document.teamId,
|
||
suspendedAt: {
|
||
[Op.is]: null,
|
||
},
|
||
};
|
||
|
||
const [collection, memberIds, collectionMemberIds] = await Promise.all([
|
||
document.$get("collection"),
|
||
Document.membershipUserIds(document.id),
|
||
document.collectionId
|
||
? Collection.membershipUserIds(document.collectionId)
|
||
: [],
|
||
]);
|
||
|
||
where = {
|
||
...where,
|
||
[Op.or]: [
|
||
{
|
||
id: {
|
||
[Op.in]: uniq([...memberIds, ...collectionMemberIds]),
|
||
},
|
||
},
|
||
collection?.permission
|
||
? {
|
||
role: {
|
||
[Op.ne]: UserRole.Guest,
|
||
},
|
||
}
|
||
: {},
|
||
],
|
||
};
|
||
|
||
if (query) {
|
||
where = {
|
||
...where,
|
||
[Op.and]: [
|
||
Sequelize.literal(
|
||
`unaccent(LOWER(name)) like unaccent(LOWER(:query))`
|
||
),
|
||
],
|
||
};
|
||
}
|
||
|
||
if (userId) {
|
||
where = {
|
||
...where,
|
||
id: userId,
|
||
};
|
||
}
|
||
|
||
const replacements = { query: `%${query}%` };
|
||
|
||
const [users, total] = await Promise.all([
|
||
User.findAll({ where, replacements, offset, limit }),
|
||
User.count({
|
||
where,
|
||
// @ts-expect-error Types are incorrect for count
|
||
replacements,
|
||
}),
|
||
]);
|
||
|
||
ctx.body = {
|
||
pagination: { ...ctx.state.pagination, total },
|
||
data: users.map((user) => presentUser(user)),
|
||
policies: presentPolicies(actor, users),
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.export",
|
||
rateLimiter(RateLimiterStrategy.TwentyFivePerMinute),
|
||
auth({ optional: true }),
|
||
validate(T.DocumentsExportSchema),
|
||
async (ctx: APIContext<T.DocumentsExportReq>) => {
|
||
const { id } = ctx.input.body;
|
||
const { user } = ctx.state.auth;
|
||
const accept = ctx.request.headers["accept"];
|
||
|
||
const { document } = await documentLoader({
|
||
id,
|
||
user,
|
||
// We need the collaborative state to generate HTML.
|
||
includeState: !accept?.includes("text/markdown"),
|
||
});
|
||
|
||
let contentType: string;
|
||
let content: string;
|
||
|
||
if (accept?.includes("text/html")) {
|
||
contentType = "text/html";
|
||
content = await DocumentHelper.toHTML(document, {
|
||
centered: true,
|
||
includeMermaid: true,
|
||
});
|
||
} else if (accept?.includes("application/pdf")) {
|
||
throw IncorrectEditionError(
|
||
"PDF export is not available in the community edition"
|
||
);
|
||
} else if (accept?.includes("text/markdown")) {
|
||
contentType = "text/markdown";
|
||
content = DocumentHelper.toMarkdown(document);
|
||
} else {
|
||
ctx.body = {
|
||
data: DocumentHelper.toMarkdown(document),
|
||
};
|
||
return;
|
||
}
|
||
|
||
// Override the extension for Markdown as it's incorrect in the mime-types
|
||
// library until a new release > 2.1.35
|
||
const extension =
|
||
contentType === "text/markdown" ? "md" : mime.extension(contentType);
|
||
|
||
const fileName = slugify(document.titleWithDefault);
|
||
const attachmentIds = ProsemirrorHelper.parseAttachmentIds(
|
||
DocumentHelper.toProsemirror(document)
|
||
);
|
||
const attachments = attachmentIds.length
|
||
? await Attachment.findAll({
|
||
where: {
|
||
teamId: document.teamId,
|
||
id: attachmentIds,
|
||
},
|
||
})
|
||
: [];
|
||
|
||
if (attachments.length === 0) {
|
||
ctx.set("Content-Type", contentType);
|
||
ctx.attachment(`${fileName}.${extension}`);
|
||
ctx.body = content;
|
||
return;
|
||
}
|
||
|
||
const zip = new JSZip();
|
||
|
||
await Promise.all(
|
||
attachments.map(async (attachment) => {
|
||
const location = path.join(
|
||
"attachments",
|
||
`${attachment.id}.${mime.extension(attachment.contentType)}`
|
||
);
|
||
zip.file(
|
||
location,
|
||
new Promise<Buffer>((resolve) => {
|
||
attachment.buffer.then(resolve).catch((err) => {
|
||
Logger.warn(`Failed to read attachment from storage`, {
|
||
attachmentId: attachment.id,
|
||
teamId: attachment.teamId,
|
||
error: err.message,
|
||
});
|
||
resolve(Buffer.from(""));
|
||
});
|
||
}),
|
||
{
|
||
date: attachment.updatedAt,
|
||
createFolders: true,
|
||
}
|
||
);
|
||
|
||
content = content.replace(
|
||
new RegExp(escapeRegExp(attachment.redirectUrl), "g"),
|
||
location
|
||
);
|
||
})
|
||
);
|
||
|
||
zip.file(`${fileName}.${extension}`, content, {
|
||
date: document.updatedAt,
|
||
});
|
||
|
||
ctx.set("Content-Type", "application/zip");
|
||
ctx.attachment(`${fileName}.zip`);
|
||
ctx.body = zip.generateNodeStream(ZipHelper.defaultStreamOptions);
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.restore",
|
||
auth({ role: UserRole.Member }),
|
||
validate(T.DocumentsRestoreSchema),
|
||
transaction(),
|
||
async (ctx: APIContext<T.DocumentsRestoreReq>) => {
|
||
const { id, collectionId, revisionId } = ctx.input.body;
|
||
const { user } = ctx.state.auth;
|
||
const { transaction } = ctx.state;
|
||
const document = await Document.findByPk(id, {
|
||
userId: user.id,
|
||
paranoid: false,
|
||
rejectOnEmpty: true,
|
||
transaction,
|
||
});
|
||
|
||
const sourceCollectionId = document.collectionId;
|
||
const destCollectionId = collectionId ?? sourceCollectionId;
|
||
|
||
const srcCollection = sourceCollectionId
|
||
? await Collection.scope({
|
||
method: ["withMembership", user.id],
|
||
}).findByPk(sourceCollectionId, { paranoid: false })
|
||
: undefined;
|
||
|
||
const destCollection = destCollectionId
|
||
? await Collection.scope({
|
||
method: ["withMembership", user.id],
|
||
}).findByPk(destCollectionId)
|
||
: undefined;
|
||
|
||
// In case of workspace templates, both source and destination collections are undefined.
|
||
if (!document.isWorkspaceTemplate && !destCollection?.isActive) {
|
||
throw ValidationError(
|
||
"Unable to restore, the collection may have been deleted or archived"
|
||
);
|
||
}
|
||
|
||
// Skip this for workspace templates and drafts of a deleted collection as they won't have sourceCollectionId.
|
||
if (sourceCollectionId && sourceCollectionId !== destCollectionId) {
|
||
authorize(user, "updateDocument", srcCollection);
|
||
await srcCollection?.removeDocumentInStructure(document, {
|
||
save: true,
|
||
transaction,
|
||
});
|
||
}
|
||
|
||
if (document.deletedAt && document.isWorkspaceTemplate) {
|
||
authorize(user, "restore", document);
|
||
|
||
await document.restore({ transaction });
|
||
await Event.createFromContext(ctx, {
|
||
name: "documents.restore",
|
||
documentId: document.id,
|
||
collectionId: document.collectionId,
|
||
data: {
|
||
title: document.title,
|
||
},
|
||
});
|
||
} else if (document.deletedAt) {
|
||
authorize(user, "restore", document);
|
||
authorize(user, "updateDocument", destCollection);
|
||
|
||
// restore a previously deleted document
|
||
await document.restoreTo(destCollectionId!, { transaction, user }); // destCollectionId is guaranteed to be defined here
|
||
await Event.createFromContext(ctx, {
|
||
name: "documents.restore",
|
||
documentId: document.id,
|
||
collectionId: document.collectionId,
|
||
data: {
|
||
title: document.title,
|
||
},
|
||
});
|
||
} else if (document.archivedAt) {
|
||
authorize(user, "unarchive", document);
|
||
authorize(user, "updateDocument", destCollection);
|
||
|
||
// restore a previously archived document
|
||
await document.restoreTo(destCollectionId!, { transaction, user }); // destCollectionId is guaranteed to be defined here
|
||
await Event.createFromContext(ctx, {
|
||
name: "documents.unarchive",
|
||
documentId: document.id,
|
||
collectionId: document.collectionId,
|
||
data: {
|
||
title: document.title,
|
||
sourceCollectionId,
|
||
},
|
||
});
|
||
} else if (revisionId) {
|
||
// restore a document to a specific revision
|
||
authorize(user, "update", document);
|
||
const revision = await Revision.findByPk(revisionId, { transaction });
|
||
authorize(document, "restore", revision);
|
||
|
||
document.restoreFromRevision(revision);
|
||
await document.save({ transaction });
|
||
|
||
await Event.createFromContext(ctx, {
|
||
name: "documents.restore",
|
||
documentId: document.id,
|
||
collectionId: document.collectionId,
|
||
data: {
|
||
title: document.title,
|
||
},
|
||
});
|
||
} else {
|
||
assertPresent(revisionId, "revisionId is required");
|
||
}
|
||
|
||
ctx.body = {
|
||
data: await presentDocument(ctx, document),
|
||
policies: presentPolicies(user, [document]),
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.search_titles",
|
||
auth(),
|
||
pagination(),
|
||
rateLimiter(RateLimiterStrategy.OneHundredPerMinute),
|
||
validate(T.DocumentsSearchTitlesSchema),
|
||
async (ctx: APIContext<T.DocumentsSearchTitlesReq>) => {
|
||
const { query, statusFilter, dateFilter, collectionId, userId } =
|
||
ctx.input.body;
|
||
const { offset, limit } = ctx.state.pagination;
|
||
const { user } = ctx.state.auth;
|
||
let collaboratorIds = undefined;
|
||
|
||
if (collectionId) {
|
||
const collection = await Collection.scope({
|
||
method: ["withMembership", user.id],
|
||
}).findByPk(collectionId);
|
||
authorize(user, "readDocument", collection);
|
||
}
|
||
|
||
if (userId) {
|
||
collaboratorIds = [userId];
|
||
}
|
||
|
||
const documents = await SearchHelper.searchTitlesForUser(user, {
|
||
query,
|
||
dateFilter,
|
||
statusFilter,
|
||
collectionId,
|
||
collaboratorIds,
|
||
offset,
|
||
limit,
|
||
});
|
||
const policies = presentPolicies(user, documents);
|
||
const data = await Promise.all(
|
||
documents.map((document) => presentDocument(ctx, document))
|
||
);
|
||
|
||
ctx.body = {
|
||
pagination: ctx.state.pagination,
|
||
data,
|
||
policies,
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.search",
|
||
auth({ optional: true }),
|
||
pagination(),
|
||
rateLimiter(RateLimiterStrategy.OneHundredPerMinute),
|
||
validate(T.DocumentsSearchSchema),
|
||
async (ctx: APIContext<T.DocumentsSearchReq>) => {
|
||
const {
|
||
query,
|
||
collectionId,
|
||
documentId,
|
||
userId,
|
||
dateFilter,
|
||
statusFilter = [],
|
||
shareId,
|
||
snippetMinWords,
|
||
snippetMaxWords,
|
||
} = ctx.input.body;
|
||
const { offset, limit } = ctx.state.pagination;
|
||
const { user } = ctx.state.auth;
|
||
|
||
let teamId;
|
||
let response;
|
||
let share;
|
||
let isPublic = false;
|
||
|
||
if (shareId) {
|
||
const teamFromCtx = await getTeamFromContext(ctx);
|
||
const { document, ...loaded } = await documentLoader({
|
||
teamId: teamFromCtx?.id,
|
||
shareId,
|
||
user,
|
||
});
|
||
|
||
share = loaded.share;
|
||
isPublic = cannot(user, "read", document);
|
||
|
||
if (!share?.includeChildDocuments) {
|
||
throw InvalidRequestError("Child documents cannot be searched");
|
||
}
|
||
|
||
teamId = share.teamId;
|
||
const team = await share.$get("team");
|
||
invariant(team, "Share must belong to a team");
|
||
|
||
response = await SearchHelper.searchForTeam(team, {
|
||
query,
|
||
collectionId: document.collectionId,
|
||
share,
|
||
dateFilter,
|
||
statusFilter,
|
||
offset,
|
||
limit,
|
||
snippetMinWords,
|
||
snippetMaxWords,
|
||
});
|
||
} else {
|
||
if (!user) {
|
||
throw AuthenticationError("Authentication error");
|
||
}
|
||
|
||
teamId = user.teamId;
|
||
|
||
if (collectionId) {
|
||
const collection = await Collection.scope({
|
||
method: ["withMembership", user.id],
|
||
}).findByPk(collectionId);
|
||
authorize(user, "readDocument", collection);
|
||
}
|
||
|
||
let documentIds = undefined;
|
||
if (documentId) {
|
||
const document = await Document.findByPk(documentId, {
|
||
userId: user.id,
|
||
});
|
||
authorize(user, "read", document);
|
||
documentIds = [
|
||
documentId,
|
||
...(await document.findAllChildDocumentIds()),
|
||
];
|
||
}
|
||
|
||
let collaboratorIds = undefined;
|
||
|
||
if (userId) {
|
||
collaboratorIds = [userId];
|
||
}
|
||
|
||
response = await SearchHelper.searchForUser(user, {
|
||
query,
|
||
collaboratorIds,
|
||
collectionId,
|
||
documentIds,
|
||
dateFilter,
|
||
statusFilter,
|
||
offset,
|
||
limit,
|
||
snippetMinWords,
|
||
snippetMaxWords,
|
||
});
|
||
}
|
||
|
||
const { results, total } = response;
|
||
const documents = results.map((result) => result.document);
|
||
|
||
const data = await Promise.all(
|
||
results.map(async (result) => {
|
||
const document = await presentDocument(ctx, result.document, {
|
||
isPublic,
|
||
shareId,
|
||
});
|
||
return { ...result, document };
|
||
})
|
||
);
|
||
|
||
// When requesting subsequent pages of search results we don't want to record
|
||
// duplicate search query records
|
||
if (query && offset === 0) {
|
||
await SearchQuery.create({
|
||
userId: user?.id,
|
||
teamId,
|
||
shareId: share?.id,
|
||
source: ctx.state.auth.type || "app", // we'll consider anything that isn't "api" to be "app"
|
||
query,
|
||
results: total,
|
||
});
|
||
}
|
||
|
||
ctx.body = {
|
||
pagination: { ...ctx.state.pagination, total },
|
||
data,
|
||
policies: user ? presentPolicies(user, documents) : null,
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.templatize",
|
||
auth({ role: UserRole.Member }),
|
||
rateLimiter(RateLimiterStrategy.TwentyFivePerMinute),
|
||
validate(T.DocumentsTemplatizeSchema),
|
||
transaction(),
|
||
async (ctx: APIContext<T.DocumentsTemplatizeReq>) => {
|
||
const { id, collectionId, publish } = ctx.input.body;
|
||
const { user } = ctx.state.auth;
|
||
const { transaction } = ctx.state;
|
||
|
||
const original = await Document.findByPk(id, {
|
||
userId: user.id,
|
||
transaction,
|
||
});
|
||
|
||
authorize(user, "update", original);
|
||
|
||
if (collectionId) {
|
||
const collection = await Collection.scope({
|
||
method: ["withMembership", user.id],
|
||
}).findByPk(collectionId, { transaction });
|
||
authorize(user, "createDocument", collection);
|
||
} else {
|
||
authorize(user, "createTemplate", user.team);
|
||
}
|
||
|
||
const document = await Document.create(
|
||
{
|
||
editorVersion: original.editorVersion,
|
||
collectionId,
|
||
teamId: user.teamId,
|
||
publishedAt: publish ? new Date() : null,
|
||
lastModifiedById: user.id,
|
||
createdById: user.id,
|
||
template: true,
|
||
icon: original.icon,
|
||
color: original.color,
|
||
title: original.title,
|
||
text: original.text,
|
||
content: original.content,
|
||
},
|
||
{
|
||
transaction,
|
||
}
|
||
);
|
||
await Event.createFromContext(ctx, {
|
||
name: "documents.create",
|
||
documentId: document.id,
|
||
collectionId: document.collectionId,
|
||
data: {
|
||
title: document.title,
|
||
template: true,
|
||
},
|
||
});
|
||
|
||
// reload to get all of the data needed to present (user, collection etc)
|
||
const reloaded = await Document.findByPk(document.id, {
|
||
userId: user.id,
|
||
transaction,
|
||
});
|
||
invariant(reloaded, "document not found");
|
||
|
||
ctx.body = {
|
||
data: await presentDocument(ctx, reloaded),
|
||
policies: presentPolicies(user, [reloaded]),
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.update",
|
||
auth(),
|
||
validate(T.DocumentsUpdateSchema),
|
||
transaction(),
|
||
async (ctx: APIContext<T.DocumentsUpdateReq>) => {
|
||
const { transaction } = ctx.state;
|
||
const { id, insightsEnabled, publish, collectionId, ...input } =
|
||
ctx.input.body;
|
||
const editorVersion = ctx.headers["x-editor-version"] as string | undefined;
|
||
|
||
const { user } = ctx.state.auth;
|
||
let collection: Collection | null | undefined;
|
||
|
||
let document = await Document.findByPk(id, {
|
||
userId: user.id,
|
||
includeState: true,
|
||
transaction,
|
||
});
|
||
collection = document?.collection;
|
||
authorize(user, "update", document);
|
||
|
||
if (collection && insightsEnabled !== undefined) {
|
||
authorize(user, "updateInsights", document);
|
||
}
|
||
|
||
if (publish) {
|
||
if (document.isDraft) {
|
||
authorize(user, "publish", document);
|
||
}
|
||
|
||
if (!document.collectionId && !document.isWorkspaceTemplate) {
|
||
assertPresent(
|
||
collectionId,
|
||
"collectionId is required to publish a draft without collection"
|
||
);
|
||
collection = await Collection.scope({
|
||
method: ["withMembership", user.id],
|
||
}).findByPk(collectionId!, { transaction });
|
||
}
|
||
|
||
if (document.parentDocumentId) {
|
||
const parentDocument = await Document.findByPk(
|
||
document.parentDocumentId,
|
||
{
|
||
userId: user.id,
|
||
transaction,
|
||
}
|
||
);
|
||
authorize(user, "createChildDocument", parentDocument, { collection });
|
||
} else if (document.isWorkspaceTemplate) {
|
||
authorize(user, "createTemplate", user.team);
|
||
} else {
|
||
authorize(user, "createDocument", collection);
|
||
}
|
||
}
|
||
|
||
document = await documentUpdater(ctx, {
|
||
document,
|
||
user,
|
||
...input,
|
||
publish,
|
||
collectionId,
|
||
insightsEnabled,
|
||
editorVersion,
|
||
});
|
||
|
||
ctx.body = {
|
||
data: await presentDocument(ctx, document),
|
||
policies: presentPolicies(user, [document]),
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.duplicate",
|
||
auth(),
|
||
validate(T.DocumentsDuplicateSchema),
|
||
transaction(),
|
||
async (ctx: APIContext<T.DocumentsDuplicateReq>) => {
|
||
const { transaction } = ctx.state;
|
||
const { id, title, publish, recursive, collectionId, parentDocumentId } =
|
||
ctx.input.body;
|
||
const { user } = ctx.state.auth;
|
||
|
||
const document = await Document.findByPk(id, {
|
||
userId: user.id,
|
||
transaction,
|
||
});
|
||
authorize(user, "read", document);
|
||
|
||
const collection = collectionId
|
||
? await Collection.scope({
|
||
method: ["withMembership", user.id],
|
||
}).findByPk(collectionId, { transaction })
|
||
: document?.collection;
|
||
|
||
if (collection) {
|
||
authorize(user, "updateDocument", collection);
|
||
} else if (document.isWorkspaceTemplate) {
|
||
authorize(user, "createTemplate", user.team);
|
||
}
|
||
|
||
if (parentDocumentId) {
|
||
const parent = await Document.findByPk(parentDocumentId, {
|
||
userId: user.id,
|
||
transaction,
|
||
});
|
||
authorize(user, "update", parent);
|
||
|
||
if (!parent.publishedAt) {
|
||
throw InvalidRequestError("Cannot duplicate document inside a draft");
|
||
}
|
||
}
|
||
|
||
const response = await documentDuplicator({
|
||
user,
|
||
collection,
|
||
document,
|
||
title,
|
||
publish,
|
||
recursive,
|
||
parentDocumentId,
|
||
ctx,
|
||
});
|
||
|
||
ctx.body = {
|
||
data: {
|
||
documents: await Promise.all(
|
||
response.map((document) => presentDocument(ctx, document))
|
||
),
|
||
},
|
||
policies: presentPolicies(user, response),
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.move",
|
||
auth(),
|
||
validate(T.DocumentsMoveSchema),
|
||
transaction(),
|
||
async (ctx: APIContext<T.DocumentsMoveReq>) => {
|
||
const { transaction } = ctx.state;
|
||
const { id, parentDocumentId, index } = ctx.input.body;
|
||
let collectionId = ctx.input.body.collectionId;
|
||
const { user } = ctx.state.auth;
|
||
const document = await Document.findByPk(id, {
|
||
userId: user.id,
|
||
transaction,
|
||
});
|
||
authorize(user, "move", document);
|
||
|
||
if (collectionId) {
|
||
const collection = await Collection.scope({
|
||
method: ["withMembership", user.id],
|
||
}).findByPk(collectionId, { transaction });
|
||
authorize(user, "updateDocument", collection);
|
||
} else if (document.template) {
|
||
authorize(user, "updateTemplate", user.team);
|
||
} else if (!parentDocumentId) {
|
||
throw InvalidRequestError("collectionId is required to move a document");
|
||
}
|
||
|
||
if (parentDocumentId) {
|
||
const parent = await Document.findByPk(parentDocumentId, {
|
||
userId: user.id,
|
||
transaction,
|
||
});
|
||
authorize(user, "update", parent);
|
||
collectionId = parent.collectionId;
|
||
|
||
if (!parent.publishedAt) {
|
||
throw InvalidRequestError("Cannot move document inside a draft");
|
||
}
|
||
}
|
||
|
||
const { documents, collectionChanged } = await documentMover({
|
||
user,
|
||
document,
|
||
collectionId: collectionId ?? null,
|
||
parentDocumentId,
|
||
index,
|
||
ip: ctx.request.ip,
|
||
transaction,
|
||
});
|
||
|
||
ctx.body = {
|
||
data: {
|
||
documents: await Promise.all(
|
||
documents.map((doc) => presentDocument(ctx, doc))
|
||
),
|
||
// Included for backwards compatibility
|
||
collections: [],
|
||
},
|
||
policies: collectionChanged ? presentPolicies(user, documents) : [],
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.archive",
|
||
auth(),
|
||
validate(T.DocumentsArchiveSchema),
|
||
transaction(),
|
||
async (ctx: APIContext<T.DocumentsArchiveReq>) => {
|
||
const { id } = ctx.input.body;
|
||
const { user } = ctx.state.auth;
|
||
const { transaction } = ctx.state;
|
||
|
||
const document = await Document.findByPk(id, {
|
||
userId: user.id,
|
||
rejectOnEmpty: true,
|
||
transaction,
|
||
});
|
||
authorize(user, "archive", document);
|
||
|
||
await document.archive(user, { transaction });
|
||
await Event.createFromContext(ctx, {
|
||
name: "documents.archive",
|
||
documentId: document.id,
|
||
collectionId: document.collectionId,
|
||
data: {
|
||
title: document.title,
|
||
},
|
||
});
|
||
|
||
ctx.body = {
|
||
data: await presentDocument(ctx, document),
|
||
policies: presentPolicies(user, [document]),
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.delete",
|
||
auth(),
|
||
validate(T.DocumentsDeleteSchema),
|
||
async (ctx: APIContext<T.DocumentsDeleteReq>) => {
|
||
const { id, permanent } = ctx.input.body;
|
||
const { user } = ctx.state.auth;
|
||
|
||
if (permanent) {
|
||
const document = await Document.findByPk(id, {
|
||
userId: user.id,
|
||
paranoid: false,
|
||
});
|
||
authorize(user, "permanentDelete", document);
|
||
|
||
await documentPermanentDeleter([document]);
|
||
await Event.createFromContext(ctx, {
|
||
name: "documents.permanent_delete",
|
||
documentId: document.id,
|
||
collectionId: document.collectionId,
|
||
data: {
|
||
title: document.title,
|
||
},
|
||
});
|
||
} else {
|
||
const document = await Document.findByPk(id, {
|
||
userId: user.id,
|
||
});
|
||
|
||
authorize(user, "delete", document);
|
||
|
||
await document.delete(user);
|
||
await Event.createFromContext(ctx, {
|
||
name: "documents.delete",
|
||
documentId: document.id,
|
||
collectionId: document.collectionId,
|
||
data: {
|
||
title: document.title,
|
||
},
|
||
});
|
||
}
|
||
|
||
ctx.body = {
|
||
success: true,
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.unpublish",
|
||
auth(),
|
||
validate(T.DocumentsUnpublishSchema),
|
||
async (ctx: APIContext<T.DocumentsUnpublishReq>) => {
|
||
const { id, detach } = ctx.input.body;
|
||
const { user } = ctx.state.auth;
|
||
|
||
const document = await Document.findByPk(id, {
|
||
userId: user.id,
|
||
});
|
||
authorize(user, "unpublish", document);
|
||
|
||
const childDocumentIds = await document.findAllChildDocumentIds({
|
||
archivedAt: {
|
||
[Op.eq]: null,
|
||
},
|
||
});
|
||
if (childDocumentIds.length > 0) {
|
||
throw InvalidRequestError(
|
||
"Cannot unpublish document with child documents"
|
||
);
|
||
}
|
||
|
||
// detaching would unset collectionId from document, so save a ref to the affected collectionId.
|
||
const collectionId = document.collectionId;
|
||
|
||
await document.unpublish(user, { detach });
|
||
await Event.createFromContext(ctx, {
|
||
name: "documents.unpublish",
|
||
documentId: document.id,
|
||
collectionId,
|
||
});
|
||
|
||
ctx.body = {
|
||
data: await presentDocument(ctx, document),
|
||
policies: presentPolicies(user, [document]),
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.import",
|
||
auth(),
|
||
rateLimiter(RateLimiterStrategy.TwentyFivePerMinute),
|
||
validate(T.DocumentsImportSchema),
|
||
multipart({ maximumFileSize: env.FILE_STORAGE_IMPORT_MAX_SIZE }),
|
||
async (ctx: APIContext<T.DocumentsImportReq>) => {
|
||
const { collectionId, parentDocumentId, publish } = ctx.input.body;
|
||
const file = ctx.input.file;
|
||
const { user } = ctx.state.auth;
|
||
|
||
const collection = await Collection.scope({
|
||
method: ["withMembership", user.id],
|
||
}).findOne({
|
||
where: {
|
||
id: collectionId,
|
||
teamId: user.teamId,
|
||
},
|
||
});
|
||
authorize(user, "createDocument", collection);
|
||
let parentDocument;
|
||
|
||
if (parentDocumentId) {
|
||
parentDocument = await Document.findByPk(parentDocumentId, {
|
||
userId: user.id,
|
||
});
|
||
authorize(user, "read", parentDocument);
|
||
}
|
||
|
||
const buffer = await fs.readFile(file.filepath);
|
||
const fileName = file.originalFilename ?? file.newFilename;
|
||
const mimeType = file.mimetype ?? "";
|
||
const acl = "private";
|
||
|
||
const key = AttachmentHelper.getKey({
|
||
acl,
|
||
id: uuidv4(),
|
||
name: fileName,
|
||
userId: user.id,
|
||
});
|
||
|
||
await FileStorage.store({
|
||
body: buffer,
|
||
contentType: mimeType,
|
||
contentLength: buffer.length,
|
||
key,
|
||
acl,
|
||
});
|
||
|
||
const job = await new DocumentImportTask().schedule({
|
||
key,
|
||
sourceMetadata: {
|
||
fileName,
|
||
mimeType,
|
||
},
|
||
userId: user.id,
|
||
collectionId,
|
||
parentDocumentId,
|
||
publish,
|
||
ip: ctx.request.ip,
|
||
});
|
||
const response: DocumentImportTaskResponse = await job.finished();
|
||
if ("error" in response) {
|
||
throw InvalidRequestError(response.error);
|
||
}
|
||
|
||
const document = await Document.findByPk(response.documentId, {
|
||
userId: user.id,
|
||
rejectOnEmpty: true,
|
||
});
|
||
|
||
ctx.body = {
|
||
data: await presentDocument(ctx, document),
|
||
policies: presentPolicies(user, [document]),
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.create",
|
||
auth(),
|
||
rateLimiter(RateLimiterStrategy.TwentyFivePerMinute),
|
||
validate(T.DocumentsCreateSchema),
|
||
transaction(),
|
||
async (ctx: APIContext<T.DocumentsCreateReq>) => {
|
||
const {
|
||
id,
|
||
title,
|
||
text,
|
||
icon,
|
||
color,
|
||
publish,
|
||
collectionId,
|
||
parentDocumentId,
|
||
fullWidth,
|
||
templateId,
|
||
template,
|
||
createdAt,
|
||
} = ctx.input.body;
|
||
const editorVersion = ctx.headers["x-editor-version"] as string | undefined;
|
||
|
||
const { transaction } = ctx.state;
|
||
const { user } = ctx.state.auth;
|
||
|
||
let collection;
|
||
|
||
let parentDocument;
|
||
|
||
if (parentDocumentId) {
|
||
parentDocument = await Document.findByPk(parentDocumentId, {
|
||
userId: user.id,
|
||
});
|
||
|
||
if (parentDocument?.collectionId) {
|
||
collection = await Collection.scope({
|
||
method: ["withMembership", user.id],
|
||
}).findOne({
|
||
where: {
|
||
id: parentDocument.collectionId,
|
||
teamId: user.teamId,
|
||
},
|
||
transaction,
|
||
});
|
||
}
|
||
|
||
authorize(user, "createChildDocument", parentDocument, {
|
||
collection,
|
||
});
|
||
} else if (collectionId) {
|
||
collection = await Collection.scope({
|
||
method: ["withMembership", user.id],
|
||
}).findOne({
|
||
where: {
|
||
id: collectionId,
|
||
teamId: user.teamId,
|
||
},
|
||
transaction,
|
||
});
|
||
authorize(user, "createDocument", collection);
|
||
} else if (!!template && !collectionId) {
|
||
authorize(user, "createTemplate", user.team);
|
||
}
|
||
|
||
let templateDocument: Document | null | undefined;
|
||
|
||
if (templateId) {
|
||
templateDocument = await Document.findByPk(templateId, {
|
||
userId: user.id,
|
||
transaction,
|
||
});
|
||
authorize(user, "read", templateDocument);
|
||
}
|
||
|
||
const document = await documentCreator({
|
||
id,
|
||
title,
|
||
text: !isNil(text)
|
||
? await TextHelper.replaceImagesWithAttachments(ctx, text, user)
|
||
: text,
|
||
icon,
|
||
color,
|
||
createdAt,
|
||
publish,
|
||
collectionId: collection?.id,
|
||
parentDocumentId,
|
||
templateDocument,
|
||
template,
|
||
fullWidth,
|
||
user,
|
||
editorVersion,
|
||
ctx,
|
||
});
|
||
|
||
if (collection) {
|
||
document.collection = collection;
|
||
}
|
||
|
||
ctx.body = {
|
||
data: await presentDocument(ctx, document),
|
||
policies: presentPolicies(user, [document]),
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.add_user",
|
||
auth(),
|
||
validate(T.DocumentsAddUserSchema),
|
||
rateLimiter(RateLimiterStrategy.OneHundredPerHour),
|
||
transaction(),
|
||
async (ctx: APIContext<T.DocumentsAddUserReq>) => {
|
||
const { transaction } = ctx.state;
|
||
const { user: actor } = ctx.state.auth;
|
||
const { id, userId, permission } = ctx.input.body;
|
||
|
||
if (userId === actor.id) {
|
||
throw ValidationError("You cannot invite yourself");
|
||
}
|
||
|
||
const [document, user] = await Promise.all([
|
||
Document.findByPk(id, {
|
||
userId: actor.id,
|
||
rejectOnEmpty: true,
|
||
transaction,
|
||
}),
|
||
User.findByPk(userId, {
|
||
rejectOnEmpty: true,
|
||
transaction,
|
||
}),
|
||
]);
|
||
|
||
authorize(actor, "read", user);
|
||
authorize(actor, "manageUsers", document);
|
||
|
||
const UserMemberships = await UserMembership.findAll({
|
||
where: {
|
||
userId,
|
||
},
|
||
attributes: ["id", "index", "updatedAt"],
|
||
limit: 1,
|
||
order: [
|
||
// using LC_COLLATE:"C" because we need byte order to drive the sorting
|
||
// find only the first star so we can create an index before it
|
||
Sequelize.literal('"user_permission"."index" collate "C"'),
|
||
["updatedAt", "DESC"],
|
||
],
|
||
transaction,
|
||
});
|
||
|
||
// create membership at the beginning of their "Shared with me" section
|
||
const index = fractionalIndex(
|
||
null,
|
||
UserMemberships.length ? UserMemberships[0].index : null
|
||
);
|
||
|
||
const [membership, isNew] = await UserMembership.findOrCreate({
|
||
where: {
|
||
documentId: id,
|
||
userId,
|
||
},
|
||
defaults: {
|
||
index,
|
||
permission: permission || user.defaultDocumentPermission,
|
||
createdById: actor.id,
|
||
},
|
||
lock: transaction.LOCK.UPDATE,
|
||
...ctx.context,
|
||
});
|
||
|
||
if (!isNew && permission) {
|
||
membership.permission = permission;
|
||
|
||
// disconnect from the source if the permission is manually updated
|
||
membership.sourceId = null;
|
||
|
||
await membership.save(ctx.context);
|
||
}
|
||
|
||
ctx.body = {
|
||
data: {
|
||
users: [presentUser(user)],
|
||
memberships: [presentMembership(membership)],
|
||
},
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.remove_user",
|
||
auth(),
|
||
validate(T.DocumentsRemoveUserSchema),
|
||
transaction(),
|
||
async (ctx: APIContext<T.DocumentsRemoveUserReq>) => {
|
||
const { transaction } = ctx.state;
|
||
const { user: actor } = ctx.state.auth;
|
||
const { id, userId } = ctx.input.body;
|
||
|
||
const [document, user] = await Promise.all([
|
||
Document.findByPk(id, {
|
||
userId: actor.id,
|
||
rejectOnEmpty: true,
|
||
transaction,
|
||
}),
|
||
User.findByPk(userId, {
|
||
rejectOnEmpty: true,
|
||
transaction,
|
||
}),
|
||
]);
|
||
|
||
if (actor.id !== userId) {
|
||
authorize(actor, "manageUsers", document);
|
||
authorize(actor, "read", user);
|
||
}
|
||
|
||
const membership = await UserMembership.findOne({
|
||
where: {
|
||
documentId: id,
|
||
userId,
|
||
},
|
||
transaction,
|
||
lock: transaction.LOCK.UPDATE,
|
||
rejectOnEmpty: true,
|
||
});
|
||
|
||
await membership.destroy(ctx.context);
|
||
|
||
ctx.body = {
|
||
success: true,
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.add_group",
|
||
auth(),
|
||
validate(T.DocumentsAddGroupSchema),
|
||
transaction(),
|
||
async (ctx: APIContext<T.DocumentsAddGroupsReq>) => {
|
||
const { id, groupId, permission } = ctx.input.body;
|
||
const { transaction } = ctx.state;
|
||
const { user } = ctx.state.auth;
|
||
|
||
const [document, group] = await Promise.all([
|
||
Document.findByPk(id, {
|
||
userId: user.id,
|
||
rejectOnEmpty: true,
|
||
transaction,
|
||
}),
|
||
Group.findByPk(groupId, {
|
||
rejectOnEmpty: true,
|
||
transaction,
|
||
}),
|
||
]);
|
||
authorize(user, "update", document);
|
||
authorize(user, "read", group);
|
||
|
||
const [membership, created] = await GroupMembership.findOrCreate({
|
||
where: {
|
||
documentId: id,
|
||
groupId,
|
||
},
|
||
defaults: {
|
||
permission: permission || user.defaultDocumentPermission,
|
||
createdById: user.id,
|
||
},
|
||
lock: transaction.LOCK.UPDATE,
|
||
...ctx.context,
|
||
});
|
||
|
||
if (!created && permission) {
|
||
membership.permission = permission;
|
||
|
||
// disconnect from the source if the permission is manually updated
|
||
membership.sourceId = null;
|
||
|
||
await membership.save(ctx.context);
|
||
}
|
||
|
||
ctx.body = {
|
||
data: {
|
||
groupMemberships: [presentGroupMembership(membership)],
|
||
},
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.remove_group",
|
||
auth(),
|
||
validate(T.DocumentsRemoveGroupSchema),
|
||
transaction(),
|
||
async (ctx: APIContext<T.DocumentsRemoveGroupReq>) => {
|
||
const { transaction } = ctx.state;
|
||
const { user } = ctx.state.auth;
|
||
const { id, groupId } = ctx.input.body;
|
||
|
||
const [document, group] = await Promise.all([
|
||
Document.findByPk(id, {
|
||
userId: user.id,
|
||
rejectOnEmpty: true,
|
||
transaction,
|
||
}),
|
||
Group.findByPk(groupId, {
|
||
rejectOnEmpty: true,
|
||
transaction,
|
||
}),
|
||
]);
|
||
authorize(user, "update", document);
|
||
authorize(user, "read", group);
|
||
|
||
const membership = await GroupMembership.findOne({
|
||
where: {
|
||
documentId: id,
|
||
groupId,
|
||
},
|
||
transaction,
|
||
lock: transaction.LOCK.UPDATE,
|
||
rejectOnEmpty: true,
|
||
});
|
||
|
||
await membership.destroy(ctx.context);
|
||
|
||
ctx.body = {
|
||
success: true,
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.memberships",
|
||
auth(),
|
||
pagination(),
|
||
validate(T.DocumentsMembershipsSchema),
|
||
async (ctx: APIContext<T.DocumentsMembershipsReq>) => {
|
||
const { id, query, permission } = ctx.input.body;
|
||
const { user: actor } = ctx.state.auth;
|
||
|
||
const document = await Document.findByPk(id, { userId: actor.id });
|
||
authorize(actor, "update", document);
|
||
|
||
let where: WhereOptions<UserMembership> = {
|
||
documentId: id,
|
||
};
|
||
let userWhere;
|
||
|
||
if (query) {
|
||
userWhere = {
|
||
name: {
|
||
[Op.iLike]: `%${query}%`,
|
||
},
|
||
};
|
||
}
|
||
|
||
if (permission) {
|
||
where = { ...where, permission };
|
||
}
|
||
|
||
const options = {
|
||
where,
|
||
include: [
|
||
{
|
||
model: User,
|
||
as: "user",
|
||
where: userWhere,
|
||
required: true,
|
||
},
|
||
],
|
||
};
|
||
|
||
const [total, memberships] = await Promise.all([
|
||
UserMembership.count(options),
|
||
UserMembership.findAll({
|
||
...options,
|
||
order: [["createdAt", "DESC"]],
|
||
offset: ctx.state.pagination.offset,
|
||
limit: ctx.state.pagination.limit,
|
||
}),
|
||
]);
|
||
|
||
ctx.body = {
|
||
pagination: { ...ctx.state.pagination, total },
|
||
data: {
|
||
memberships: memberships.map(presentMembership),
|
||
users: memberships.map((membership) => presentUser(membership.user)),
|
||
},
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.group_memberships",
|
||
auth(),
|
||
pagination(),
|
||
validate(T.DocumentsMembershipsSchema),
|
||
async (ctx: APIContext<T.DocumentsMembershipsReq>) => {
|
||
const { id, query, permission } = ctx.input.body;
|
||
const { user } = ctx.state.auth;
|
||
|
||
const document = await Document.findByPk(id, { userId: user.id });
|
||
authorize(user, "update", document);
|
||
|
||
let where: WhereOptions<GroupMembership> = {
|
||
documentId: id,
|
||
};
|
||
let groupWhere;
|
||
|
||
if (query) {
|
||
groupWhere = {
|
||
name: {
|
||
[Op.iLike]: `%${query}%`,
|
||
},
|
||
};
|
||
}
|
||
|
||
if (permission) {
|
||
where = { ...where, permission };
|
||
}
|
||
|
||
const options = {
|
||
where,
|
||
include: [
|
||
{
|
||
model: Group,
|
||
as: "group",
|
||
where: groupWhere,
|
||
required: true,
|
||
},
|
||
],
|
||
};
|
||
|
||
const [total, memberships] = await Promise.all([
|
||
GroupMembership.count(options),
|
||
GroupMembership.findAll({
|
||
...options,
|
||
order: [["createdAt", "DESC"]],
|
||
offset: ctx.state.pagination.offset,
|
||
limit: ctx.state.pagination.limit,
|
||
}),
|
||
]);
|
||
|
||
const groupMemberships = memberships.map(presentGroupMembership);
|
||
|
||
ctx.body = {
|
||
pagination: { ...ctx.state.pagination, total },
|
||
data: {
|
||
groupMemberships,
|
||
groups: await Promise.all(
|
||
memberships.map((membership) => presentGroup(membership.group))
|
||
),
|
||
},
|
||
};
|
||
}
|
||
);
|
||
|
||
router.post(
|
||
"documents.empty_trash",
|
||
auth({ role: UserRole.Admin }),
|
||
async (ctx: APIContext) => {
|
||
const { user } = ctx.state.auth;
|
||
|
||
const collectionIds = await user.collectionIds({
|
||
paranoid: false,
|
||
});
|
||
const documents = await Document.scope("withDrafts").findAll({
|
||
attributes: ["id"],
|
||
where: {
|
||
deletedAt: {
|
||
[Op.ne]: null,
|
||
},
|
||
[Op.or]: [
|
||
{
|
||
collectionId: {
|
||
[Op.in]: collectionIds,
|
||
},
|
||
},
|
||
{
|
||
createdById: user.id,
|
||
collectionId: {
|
||
[Op.is]: null,
|
||
},
|
||
},
|
||
],
|
||
},
|
||
paranoid: false,
|
||
});
|
||
|
||
if (documents.length) {
|
||
await new EmptyTrashTask().schedule({
|
||
documentIds: documents.map((doc) => doc.id),
|
||
});
|
||
}
|
||
|
||
await Event.createFromContext(ctx, {
|
||
name: "documents.empty_trash",
|
||
});
|
||
|
||
ctx.body = {
|
||
success: true,
|
||
};
|
||
}
|
||
);
|
||
|
||
// Remove this helper once apiVersion is removed (#6175)
|
||
function getAPIVersion(ctx: APIContext) {
|
||
return Number(
|
||
ctx.headers["x-api-version"] ??
|
||
(typeof ctx.input.body === "object" &&
|
||
ctx.input.body &&
|
||
"apiVersion" in ctx.input.body &&
|
||
ctx.input.body.apiVersion) ??
|
||
0
|
||
);
|
||
}
|
||
|
||
export default router;
|