【SwiftUI】EnvironmentObject を ViewModel に渡す

EnvironmentObjectをViewModelに渡す方法について記載します。

1. 環境

Xcode : 12.2
Swift : 5

2. 構成

ContentView
ChildViewModel で設定した値を表示します。

ChildView
ContentView の子ビュー。
ボタンを押したら、ChildViewModel の関数を呼び出します。

ChildViewModel
EnvironmentObject に値を設定します。
ここで設定した値が、ContentView に表示されます。

3. ポイント

特別なことはなく ChildViewModel で EnvironmentObject を受け取る関数を作り、ChildView から VM の関数を呼び出し設定します。

ChildViewModel

// EnvironmentObject を設定
public func setShareData( shareData: ShareData ){
    self.shareData = shareData
}

// EnvironmentObject を設定

public func setShareData( shareData: ShareData ){

self.shareData = shareData

}

ChildView

}.onAppear(){
    // ViewModel に EnvironmentObject を設定
    childViewModel.setShareData(shareData: shareData)
}

}.onAppear(){

// ViewModel に EnvironmentObject を設定

childViewModel.setShareData(shareData: shareData)

}

4. コード全体

EnvironmentObject（ShareData.swift）

import Foundation

class ShareData: ObservableObject {
    @Published var message = ""
}

import Foundation

class ShareData: ObservableObject {

@Published var message = ""

}

ContentView

import SwiftUI

struct ContentView: View {
    
    @EnvironmentObject var shareData: ShareData
    
    var body: some View {
        
        ChildView()
            .environmentObject(shareData)
        
        Text(shareData.message) // ViewModel で設定した値を表示

    }
}

struct ContentView_Previews: PreviewProvider {
    
    static var previews: some View {
        ContentView()
            .environmentObject(ShareData())

    }
}

import SwiftUI

struct ContentView: View {

@EnvironmentObject var shareData: ShareData

var body: some View {

ChildView()

.environmentObject(shareData)

Text(shareData.message) // ViewModel で設定した値を表示

}

struct ContentView_Previews: PreviewProvider {

static var previews: some View {

ContentView()

.environmentObject(ShareData())

}

ChildView

import SwiftUI

struct ChildView: View {
    
    @EnvironmentObject var shareData: ShareData
    @ObservedObject var childViewModel: ChildViewModel
    
    init(){
        childViewModel = ChildViewModel()
    }
    
    var body: some View {
        
        VStack {
            Button(action: {
                // ViewModel の関数を呼び出す
                childViewModel.setMessage(message: "VMでメッセージを設定")
            }, label: {
                Text("メッセージを表示")
                .frame(minWidth: 0, maxWidth: .infinity)
                .padding(20)
                    .overlay(RoundedRectangle(cornerRadius: 25.0).stroke(Color.blue, lineWidth: 3))
            })
            .background(RoundedRectangle(cornerRadius: 25.0).fill(Color.white))
            .padding(20)

        }.onAppear(){
            // ViewModel に EnvironmentObject を設定
            childViewModel.setShareData(shareData: shareData)
        }

    }
}

struct ChildView_Previews: PreviewProvider {
    static var previews: some View {
        ChildView()
    }
}

import SwiftUI

struct ChildView: View {

@EnvironmentObject var shareData: ShareData

@ObservedObject var childViewModel: ChildViewModel

init(){

childViewModel = ChildViewModel()

}

var body: some View {

VStack {

Button(action: {

// ViewModel の関数を呼び出す

childViewModel.setMessage(message: "VMでメッセージを設定")

}, label: {

Text("メッセージを表示")

.frame(minWidth: 0, maxWidth: .infinity)

.padding(20)

.overlay(RoundedRectangle(cornerRadius: 25.0).stroke(Color.blue, lineWidth: 3))

})

.background(RoundedRectangle(cornerRadius: 25.0).fill(Color.white))

.padding(20)

}.onAppear(){

// ViewModel に EnvironmentObject を設定

childViewModel.setShareData(shareData: shareData)

}

struct ChildView_Previews: PreviewProvider {

static var previews: some View {

ChildView()

}

ChildViewModel

import SwiftUI
import Combine

class ChildViewModel: ObservableObject, Identifiable {

    @Published var shareData: ShareData?
    
    // EnvironmentObject を設定
    public func setShareData( shareData: ShareData ){
        self.shareData = shareData
    }
    
    // EnvironmentObject に値を設定
    public func setMessage( message: String ){
        self.shareData?.message = message
    }
    
}

import SwiftUI

import Combine

class ChildViewModel: ObservableObject, Identifiable {

@Published var shareData: ShareData?

// EnvironmentObject を設定

public func setShareData( shareData: ShareData ){

self.shareData = shareData

}

// EnvironmentObject に値を設定

public func setMessage( message: String ){

self.shareData?.message = message

}