router.go 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899
  1. package dynamicproxy
  2. import (
  3. "errors"
  4. "log"
  5. "net/url"
  6. "strings"
  7. "imuslab.com/zoraxy/mod/dynamicproxy/dpcore"
  8. )
  9. /*
  10. Dynamic Proxy Router Functions
  11. This script handle the proxy rules router spawning
  12. and preparation
  13. */
  14. // Prepare proxy route generate a proxy handler service object for your endpoint
  15. func (router *Router) PrepareProxyRoute(endpoint *ProxyEndpoint) (*ProxyEndpoint, error) {
  16. for _, thisOrigin := range endpoint.ActiveOrigins {
  17. //Create the proxy routing handler
  18. err := thisOrigin.StartProxy()
  19. if err != nil {
  20. log.Println("Unable to setup upstream " + thisOrigin.OriginIpOrDomain + ": " + err.Error())
  21. continue
  22. }
  23. }
  24. endpoint.parent = router
  25. //Prepare proxy routing handler for each of the virtual directories
  26. for _, vdir := range endpoint.VirtualDirectories {
  27. domain := vdir.Domain
  28. if len(domain) == 0 {
  29. //invalid vdir
  30. continue
  31. }
  32. if domain[len(domain)-1:] == "/" {
  33. domain = domain[:len(domain)-1]
  34. }
  35. //Parse the web proxy endpoint
  36. webProxyEndpoint := domain
  37. if !strings.HasPrefix("http://", domain) && !strings.HasPrefix("https://", domain) {
  38. //TLS is not hardcoded in proxy target domain
  39. if vdir.RequireTLS {
  40. webProxyEndpoint = "https://" + webProxyEndpoint
  41. } else {
  42. webProxyEndpoint = "http://" + webProxyEndpoint
  43. }
  44. }
  45. path, err := url.Parse(webProxyEndpoint)
  46. if err != nil {
  47. return nil, err
  48. }
  49. proxy := dpcore.NewDynamicProxyCore(path, vdir.MatchingPath, &dpcore.DpcoreOptions{
  50. IgnoreTLSVerification: vdir.SkipCertValidations,
  51. })
  52. vdir.proxy = proxy
  53. vdir.parent = endpoint
  54. }
  55. return endpoint, nil
  56. }
  57. // Add Proxy Route to current runtime. Call to PrepareProxyRoute before adding to runtime
  58. func (router *Router) AddProxyRouteToRuntime(endpoint *ProxyEndpoint) error {
  59. if !router.loadBalancer.UpstreamsReady(endpoint.ActiveOrigins) {
  60. //This endpoint is not prepared
  61. return errors.New("proxy endpoint not ready. Use PrepareProxyRoute before adding to runtime")
  62. }
  63. // Push record into running subdomain endpoints
  64. router.ProxyEndpoints.Store(endpoint.RootOrMatchingDomain, endpoint)
  65. return nil
  66. }
  67. // Set given Proxy Route as Root. Call to PrepareProxyRoute before adding to runtime
  68. func (router *Router) SetProxyRouteAsRoot(endpoint *ProxyEndpoint) error {
  69. if !router.loadBalancer.UpstreamsReady(endpoint.ActiveOrigins) {
  70. //This endpoint is not prepared
  71. return errors.New("proxy endpoint not ready. Use PrepareProxyRoute before adding to runtime")
  72. }
  73. // Push record into running root endpoints
  74. router.Root = endpoint
  75. return nil
  76. }
  77. // ProxyEndpoint remove provide global access by key
  78. func (router *Router) RemoveProxyEndpointByRootname(rootnameOrMatchingDomain string) error {
  79. targetEpt, err := router.LoadProxy(rootnameOrMatchingDomain)
  80. if err != nil {
  81. return err
  82. }
  83. return targetEpt.Remove()
  84. }