src/main/java/net/pterodactylus/fcp/quelaton/LoadPluginCommandImpl.java

   1 package net.pterodactylus.fcp.quelaton;
   2
   3 import java.io.IOException;
   4 import java.util.Optional;
   5 import java.util.concurrent.ExecutionException;
   6 import java.util.concurrent.ExecutorService;
   7 import java.util.function.Supplier;
   8
   9 import net.pterodactylus.fcp.LoadPlugin;
  10 import net.pterodactylus.fcp.LoadPlugin.OfficialSource;
  11 import net.pterodactylus.fcp.LoadPlugin.UrlType;
  12 import net.pterodactylus.fcp.PluginInfo;
  13 import net.pterodactylus.fcp.ProtocolError;
  14
  15 import com.google.common.util.concurrent.ListenableFuture;
  16 import com.google.common.util.concurrent.ListeningExecutorService;
  17 import com.google.common.util.concurrent.MoreExecutors;
  18
  19 /**
  20  * Default {@link LoadPluginCommand} implementation based on {@link FcpDialog}.
  21  *
  22  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  23  */
  24 public class LoadPluginCommandImpl implements LoadPluginCommand {
  25
  26         private final ListeningExecutorService threadPool;
  27         private final ConnectionSupplier connectionSupplier;
  28         private final LoadPlugin loadPlugin;
  29
  30         public LoadPluginCommandImpl(ExecutorService threadPool, ConnectionSupplier connectionSupplier, Supplier<String> identifierGenerator) {
  31                 this.threadPool = MoreExecutors.listeningDecorator(threadPool);
  32                 this.connectionSupplier = connectionSupplier;
  33                 loadPlugin = new LoadPlugin(identifierGenerator.get());
  34         }
  35
  36         @Override
  37         public LoadPluginCommand addToConfig() {
  38                 loadPlugin.setStore(true);
  39                 return this;
  40         }
  41
  42         @Override
  43         public Executable<Optional<PluginInfo>> officialFromFreenet(String pluginIdentifier) {
  44                 loadPlugin.setUrlType(UrlType.OFFICIAL);
  45                 loadPlugin.setOfficialSource(OfficialSource.FREENET);
  46                 loadPlugin.setPluginUrl(pluginIdentifier);
  47                 return this::execute;
  48         }
  49
  50         @Override
  51         public Executable<Optional<PluginInfo>> officialFromHttps(String pluginIdentifier) {
  52                 loadPlugin.setUrlType(UrlType.OFFICIAL);
  53                 loadPlugin.setOfficialSource(OfficialSource.HTTPS);
  54                 loadPlugin.setPluginUrl(pluginIdentifier);
  55                 return this::execute;
  56         }
  57
  58         @Override
  59         public Executable<Optional<PluginInfo>> fromFreenet(String key) {
  60                 loadPlugin.setUrlType(UrlType.FREENET);
  61                 loadPlugin.setPluginUrl(key);
  62                 return this::execute;
  63         }
  64
  65         @Override
  66         public Executable<Optional<PluginInfo>> fromFile(String filename) {
  67                 loadPlugin.setUrlType(UrlType.FILE);
  68                 loadPlugin.setPluginUrl(filename);
  69                 return this::execute;
  70         }
  71
  72         @Override
  73         public Executable<Optional<PluginInfo>> fromUrl(String url) {
  74                 loadPlugin.setUrlType(UrlType.URL);
  75                 loadPlugin.setPluginUrl(url);
  76                 return this::execute;
  77         }
  78
  79         private ListenableFuture<Optional<PluginInfo>> execute() {
  80                 return threadPool.submit(this::executeDialog);
  81         }
  82
  83         private Optional<PluginInfo> executeDialog() throws IOException, ExecutionException, InterruptedException {
  84                 try (LoadPluginDialog loadPluginDialog = new LoadPluginDialog()) {
  85                         return loadPluginDialog.send(loadPlugin).get();
  86                 }
  87         }
  88
  89         private class LoadPluginDialog extends FcpDialog<Optional<PluginInfo>> {
  90
  91                 public LoadPluginDialog() throws IOException {
  92                         super(threadPool, connectionSupplier.get(), Optional.<PluginInfo>empty());
  93                 }
  94
  95                 @Override
  96                 protected void consumePluginInfo(PluginInfo pluginInfo) {
  97                         setResult(Optional.ofNullable(pluginInfo));
  98                 }
  99
 100                 @Override
 101                 protected void consumeProtocolError(ProtocolError protocolError) {
 102                         finish();
 103                 }
 104
 105         }
 106
 107 }