Automatically clean up after an error is thrown (#31)

Co-authored-by: Sindre Sorhus <sindresorhus@gmail.com>

index.js

@@ -18,9 +18,12 @@ const writeStream = async (filePath, data) => pipeline(data, fs.createWriteStrea
 const createTask = (tempyFunction, {extraArguments = 0} = {}) => async (...arguments_) => {
 	const [callback, options] = arguments_.slice(extraArguments);
 	const result = await tempyFunction(...arguments_.slice(0, extraArguments), options);
-	const returnValue = await callback(result);
-	await del(result, {force: true});
-	return returnValue;
+
+	try {
+		return await callback(result);
+	} finally {
+		await del(result, {force: true});
+	}
 };
 
 module.exports.file = options => {

test.js

@@ -44,6 +44,19 @@ test('.file.task()', async t => {
 	t.false(await pathExists(temporaryFilePath));
 });
 
+test('.task() - cleans up even if callback throws', async t => {
+	let temporaryDirectoryPath;
+	await t.throwsAsync(tempy.directory.task(async temporaryDirectory => {
+		temporaryDirectoryPath = temporaryDirectory;
+		throw new Error('Catch me if you can!');
+	}), {
+		instanceOf: Error,
+		message: 'Catch me if you can!'
+	});
+
+	t.false(await pathExists(temporaryDirectoryPath));
+});
+
 test('.directory()', t => {
 	const prefix = 'name_';