Source File
	smb.go

Belonging Package
	github.com/limanmys/render-engine/internal/bridge

package bridge

import (
	"errors"
	"io"
	"net"
	"os"
	"time"

	"github.com/hirochachacha/go-smb2"
)

// OpenSmbConnection creates a new smb connection session
func OpenSmbConnection(host, username, password string) (*smb2.Session, error) {
	dialer := net.Dialer{Timeout: time.Second * 5}
	conn, err := dialer.Dial("tcp", host+":445")
	if err != nil {
		return nil, err
	}

	d := &smb2.Dialer{
		Initiator: &smb2.NTLMInitiator{
			User:     username,
			Password: password,
		},
	}

	s, err := d.Dial(conn)
	if err != nil {
		return nil, err
	}
	return s, nil
}

// SmbPutFile uploads file from local path to remote path
func (s *Session) SmbPutFile(localPath, remotePath, remoteDisk string) error {
	if s.SMB == nil {
		return errors.New("smb connection does not exist")
	}

	fs, err := s.SMB.Mount(remoteDisk + "$")
	if err != nil {
		return err
	}
	defer fs.Umount()

	f, err := fs.Create(remotePath)
	if err != nil {
		return err
	}
	defer f.Close()

	srcFile, err := os.Open(localPath)
	if err != nil {
		return err
	}
	defer srcFile.Close()

	_, err = io.Copy(f, srcFile)
	if err != nil {
		return err
	}

	return nil
}

// SmbGetFile downloads file from remote path to local path
func (s *Session) SmbGetFile(localPath, remotePath, remoteDisk string) error {
	if s.SMB == nil {
		return errors.New("smb connection does not exist")
	}

	fs, err := s.SMB.Mount(remoteDisk + "$")
	if err != nil {
		return err
	}
	defer fs.Umount()

	f, err := fs.Open(remotePath)
	if err != nil {
		return err
	}
	defer f.Close()

	_, err = os.Stat(localPath)
	if os.IsNotExist(err) {
		_, err = os.Create(localPath)
		if err != nil {
			return err
		}
	}

	srcFile, err := os.OpenFile(localPath, os.O_APPEND|os.O_WRONLY, os.ModeAppend)
	if err != nil {
		return err
	}
	defer srcFile.Close()

	_, err = io.Copy(srcFile, f)
	if err != nil {
		return err
	}

	return nil
}

The pages are generated with Golds v0.6.7. (GOOS=linux GOARCH=amd64)
Golds is a Go 101 project developed by Tapir Liu.
PR and bug reports are welcome and can be submitted to the issue list.
Please follow @Go100and1 (reachable from the left QR code) to get the latest news of Golds.