When a peace officer has stopped a person for temporary questioning pursuant to Section 107-14 of this Code and reasonably suspects that he or another is in danger of attack, he may search the person for weapons. If the officer discovers a weapon, he may take it until the completion of the questioning, at which time he shall either return the weapon, if lawfully possessed, or arrest the person so questioned.