Refactor and add MediaDownloadSize to General

1 files changed, 5 insertions, 11 deletions

diff --git a/bridge/sshchat/sshchat.go b/bridge/sshchat/sshchat.go
index becb688c..81e83244 100644
--- a/bridge/sshchat/sshchat.go
+++ b/bridge/sshchat/sshchat.go
@@ -10,11 +10,9 @@ import (
 )
 
 type Bsshchat struct {
-	r       *bufio.Scanner
-	w       io.WriteCloser
-	Config  *config.Protocol
-	Remote  chan config.Message
-	Account string
+	r *bufio.Scanner
+	w io.WriteCloser
+	*config.BridgeConfig
 }
 
 var flog *log.Entry
@@ -24,12 +22,8 @@ func init() {
 	flog = log.WithFields(log.Fields{"module": protocol})
 }
 
-func New(cfg config.Protocol, account string, c chan config.Message) *Bsshchat {
-	b := &Bsshchat{}
-	b.Config = &cfg
-	b.Account = account
-	b.Remote = c
-	return b
+func New(cfg *config.BridgeConfig) *Bsshchat {
+	return &Bsshchat{BridgeConfig: cfg}
 }
 
 func (b *Bsshchat) Connect() error {