How can I make conditional wait methods in a one function?

Question

This is the most known way to make the wait function

public BaseFunction(WebDriver driver) {
    this.driver = driver;
    wait = new WebDriverWait(this.drive