//
// Cforall Version 1.0.0 Copyright (C) 2021 University of Waterloo
//
// The contents of this file are covered under the licence agreement in the
// file "LICENCE" distributed with Cforall.
//
// LinkOnce.cpp -- Translate the cfa_linkonce attribute.
//
// Author           : Andrew Beach
// Created On       : Thur May 13 10:10:00 2021
// Last Modified By : Andrew Beach
// Last Modified On : Wed Oct  4 10:52:00 2023
// Update Count     : 1
//

#include "LinkOnce.hpp"

#include <algorithm>

#include "AST/Attribute.hpp"
#include "AST/Decl.hpp"
#include "AST/Expr.hpp"
#include "AST/Pass.hpp"

namespace CodeGen {

namespace {

bool is_cfa_linkonce( ast::Attribute const * attr ) {
	return "cfa_linkonce" == attr->name;
}

bool is_section_attribute( ast::Attribute const * attr ) {
	return "section" == attr->name;
}

struct LinkOnceCore : public ast::WithShortCircuiting {
	void previsit( ast::Decl const * ) {
		visit_children = false;
	}

	ast::DeclWithType const * postvisit( ast::DeclWithType const * decl ) {
		// Check to see if we have to mutate, because should be uncommon.
		{
			auto & attributes = decl->attributes;
			auto found = std::find_if( attributes.begin(), attributes.end(),
					is_cfa_linkonce );
			if ( attributes.end() == found ) return decl;
		}
		auto mutDecl = mutate( decl );
		auto & attributes = mutDecl->attributes;

		// Remove all conflicting section attributes.
		erase_if( attributes, is_section_attribute );

		// Get the attribute, and overwrite it as a section attribute.
		auto found = std::find_if( attributes.begin(), attributes.end(),
				is_cfa_linkonce );
		assert( attributes.end() != found );
		ast::Attribute * attribute = found->get_and_mutate();
		assert( attribute->params.empty() );
		assert( !decl->mangleName.empty() );

		attribute->name = "section";
		attribute->params.push_back(
			ast::ConstantExpr::from_string( mutDecl->location,
				".gnu.linkonce." + decl->mangleName
			)
		);

		// Unconditionnaly add "visibility(default)" to anything with
		// .gnu.linkonce visibility is a mess otherwise.
		attributes.push_back( new ast::Attribute( "visibility", {
			ast::ConstantExpr::from_string( mutDecl->location, "default" )
		} ) );
		return mutDecl;
	}
};

} // namespace

void translateLinkOnce( ast::TranslationUnit & translationUnit ) {
	ast::Pass<LinkOnceCore>::run( translationUnit );
}

} // namespace CodeGen