mirror of https://github.com/Xhofe/alist
37 lines
714 B
Go
37 lines
714 B
Go
package ftp
|
|
|
|
import (
|
|
"github.com/Xhofe/alist/drivers/base"
|
|
"github.com/Xhofe/alist/model"
|
|
"github.com/jlaffaye/ftp"
|
|
)
|
|
|
|
var connMap map[string]*ftp.ServerConn
|
|
|
|
func (driver FTP) Login(account *model.Account) (*ftp.ServerConn, error) {
|
|
conn, ok := connMap[account.Name]
|
|
if ok {
|
|
_, err := conn.CurrentDir()
|
|
if err == nil {
|
|
return conn, nil
|
|
} else {
|
|
delete(connMap, account.Name)
|
|
}
|
|
}
|
|
conn, err := ftp.Connect(account.SiteUrl)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
err = conn.Login(account.Username, account.Password)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
connMap[account.Name] = conn
|
|
return conn, nil
|
|
}
|
|
|
|
func init() {
|
|
base.RegisterDriver(&FTP{})
|
|
connMap = make(map[string]*ftp.ServerConn)
|
|
}
|