create a grid where the squares can be moved with matched geometry effect in swiftui in swift

main.swift
struct SquareView: View {
    @Namespace var animation
    @State var isExpanded = false
    
    var body: some View {
        VStack {
            if !isExpanded {
                Rectangle()
                    .fill(Color.blue)
                    .frame(width: 100, height: 100)
                    .matchedGeometryEffect(id: "blueSquare", in: animation)
                    .onTapGesture {
                        withAnimation {
                            isExpanded.toggle()
                        }
                    }
            } else {
                VStack {
                    Rectangle()
                        .fill(Color.blue)
                        .frame(width: 300, height: 300)
                        .matchedGeometryEffect(id: "blueSquare", in: animation)
                        .onTapGesture {
                            withAnimation {
                                isExpanded.toggle()
                            }
                        }
                }
            }
        }
    }
}

struct GridView: View {
    var body: some View {
        let columns: [GridItem] = Array(repeating: .init(.flexible()), count: 3)
        LazyVGrid(columns: columns, spacing: 15) {
            ForEach(0..<9) { index in
                SquareView()
            }
        }
    }
}
1312 chars
44 lines