Reference implementation for proposed changes to create custom tasks mutation

services/api/src/resolvers.js

@@ -81,6 +81,7 @@ const {
 
 const {
   addAdvancedTaskDefinition,
+  customTaskDefinitionResolver,
   updateAdvancedTaskDefinition,
   advancedTaskDefinitionById,
   resolveTasksForEnvironment,
@@ -491,6 +492,10 @@ const resolvers = {
     deployTargetConfigsByDeployTarget: getDeployTargetConfigsByDeployTarget,
     getEnvVariablesByProjectEnvironmentName,
   },
+  CustomTaskMutations: {
+    createImage: customTaskDefinitionResolver.image,
+    createCommand: customTaskDefinitionResolver.command,
+  },
   Mutation: {
     addProblem,
     addProblemHarborScanMatch,
@@ -573,6 +578,7 @@ const resolvers = {
     deleteEnvVariableByName,
     addTask,
     addAdvancedTaskDefinition,
+    customTasks: () => ({}),
     updateAdvancedTaskDefinition,
     deleteAdvancedTaskDefinition,
     invokeRegisteredTask,

services/api/src/resources/task/task_definition_resolvers.ts

@@ -213,6 +213,49 @@ export const advancedTaskDefinitionArgumentById = async (
   return R.prop(0, rows);
 };
 
+
+/**
+ *
+ */
+export const customTaskDefinitionResolver = {
+  image: async (parent, args, context, info) => {
+    // console.log(parent);
+    const { input } = args;
+    const { sqlClientPool, hasPermission, models, userActivityLogger } = context;
+    let target = await customTaskDefinitionResolver.resolveTarget(sqlClientPool, input.target.projectName, input.target.environmentName || null);
+    let customTask = { ... input, ... target, ...{advancedTaskDefinitionArguments: input.arguments, type: "IMAGE"}};
+    let ret = await addAdvancedTaskDefinition(parent, {input:customTask}, context);
+    return ret;
+  },
+  command: async (parent, args, context, info) => {
+    const { input } = args;
+    const { sqlClientPool, hasPermission, models, userActivityLogger } = context;
+    let target = await customTaskDefinitionResolver.resolveTarget(sqlClientPool, input.target.projectName, input.target.environmentName || null);
+    let customTask = { ... input, ... target, ...{advancedTaskDefinitionArguments: input.arguments, type: "COMMAND"}};
+    let ret = await addAdvancedTaskDefinition(parent, {input:customTask}, context);
+    return ret;
+    },
+    resolveTarget: async(sqlClientPool, projectName, environmentName = null) => {
+      let pid = await projectHelpers(sqlClientPool).getProjectIdByName(projectName);
+      if(pid == null) {
+        throw Error(`Unable to find project "${projectName}"`);
+      }
+
+      if(environmentName == null) {
+        // we're dealing with a project only, so we set up the target to return the projectID in the required form
+        return {project: pid};
+      }
+
+      // find the environment id
+      let environments = await environmentHelpers(sqlClientPool).getEnvironmentsByEnvironmentInput({name: environmentName, project: { id: pid }});
+      let environment = environments[0];
+      if(environment == null) {
+        throw Error(`Unable to find environment "${environmentName}" for project "${projectName}"`);
+      }
+      return {environment: environment.id};
+    }
+};
+
 export const addAdvancedTaskDefinition = async (
   root,
   {

services/api/src/typeDefs.js

@@ -1050,6 +1050,37 @@ const typeDefs = gql`
     project: String!
   }
 
+  input CustomTaskTargetInput {
+    projectName: String!
+    environmentName: String
+  }
+
+  input CreateImageInput {
+    name: String!
+    description: String!
+    image: String!
+    permission: TaskPermission!
+    target: CustomTaskTargetInput!
+    advancedTaskDefinitionArguments: [AdvancedTaskDefinitionArgumentInput]
+    confirmationText: String
+  }
+
+  input CreateCommandInput {
+    name: String!
+    description: String!
+    service: String!
+    command: String!
+    permission: TaskPermission!
+    target: CustomTaskTargetInput!
+    advancedTaskDefinitionArguments: [AdvancedTaskDefinitionArgumentInput]
+    confirmationText: String
+  }
+
+  type CustomTaskMutations {
+    createImage(input: CreateImageInput!): AdvancedTaskDefinitionImage
+    createCommand(input: CreateCommandInput!): AdvancedTaskDefinitionCommand
+  }
+
   type Query {
     """
     Returns the current user
@@ -1916,6 +1947,7 @@ const typeDefs = gql`
   }
 
   type Mutation {
+    customTasks: CustomTaskMutations!
     """
     Add Environment or update if it is already existing
     """