morpheus.meta

55.diff (6732B)

---

 diff --git a/main.go b/main.go
 index 9f05cfe..cb2706c 100644
 --- a/main.go
 +++ b/main.go
 @@ -8,6 +8,7 @@ import (
  
  	"github.com/Nordgedanken/Morpheus/matrix"
  	"github.com/Nordgedanken/Morpheus/matrix/db"
 +	"github.com/Nordgedanken/Morpheus/matrix/scalar"
  	"github.com/Nordgedanken/Morpheus/ui"
  	"github.com/Nordgedanken/dugong"
  	"github.com/matrix-org/gomatrix"
 @@ -118,6 +119,7 @@ func main() {
  				log.Errorln("mainUI: ", mainUIErr)
  				return
  			}
 +			scalar.ReqAndSaveScalarToken(mainUIStruct.GetCli())
  
  			mainUIStruct.GetWidget().Resize2(windowWidth, windowHeight)
  			window.SetCentralWidget(mainUIStruct.GetWidget())
 diff --git a/matrix/scalar/client.go b/matrix/scalar/client.go
 new file mode 100644
 index 0000000..e6c4810
 --- /dev/null
 +++ b/matrix/scalar/client.go
 @@ -0,0 +1,127 @@
 +package scalar
 +
 +import (
 +	"github.com/Nordgedanken/Morpheus/matrix/db"
 +	"github.com/dgraph-io/badger"
 +	"github.com/matrix-org/gomatrix"
 +	log "github.com/sirupsen/logrus"
 +)
 +
 +type openIDTokenResponse struct {
 +	AccessToken      string `json:"access_token"`
 +	TokenType        string `json:"token_type"`
 +	ExpiresIn        int    `json:"expires_in"`
 +	MatrixServerName string `json:"matrix_server_name"`
 +}
 +
 +func getOpenIDToken(cli *gomatrix.Client) (resp *openIDTokenResponse, err error) {
 +	urlPath := cli.BuildURL("user", cli.UserID, "openid", "request_token")
 +	_, err = makeRequest(cli, "POST", urlPath, "{}", &resp)
 +	return
 +}
 +
 +type scalarRegisterResp struct {
 +	ScalarToken string `json:"scalar_token"`
 +}
 +
 +//getScalarToken returns the scalar token for the current user. Used inside the Integration manager.
 +func getScalarToken(cli *gomatrix.Client, openIDToken *openIDTokenResponse) (resp *scalarRegisterResp, err error) {
 +	urlPath := "https://scalar.vector.im/api/register"
 +	_, err = cli.MakeRequest("POST", urlPath, openIDToken, &resp)
 +	return
 +}
 +
 +func ReqAndSaveScalarToken(cli *gomatrix.Client) (err error) {
 +	cacheDB, DBOpenErr := db.OpenCacheDB()
 +	if DBOpenErr != nil {
 +		log.Fatalln(DBOpenErr)
 +	}
 +
 +	var integToken string
 +
 +	// Get cache
 +	DBErr := cacheDB.View(func(txn *badger.Txn) error {
 +		roomAvatarDataItem, QueryErr := txn.Get([]byte("user|" + cli.UserID + "|integToken"))
 +		if QueryErr != nil && QueryErr != badger.ErrKeyNotFound {
 +			return QueryErr
 +		}
 +		if QueryErr != badger.ErrKeyNotFound {
 +			integToken = roomAvatarDataItem.ToString()
 +			return nil
 +		}
 +		return nil
 +	})
 +	if DBErr != nil {
 +		err = DBErr
 +		return
 +	}
 +
 +	if integToken != "" {
 +		openIdToken, OpenIDerr := getOpenIDToken(cli)
 +		if OpenIDerr != nil {
 +			err = OpenIDerr
 +			return
 +		}
 +		var ScalarRegisterErr error
 +		scalarToken, ScalarRegisterErr := getScalarToken(cli, openIdToken)
 +		if ScalarRegisterErr != nil {
 +			err = ScalarRegisterErr
 +		}
 +		integToken = scalarToken.ScalarToken
 +
 +		// Update cache
 +		DBSetErr := cacheDB.Update(func(txn *badger.Txn) error {
 +			DBSetErr := txn.Set([]byte("user|"+cli.UserID+"|integToken"), []byte(integToken))
 +			return DBSetErr
 +		})
 +		if DBSetErr != nil {
 +			err = DBSetErr
 +			return
 +		}
 +	}
 +
 +	return
 +}
 +
 +func GetScalarURL(cli *gomatrix.Client) (err error, urlPath string) {
 +	cacheDB, DBOpenErr := db.OpenCacheDB()
 +	if DBOpenErr != nil {
 +		log.Fatalln(DBOpenErr)
 +	}
 +
 +	var integToken string
 +
 +	// Get cache
 +	DBErr := cacheDB.View(func(txn *badger.Txn) error {
 +		roomAvatarDataItem, QueryErr := txn.Get([]byte("user|" + cli.UserID + "|integToken"))
 +		if QueryErr != nil && QueryErr != badger.ErrKeyNotFound {
 +			return QueryErr
 +		}
 +		if QueryErr != badger.ErrKeyNotFound {
 +			integToken = roomAvatarDataItem.ToString()
 +			return nil
 +		}
 +		return nil
 +	})
 +	if DBErr != nil {
 +		err = DBErr
 +		return
 +	}
 +
 +	urlPath = "https://scalar.vector.im/?access_token=" + integToken
 +
 +	return
 +}
 +
 +// How to listen for Scalar events:
 +/*
 +webframe := webkit.NewQWebView(nil)
 +webframe.Load(core.NewQUrl3(GetScalarURL(cli), core.QUrl__TolerantMode))
 +webframe.ConnectLoadFinished(func(ok bool) {
 +	javascript := "window.addEventListener(\"message\", qtJS-onMessage, false);"
 +
 +	mainFrame := webframe.Page().MainFrame()
 +// Replace second argument with actual QObject
 +	mainFrame.AddToJavaScriptWindowObject("qtJS", nil, webkit.QWebFrame__AutoOwnership)
 +	mainFrame.EvaluateJavaScript(javascript)
 +})*/
 diff --git a/matrix/scalar/utils.go b/matrix/scalar/utils.go
 new file mode 100644
 index 0000000..ab238c3
 --- /dev/null
 +++ b/matrix/scalar/utils.go
 @@ -0,0 +1,66 @@
 +package scalar
 +
 +import (
 +	"bytes"
 +	"encoding/json"
 +	"io/ioutil"
 +	"net/http"
 +
 +	"github.com/matrix-org/gomatrix"
 +)
 +
 +//makeRequest does the same as cli.MakeRequest but with a string as body instead of a struct
 +func makeRequest(cli *gomatrix.Client, method string, httpURL string, reqBody string, resBody interface{}) ([]byte, error) {
 +	var req *http.Request
 +	var err error
 +	if reqBody != "" {
 +		byteString := []byte(reqBody)
 +		req, err = http.NewRequest(method, httpURL, bytes.NewBuffer(byteString))
 +	} else {
 +		req, err = http.NewRequest(method, httpURL, nil)
 +	}
 +
 +	if err != nil {
 +		return nil, err
 +	}
 +	req.Header.Set("Content-Type", "application/json")
 +	res, err := cli.Client.Do(req)
 +	if res != nil {
 +		defer res.Body.Close()
 +	}
 +	if err != nil {
 +		return nil, err
 +	}
 +	contents, err := ioutil.ReadAll(res.Body)
 +	if res.StatusCode/100 != 2 { // not 2xx
 +		var wrap error
 +		var respErr gomatrix.RespError
 +		if _ = json.Unmarshal(contents, &respErr); respErr.ErrCode != "" {
 +			wrap = respErr
 +		}
 +
 +		// If we failed to decode as RespError, don't just drop the HTTP body, include it in the
 +		// HTTP error instead (e.g proxy errors which return HTML).
 +		msg := "Failed to " + method + " JSON to " + req.URL.Path
 +		if wrap == nil {
 +			msg = msg + ": " + string(contents)
 +		}
 +
 +		return contents, gomatrix.HTTPError{
 +			Code:         res.StatusCode,
 +			Message:      msg,
 +			WrappedError: wrap,
 +		}
 +	}
 +	if err != nil {
 +		return nil, err
 +	}
 +
 +	if resBody != nil {
 +		if err = json.Unmarshal(contents, &resBody); err != nil {
 +			return nil, err
 +		}
 +	}
 +
 +	return contents, nil
 +}
 diff --git a/ui/LoginUI.go b/ui/LoginUI.go
 index 7a77a04..fddb138 100644
 --- a/ui/LoginUI.go
 +++ b/ui/LoginUI.go
 @@ -4,6 +4,7 @@ import (
  	"sync"
  
  	"github.com/Nordgedanken/Morpheus/matrix"
 +	"github.com/Nordgedanken/Morpheus/matrix/scalar"
  	"github.com/Nordgedanken/Morpheus/matrix/globalTypes"
  	"github.com/matrix-org/gomatrix"
  	log "github.com/sirupsen/logrus"
 @@ -173,6 +174,9 @@ func (l *LoginUI) login() (err error) {
  				err = mainUIErr
  				return
  			}
 +
 +			scalar.ReqAndSaveScalarToken(MainUIStruct.GetCli())
 +
  			l.window.SetCentralWidget(MainUIStruct.GetWidget())
  			l.window.Resize(l.widget.Size())
  		}