multi: add MacaroonPath method to subserver manager

Author: ellemouton

rpc_proxy.go

@@ -467,46 +467,21 @@ func (p *rpcProxy) basicAuthToMacaroon(basicAuth, requestURI string,
 		return nil, ctxErr
 	}
 
-	var (
-		macPath string
-		macData []byte
-	)
-	subserver, err := p.permsMgr.SubServerHandler(requestURI)
-	if err != nil {
-		return nil, err
-	}
-
-	switch subserver {
-	case subservers.LND:
-		_, _, _, macPath, macData = p.cfg.lndConnectParams()
+	var macData []byte
+	handled, macPath := p.subServerMgr.MacaroonPath(requestURI)
 
-	case subservers.FARADAY:
-		if p.cfg.faradayRemote {
-			macPath = p.cfg.Remote.Faraday.MacaroonPath
-		} else {
-			macPath = p.cfg.Faraday.MacaroonPath
-		}
-
-	case subservers.LOOP:
-		if p.cfg.loopRemote {
-			macPath = p.cfg.Remote.Loop.MacaroonPath
-		} else {
-			macPath = p.cfg.Loop.MacaroonPath
-		}
+	switch {
+	case handled:
 
-	case subservers.POOL:
-		if p.cfg.poolRemote {
-			macPath = p.cfg.Remote.Pool.MacaroonPath
-		} else {
-			macPath = p.cfg.Pool.MacaroonPath
-		}
+	case p.permsMgr.IsSubServerURI(subservers.LND, requestURI):
+		_, _, _, macPath, macData = p.cfg.lndConnectParams()
 
-	case subservers.LIT:
+	case p.permsMgr.IsSubServerURI(subservers.LIT, requestURI):
 		macPath = p.cfg.MacaroonPath
 
 	default:
-		return nil, fmt.Errorf("unknown subserver handler: %v",
-			subserver)
+		return nil, fmt.Errorf("unknown gRPC web request: %v",
+			requestURI)
 	}
 
 	switch {

subservers/manager.go

@@ -189,6 +189,27 @@ func (s *Manager) ValidateMacaroon(ctx context.Context,
 	return false, nil
 }
 
+// MacaroonPath checks if any of the manager's sub-servers owns the given uri
+// and if so, the appropriate macaroon path is returned for that sub-server.
+func (s *Manager) MacaroonPath(uri string) (bool, string) {
+	s.mu.RLock()
+	defer s.mu.RUnlock()
+
+	for _, ss := range s.servers {
+		if !s.permsMgr.IsSubServerURI(ss.Name(), uri) {
+			continue
+		}
+
+		if ss.Remote() {
+			return true, ss.RemoteConfig().MacaroonPath
+		}
+
+		return true, ss.MacPath()
+	}
+
+	return false, ""
+}
+
 // Stop stops all the manager's sub-servers
 func (s *Manager) Stop() error {
 	var returnErr error