Added a DeletedIconOverlay.cs
[pithos-ms-client] / trunk / Pithos.Core / Agents / StatusAgent.cs
index 0f99f86..89cce58 100644 (file)
@@ -6,6 +6,7 @@ using System.Diagnostics;
 using System.Diagnostics.Contracts;
 using System.IO;
 using System.Linq;
+using System.Text;
 using System.Threading;
 using System.Threading.Tasks;
 using Castle.ActiveRecord;
@@ -40,7 +41,31 @@ namespace Pithos.Core.Agents
 
             if (!File.Exists(Path.Combine(_pithosDataPath ,"pithos.db")))
                 ActiveRecordStarter.CreateSchema();
-        }        
+
+            CreateTrigger();
+            
+        }
+
+        private void CreateTrigger()
+        {
+            using (var connection = GetConnection())
+            using (var triggerCommand = connection.CreateCommand())
+            {
+                var cmdText = new StringBuilder()
+                    .AppendLine("CREATE TRIGGER IF NOT EXISTS update_last_modified UPDATE ON FileState FOR EACH ROW")
+                    .AppendLine("BEGIN")
+                    .AppendLine("UPDATE FileState SET Modified=datetime('now')  WHERE Id=old.Id;")
+                    .AppendLine("END;")
+                    .AppendLine("CREATE TRIGGER IF NOT EXISTS insert_last_modified INSERT ON FileState FOR EACH ROW")
+                    .AppendLine("BEGIN")
+                    .AppendLine("UPDATE FileState SET Modified=datetime('now')  WHERE Id=new.Id;")
+                    .AppendLine("END;")
+                    .ToString();
+                triggerCommand.CommandText = cmdText;                
+                triggerCommand.ExecuteNonQuery();
+            }
+        }
+
 
         private static InPlaceConfigurationSource GetConfiguration(string pithosDbPath)
         {
@@ -147,7 +172,7 @@ namespace Pithos.Core.Agents
                 if (fileState == null)
                 {
                     //This is a new file
-                    var fullPath = pair.File.FullName.ToLower();
+                    var fullPath = pair.File.FullName;
                     var createState = FileState.CreateForAsync(fullPath, BlockSize, BlockHash);
                     createState.ContinueWith(state => _persistenceAgent.Post(state.Result.Create));
                 }                
@@ -174,85 +199,99 @@ namespace Pithos.Core.Agents
 
         private int UpdateStatusDirect(Guid id, FileStatus status)
         {
-            try
+            using (log4net.ThreadContext.Stacks["StatusAgent"].Push("UpdateStatusDirect"))
             {
 
-                var connectionString = GetConnectionString();
-                using (var connection = new SQLiteConnection(connectionString))
-                using (var command = new SQLiteCommand("update FileState set FileStatus= :fileStatus where Id = :id  ",
-                                                    connection))
+                try
                 {
                     
+                    using (var connection = GetConnection())
+                    using (
+                        var command = new SQLiteCommand("update FileState set FileStatus= :fileStatus where Id = :id  ",
+                                                        connection))
+                    {                                                
+                        command.Parameters.AddWithValue("fileStatus", status);
+
+                        command.Parameters.AddWithValue("id", id);
+                        
+                        var affected = command.ExecuteNonQuery();
+                        
+                        return affected;
+                    }
 
-                    command.Parameters.AddWithValue("fileStatus", status);
-
-                    command.Parameters.AddWithValue("id", id);
-                    connection.Open();
-                    var affected = command.ExecuteNonQuery();
-                    return affected;
+                }
+                catch (Exception exc)
+                {
+                    Log.Error(exc.ToString());
+                    throw;
                 }
             }
-            catch (Exception exc)
-            {
-                Log.Error(exc.ToString());
-                throw;
-            }
-
         }
         
         private int UpdateStatusDirect(string path, FileStatus status)
         {
-            try
+            using (log4net.ThreadContext.Stacks["StatusAgent"].Push("UpdateStatusDirect"))
             {
 
-                var connectionString = GetConnectionString();
-                using (var connection = new SQLiteConnection(connectionString))
-                using (var command = new SQLiteCommand("update FileState set FileStatus= :fileStatus where FilePath = :path ",
-                                                    connection))
+                try
                 {
+
                     
+                    using (var connection = GetConnection())
+                    using (
+                        var command =
+                            new SQLiteCommand("update FileState set FileStatus= :fileStatus where FilePath = :path COLLATE NOCASE",
+                                              connection))
+                    {
 
-                    command.Parameters.AddWithValue("fileStatus", status);
 
-                    command.Parameters.AddWithValue("path", path.ToLower());
-                    connection.Open();
-                    var affected = command.ExecuteNonQuery();
-                    return affected;
+                        command.Parameters.AddWithValue("fileStatus", status);
+
+                        command.Parameters.AddWithValue("path", path);
+                        
+                        var affected = command.ExecuteNonQuery();
+                        return affected;
+                    }
+                }
+                catch (Exception exc)
+                {
+                    Log.Error(exc.ToString());
+                    throw;
                 }
             }
-            catch (Exception exc)
-            {
-                Log.Error(exc.ToString());
-                throw;
-            }
-
         }
 
         private int UpdateStatusDirect(string absolutePath, FileStatus fileStatus, FileOverlayStatus overlayStatus)
         {
-            try
+            using (log4net.ThreadContext.Stacks["StatusAgent"].Push("UpdateStatusDirect"))
             {
 
-                var connectionString = GetConnectionString();
-                using (var connection = new SQLiteConnection(connectionString))
-                using (var command = new SQLiteCommand("update FileState set OverlayStatus= :overlayStatus, FileStatus= :fileStatus where FilePath = :path  ",
-                                                    connection))
+                try
                 {
+
                     
-                    command.Parameters.AddWithValue("path", absolutePath.ToLower());
-                    command.Parameters.AddWithValue("fileStatus", fileStatus);
-                    command.Parameters.AddWithValue("overlayStatus", overlayStatus);
-                    connection.Open();
-                    var affected = command.ExecuteNonQuery();
-                    return affected;
+                    using (var connection = GetConnection())
+                    using (
+                        var command =
+                            new SQLiteCommand(
+                                "update FileState set OverlayStatus= :overlayStatus, FileStatus= :fileStatus where FilePath = :path COLLATE NOCASE ",
+                                connection))
+                    {
+
+                        command.Parameters.AddWithValue("path", absolutePath);
+                        command.Parameters.AddWithValue("fileStatus", fileStatus);
+                        command.Parameters.AddWithValue("overlayStatus", overlayStatus);
+                        
+                        var affected = command.ExecuteNonQuery();
+                        return affected;
+                    }
+                }
+                catch (Exception exc)
+                {
+                    Log.Error(exc.ToString());
+                    throw;
                 }
             }
-            catch (Exception exc)
-            {
-                Log.Error(exc.ToString());
-                throw;
-            }
-
         }
         
 
@@ -292,6 +331,74 @@ namespace Pithos.Core.Agents
         private readonly string _pithosDataPath;
 
 
+        public FileState GetStateByFilePath(string path)
+        {
+            if (String.IsNullOrWhiteSpace(path))
+                throw new ArgumentNullException("path");
+            if (!Path.IsPathRooted(path))
+                throw new ArgumentException("The path must be rooted", "path");
+            Contract.EndContractBlock();
+
+            try
+            {
+                
+                using (var connection = GetConnection())
+                using (var command = new SQLiteCommand("select Id, FilePath, OverlayStatus,FileStatus ,Checksum   ,Version    ,VersionTimeStamp,IsShared   ,SharedBy   ,ShareWrite  from FileState where FilePath=:path COLLATE NOCASE", connection))
+                {
+                    
+                    command.Parameters.AddWithValue("path", path);
+                    
+                    using (var reader = command.ExecuteReader())
+                    {
+                        if (reader.Read())
+                        {
+                            //var values = new object[reader.FieldCount];
+                            //reader.GetValues(values);
+                            var state = new FileState
+                                            {
+                                                Id = reader.GetGuid(0),
+                                                FilePath = reader.IsDBNull(1)?"":reader.GetString(1),
+                                                OverlayStatus =reader.IsDBNull(2)?FileOverlayStatus.Unversioned: (FileOverlayStatus) reader.GetInt64(2),
+                                                FileStatus = reader.IsDBNull(3)?FileStatus.Missing:(FileStatus) reader.GetInt64(3),
+                                                Checksum = reader.IsDBNull(4)?"":reader.GetString(4),
+                                                Version = reader.IsDBNull(5)?default(long):reader.GetInt64(5),
+                                                VersionTimeStamp = reader.IsDBNull(6)?default(DateTime):reader.GetDateTime(6),
+                                                IsShared = !reader.IsDBNull(7) && reader.GetBoolean(7),
+                                                SharedBy = reader.IsDBNull(8)?"":reader.GetString(8),
+                                                ShareWrite = !reader.IsDBNull(9) && reader.GetBoolean(9)
+                                            };
+/*
+                            var state = new FileState
+                                            {
+                                                Id = (Guid) values[0],
+                                                FilePath = (string) values[1],
+                                                OverlayStatus = (FileOverlayStatus) (long)values[2],
+                                                FileStatus = (FileStatus) (long)values[3],
+                                                Checksum = (string) values[4],
+                                                Version = (long?) values[5],
+                                                VersionTimeStamp = (DateTime?) values[6],
+                                                IsShared = (long)values[7] == 1,
+                                                SharedBy = (string) values[8],
+                                                ShareWrite = (long)values[9] == 1
+                                            };
+*/
+                            return state;
+                        }
+                        else
+                        {
+                            return null;
+                        }
+
+                    }                    
+                }
+            }
+            catch (Exception exc)
+            {
+                Log.ErrorFormat(exc.ToString());
+                throw;
+            }            
+        }
+
         public FileOverlayStatus GetFileOverlayStatus(string path)
         {
             if (String.IsNullOrWhiteSpace(path))
@@ -302,13 +409,13 @@ namespace Pithos.Core.Agents
 
             try
             {
-                var connectionString = GetConnectionString();
-                using (var connection = new SQLiteConnection(connectionString))
-                using (var command = new SQLiteCommand("select OverlayStatus from FileState where FilePath=:path", connection))
+                
+                using (var connection = GetConnection())
+                using (var command = new SQLiteCommand("select OverlayStatus from FileState where FilePath=:path  COLLATE NOCASE", connection))
                 {
                     
-                    command.Parameters.AddWithValue("path", path.ToLower());
-                    connection.Open();
+                    command.Parameters.AddWithValue("path", path);
+                    
                     var s = command.ExecuteScalar();
                     return (FileOverlayStatus) Convert.ToInt32(s);
                 }
@@ -322,10 +429,23 @@ namespace Pithos.Core.Agents
 
         private string GetConnectionString()
         {
-            var connectionString = String.Format(@"Data Source={0}\pithos.db;Version=3;Enlist=N;Pooling=True;Default Isolation Level=ReadCommited", _pithosDataPath);
+            var connectionString = String.Format(@"Data Source={0}\pithos.db;Version=3;Enlist=N;Pooling=True", _pithosDataPath);
             return connectionString;
         }
 
+        private SQLiteConnection GetConnection()
+        {
+            var connectionString = GetConnectionString();
+            var connection = new SQLiteConnection(connectionString);
+            connection.Open();
+            using(var cmd =connection.CreateCommand())
+            {
+                cmd.CommandText = "PRAGMA journal_mode=WAL";
+                cmd.ExecuteNonQuery();
+            }
+            return connection;
+        }
+
         public void SetFileOverlayStatus(string path, FileOverlayStatus overlayStatus)
         {
             if (String.IsNullOrWhiteSpace(path))
@@ -334,7 +454,7 @@ namespace Pithos.Core.Agents
                 throw new ArgumentException("The path must be rooted","path");
             Contract.EndContractBlock();
 
-            _persistenceAgent.Post(() => FileState.StoreOverlayStatus(path.ToLower(),overlayStatus));
+            _persistenceAgent.Post(() => FileState.StoreOverlayStatus(path,overlayStatus));
         }
 
        /* public void RenameFileOverlayStatus(string oldPath, string newPath)
@@ -363,9 +483,10 @@ namespace Pithos.Core.Agents
             Debug.Assert(!path.Contains(FolderConstants.CacheFolder));
             Debug.Assert(!path.EndsWith(".ignore"));
 
-            _persistenceAgent.Post(() => UpdateStatusDirect(path.ToLower(), fileStatus, overlayStatus));
+            _persistenceAgent.Post(() => UpdateStatusDirect(path, fileStatus, overlayStatus));
         }
 
+/*
         public void StoreInfo(string path,ObjectInfo objectInfo)
         {
             if (String.IsNullOrWhiteSpace(path))
@@ -404,7 +525,9 @@ namespace Pithos.Core.Agents
             });
 
         }
-        public void StoreInfoDirect(string path, ObjectInfo objectInfo)
+*/
+        
+        public void StoreInfo(string path, ObjectInfo objectInfo)
         {
             if (String.IsNullOrWhiteSpace(path))
                 throw new ArgumentNullException("path");
@@ -414,76 +537,57 @@ namespace Pithos.Core.Agents
                 throw new ArgumentNullException("objectInfo", "objectInfo can't be empty");
             Contract.EndContractBlock();
 
-            _persistenceAgent.Post(() =>
-                                       {
-                                           try
-                                           {
-
-                                               var connectionString = GetConnectionString();
-                                               using (var connection = new SQLiteConnection(connectionString))
-                                               {
-                                                   if (StateExists(path))
-                                                   var command =
-                                                       new SQLiteCommand(
-                                                           "update FileState set FileStatus= :fileStatus where Id = :id  ",
-                                                           connection);
-
-                                                   
-                                                   var filePath = path.ToLower();
-                                                   //Load the existing files state and set its properties in one session            
-                                                   using (new SessionScope())
-                                                   {
-                                                       //Forgetting to use a sessionscope results in two sessions being created, one by 
-                                                       //FirstOrDefault and one by Save()
-                                                       var state = FileState.FindByFilePath(filePath);
-
-                                                       //Create a new empty state object if this is a new file
-                                                       state = state ?? new FileState();
-
-                                                       state.FilePath = filePath;
-                                                       state.Checksum = objectInfo.Hash;
-                                                       state.Version = objectInfo.Version;
-                                                       state.VersionTimeStamp = objectInfo.VersionTimestamp;
-
-                                                       state.FileStatus = FileStatus.Unchanged;
-                                                       state.OverlayStatus = FileOverlayStatus.Normal;
-
-
-                                                       //Do the save
-                                                       state.Save();
-                                                   }
-
-
-                                                   command.Parameters.AddWithValue("fileStatus", status);
-
-                                                   command.Parameters.AddWithValue("id", id);
-                                                   connection.Open();
-                                                   var affected = command.ExecuteNonQuery();
-                                                   return;
-                                               }
-                                           }
-                                           catch (Exception exc)
-                                           {
-                                               Log.Error(exc.ToString());
-                                               throw;
-                                           }
-
-                                       });
+            _persistenceAgent.Post(() => StoreInfoDirect(path, objectInfo));
+
+        }
+
+        private void StoreInfoDirect(string path, ObjectInfo objectInfo)
+        {
+            try
+            {
+                
+                using (var connection = GetConnection())
+                using (var command = new SQLiteCommand(connection))
+                {
+                    if (StateExists(path, connection))
+                        command.CommandText =
+                            "update FileState set FileStatus= :fileStatus where FilePath = :path  COLLATE NOCASE ";
+                    else
+                    {
+                        command.CommandText =
+                            "INSERT INTO FileState (Id,FilePath,Checksum,Version,VersionTimeStamp,FileStatus,OverlayStatus) VALUES (:id,:path,:checksum,:version,:versionTimeStamp,:fileStatus,:overlayStatus)";
+                        command.Parameters.AddWithValue("id", Guid.NewGuid());
+                    }
+
+                    command.Parameters.AddWithValue("path", path);
+                    command.Parameters.AddWithValue("checksum", objectInfo.Hash);
+                    command.Parameters.AddWithValue("version", objectInfo.Version);
+                    command.Parameters.AddWithValue("versionTimeStamp",
+                                                    objectInfo.VersionTimestamp);
+                    command.Parameters.AddWithValue("fileStatus", FileStatus.Unchanged);
+                    command.Parameters.AddWithValue("overlayStatus",
+                                                    FileOverlayStatus.Normal);
 
+                    var affected = command.ExecuteNonQuery();
+                    return;
+                }
+            }
+            catch (Exception exc)
+            {
+                Log.Error(exc.ToString());
+                throw;
+            }
         }
 
         private bool StateExists(string filePath,SQLiteConnection connection)
         {
-            using(var command=new SQLiteCommand("Select count(*) from FileState where FilePath=:path",connection))
+            using (var command = new SQLiteCommand("Select count(*) from FileState where FilePath=:path  COLLATE NOCASE", connection))
             {
-                command.Parameters.AddWithValue("path", filePath.ToLower());
-                var result=command.ExecuteScalar();
-                    new SQLiteCommand(
-                        "update FileState set FileStatus= :fileStatus where Id = :id  ",
-                        connection);
+                command.Parameters.AddWithValue("path", filePath);
+                var result = command.ExecuteScalar();
+                return ((long)result >= 1);
+            }
 
-                                                   
-            
         }
 
         public void SetFileStatus(string path, FileStatus status)
@@ -494,7 +598,7 @@ namespace Pithos.Core.Agents
                 throw new ArgumentException("The path must be rooted", "path");
             Contract.EndContractBlock();
             
-            _persistenceAgent.Post(() => UpdateStatusDirect(path.ToLower(), status));
+            _persistenceAgent.Post(() => UpdateStatusDirect(path, status));
         }
 
         public FileStatus GetFileStatus(string path)
@@ -505,17 +609,23 @@ namespace Pithos.Core.Agents
                 throw new ArgumentException("The path must be rooted", "path");
             Contract.EndContractBlock();
 
-            var connectionString = GetConnectionString();
-            using (var connection = new SQLiteConnection(connectionString))
+            
+            using (var connection = GetConnection())
             {
-                var command = new SQLiteCommand("select FileStatus from FileState where FilePath=:path", connection);
-                command.Parameters.AddWithValue("path", path.ToLower());
-                connection.Open();
-                var s = command.ExecuteScalar();
-                return (FileStatus)Convert.ToInt32(s);
+                var command = new SQLiteCommand("select FileStatus from FileState where FilePath=:path  COLLATE NOCASE", connection);
+                command.Parameters.AddWithValue("path", path);
+                
+                var statusValue = command.ExecuteScalar();
+                if (statusValue==null)
+                    return FileStatus.Missing;
+                return (FileStatus)Convert.ToInt32(statusValue);
             }
         }
 
+        /// <summary>
+        /// Deletes the status of the specified file
+        /// </summary>
+        /// <param name="path"></param>
         public void ClearFileStatus(string path)
         {
             if (String.IsNullOrWhiteSpace(path))
@@ -527,29 +637,87 @@ namespace Pithos.Core.Agents
             _persistenceAgent.Post(() => DeleteDirect(path));   
         }
 
+        /// <summary>
+        /// Deletes the status of the specified folder and all its contents
+        /// </summary>
+        /// <param name="path"></param>
+        public void ClearFolderStatus(string path)
+        {
+            if (String.IsNullOrWhiteSpace(path))
+                throw new ArgumentNullException("path");
+            if (!Path.IsPathRooted(path))
+                throw new ArgumentException("The path must be rooted", "path");
+            Contract.EndContractBlock();
+
+            _persistenceAgent.Post(() => DeleteFolderDirect(path));   
+        }
+
+        public IEnumerable<FileState> GetChildren(FileState fileState)
+        {
+            if (fileState == null)
+                throw new ArgumentNullException("fileState");
+            Contract.EndContractBlock();
+
+            var children = from state in FileState.Queryable
+                           where state.FilePath.StartsWith(fileState.FilePath + "\\")
+                           select state;
+            return children;
+        }
+
         private int DeleteDirect(string filePath)
         {
-            try
+            using (log4net.ThreadContext.Stacks["StatusAgent"].Push("DeleteDirect"))
             {
 
-                var connectionString = GetConnectionString();
-                using (var connection = new SQLiteConnection(connectionString))
+                try
                 {
-                    var command = new SQLiteCommand("delete FileState where FilePath = :path",
-                                                    connection);
 
-                    command.Parameters.AddWithValue("path", filePath.ToLower());
-                    connection.Open();
-                    var affected = command.ExecuteNonQuery();
-                    return affected;
+                    
+                    using (var connection = GetConnection())
+                    {
+                        var command = new SQLiteCommand("delete from FileState where FilePath = :path  COLLATE NOCASE",
+                                                        connection);
+
+                        command.Parameters.AddWithValue("path", filePath);
+                        
+                        var affected = command.ExecuteNonQuery();
+                        return affected;
+                    }
+                }
+                catch (Exception exc)
+                {
+                    Log.Error(exc.ToString());
+                    throw;
                 }
             }
-            catch (Exception exc)
+        }
+
+        private int DeleteFolderDirect(string filePath)
+        {
+            using (log4net.ThreadContext.Stacks["StatusAgent"].Push("DeleteDirect"))
             {
-                Log.Error(exc.ToString());
-                throw;
-            }
 
+                try
+                {
+
+                    
+                    using (var connection = GetConnection())
+                    {
+                        var command = new SQLiteCommand("delete from FileState where FilePath = :path or FilePath like :path + '/%'  COLLATE NOCASE",
+                                                        connection);
+
+                        command.Parameters.AddWithValue("path", filePath);
+                        
+                        var affected = command.ExecuteNonQuery();
+                        return affected;
+                    }
+                }
+                catch (Exception exc)
+                {
+                    Log.Error(exc.ToString());
+                    throw;
+                }
+            }
         }
 
         public void UpdateFileChecksum(string path, string checksum)
@@ -560,7 +728,7 @@ namespace Pithos.Core.Agents
                 throw new ArgumentException("The path must be rooted", "path");            
             Contract.EndContractBlock();
 
-            _persistenceAgent.Post(() => FileState.UpdateChecksum(path.ToLower(), checksum));
+            _persistenceAgent.Post(() => FileState.UpdateChecksum(path, checksum));
         }
 
     }