شجاع‌الدین بازرگانی، معاون امور حقوقی و مجلس وزارت نفت