Segregation interfaces for ViewModels integration with DI

[jordizspmobile]

Hi,

I have a ViewModel that conforms to different interfaces:

protocol ViewModelAContract {
       ...
}

protocol ViewModelSectionAContract {
       ...
}

protocol ViewModelSectionBContract {
       ...
}

class ViewModelA: ViewModelAContract, ViewModelSectionAContract, ViewModelSectionBContract {
	...
}

My module is the following:

extension Container {
    var viewModel: Factory<ViewModelAContract> {
        self { @MainActor in ViewModelA() }
    }
}

How can I retrieve this viewModel, for example, as ViewModelSectionAContract? Without having to cast it everywhere. Is there a way to retrieve this ViewModel with the same memory reference but as another protocol?

[hmlongco]

You can either expose each individually...

extension Container {
    var viewModelA: Factory<ViewModelAContract> {
        self { self.viewModel() }
    }
    var viewModelSectionAContract: Factory<ViewModelSectionAContract> {
        self { self.viewModel() }
    }
    var viewModelSectionBContract: Factory<ViewModelSectionBContract> {
        self { self.viewModel() }
    }
    var viewModel: Factory<ViewModelA> {
        self { ViewModelA() }.shared
    }
}

Or pass the container...

class TestContract {
    let viewModel: ViewModelSectionAContract
    init(container: Container) {
        viewModel = container.viewModel()
    }
}

Or simply by assignment

class TestContract {
    var viewModel: ViewModelSectionAContract = Container.shared.viewModel()
}

[jordizspmobile]

Perfect, thanks ^^