【SwiftUI】 UITextViewでboundingRectが上手く動きません・・・

Question

UITextViewをUIViewRepresentableでラッピングして以下の様に使用しています。また文字をハイライトできる様にboundingRectで文字の範囲を取得してUIViewにaddしています。

 struct CustomTextView: UIViewRepresentable {
    
    @ObservedObject var tmpTextVM: TmpTextViewModel
        
        func makeUIView(context: UIViewRepresentableContext<CustomTextView>) -> UITextView {
            let textView = UITextView()
            textView.backgroundColor = UIColor.clear
            textView.isScrollEnabled = false
            textView.text = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
            textView.font = UIFont(name: "ArialMT", size: 20)
            textView.setContentCompressionResistancePriority(.defaultLow, for: .horizontal)
            return textView
        }
    
        func updateUIView(_ uiView: UITextView, context: Context) {
            
            uiView.textColor = tmpTextVM.color

            
            let range = (uiView.text as NSString).range(of: uiView.text)
            var rect = uiView.layoutManager.boundingRect(forGlyphRange: range, in: uiView.textContainer)
            
            rect.size.width += 25
            rect.size.height += 10
            rect.origin.x += uiView.textContainerInset.left - 10
            rect.origin.y += uiView.textContainerInset.top - 5
            
            uiView.viewWithTag(300)?.removeFromSuperview()

            let view = UIView()
            
            view.backgroundColor = UIColor.blue
            
            view.layer.cornerRadius = 10
            view.frame = rect
            view.tag = 300
            
            uiView.addSubview(view)
            uiView.sendSubviewToBack(view)
            
        }
}

これを使用する際には以下の様にします。

struct TmpView7View: View {
    
    @ObservedObject var tmpTextVM: TmpTextViewModel
    
    var body: some View {
        VStack {
            
            Button(action: {
                  tmpTextVM.changeColor(UIColor.red)
              }){
                  Image(systemName:"eyedropper.full")
              }
            
            CustomTextView(tmpTextVM: tmpTextVM)
                .frame(width:300, height: 300, alignment: .topLeading)
                .border(Color.red, width: 1)
        }
    }
}

ただこれだと文字の高さが1行分だけしか取得できていない様でこの様になります。

上のボタンでを押すとObservedObjectの値が更新されて再度updateUIViewが動き、そうすると以下の様に全ての文字がハイライトされます。

最初から正しい文字の高さを取得するにはどうすれば良いでしょうか？
最初にmakeUIViewが呼ばれたた後にもupdateUIViewは呼ばれていますがその際には正しい高さが取得できなくてその次のupdateUIViewで正しく高さが取得できる様です。

因みにあまり関係ありませんがObservedObjectのクラスは以下の様になっています。これでほぼコードの全てなので実際に動きます。

class TmpTextViewModel: ObservableObject {
    
    @Published private var tmpTextModel: TmpTextModel = TmpTextModel()
   
    var color: UIColor {tmpTextModel.color}
    
    func changeColor(_ color: UIColor) {
        tmpTextModel.color = color
    }
    
}

struct TmpTextModel {
    
    var color:UIColor = UIColor.purple
    
}