backend/src/controllers/RunnerOrganizationController.ts

import { Authorized, Body, Delete, Get, JsonController, OnUndefined, Param, Post, Put, QueryParam } from 'routing-controllers';
import { OpenAPI, ResponseSchema } from 'routing-controllers-openapi';
import { getConnectionManager, Repository } from 'typeorm';
import { RunnerOrganizationHasRunnersError, RunnerOrganizationHasTeamsError, RunnerOrganizationIdsNotMatchingError, RunnerOrganizationNotFoundError } from '../errors/RunnerOrganizationErrors';
import { CreateRunnerOrganization } from '../models/actions/create/CreateRunnerOrganization';
import { UpdateRunnerOrganization } from '../models/actions/update/UpdateRunnerOrganization';
import { RunnerOrganization } from '../models/entities/RunnerOrganization';
import { ResponseEmpty } from '../models/responses/ResponseEmpty';
import { ResponseRunnerOrganization } from '../models/responses/ResponseRunnerOrganization';
import { RunnerController } from './RunnerController';
import { RunnerTeamController } from './RunnerTeamController';


@JsonController('/organizations')
@OpenAPI({ security: [{ "AuthToken": [] }, { "RefreshTokenCookie": [] }] })
export class RunnerOrganizationController {
	private runnerOrganizationRepository: Repository<RunnerOrganization>;

	/**
	 * Gets the repository of this controller's model/entity.
	 */
	constructor() {
		this.runnerOrganizationRepository = getConnectionManager().get().getRepository(RunnerOrganization);
	}

	@Get()
	@Authorized("ORGANIZATION:GET")
	@ResponseSchema(ResponseRunnerOrganization, { isArray: true })
	@OpenAPI({ description: 'Lists all organizations. <br> This includes their address, contact and teams (if existing/associated).' })
	async getAll() {
		let responseTeams: ResponseRunnerOrganization[] = new Array<ResponseRunnerOrganization>();
		const runners = await this.runnerOrganizationRepository.find({ relations: ['contact', 'teams'] });
		runners.forEach(runner => {
			responseTeams.push(new ResponseRunnerOrganization(runner));
		});
		return responseTeams;
	}

	@Get('/:id')
	@Authorized("ORGANIZATION:GET")
	@ResponseSchema(ResponseRunnerOrganization)
	@ResponseSchema(RunnerOrganizationNotFoundError, { statusCode: 404 })
	@OnUndefined(RunnerOrganizationNotFoundError)
	@OpenAPI({ description: 'Lists all information about the organization whose id got provided.' })
	async getOne(@Param('id') id: number) {
		let runnerOrg = await this.runnerOrganizationRepository.findOne({ id: id }, { relations: ['contact', 'teams'] });
		if (!runnerOrg) { throw new RunnerOrganizationNotFoundError(); }
		return new ResponseRunnerOrganization(runnerOrg);
	}

	@Post()
	@Authorized("ORGANIZATION:CREATE")
	@ResponseSchema(ResponseRunnerOrganization)
	@OpenAPI({ description: 'Create a new organsisation.' })
	async post(@Body({ validate: true }) createRunnerOrganization: CreateRunnerOrganization) {
		let runnerOrganization;
		try {
			runnerOrganization = await createRunnerOrganization.toEntity();
		} catch (error) {
			throw error;
		}

		runnerOrganization = await this.runnerOrganizationRepository.save(runnerOrganization);

		return new ResponseRunnerOrganization(await this.runnerOrganizationRepository.findOne(runnerOrganization, { relations: ['contact', 'teams'] }));
	}

	@Put('/:id')
	@Authorized("ORGANIZATION:UPDATE")
	@ResponseSchema(ResponseRunnerOrganization)
	@ResponseSchema(RunnerOrganizationNotFoundError, { statusCode: 404 })
	@ResponseSchema(RunnerOrganizationIdsNotMatchingError, { statusCode: 406 })
	@OpenAPI({ description: "Update the organization whose id you provided. <br> Please remember that ids can't be changed." })
	async put(@Param('id') id: number, @Body({ validate: true }) updateOrganization: UpdateRunnerOrganization) {
		let oldRunnerOrganization = await this.runnerOrganizationRepository.findOne({ id: id });

		if (!oldRunnerOrganization) {
			throw new RunnerOrganizationNotFoundError();
		}

		if (oldRunnerOrganization.id != updateOrganization.id) {
			throw new RunnerOrganizationIdsNotMatchingError();
		}

		await this.runnerOrganizationRepository.save(await updateOrganization.update(oldRunnerOrganization));

		return new ResponseRunnerOrganization(await this.runnerOrganizationRepository.findOne(id, { relations: ['contact', 'teams'] }));
	}

	@Delete('/:id')
	@Authorized("ORGANIZATION:DELETE")
	@ResponseSchema(ResponseRunnerOrganization)
	@ResponseSchema(ResponseEmpty, { statusCode: 204 })
	@ResponseSchema(RunnerOrganizationHasTeamsError, { statusCode: 406 })
	@ResponseSchema(RunnerOrganizationHasRunnersError, { statusCode: 406 })
	@OnUndefined(204)
	@OpenAPI({ description: 'Delete the organsisation whose id you provided. <br> If the organization still has runners and/or teams associated this will fail. <br> To delete the organization with all associated runners and teams set the force QueryParam to true (cascading deletion might take a while). <br> This won\'t delete the associated contact. <br> If no organization with this id exists it will just return 204(no content).' })
	async remove(@Param("id") id: number, @QueryParam("force") force: boolean) {
		let organization = await this.runnerOrganizationRepository.findOne({ id: id });
		if (!organization) { return null; }
		let runnerOrganization = await this.runnerOrganizationRepository.findOne(organization, { relations: ['contact', 'runners', 'teams'] });

		if (!force) {
			if (runnerOrganization.teams.length != 0) {
				throw new RunnerOrganizationHasTeamsError();
			}
		}
		const teamController = new RunnerTeamController()
		for (let team of runnerOrganization.teams) {
			await teamController.remove(team.id, true);
		}

		if (!force) {
			if (runnerOrganization.runners.length != 0) {
				throw new RunnerOrganizationHasRunnersError();
			}
		}
		const runnerController = new RunnerController()
		for (let runner of runnerOrganization.runners) {
			await runnerController.remove(runner.id, true);
		}

		const responseOrganization = new ResponseRunnerOrganization(runnerOrganization);
		await this.runnerOrganizationRepository.delete(organization);
		return responseOrganization;
	}
}