If we get divorced, and I'm still an alien with no job, will that affect my chance to get child custody and bring him back my country?

I am married with my husband for a year. We have a 3 month old son. I was a foreign student, and we had applied for my status change in March 2016, but we have not received a letter for the interview yet. I checked that I will not be interviewed for conditional green card until April 2017. My husband and I might not work out. He doesn't care for the baby. I have been taking care of the baby almost 24/7. I have not found a job yet. And I might have to go back my country if my husband doesn't go to the interview with me. As I said, my husband doesn't take care of the baby, and has anger issues. How can I prove that to the judge? I'm so afraid he will get the custody and I have to go back my country without my baby.