import construction.Constructible;
import resources;

tidy class RetrofitConstructible : Constructible {
	Object@ fleet;
	Object@ constructFrom;

	RetrofitConstructible(Object& obj, Object@ Fleet, int cost, double labor, int extraMaint) {
		@fleet = Fleet;
		buildCost = cost;
		maintainCost = extraMaint;
		totalLabor = labor;
	}

	RetrofitConstructible(SaveFile& file) {
		Constructible::load(file);
		file >> fleet;
		if(file >= SV_0149)
			file >> constructFrom;
	}

	void save(SaveFile& file) {
		Constructible::save(file);
		file << fleet;
		file << constructFrom;
	}

	bool repeat(Object& obj) {
		return false;
	}

	ConstructibleType get_type() {
		return CT_Retrofit;
	}

	string get_name() {
		return format(locale::BUILD_RETROFIT, fleet.name);
	}

	void cancel(Object& obj) {
		fleet.stopFleetRetrofit(obj);
		Constructible::cancel(obj);
	}

	void complete(Object& obj) {
		fleet.finishFleetRetrofit(obj);
	}

	TickResult tick(Object& obj, double time) override {
		if(obj.owner !is fleet.owner || obj.region is null) {
			cancel(obj);
			return TR_Remove;
		}
		if(constructFrom !is null) {
			if(constructFrom.region !is fleet.region) {
				cancel(obj);
				return TR_Remove;
			}
		}
		else {
			if(obj.region !is fleet.region) {
				cancel(obj);
				return TR_Remove;
			}
		}
		return TR_UsedLabor;
	}

	void write(Message& msg) {
		Constructible::write(msg);
		msg << fleet;
	}
};