Brestfeeding and medication

I have horrible ADHD its been a rough road off of my medication while being pregant. My little guy is due in 6 weeks and I would like to brestfeed but return to my medication. I am a first time mom and I am assuming pumping can work for me? I cant be the only one wondering about breastfeeding and meds. Any help or advice I would love! Thanks!