(function($, undefined) {
	// Needed for conditional logic
	var Field = acf.models.SelectField.extend({
		type: 'country'
	});
	acf.registerFieldType(Field);
	acf.registerConditionForFieldType('contains', 'country');
	acf.registerConditionForFieldType('selectEqualTo', 'country');
	acf.registerConditionForFieldType('selectNotEqualTo', 'country');

	/**
	 * Format country (Select3 v4)
	 *
	 * @param  {object} state
	 * @return {jQuery}
	 */
	function format_country(state) {
		if (!state.id) {
			return state.text;
		}
		return $(
			'<span class="acf-country-flag-icon famfamfam-flags ' +
				state.id.toLowerCase() +
				'"></span> <span class="acf-country-flag-name">' +
				state.text +
				'</span>'
		);
	}

	/**
	 * Country args for select2
	 *
	 * @param  {object} args
	 * @param  {jQuery} $select
	 * @param  {object} settings
	 * @param  {jQuery} field
	 * @param  {object} instance
	 * @return {object}
	 */
	acf.addFilter('select2_args', function(
		args,
		$select,
		settings,
		field,
		instance
	) {
		if (instance.data.field.get('type') !== 'country') {
			return args;
		}

		// Select2 version
		$.extend(args, {
			templateResult: format_country,
			templateSelection: format_country
		});
		return args;
	});
})(jQuery);