Function InsertOrUpdateRemote

cli/internal/dal/remote.go:161–218

(remoteInfo workspace.RemoteInfo)

}

func InsertOrUpdateRemote(remoteInfo workspace.RemoteInfo) (id int, err error) {

	var single *workspace.RemoteInfo

	if remoteInfo.ID > 0 {
		single, err = GetRemoteById(remoteInfo.ID)
		if err != nil {
			return id, err
		}
	} else {
		single, err = GetRemoteByHost(remoteInfo.Addr, remoteInfo.UserName)
		if err != nil {
			return id, err
		}
	}

	//1. init
	db := getDb()
	defer db.Close()

	passwordEncrypt := ""
	if len(remoteInfo.Password) > 0 {
		passwordEncrypt = aes4go.Encrypt(remoteInfo.Password, aesDecryptKey)
	}

	//2. insert or update
	if single != nil { //2.1. update
		stmt, err := db.Prepare(`update remote
		set r_addr=?, r_port=?, r_username=?, r_auth_type=?, r_password=?  
		where r_id=? or r_addr=?`)
		if err != nil {
			return id, err
		}
		_, err = stmt.Exec(remoteInfo.Addr, remoteInfo.SSHPort, remoteInfo.UserName, remoteInfo.AuthType, passwordEncrypt,
			remoteInfo.ID, remoteInfo.Addr)
		if err != nil {
			return -1, err
		}

		id = single.ID

	} else { //2.2. insert
		stmt, err := db.Prepare(`INSERT INTO remote(r_addr, r_port, r_username, r_auth_type, r_password)  
                                        values(?, ?, ?, ?, ?)`)
		if err != nil {
			return id, err
		}
		res, err := stmt.Exec(remoteInfo.Addr, remoteInfo.SSHPort, remoteInfo.UserName, remoteInfo.AuthType, passwordEncrypt)
		if err != nil {
			return id, err
		}
		id64, err := res.LastInsertId()
		id = int(id64)
		return id, err
	}

	return id, err
}