DriveVision/DriveScanner.cs at main · FSMaker77/DriveVision · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
using System;
using System.IO;
using System.Management;
using System.Collections.Generic;
using System.Diagnostics;

namespace DriveVision
{
    public class DriveCard
    {
        public string Lettera { get; set; } = string.Empty;
        public string NomeVolume { get; set; } = string.Empty;
        public string FileSystem { get; set; } = string.Empty;
        public string Icona { get; set; } = string.Empty;
        public string DescrizioneMotore { get; set; } = string.Empty;
        public double SpazioTotaleGB { get; set; }
    }

    public class DriveScanner
    {
        public List<DriveCard> OttieniDischiDisponibili()
        {
            var listaCard = new List<DriveCard>();
            DriveInfo[] dischi = DriveInfo.GetDrives();

            foreach (DriveInfo disco in dischi)
            {
                if (!disco.IsReady) continue;

                var card = new DriveCard
                {
                    Lettera = disco.Name,
                    NomeVolume = string.IsNullOrWhiteSpace(disco.VolumeLabel) ? Translator.Get("DiscoLocale") : disco.VolumeLabel,
                    FileSystem = disco.DriveFormat,
                    SpazioTotaleGB = Math.Round(disco.TotalSize / 1073741824.0, 1)
                };

                if (disco.DriveType == DriveType.Removable)
                {
                    card.Icona = "🐢";
                    card.DescrizioneMotore = Translator.Get("LentoUSB");
                }
                else if (disco.DriveType == DriveType.Fixed)
                {
                    if (disco.DriveFormat == "NTFS")
                    {
                        bool isSsd = ControllaSeSsd(disco.Name);
                        if (isSsd)
                        {
                            card.Icona = "🚀";
                            card.DescrizioneMotore = Translator.Get("UltraVeloce");
                        }
                        else
                        {
                            card.Icona = "🐇";
                            card.DescrizioneMotore = Translator.Get("Veloce");
                        }
                    }
                    else
                    {
                        card.Icona = "🐢";
                        card.DescrizioneMotore = Translator.Get("StandardNoMFT");
                    }
                }
                listaCard.Add(card);
            }
            return listaCard;
        }

        private bool ControllaSeSsd(string letteraDrive)
        {
            try
            {
                string lettera = letteraDrive.Substring(0, 1);

                var partitionSearcher = new ManagementObjectSearcher(
                    @"Root\Microsoft\Windows\Storage",
                    $"SELECT DiskNumber FROM MSFT_Partition WHERE DriveLetter = '{lettera}'");

                foreach (ManagementObject partition in partitionSearcher.Get())
                {
                    if (partition["DiskNumber"] != null)
                    {
                        string diskNumber = partition["DiskNumber"].ToString() ?? string.Empty;

                        var diskSearcher = new ManagementObjectSearcher(
                            @"Root\Microsoft\Windows\Storage",
                            $"SELECT MediaType FROM MSFT_PhysicalDisk WHERE DeviceId = '{diskNumber}'");

                        foreach (ManagementObject disk in diskSearcher.Get())
                        {
                            if (disk["MediaType"] != null && (disk["MediaType"].ToString() ?? string.Empty) == "4")
                            {
                                return true;
                            }
                        }
                    }
                }
            }
            catch (Exception ex)
            {
                System.Diagnostics.Debug.WriteLine($"Errore WMI MediaType per {letteraDrive}: {ex.Message}");
            }

            return false;
        }
    }
}