Util/FSpotTools.cs

   1 //
   2 // FSpotTools.cs
   3 //
   4 // Copyright (C) 2004 Novell, Inc.
   5 //
   6
   7 //
   8 // Permission is hereby granted, free of charge, to any person obtaining a
   9 // copy of this software and associated documentation files (the "Software"),
  10 // to deal in the Software without restriction, including without limitation
  11 // the rights to use, copy, modify, merge, publish, distribute, sublicense,
  12 // and/or sell copies of the Software, and to permit persons to whom the
  13 // Software is furnished to do so, subject to the following conditions:
  14 //
  15 // The above copyright notice and this permission notice shall be included in
  16 // all copies or substantial portions of the Software.
  17 //
  18 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  21 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  22 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  23 // FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  24 // DEALINGS IN THE SOFTWARE.
  25 //
  26
  27 using System;
  28 using System.Collections;
  29 using System.IO;
  30 using Mono.Data.SqliteClient;
  31
  32 namespace Beagle.Util {
  33
  34         public class FSpotTools {
  35
  36                 public class Tag {
  37                         public uint   Id;
  38                         public string Name;
  39                         public uint   CategoryId;
  40                         public bool   IsCategory;
  41                         public int    SortPriority;
  42                         public Tag    Category;
  43                         // FIXME: Icon
  44                 }
  45
  46                 public class Photo {
  47                         public uint   Id;
  48                         public string Path;
  49                         public string Description;
  50                         public Tag[]  Tags;
  51                 }
  52
  53                 static private bool tried_connection;
  54                 static private SqliteConnection connection;
  55                 static Hashtable tagCache = null;
  56                 static Hashtable directoryCache = null;
  57
  58
  59                 static private string PhotoStorePath {
  60                         get {
  61                                 string home = Environment.GetEnvironmentVariable ("HOME");
  62                                 return Path.Combine (home, ".gnome2/f-spot/photos.db");
  63                         }
  64                 }
  65
  66                 static private SqliteConnection PhotoStoreConnection {
  67                         get {
  68                                 if (! tried_connection && File.Exists (PhotoStorePath)) {
  69                                         connection = new SqliteConnection ();
  70                                         connection.ConnectionString = "version=" + ExternalStringsHack.SqliteVersion
  71                                                 + ",URI=file:" + PhotoStorePath;
  72
  73                                         // Try to open the f-spot store.  This
  74                                         // will fail if there is a version
  75                                         // mismatch.
  76                                         try {
  77                                                 connection.Open ();
  78                                         } catch (ApplicationException) {
  79                                                 Logger.Log.Warn ("Unable to open F-Spot database: sqlite version mismatch");
  80                                         }
  81
  82                                         tried_connection = true;
  83                                 }
  84
  85                                 return connection;
  86                         }
  87                 }
  88
  89                 static private bool HavePhotoStore {
  90                         get {
  91                                 return PhotoStoreConnection != null;
  92                         }
  93                 }
  94
  95                 // FIXME: We should expire this cache if the underlying db has changed.
  96                 static private Tag GetTagById (uint id)
  97                 {
  98                         if (! HavePhotoStore)
  99                                 return null;
 100
 101                         if (tagCache == null) {
 102                                 tagCache = new Hashtable ();
 103
 104                                 SqliteCommand command = new SqliteCommand ();
 105                                 command.Connection = PhotoStoreConnection;
 106                                 command.CommandText = "SELECT id, name, category_id, is_category, sort_priority FROM tags";
 107
 108                                 SqliteDataReader reader = command.ExecuteReader ();
 109                                 while (reader.Read ()) {
 110                                         Tag tag = new Tag ();
 111                                         tag.Id = Convert.ToUInt32 (reader [0]);
 112                                         tag.Name = (string) reader [1];
 113                                         tag.CategoryId = Convert.ToUInt32 (reader [2]);
 114                                         tag.IsCategory = (((string)reader [3]) == "1");
 115                                         tag.SortPriority = Convert.ToInt32 (reader [4]);
 116                                         tagCache [tag.Id] = tag;
 117                                 }
 118
 119                                 // Walk across all tags, linking to the category's Tag
 120                                 // object.  Since the tagCache is fully populated, it is
 121                                 // safe to call GetTagById here.
 122                                 foreach (Tag tag in tagCache.Values)
 123                                         tag.Category = GetTagById (tag.CategoryId);
 124
 125                                 command.Dispose ();
 126                         }
 127
 128                         return (Tag) tagCache [id];
 129                 }
 130
 131                 // FIXME: We should expire this cache if the underlying db has changed.
 132                 static bool IsPossibleDirectory (string directory)
 133                 {
 134                         if (! HavePhotoStore)
 135                                 return false;
 136
 137                         if (directoryCache == null) {
 138                                 directoryCache = new Hashtable ();
 139
 140                                 SqliteCommand command = new SqliteCommand ();
 141                                 command.Connection = PhotoStoreConnection;
 142                                 command.CommandText = "SELECT DISTINCT directory_path FROM photos";
 143
 144                                 SqliteDataReader reader = command.ExecuteReader ();
 145                                 while (reader.Read ()) {
 146                                         directoryCache [reader [0]] = true;
 147                                 }
 148
 149                                 command.Dispose ();
 150                         }
 151
 152                         return directoryCache.Contains (directory);
 153                 }
 154
 155                 static public Photo GetPhoto (string path)
 156                 {
 157                         if (! HavePhotoStore)
 158                                 return null;
 159
 160                         path = Path.GetFullPath (path);
 161                         string dir = Path.GetDirectoryName (path);
 162                         string name = Path.GetFileName (path);
 163
 164                         if (! IsPossibleDirectory (dir))
 165                                 return null;
 166
 167                         SqliteCommand command;
 168                         SqliteDataReader reader;
 169
 170                         command = new SqliteCommand ();
 171                         command.Connection = PhotoStoreConnection;
 172                         command.CommandText = String.Format ("SELECT id, description         " +
 173                                                              "FROM photos                    " +
 174                                                              "WHERE directory_path = \"{0}\" " +
 175                                                              "  AND name = \"{1}\"",
 176                                                              dir, name);
 177
 178                         Photo photo = null;
 179                         reader = command.ExecuteReader ();
 180                         if (reader.Read ()) {
 181                                 photo = new Photo ();
 182                                 photo.Path = path;
 183                                 photo.Id = Convert.ToUInt32 (reader [0]);
 184                                 photo.Description = (string) reader [1];
 185                         }
 186
 187                         command.Dispose ();
 188
 189                         if (photo != null) {
 190                                 command = new SqliteCommand ();
 191                                 command.Connection = PhotoStoreConnection;
 192                                 command.CommandText = String.Format ("SELECT tag_id       " +
 193                                                                      "FROM photo_tags     " +
 194                                                                      "WHERE photo_id = {0}",
 195                                                                      photo.Id);
 196
 197                                 Hashtable tagHash = new Hashtable ();
 198
 199                                 // Mark the photo with both any tags and all parents.
 200                                 // Maybe this isn't the right thing to do, but it seems
 201                                 // to most closely mirror the tag semantics implied by f-spot.
 202                                 reader = command.ExecuteReader ();
 203                                 while (reader.Read ()) {
 204                                         uint id = Convert.ToUInt32 (reader [0]);
 205                                         Tag tag = GetTagById (id);
 206                                         while (tag != null) {
 207                                                 tagHash [tag] = tag;
 208                                                 tag = tag.Category;
 209                                         }
 210                                 }
 211
 212                                 photo.Tags = new Tag [tagHash.Count];
 213                                 int i = 0;
 214                                 foreach (Tag t in tagHash.Values) {
 215                                         photo.Tags [i] = t;
 216                                         ++i;
 217                                 }
 218                         }
 219
 220                         return photo;
 221                 }
 222
 223         }
 224 }