@pnp/cli-microsoft365/dist/m365/purview/commands/threatassessment/threatassessment-add.js

Manage Microsoft 365 and SharePoint Framework projects on any platform

import { z } from 'zod';
import { globalOptionsZod } from '../../../../Command.js';
import request from '../../../../request.js';
import { accessToken } from '../../../../utils/accessToken.js';
import GraphCommand from '../../../base/GraphCommand.js';
import commands from '../../commands.js';
import auth from '../../../../Auth.js';
import fs from 'fs';
import path from 'path';
const allowedTypes = ['file', 'url'];
const allowedExpectedAssessments = ['block', 'unblock'];
const allowedCategories = ['spam', 'phishing', 'malware'];
export const options = z.strictObject({
    ...globalOptionsZod.shape,
    type: z.enum(allowedTypes).alias('t'),
    expectedAssessment: z.enum(allowedExpectedAssessments).alias('e'),
    category: z.enum(allowedCategories).alias('c'),
    path: z.string().refine(val => fs.existsSync(val), {
        error: 'Specified file does not exist.'
    }).optional().alias('p'),
    url: z.string().optional().alias('u')
});
class PurviewThreatAssessmentAddCommand extends GraphCommand {
    get name() {
        return commands.THREATASSESSMENT_ADD;
    }
    get description() {
        return 'Creates a threat assessment';
    }
    get schema() {
        return options;
    }
    getRefinedSchema(schema) {
        return schema
            .refine(opts => {
            if (opts.type === 'file') {
                return opts.path !== undefined;
            }
            return true;
        }, {
            error: `'path' is required when type is 'file'.`,
            path: ['path'],
            params: {
                customCode: 'required'
            }
        })
            .refine(opts => {
            if (opts.type === 'url') {
                return opts.url !== undefined;
            }
            return true;
        }, {
            error: `'url' is required when type is 'url'.`,
            path: ['url'],
            params: {
                customCode: 'required'
            }
        });
    }
    async commandAction(logger, args) {
        try {
            if (accessToken.isAppOnlyAccessToken(auth.connection.accessTokens[this.resource].accessToken)) {
                throw 'This command currently does not support app only permissions.';
            }
            if (this.verbose) {
                await logger.logToStderr(`Adding threat assessment of type ${args.options.type} with expected assessment ${args.options.expectedAssessment} and category ${args.options.category}`);
            }
            const requestBody = {
                expectedAssessment: args.options.expectedAssessment,
                category: args.options.category,
                url: args.options.url,
                contentData: args.options.path && fs.readFileSync(args.options.path).toString('base64'),
                fileName: args.options.path && path.basename(args.options.path)
            };
            switch (args.options.type) {
                case 'file':
                    requestBody['@odata.type'] = '#microsoft.graph.fileAssessmentRequest';
                    break;
                case 'url':
                    requestBody['@odata.type'] = '#microsoft.graph.urlAssessmentRequest';
                    break;
            }
            const requestOptions = {
                url: `${this.resource}/v1.0/informationProtection/threatAssessmentRequests`,
                headers: {
                    accept: 'application/json;odata.metadata=none'
                },
                data: requestBody,
                responseType: 'json'
            };
            const response = await request.post(requestOptions);
            await logger.log(response);
        }
        catch (err) {
            this.handleRejectedODataPromise(err);
        }
    }
}
export default new PurviewThreatAssessmentAddCommand();
//# sourceMappingURL=threatassessment-add.js.map