Нужна функция для вывода только партнерской части ссылки

В функции get_ref_url нашел две возможности для шорткода [ ref-url type='tail' ] и [ ref-url type='current' ]. Одна выводит через tail, вторая адрес страницы

Можете добавить что-то типа [ ref-url type='ref' ], чтобы можно было выводить для людей такого рода ссылки
Ваша партнерская ссылка на страницу Page1:

<a href="http://site.ru/page1/[ ref-url type='ref' ]">http://site.ru/page1/[ ref-url type='ref' ]</a>

то есть просто дополнительный реферальный кусок ссылки выводить.

Вот такую конструкцию написал:

if($type=='ref') $url = '?'.$ps_options['get_name'].'='.$data;

Вроде работает) Надеюсь на работоспособность плагина это не повлияет.
Скажите, а как-то можно это вынести в какое-нть место, чтобы при обновлении компонента не слетела эта строка?

Спасибо Андрей за идею) Хорошая, только не представляю, как это реализовать. Первый раз в жизни на php что-то написал.