Problema al reemplazar caracteres por divs

[SeaPirates]

Hola, después de intentar reemplazar cada caracter por un color para hacer un bbcode de códigos con colores, encontré la función "highlight_string()", el problema es que necesito cambiar los colores de funciones, etc.

¿Como puedo hacerlo? Hay algún modo de reemplazar partes del código por colores, como por ejemplo con str_replace?

Traté de hacerlo con str_replace pero me reemplaza todo..

Código PHP:

$string = str_replace(
        array(
        '<',
        '>',
        '"',
        '\'',
        '=',
        'function'
        ),
        
        array(
        '<span color="color_1"><</span>',
        '<span color="color_1">s</spa>',
        '<span color="color_2">"</spa>',
        '<span color="color_2">\'</spa>',
        '<span color="color_3">=</spa>',
        '<span color="color_4">function</spa>'
        ),
        $string
    );