From ad59e46740ef636b95667615e3881fcee6fbbcb9 Mon Sep 17 00:00:00 2001
From: liach
Date: Fri, 8 Nov 2019 16:35:19 -0600
Subject: Allow multiple services for enigma (#168)

* Allow multiple services for enigma

Signed-off-by: liach <liach@users.noreply.github.com>

* Delete bad dummy

Signed-off-by: liach <liach@users.noreply.github.com>
---
 src/main/java/cuchaz/enigma/EnigmaServices.java | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

(limited to 'src/main/java/cuchaz/enigma/EnigmaServices.java')

diff --git a/src/main/java/cuchaz/enigma/EnigmaServices.java b/src/main/java/cuchaz/enigma/EnigmaServices.java
index 86507bc..45062d7 100644
--- a/src/main/java/cuchaz/enigma/EnigmaServices.java
+++ b/src/main/java/cuchaz/enigma/EnigmaServices.java
@@ -1,21 +1,21 @@
 package cuchaz.enigma;
 
-import com.google.common.collect.ImmutableMap;
+import com.google.common.collect.ImmutableListMultimap;
 import cuchaz.enigma.api.service.EnigmaService;
 import cuchaz.enigma.api.service.EnigmaServiceType;
 
-import java.util.Optional;
+import java.util.Collections;
+import java.util.List;
 
 public final class EnigmaServices {
-	private final ImmutableMap<EnigmaServiceType<?>, EnigmaService> services;
+	private final ImmutableListMultimap<EnigmaServiceType<?>, EnigmaService> services;
 
-	EnigmaServices(ImmutableMap<EnigmaServiceType<?>, EnigmaService> services) {
+	EnigmaServices(ImmutableListMultimap<EnigmaServiceType<?>, EnigmaService> services) {
 		this.services = services;
 	}
 
 	@SuppressWarnings("unchecked")
-	public <T extends EnigmaService> Optional<T> get(EnigmaServiceType<T> type) {
-		EnigmaService service = services.get(type);
-		return Optional.ofNullable((T) service);
+	public <T extends EnigmaService> List<T> get(EnigmaServiceType<T> type) {
+		return (List<T>) services.get(type);
 	}
 }
-- 
cgit v1.2.3