Novak Djokovic has almost fully recovered from the injury that ended his Australian Open campaign in the semi-finals.