While Iran has told its proxies in Iraq to lay low in the final weeks of Trump's presidency, a senior US diplomat has warned Tehran and its allies the US will respond if attacked.