9/6/2019 10:58 AM PT

Exclusive Details

TMZ

10:57 AM PT -- Sources close to the situation tell TMZ the shooting had nothing to do with Olsson Lilja's representation of A$AP. Our sources say it was a domestic dispute with a woman close to him, and she is now in custody.

8:20 AM PT -- The Swedish Prosecution Authority tells TMZ ... a suspect has been arrested on instigation of attempted murder, but could not give further details at this time per Swedish law.

The lawyer who initially represented A$AP Rocky in his Swedish assault case was shot by someone who appears to have been lying in wait for him ... TMZ has learned.

We're told Henrik Olsson Lilja was leaving his apartment building early Friday morning in downtown Stockholm when a gunman walked up to him and shot him in the head and the chest. Cops tell us the lawyer was inside the building at the time of the attack.

The gunman jumped into a waiting SUV, which then sped off.

We don't know the lawyer's condition, but we're told he actually called the police. TMZ has seen a photo of him on the street and the head injury is too gruesome to post.