any waitForJs function to wait some javascript code returns true

[ypapax]

Is there any function that returns only after some js code return true.

func waitForRedButton(session *webdriver.Session){
     session.waitForJs(`$('#redButton').css('color')=='red'`)
     // next code should be executed only after `#redButton` becomes red
}