After spending the past several weeks on New York’s tony Upper East Side, Michael Cohen has found himself back in federal prison after a dispute with the government over what his lawyers say were restrictions designed to prevent him from finishing a book about the president.