acoplar imagenes en una lista de archivos

[Joselitro]

hola tengo en mi web una lista de archivos y directorios como la siguiente

Código HTML:

<?php
$dir = (isset($_GET['dir']))?$_GET['dir']:"/";
$directorio=opendir($dir); 
echo "<b>Directorio actual:</b><br>$dir<br>"; 
echo "<b>Archivos:</b><br>"; 
while ($archivo = readdir($directorio)) { 
  if($archivo == '.')
    echo "<a href=\"?dir=.\">$archivo</a><br>"; 
  elseif($archivo == '..'){ 
    if($dir != '.'){ 
      $carpetas = split("/",$dir); 
      array_pop($carpetas); 
      $dir2 = join("/",$carpetas); 
      echo "<a href=\"?dir=$dir2\">$archivo</a><br>"; 
    } 
  }
  elseif(is_dir("$dir/$archivo"))
    echo "<a href=\"?dir=$dir/$archivo\"><img src='url de la imagen' alt='$archivo'>$archivo</a><br>"; 
  else echo "$archivo"; 
} 
closedir($directorio); 
?> 

funciona bien pero me gustaria saber como puedo ponerle una imagen a un archivo dependiendo de su extension lo e intentado pero no se mucho de php T_T ¿se puede?