Saneky Nodes Color

[Omer Paran]

I need to set by myself the colors for each node . 

I need to group nodes by colors, in order to show flow of information from A to B To C , the nodes of A,B,C need to be all blue for example

The way it is handled right now is that I define an array:

var nodeColors = ['black','red','purple','orange','green'];

pass it to :

sankey: {
        node: {
          colors: nodeColors,
      }
  };

and colors are chosen with no regard to the data flow 

Is there a way to define a color per node ? also per link ?

thank you

Omer.