src/LanSpider/TextNetworkBrowser.cs

   1 using System;
   2 using System.Collections.Generic;
   3 using System.IO;
   4 using System.Linq;
   5
   6 namespace LanSpider
   7 {
   8     /// <summary>
   9     /// Fake network browser which takes information from file.
  10     /// </summary>
  11     internal class TextNetworkBrowser : INetworkBrowser
  12     {
  13         private readonly string _filename;
  14
  15         public TextNetworkBrowser( string filename )
  16         {
  17             _filename = filename;
  18         }
  19
  20         public IEnumerable<string> GetNetworkComputers()
  21         {
  22             return ReadItems( 2 );
  23         }
  24
  25         public IEnumerable<string> GetShares( string serverName )
  26         {
  27             return ReadItems( serverName, 3, true );
  28         }
  29
  30         public IEnumerable<FileInfo> GetSharedFiles( string shareName )
  31         {
  32             return from filename in ReadItems( shareName, true )
  33                    select new FileInfo( filename );
  34         }
  35
  36         #region Overloads
  37
  38         private IEnumerable<string> ReadItems( string mask, int clipDepth )
  39         {
  40             return ReadItems( mask, clipDepth, false );
  41         }
  42
  43         private IEnumerable<string> ReadItems( int clipDepth )
  44         {
  45             return ReadItems( clipDepth, false );
  46         }
  47
  48         private IEnumerable<string> ReadItems( int clipDepth, bool entriesGroupped )
  49         {
  50             return ReadItems( null, clipDepth, entriesGroupped );
  51         }
  52
  53         private IEnumerable<string> ReadItems( string mask, bool entriesGroupped )
  54         {
  55             return ReadItems( mask, 0, entriesGroupped );
  56         }
  57
  58         #endregion Overloads
  59
  60         private IEnumerable<string> ReadItems( string mask, int clipDepth, bool entriesGroupped )
  61         {
  62             List<string> items = new List<string>();
  63
  64             using ( StreamReader reader = File.OpenText( _filename ) )
  65             {
  66                 string line;
  67
  68                 while ( !( String.IsNullOrEmpty( line = reader.ReadLine() ) ) )
  69                 {
  70                     string path = line.Trim( new[] { '[', ']', ' ', '\t' } );
  71                     if ( String.IsNullOrEmpty( mask ) || path.StartsWith( mask ) )
  72                     {
  73                         string item;
  74
  75                         if ( clipDepth > 0 )
  76                         {
  77                             int lastIndex = 0;
  78                             for ( int i = 0; i < clipDepth; ++i )
  79                             {
  80                                 lastIndex = path.IndexOf( '\\', lastIndex + 1 );
  81                             }
  82
  83                             item = path.Substring( 0, lastIndex );
  84                         }
  85                         else
  86                         {
  87                             item = path;
  88                         }
  89
  90                         if ( !items.Contains( item ) )
  91                         {
  92                             items.Add( item );
  93                         }
  94                     }
  95                     else if ( entriesGroupped && items.Count > 0 )
  96                     {
  97                         break;
  98                     }
  99                 }
 100             }
 101
 102             return items;
 103         }
 104     }
 105 }