From 3d156a8435077973032e61263ebe0fe833a35037 Mon Sep 17 00:00:00 2001
From: Giuseppe Lo Presti <giuseppe.lopresti@cern.ch>
Date: Mon, 28 Jun 2021 15:51:08 +0200
Subject: [PATCH] Further work to be squashed to previous commit

---
 pkg/app/provider/wopi/wopi.go | 69 ++++++++++++++++++++++++++++-------
 1 file changed, 55 insertions(+), 14 deletions(-)

diff --git a/pkg/app/provider/wopi/wopi.go b/pkg/app/provider/wopi/wopi.go
index a2fc7eabbe1..df8fb01884a 100644
--- a/pkg/app/provider/wopi/wopi.go
+++ b/pkg/app/provider/wopi/wopi.go
@@ -46,11 +46,10 @@ func init() {
 type config struct {
 	IOPSecret    string `mapstructure:"iop_secret" docs:";The IOP secret used to connect to the wopiserver."`
 	WopiURL      string `mapstructure:"wopi_url" docs:";The wopiserver's URL."`
-	MSOOURL      string `mapstructure:"msoo_url" docs:";The MS Office Online URL."`
-	CodeURL      string `mapstructure:"code_url" docs:";The Collabora Online URL."`
-	CodiMDURL    string `mapstructure:"codimd_url" docs:";The CodiMD URL."`
-	CodiMDIntURL string `mapstructure:"codimd_int_url" docs:";The CodiMD internal URL."`
-	CodiMDApiKey string `mapstructure:"codimd_url" docs:";The CodiMD URL."`
+	AppName      string `mapstructure:"app_name" docs:";The App user-friendly name."`
+	AppURL       string `mapstructure:"app_url" docs:";The App URL."`
+	AppIntURL    string `mapstructure:"app_int_url" docs:";The App internal URL in case of dockerized deployments. Defaults to AppURL"`
+	AppApiKey    string `mapstructure:"app_apikey" docs:";The API key used by the App, if applicable."`
 }
 
 func parseConfig(m map[string]interface{}) (*config, error) {
@@ -66,7 +65,7 @@ type wopiProvider struct {
 	wopiClient *http.Client
 }
 
-func (p *wopiProvider) GetAppURL(ctx context.Context, resource *provider.ResourceInfo, viewMode appprovider.OpenInAppRequest_ViewMode, app, token string) (string, error) {
+func (p *wopiProvider) GetAppURL(ctx context.Context, resource *provider.ResourceInfo, viewMode appprovider.OpenInAppRequest_ViewMode, app string, token string) (string, error) {
 	log := appctx.GetLogger(ctx)
 
 	wopiurl, err := url.Parse(p.conf.WopiURL)
@@ -99,15 +98,14 @@ func (p *wopiProvider) GetAppURL(ctx context.Context, resource *provider.Resourc
 	q.Add("appname", app)
 	if app == "CodiMD" {
 		// This is served by the WOPI bridge extensions
-		q.Add("appediturl", p.conf.CodiMDURL)
-		if p.conf.CodiMDIntURL != "" {
-			q.Add("appinturl", p.conf.CodiMDIntURL)
+		q.Add("appediturl", p.conf.AppURL)
+		if p.conf.AppIntURL != "" {
+			q.Add("appinturl", p.conf.AppIntURL)
 		}
-		httpReq.Header.Set("ApiKey", p.conf.CodiMDApiKey)
+		httpReq.Header.Set("ApiKey", p.conf.AppApiKey)
 	} else {
-		// TODO get AppRegistry
-		//q.Add("appediturl", AppRegistry.get(app).getEditUrl())
-		//q.Add("appviewurl", AppRegistry.get(app).getViewUrl())
+		q.Add("appediturl", p.conf.AppEditURL)
+		q.Add("appviewurl", p.conf.AppViewURL)
 	}
 
 	if p.conf.IOPSecret == "" {
@@ -135,12 +133,55 @@ func (p *wopiProvider) GetAppURL(ctx context.Context, resource *provider.Resourc
 
 // New returns an implementation of the app.Provider interface that
 // connects to an application in the backend.
-func New(m map[string]interface{}) (app.Provider, error) {
+func New(m map[string]interface{})() (app.Provider, error) {
 	c, err := parseConfig(m)
 	if err != nil {
 		return nil, err
 	}
 
+	// Initialize WOPI URLs by discovery
+	httpcl := rhttp.GetHTTPClient(
+		rhttp.Timeout(time.Duration(5 * int64(time.Second))),
+	)
+	if c.AppIntURL == nil {
+		c.AppIntURL = c.AppURL
+	}
+	appurl, err := url.Parse(c.AppIntURL)
+	if err != nil {
+		return nil, err
+	}
+	appurl.Path = path.Join(c.AppIntURL.Path, "/hosting/discovery")
+	discReq, err := rhttp.NewRequest(ctx, "GET", appurl, nil)
+	if err != nil {
+		return nil, err
+	}
+	discRes, err := httpcl.Do(discReq)
+	if err != nil {
+		return nil, err
+	}
+	defer discRes.Body.Close()
+	if discRes.StatusCode == http.StatusNotFound) {
+		// this may be a bridge-supported app, let's find out
+		discReq, err := rhttp.NewRequest(ctx, "GET", c.AppIntURL, nil)
+		if err != nil {
+			return nil, err
+		}
+		discRes, err := httpcl.Do(discReq)
+		if err != nil {
+			return nil, err
+		}
+		defer discRes.Body.Close()
+		// scrape app's home page to find the appname
+		if !discRes.Body.Find(c.AppName) {
+		// || (c.AppName != "CodiMD" && c.AppName != "Etherpad") {
+			return nil, errors.New("Application server at " + c.AppURL + " is not supported by the WOPI AppProvider")
+		}
+	} else {
+		// TODO parse XML from discRes.Body
+		c.AppEditURL := "bla"
+		c.AppViewURL := "bla"
+	}
+
 	wopiClient := rhttp.GetHTTPClient(
 		rhttp.Timeout(time.Duration(5 * int64(time.Second))),
 	)