chat-widget

Author: vickeykumar

src/Makefile

@@ -20,7 +20,7 @@ gotty:	asset ${TARGET}/main.go utils/*.go server/*.go webtty/*.go filebrowser/*.
 	#godep go build ${BUILD_OPTIONS}
 
 .PHONY: asset
-asset: bindata/static/js/gotty-bundle.js bindata/static/js/jsconsole.js bindata/static/index.html bindata/static/doc.html bindata/static/editblog.html bindata/static/about.html bindata/static/NewFile.html bindata/static/images bindata/static/css bindata/static/css/xterm.css bindata/static/meta
+asset: bindata/static/js/gotty-bundle.js bindata/static/js/chat-widget.js bindata/static/js/jsconsole.js bindata/static/index.html bindata/static/doc.html bindata/static/editblog.html bindata/static/about.html bindata/static/NewFile.html bindata/static/images bindata/static/css bindata/static/css/xterm.css bindata/static/meta
 	$(ROOT)/bin/go-bindata -prefix bindata -pkg server -ignore=\\.gitkeep -o server/asset.go bindata/...
 	GO111MODULE=off $(GOROOT)/bin/gofmt -w server/asset.go
 
@@ -108,6 +108,16 @@ jsconsole/build/static/js/jsconsole.js: jsconsole/node_modules/webpack $(JS_FILE
 bindata/static/js/jsconsole.js: jsconsole/build/static/js/jsconsole.js
 	cp jsconsole/build/static/js/jsconsole.js bindata/static/js/jsconsole.js
 
+CHAT_JS_FILES := $(wildcard resources/chat-widget/src/widget.*) $(wildcard resources/chat-widget/*.json) $(wildcard resources/chat-widget/*.js) resources/chat-widget/src/index.ts
+
+resources/chat-widget/dist/index.umd.js: $(CHAT_JS_FILES)
+	@echo "building chat-widget: "
+	cd resources/chat-widget && \
+	npm install
+
+bindata/static/js/chat-widget.js: resources/chat-widget/dist/index.umd.js
+	cp resources/chat-widget/dist/index.umd.js bindata/static/js/chat-widget.js
+
 deb: all
 	mkdir -p ${DEBIAN_ROOT}/gotty/usr/lib/systemd/system
 	mkdir -p ${DEBIAN_ROOT}/gotty/usr/local/bin
@@ -141,6 +151,9 @@ clean:
 cleanjs:
 	rm -rf js/node_modules
 	rm -rf jsconsole/node_modules
+	rm -rf resources/chat-widget/node_modules
+	rm -rf js/dist/*
+	rm -rf resources/chat-widget/dist/*
 	@echo ".... Clean Done"
 
 

src/cookie/cookie.go

@@ -13,6 +13,8 @@ import (
 )
 
 const MAX_CONN_PER_BROWSER = 1
+// this key will be used for encryption and decdryption of tokens 
+var SECRET_KEY []byte = encoder.GenerateLargePrime().Bytes()
 
 // handling for cookies and incrementing session cookie counter
 func IncrementCounterCookies(rw http.ResponseWriter, req *http.Request) {
@@ -68,8 +70,8 @@ func GetCounterCookieValue(req *http.Request) int {
 var session_store *sessions.CookieStore
 
 // initilize the cookiestore with secret stored in session DB
-func Init_SessionStore(secret string) {
-	session_store = sessions.NewCookieStore([]byte(secret))
+func Init_SessionStore(secret []byte) {
+	session_store = sessions.NewCookieStore(secret)
 }
 
 func Get_SessionStore() *sessions.CookieStore {
@@ -82,14 +84,15 @@ func init() {
     	panic(errors.New("uninitialized session_db_handle!!"))
     }
     secret , err := session_db_handle.Fetch([]byte(user.SESSION_KEY))
-    log.Println("secret: ", secret, err)
+    log.Println("secret fetched: ", err)
     if err != nil {
 	// failed to fetch secret, generate a temporary secret for this instance
     	secret = encoder.GenerateLargePrime().Bytes()
-    	log.Println("Failed to fetch secret for session_cookie, generated temporary secret: ", secret, err)
+    	log.Println("Failed to fetch secret for session_cookie, generated temporary secret. ", err)
     }
+    SECRET_KEY = secret
     // init one time session store using SESSION_KEY
-    Init_SessionStore(string(secret))
+    Init_SessionStore(secret)
 
 }
 
@@ -107,14 +110,23 @@ func Set_SessionCookie(rw http.ResponseWriter, req *http.Request, session user.U
 		session_cookie.Values ["expirationTime"] = session.ExpirationTime
 		session_cookie.Values [utils.HOME_DIR_KEY] = user.GetHomeDir(session.Uid)
 
+		// number of api request remaining in seconds
+		var req_count_rem float64 = float64(session.ExpirationTime-utils.GetUnixMilli())/1000
 		// set maxage of the session
 		session_cookie.Options = &sessions.Options{
 			Path:     "/",
-			MaxAge:   int(session.ExpirationTime-utils.GetUnixMilli())/1000,
+			MaxAge:   int(req_count_rem),
 		}
 
-		log.Println("session cookie save: ", int(session.ExpirationTime-utils.GetUnixMilli())/1000, session_cookie)
+		if session.LoggedIn {
+			session_cookie.Values[utils.OPENAI_REQUEST_COUNT_KEY] = (req_count_rem/60)*utils.USER_FACTOR
+		} else {
+			session_cookie.Values[utils.OPENAI_REQUEST_COUNT_KEY] = (req_count_rem/60)*utils.GUEST_FACTOR
+		}
+
+		log.Println("session cookie save: ", int(session.ExpirationTime-utils.GetUnixMilli())/1000)
 		return session_store.Save(req, rw, session_cookie)
+		
 }
 
 
@@ -127,6 +139,8 @@ func Delete_SessionCookie(rw http.ResponseWriter, req *http.Request, session use
 		session_cookie.Values["uid"] = session.Uid
 		session_cookie.Values["sessionID"] = session.SessionID
 		session_cookie.Values ["loggedIn"] = false
+		delete(session_cookie.Values, utils.OPENAI_REQUEST_COUNT_KEY)
+		delete(session_cookie.Values, utils.OPENAI_REQUEST_LAST_ACCESS)
 
 		// set maxage of the session
 		session_cookie.Options = &sessions.Options{
@@ -187,6 +201,78 @@ func IsSessionExpired(req *http.Request) bool {
 	return false
 }
 
+func GetOpenApiRequestCount(req *http.Request) (count float64) {
+	session_cookie, _ := session_store.Get(req, "user-session")
+	val := session_cookie.Values[utils.OPENAI_REQUEST_COUNT_KEY]
+	count, ok := val.(float64);
+	if !ok {
+		return float64(0)
+	}
+	return count
+}
+
+func SetOpenApiRequestCount(rw http.ResponseWriter, req *http.Request, count float64) (err error) {
+	session_cookie, _ := session_store.Get(req, "user-session")
+	session_cookie.Values[utils.OPENAI_REQUEST_COUNT_KEY] = count
+	return session_store.Save(req, rw, session_cookie)
+}
+
+func GetOpenApiLastAccessTime(req *http.Request) (lastaccesstime int64) {
+	session_cookie, _ := session_store.Get(req, "user-session")
+	val := session_cookie.Values[utils.OPENAI_REQUEST_LAST_ACCESS]
+	lastaccesstime, ok := val.(int64);
+	if !ok {
+		return utils.GetUnixMilli() - utils.DEADLINE_MINUTES*60*1000
+	}
+	return lastaccesstime
+}
+
+/* 
+   to be called whenever openAIapi request is made, to recharge the request count
+   based on current time lapsed, it should eb able to give n number of requests per minute,
+   as given by user or GUEST_FACTOR
+*/
+func UpdateOpenApiRequestCountBalance(rw http.ResponseWriter, req *http.Request) (err error) {
+	session_cookie, _ := session_store.Get(req, "user-session")
+	// recharge req_count balance in sec
+	current_time_mili := utils.GetUnixMilli()
+	var req_count_balance_sec float64 = float64(current_time_mili-GetOpenApiLastAccessTime(req))/1000
+
+	var req_count_balance float64 = 0
+	var max_cap float64 = utils.GUEST_FACTOR*utils.DEADLINE_MINUTES // max num of request per minute a user can make
+	if Is_UserLoggedIn(req) {
+		req_count_balance = GetOpenApiRequestCount(req)+(req_count_balance_sec/60)*utils.USER_FACTOR
+		max_cap = utils.USER_FACTOR*utils.DEADLINE_MINUTES
+	} else {
+		req_count_balance = GetOpenApiRequestCount(req)+(req_count_balance_sec/60)*utils.GUEST_FACTOR
+		max_cap = utils.GUEST_FACTOR*utils.DEADLINE_MINUTES
+	}
+	if max_cap >= req_count_balance {
+		session_cookie.Values[utils.OPENAI_REQUEST_COUNT_KEY] = req_count_balance
+	} else {
+		session_cookie.Values[utils.OPENAI_REQUEST_COUNT_KEY] = max_cap
+	}
+	// not exceeding request balance more that maxcap req per user
+	session_cookie.Values[utils.OPENAI_REQUEST_LAST_ACCESS] = current_time_mili
+	return session_store.Save(req, rw, session_cookie)
+}
+
+func GetOpenApiAccessToken(req *http.Request) (acc_token, secret []byte) {
+	session_cookie, _ := session_store.Get(req, "user-session")
+	val := session_cookie.Values[utils.ACCESS_TOKEN_KEY]
+	secretval := session_cookie.Values[utils.ACCESS_SECRET_KEY]
+	acc_token, _ = val.([]byte);
+	secret, _ = secretval.([]byte);
+	return acc_token, secret
+}
+
+func SetOpenApiAccessToken(rw http.ResponseWriter, req *http.Request, acc_token, secret []byte) (err error) {
+	session_cookie, _ := session_store.Get(req, "user-session")
+	session_cookie.Values[utils.ACCESS_TOKEN_KEY] = acc_token
+	session_cookie.Values[utils.ACCESS_SECRET_KEY] = secret
+	return session_store.Save(req, rw, session_cookie)
+}
+
 func Get_SessionCookie(req *http.Request) (session user.UserSession) {
 		session.Uid = Get_Uid(req)
 		session.SessionID = Get_SessionID(req)
@@ -205,6 +291,7 @@ func UpdateGuestSessionCookieAge(rw http.ResponseWriter, req *http.Request, newa
 
 	if !Is_UserLoggedIn(req) || IsSessionExpired(req) {
 		// only update age if user not logged in
+
 		maxage = newage
 	}
 	// update maxage (sec) of the session 
@@ -213,7 +300,7 @@ func UpdateGuestSessionCookieAge(rw http.ResponseWriter, req *http.Request, newa
 		MaxAge:   maxage,
 	}
 	err = session_store.Save(req, rw, session_cookie)
-	log.Println("session cookie save: ", maxage, session_cookie, "Error: ", err)
+	log.Println("session cookie save: ", maxage, "Error: ", err)
 	return err
 }
 
@@ -244,7 +331,7 @@ func GetOrUpdateHomeDir(rw http.ResponseWriter, req *http.Request, Uid string) (
 		var ok bool
 
 		homedir, ok = session_cookie.Values[utils.HOME_DIR_KEY].(string);
-		log.Println("previous homedir: ", homedir, ok, session_cookie)
+		log.Println("previous homedir: ", homedir, ok)
 		// check if same uid amd valid home dir, if not generate a new homedir
 		if Uid!=Get_Uid(req) || !ok || homedir=="" {
 			homedir = user.GetHomeDir(Uid)

src/encoder/encode.go

@@ -3,18 +3,25 @@ package encoder
 import (
 	"encoding/base64"
 	"log"
-	"crypto/rand"
-    	"math/big"
+	"crypto/aes"
+    "crypto/cipher"
+    "crypto/rand"
+    "math/big"
+    "io"
+    "fmt"
 )
 
-// use 32 bits for type int
-const SECRET_BITSIZE = 16
+// use 16 byte, default 128 bit for AES
+const SECRET_BITSIZE = 16*8
 
-func GenerateLargePrime() *big.Int {
-     // generate large prime
+func GenerateLargePrime(bits ...int) *big.Int {
+     var bitsize int = SECRET_BITSIZE
+     if len(bits) > 0 {
+        bitsize = bits[0]
+     }
 
     // Generate a random prime number
-    prime, err := rand.Prime(rand.Reader, SECRET_BITSIZE)
+    prime, err := rand.Prime(rand.Reader, bitsize)
     if err != nil {
         log.Println("Error:", err)
         // any prime number, probably we won't reach here
@@ -49,4 +56,56 @@ func DecodeToPID(jid string) int {
         return -1
 }
 
+// Encrypt encrypts the plaintext using AES-GCM with the given key and returns a URL-safe base64-encoded ciphertext.
+func Encrypt(plainTextBytes, keyBytes []byte) ([]byte, error) {
 
+    block, err := aes.NewCipher(keyBytes)
+    if err != nil {
+        return []byte(""), err
+    }
+
+    aesGCM, err := cipher.NewGCM(block)
+    if err != nil {
+        return []byte(""), err
+    }
+
+    nonce := make([]byte, aesGCM.NonceSize())
+    if _, err := io.ReadFull(rand.Reader, nonce); err != nil {
+        return []byte(""), err
+    }
+
+    cipherText := aesGCM.Seal(nonce, nonce, plainTextBytes, nil)
+    urlSafeCipherText := RawURLEncoding.EncodeToString(cipherText)
+    return []byte(urlSafeCipherText), nil
+}
+
+// Decrypt decrypts the URL-safe base64-encoded ciphertext using AES-GCM with the given key.
+func Decrypt(cipherText, keyBytes []byte) ([]byte, error) {
+    cipherTextBytes, err := RawURLEncoding.DecodeString(string(cipherText))
+    if err != nil {
+        return []byte(""), err
+    }
+
+    block, err := aes.NewCipher(keyBytes)
+    if err != nil {
+        return []byte(""), err
+    }
+
+    aesGCM, err := cipher.NewGCM(block)
+    if err != nil {
+        return []byte(""), err
+    }
+
+    nonceSize := aesGCM.NonceSize()
+    if len(cipherTextBytes) < nonceSize {
+        return []byte(""), fmt.Errorf("ciphertext too short")
+    }
+
+    nonce, cipherTextBytes := cipherTextBytes[:nonceSize], cipherTextBytes[nonceSize:]
+    plainTextBytes, err := aesGCM.Open(nil, nonce, cipherTextBytes, nil)
+    if err != nil {
+        return []byte(""), err
+    }
+
+    return plainTextBytes, nil
+}

src/js/src/firetty.ts

@@ -1,4 +1,5 @@
-import * as firebase from 'firebase';
+import * as firebase from 'firebase/app';
+import 'firebase/database';
 import { Terminal, eventHandler, eventhandlertype, CloserArgs} from "./webtty";
 
 

src/resources/chat-widget/.gitignore

@@ -0,0 +1,2 @@
+dist/
+node_modules/

src/resources/chat-widget/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Rowy
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.