/**

 * cpp-finally

 * Copyright © 2024 František Kučera (Frantovo.cz, GlobalCode.info)

 *

 * This program is free software: you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation, version 3 of the License.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program. If not, see <http://www.gnu.org/licenses/>.

 */

#include <iostream>

#include <exception>

#include <functional>

class Finally {

private:

	Finally(const Finally&) = delete;

	Finally& operator=(const Finally&) = delete;

public:

	std::function<void(void) > fx;

	Finally(std::function<void(void) > fx) : fx(fx) {

	}

	virtual ~Finally() {

		fx();

	}

};

void fxThrowing(bool fail, void* data) {

	std::cout << "  doing something with data " << data << std::endl;

	if (fail) throw std::logic_error("error from fxThrowing()");

}

void fxAllocating(bool fail) {

	void* buf = malloc(486);

	void* tmp = malloc(123);

	std::cout << "  allocated memory at: " << buf << std::endl;

	std::cout << "  allocated memory at: " << tmp << std::endl;

	Finally finally([&]() {

		free(buf);

		free(tmp);

		std::cout << "  freed memory at: " << buf << std::endl;

		std::cout << "  freed memory at: " << tmp << std::endl;

	});

	fxThrowing(fail, buf);

	fxThrowing(fail, tmp);

}

int main(int argc, char** argv) {

	bool fail = argc == 2 && std::string("fail") == argv[1];

	const char* name = "good-lambda";

	std::cout << name << " (fail=" << fail << ")\n";

	try {

		fxAllocating(fail);

	} catch (const std::exception& e) {

		std::cout << "  caught exception: " << e.what() << std::endl;

	}

}