cangui/golang_pvnet
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

remove simply -> go github repo

Browse Source
This commit is contained in:
cangui 2024-12-13 19:33:06 +01:00
parent 0adfd9e37a
commit f73006cb93
4 changed files with 5 additions and 271 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
go.mod
View File

@ -1,6 +1,6 @@
module canguidev.fr/pvnet module canguidev.fr/pvnet
go 1.23.2 go 1.23.4
require github.com/TwiN/go-choice v1.2.0 require github.com/TwiN/go-choice v1.2.0
@ -19,6 +19,7 @@ require (
) )
require ( require (
github.com/cangui2/simplyssh v0.1.0
github.com/cpuguy83/go-md2man/v2 v2.0.5 // indirect github.com/cpuguy83/go-md2man/v2 v2.0.5 // indirect
github.com/fatih/color v1.18.0 github.com/fatih/color v1.18.0
github.com/gdamore/encoding v1.0.1 // indirect github.com/gdamore/encoding v1.0.1 // indirect

2
go.sum
View File

@ -1,6 +1,8 @@
github.com/BurntSushi/toml v1.4.0/go.mod h1:ukJfTF/6rtPPRCnwkur4qwRxa8vTRFBF0uk2lLoLwho= github.com/BurntSushi/toml v1.4.0/go.mod h1:ukJfTF/6rtPPRCnwkur4qwRxa8vTRFBF0uk2lLoLwho=
github.com/TwiN/go-choice v1.2.0 h1:hMEJ09UPLwuowHhfXpooBNsIiV7siPfanjU76buni/Y= github.com/TwiN/go-choice v1.2.0 h1:hMEJ09UPLwuowHhfXpooBNsIiV7siPfanjU76buni/Y=
github.com/TwiN/go-choice v1.2.0/go.mod h1:LlMvhuqgWfGSdqUN3z3sDlDpSNNjjBpve+CBIC+2Nlg= github.com/TwiN/go-choice v1.2.0/go.mod h1:LlMvhuqgWfGSdqUN3z3sDlDpSNNjjBpve+CBIC+2Nlg=
github.com/cangui2/simplyssh v0.1.0 h1:AMNao+vSIDcc0dt/W5VoskDx0sKByo5aev7nwPm7zKM=
github.com/cangui2/simplyssh v0.1.0/go.mod h1:Cs4yhDqGfIk5JyDiNbm4kg2PGieJ9ncejQ6JIzld1uE=
github.com/cpuguy83/go-md2man/v2 v2.0.5 h1:ZtcqGrnekaHpVLArFSe4HK5DoKx1T0rq2DwVB0alcyc= github.com/cpuguy83/go-md2man/v2 v2.0.5 h1:ZtcqGrnekaHpVLArFSe4HK5DoKx1T0rq2DwVB0alcyc=
github.com/cpuguy83/go-md2man/v2 v2.0.5/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o= github.com/cpuguy83/go-md2man/v2 v2.0.5/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=

2
main.go
View File

@ -8,7 +8,7 @@ import (
"strings" "strings"
"time" "time"
"canguidev.fr/pvnet/function" "canguidev.fr/pvnet/function"
simplyssh "canguidev.fr/pvnet/simply-ssh" "github.com/cangui2/simplyssh"
"github.com/TwiN/go-choice" "github.com/TwiN/go-choice"
"github.com/joho/godotenv" "github.com/joho/godotenv"
"github.com/pkg/sftp" "github.com/pkg/sftp"

269
simply-ssh/main.go
View File

@ -1,269 +0,0 @@
package simplyssh
import (
"bytes"
"fmt"
"io"
"log"
"os"
"path/filepath"
"strings"
"time"
"github.com/pkg/sftp"
"golang.org/x/crypto/ssh"
)
type hostConnection struct {
Id uint
Host string
HostName string
Port string
User string
}
func GetHost(path string) []hostConnection {
fileData, err := os.ReadFile(path)
if err != nil {
fmt.Println("Error reading file:", err)
return nil
}
allHosts := []hostConnection{}
word := []byte{}
breakLine := "\n"
//chargement initiale de lobject
var currentHost hostConnection
var i int=0
for _, data := range fileData {
if !bytes.Equal([]byte{data}, []byte(breakLine)) {
word = append(word, data)
} else {
line := strings.TrimSpace(string(word))
if strings.HasPrefix(line, "Host ") {
// Si un nouveau Host est trouvé, ajouter l'ancien à la liste
if currentHost.Host != "" {
allHosts = append(allHosts, currentHost)
}
// Initialiser un nouveau hostConnection
currentHost = hostConnection{}
currentHost.Id=uint(i)
parts := strings.Fields(line)
if len(parts) > 1 {
currentHost.Host = parts[1]
}
} else if strings.HasPrefix(line, "HostName ") {
parts := strings.Fields(line)
if len(parts) > 1 {
currentHost.HostName = parts[1]
}
} else if strings.HasPrefix(line, "Port ") {
parts := strings.Fields(line)
if len(parts) > 1 {
currentHost.Port = parts[1]
}
} else if strings.HasPrefix(line, "User ") {
parts := strings.Fields(line)
if len(parts) > 1 {
currentHost.User = parts[1]
i++
}
}
word = word[:0]
}
}
// Ajouter le dernier host traité
if currentHost.Host != "" {
allHosts = append(allHosts, currentHost)
}
return allHosts
}
func Connect(pathKey, hostName, port, user string) *ssh.Client {
fmt.Println(pathKey)
sshConfig := sshConfig(user, pathKey)
connection, err := ssh.Dial("tcp", hostName+":"+port, sshConfig)
if err != nil {
log.Fatal("Failed to dial: %s", err)
return nil
}
fmt.Println("Connection ok ")
return connection
}
func sshConfig(usernamme, path string) *ssh.ClientConfig {
sshConfig := &ssh.ClientConfig{
User: usernamme,
Auth: []ssh.AuthMethod{
PublicKeyFile(path),
},
HostKeyCallback: ssh.InsecureIgnoreHostKey(), // Insecure, use for testing only
}
return sshConfig
}
func PublicKeyFile(file string) ssh.AuthMethod {
fmt.Println(file)
buffer, err := os.ReadFile(file)
if err != nil {
fmt.Errorf("Failed to buffer: %s", err)
return nil
}
key, err := ssh.ParsePrivateKey(buffer)
if err != nil {
fmt.Errorf("Failed to key: %s", err)
return nil
}
return ssh.PublicKeys(key)
}
type ProgressReader struct {
reader io.Reader
totalSize int64
bytesRead int64
lastUpdate time.Time
}
func NewProgressReader(r io.Reader, totalSize int64) *ProgressReader {
return &ProgressReader{
reader: r,
totalSize: totalSize,
}
}
func (pr *ProgressReader) Read(p []byte) (int, error) {
n, err := pr.reader.Read(p)
pr.bytesRead += int64(n)
// Mettre à jour la progression toutes les 500 ms par exemple
now := time.Now()
if now.Sub(pr.lastUpdate) > 500*time.Millisecond || err == io.EOF {
pr.lastUpdate = now
if pr.totalSize > 0 {
percent := float64(pr.bytesRead) / float64(pr.totalSize) * 100
fmt.Printf("\r%.2f%% téléchargé...", percent)
} else {
// Si on ne connaît pas la taille, on ne peut afficher que les octets
fmt.Printf("\r%v bytes téléchargés...", pr.bytesRead)
}
}
return n, err
}
func DownloadDirectory(sftpClient *sftp.Client, remoteDir, localDir string) error {
entries, err := sftpClient.ReadDir(remoteDir)
if err != nil {
return err
}
// Crée le répertoire local s’il n’existe pas
os.MkdirAll(localDir, 0755)
for _, entry := range entries {
remotePath := remoteDir + "/" + entry.Name()
localPath := localDir + "/" + entry.Name()
if entry.IsDir() {
// Téléchargement récursif
err = DownloadDirectory(sftpClient, remotePath, localPath)
if err != nil {
return err
}
} else {
// Téléchargement de fichier
remoteFile, err := sftpClient.Open(remotePath)
if err != nil {
return err
}
defer remoteFile.Close()
localFile, err := os.Create(localPath)
if err != nil {
return err
}
defer localFile.Close()
// Récupère la taille du fichier distant
info, err := remoteFile.Stat()
if err != nil {
return err
}
totalSize := info.Size()
// Crée un ProgressReader qui va afficher la progression
progressReader := NewProgressReader(remoteFile, totalSize)
_, err = io.Copy(localFile, progressReader)
if err != nil {
return err
}
fmt.Println("\nTéléchargement terminé:", localPath)
}
}
return nil
}
func DownloadFile(sftpClient *sftp.Client, remoteFilePath, localFilePath string) error {
// Ouvrir le fichier distant
remoteFile, err := sftpClient.Open(remoteFilePath)
if err != nil {
return err
}
defer remoteFile.Close()
// Créer le répertoire local si nécessaire
localDir := filepath.Dir(localFilePath)
if err := os.MkdirAll(localDir, 0755); err != nil {
return err
}
// Créer le fichier local ou ecrase le dossier
localFile, err := os.Create(localFilePath)
if err != nil {
return err
}
defer localFile.Close()
// Récupère la taille du fichier distant
info, err := remoteFile.Stat()
if err != nil {
return err
}
totalSize := info.Size()
// Crée un ProgressReader qui va afficher la progression
progressReader := NewProgressReader(remoteFile, totalSize)
// Copie du fichier avec affichage de la progression
_, err = io.Copy(localFile, progressReader)
if err != nil {
return err
}
fmt.Println("\nTéléchargement terminé:", localFilePath)
return nil
}
func ExecuteCommand(command string,session *ssh.Session){
// initialisation du buffer pour recupere les infos reçu de l' hôte
var b bytes.Buffer
session.Stdout = &b
if err := session.Run(command); err != nil {
fmt.Println(b)
log.Fatal("Failed cmd: ", err)
}
fmt.Println(b)
}