misc: addressed review comments

This commit is contained in:
Sheen Capadngan
2024-07-04 13:33:32 +08:00
parent f7626d03bf
commit cb664bb042
7 changed files with 77 additions and 119 deletions
@@ -5,14 +5,15 @@ import { WebhookType } from "@app/services/webhook/webhook-types";
import { TableName } from "../schemas"; import { TableName } from "../schemas";
export async function up(knex: Knex): Promise<void> { export async function up(knex: Knex): Promise<void> {
const hasEncryptedURL = await knex.schema.hasColumn(TableName.Webhook, "encryptedUrl"); const hasUrlCipherText = await knex.schema.hasColumn(TableName.Webhook, "urlCipherText");
const hasUrlIV = await knex.schema.hasColumn(TableName.Webhook, "urlIV"); const hasUrlIV = await knex.schema.hasColumn(TableName.Webhook, "urlIV");
const hasUrlTag = await knex.schema.hasColumn(TableName.Webhook, "urlTag"); const hasUrlTag = await knex.schema.hasColumn(TableName.Webhook, "urlTag");
const hasType = await knex.schema.hasColumn(TableName.Webhook, "type"); const hasType = await knex.schema.hasColumn(TableName.Webhook, "type");
if (await knex.schema.hasTable(TableName.Webhook)) {
await knex.schema.alterTable(TableName.Webhook, (tb) => { await knex.schema.alterTable(TableName.Webhook, (tb) => {
if (!hasEncryptedURL) { if (!hasUrlCipherText) {
tb.text("encryptedUrl"); tb.text("urlCipherText");
} }
if (!hasUrlIV) { if (!hasUrlIV) {
tb.string("urlIV"); tb.string("urlIV");
@@ -25,16 +26,18 @@ export async function up(knex: Knex): Promise<void> {
} }
}); });
} }
}
export async function down(knex: Knex): Promise<void> { export async function down(knex: Knex): Promise<void> {
const hasEncryptedURL = await knex.schema.hasColumn(TableName.Webhook, "encryptedUrl"); const hasUrlCipherText = await knex.schema.hasColumn(TableName.Webhook, "urlCipherText");
const hasUrlIV = await knex.schema.hasColumn(TableName.Webhook, "urlIV"); const hasUrlIV = await knex.schema.hasColumn(TableName.Webhook, "urlIV");
const hasUrlTag = await knex.schema.hasColumn(TableName.Webhook, "urlTag"); const hasUrlTag = await knex.schema.hasColumn(TableName.Webhook, "urlTag");
const hasType = await knex.schema.hasColumn(TableName.Webhook, "type"); const hasType = await knex.schema.hasColumn(TableName.Webhook, "type");
if (await knex.schema.hasTable(TableName.Webhook)) {
await knex.schema.alterTable(TableName.Webhook, (t) => { await knex.schema.alterTable(TableName.Webhook, (t) => {
if (hasEncryptedURL) { if (hasUrlCipherText) {
t.dropColumn("encryptedUrl"); t.dropColumn("urlCipherText");
} }
if (hasUrlIV) { if (hasUrlIV) {
t.dropColumn("urlIV"); t.dropColumn("urlIV");
@@ -47,3 +50,4 @@ export async function down(knex: Knex): Promise<void> {
} }
}); });
} }
}
+1 -1
View File
@@ -22,7 +22,7 @@ export const WebhooksSchema = z.object({
createdAt: z.date(), createdAt: z.date(),
updatedAt: z.date(), updatedAt: z.date(),
envId: z.string().uuid(), envId: z.string().uuid(),
encryptedUrl: z.string().nullable().optional(), urlCipherText: z.string().nullable().optional(),
urlIV: z.string().nullable().optional(), urlIV: z.string().nullable().optional(),
urlTag: z.string().nullable().optional(), urlTag: z.string().nullable().optional(),
type: z.string().default("general").nullable().optional() type: z.string().default("general").nullable().optional()
@@ -14,7 +14,7 @@ export const sanitizedWebhookSchema = WebhooksSchema.omit({
tag: true, tag: true,
algorithm: true, algorithm: true,
keyEncoding: true, keyEncoding: true,
encryptedUrl: true, urlCipherText: true,
urlIV: true, urlIV: true,
urlTag: true urlTag: true
}).merge( }).merge(
+13 -40
View File
@@ -4,9 +4,8 @@ import { AxiosError } from "axios";
import picomatch from "picomatch"; import picomatch from "picomatch";
import { SecretKeyEncoding, TWebhooks } from "@app/db/schemas"; import { SecretKeyEncoding, TWebhooks } from "@app/db/schemas";
import { getConfig } from "@app/lib/config/env";
import { request } from "@app/lib/config/request"; import { request } from "@app/lib/config/request";
import { decryptSymmetric, decryptSymmetric128BitHexKeyUTF8 } from "@app/lib/crypto"; import { infisicalSymmetricDecrypt } from "@app/lib/crypto/encryption";
import { BadRequestError } from "@app/lib/errors"; import { BadRequestError } from "@app/lib/errors";
import { logger } from "@app/lib/logger"; import { logger } from "@app/lib/logger";
@@ -17,54 +16,28 @@ import { WebhookType } from "./webhook-types";
const WEBHOOK_TRIGGER_TIMEOUT = 15 * 1000; const WEBHOOK_TRIGGER_TIMEOUT = 15 * 1000;
export const decryptWebhookDetails = (webhook: TWebhooks) => { export const decryptWebhookDetails = (webhook: TWebhooks) => {
const appCfg = getConfig(); const { keyEncoding, iv, encryptedSecretKey, tag, urlCipherText, urlIV, urlTag, url } = webhook;
const { keyEncoding, iv, encryptedSecretKey, tag, encryptedUrl, urlIV, urlTag, url } = webhook;
const encryptionKey = appCfg.ENCRYPTION_KEY;
const rootEncryptionKey = appCfg.ROOT_ENCRYPTION_KEY;
let decryptedSecretKey = ""; let decryptedSecretKey = "";
let decryptedUrl = url; let decryptedUrl = url;
if (rootEncryptionKey && keyEncoding === SecretKeyEncoding.BASE64) {
// case: encoding scheme is base64
if (encryptedSecretKey) { if (encryptedSecretKey) {
decryptedSecretKey = decryptSymmetric({ decryptedSecretKey = infisicalSymmetricDecrypt({
keyEncoding: keyEncoding as SecretKeyEncoding,
ciphertext: encryptedSecretKey, ciphertext: encryptedSecretKey,
iv: iv as string, iv: iv as string,
tag: tag as string, tag: tag as string
key: rootEncryptionKey
}); });
} }
if (encryptedUrl) { if (urlCipherText) {
decryptedUrl = decryptSymmetric({ decryptedUrl = infisicalSymmetricDecrypt({
ciphertext: encryptedUrl, keyEncoding: keyEncoding as SecretKeyEncoding,
ciphertext: urlCipherText,
iv: urlIV as string, iv: urlIV as string,
tag: urlTag as string, tag: urlTag as string
key: rootEncryptionKey
}); });
} }
} else if (encryptionKey && keyEncoding === SecretKeyEncoding.UTF8) {
// case: encoding scheme is utf8
if (encryptedSecretKey) {
decryptedSecretKey = decryptSymmetric128BitHexKeyUTF8({
ciphertext: encryptedSecretKey,
iv: iv as string,
tag: tag as string,
key: encryptionKey
});
}
if (encryptedUrl) {
decryptedUrl = decryptSymmetric128BitHexKeyUTF8({
ciphertext: encryptedUrl,
iv: urlIV as string,
tag: urlTag as string,
key: encryptionKey
});
}
}
return { return {
secretKey: decryptedSecretKey, secretKey: decryptedSecretKey,
@@ -109,17 +82,17 @@ export const getWebhookPayload = (
{ {
title: "Workspace ID", title: "Workspace ID",
value: workspaceId, value: workspaceId,
short: true short: false
}, },
{ {
title: "Environment", title: "Environment",
value: environment, value: environment,
short: true short: false
}, },
{ {
title: "Secret Path", title: "Secret Path",
value: secretPath, value: secretPath,
short: true short: false
} }
] ]
} }
+10 -31
View File
@@ -1,10 +1,9 @@
import { ForbiddenError } from "@casl/ability"; import { ForbiddenError } from "@casl/ability";
import { SecretEncryptionAlgo, SecretKeyEncoding, TWebhooksInsert } from "@app/db/schemas"; import { TWebhooksInsert } from "@app/db/schemas";
import { TPermissionServiceFactory } from "@app/ee/services/permission/permission-service"; import { TPermissionServiceFactory } from "@app/ee/services/permission/permission-service";
import { ProjectPermissionActions, ProjectPermissionSub } from "@app/ee/services/permission/project-permission"; import { ProjectPermissionActions, ProjectPermissionSub } from "@app/ee/services/permission/project-permission";
import { getConfig } from "@app/lib/config/env"; import { infisicalSymmetricEncypt } from "@app/lib/crypto/encryption";
import { encryptSymmetric, encryptSymmetric128BitHexKeyUTF8 } from "@app/lib/crypto";
import { BadRequestError } from "@app/lib/errors"; import { BadRequestError } from "@app/lib/errors";
import { TProjectEnvDALFactory } from "../project-env/project-env-dal"; import { TProjectEnvDALFactory } from "../project-env/project-env-dal";
@@ -39,10 +38,6 @@ export const webhookServiceFactory = ({ webhookDAL, projectEnvDAL, permissionSer
webhookSecretKey, webhookSecretKey,
type type
}: TCreateWebhookDTO) => { }: TCreateWebhookDTO) => {
const appCfg = getConfig();
const encryptionKey = appCfg.ENCRYPTION_KEY;
const rootEncryptionKey = appCfg.ROOT_ENCRYPTION_KEY;
const { permission } = await permissionService.getProjectPermission( const { permission } = await permissionService.getProjectPermission(
actor, actor,
actorId, actorId,
@@ -63,37 +58,21 @@ export const webhookServiceFactory = ({ webhookDAL, projectEnvDAL, permissionSer
}; };
if (webhookSecretKey) { if (webhookSecretKey) {
if (rootEncryptionKey) { const { ciphertext, iv, tag, algorithm, encoding } = infisicalSymmetricEncypt(webhookSecretKey);
const { ciphertext, iv, tag } = encryptSymmetric(webhookSecretKey, rootEncryptionKey);
insertDoc.encryptedSecretKey = ciphertext; insertDoc.encryptedSecretKey = ciphertext;
insertDoc.iv = iv; insertDoc.iv = iv;
insertDoc.tag = tag; insertDoc.tag = tag;
insertDoc.algorithm = SecretEncryptionAlgo.AES_256_GCM; insertDoc.algorithm = algorithm;
insertDoc.keyEncoding = SecretKeyEncoding.BASE64; insertDoc.keyEncoding = encoding;
} else if (encryptionKey) {
const { ciphertext, iv, tag } = encryptSymmetric128BitHexKeyUTF8(webhookSecretKey, encryptionKey);
insertDoc.encryptedSecretKey = ciphertext;
insertDoc.iv = iv;
insertDoc.tag = tag;
insertDoc.algorithm = SecretEncryptionAlgo.AES_256_GCM;
insertDoc.keyEncoding = SecretKeyEncoding.UTF8;
}
} }
if (rootEncryptionKey) { if (webhookUrl) {
const { ciphertext, iv, tag } = encryptSymmetric(webhookUrl, rootEncryptionKey); const { ciphertext, iv, tag, algorithm, encoding } = infisicalSymmetricEncypt(webhookUrl);
insertDoc.encryptedUrl = ciphertext; insertDoc.urlCipherText = ciphertext;
insertDoc.urlIV = iv; insertDoc.urlIV = iv;
insertDoc.urlTag = tag; insertDoc.urlTag = tag;
insertDoc.algorithm = SecretEncryptionAlgo.AES_256_GCM; insertDoc.algorithm = algorithm;
insertDoc.keyEncoding = SecretKeyEncoding.BASE64; insertDoc.keyEncoding = encoding;
} else if (encryptionKey) {
const { ciphertext, iv, tag } = encryptSymmetric128BitHexKeyUTF8(webhookUrl, encryptionKey);
insertDoc.encryptedUrl = ciphertext;
insertDoc.urlIV = iv;
insertDoc.urlTag = tag;
insertDoc.algorithm = SecretEncryptionAlgo.AES_256_GCM;
insertDoc.keyEncoding = SecretKeyEncoding.UTF8;
} }
const webhook = await webhookDAL.create(insertDoc); const webhook = await webhookDAL.create(insertDoc);
+7
View File
@@ -1,5 +1,11 @@
export enum WebhookType {
GENERAL = "general",
SLACK = "slack"
}
export type TWebhook = { export type TWebhook = {
id: string; id: string;
type: WebhookType;
projectId: string; projectId: string;
environment: { environment: {
slug: string; slug: string;
@@ -22,6 +28,7 @@ export type TCreateWebhookDto = {
webhookUrl: string; webhookUrl: string;
webhookSecretKey?: string; webhookSecretKey?: string;
secretPath: string; secretPath: string;
type: WebhookType;
}; };
export type TUpdateWebhookDto = { export type TUpdateWebhookDto = {
@@ -14,11 +14,7 @@ import {
SelectItem SelectItem
} from "@app/components/v2"; } from "@app/components/v2";
import { SecretPathInput } from "@app/components/v2/SecretPathInput"; import { SecretPathInput } from "@app/components/v2/SecretPathInput";
import { WebhookType } from "@app/hooks/api/webhooks/types";
enum WebhookType {
GENERAL = "general",
SLACK = "slack"
}
const formSchema = z const formSchema = z
.object({ .object({
@@ -116,7 +112,6 @@ export const AddWebhookForm = ({
<Controller <Controller
control={control} control={control}
name="type" name="type"
defaultValue={WebhookType.GENERAL}
render={({ field: { onChange, ...field }, fieldState: { error } }) => ( render={({ field: { onChange, ...field }, fieldState: { error } }) => (
<FormControl <FormControl
label="Type" label="Type"