connection = $connection; $this->preview = \OC::$server->get(IPreview::class); $this->livePhoto = new LivePhoto($connection); } /** * Check if a file has a valid mimetype for processing. * * @return int 0 for invalid, 1 for image, 2 for video */ public function getFileType(File $file): int { $mime = $file->getMimeType(); if (\in_array($mime, Application::IMAGE_MIMES, true)) { // Make sure preview generator supports the mime type if (!$this->preview->isMimeSupported($mime)) { return 0; } return 1; } if (\in_array($mime, Application::VIDEO_MIMES, true)) { return 2; } return 0; } /** * Process a file to insert Exif data into the database. * * @return int 2 if processed, 1 if skipped, 0 if not valid */ public function processFile( File &$file, bool $force = false ): int { // There is no easy way to UPSERT in a standard SQL way, so just // do multiple calls. The worst that can happen is more updates, // but that's not a big deal. // https://stackoverflow.com/questions/15252213/sql-standard-upsert-call // Check if we want to process this file $fileType = $this->getFileType($file); $isvideo = (2 === $fileType); if (!$fileType) { return 0; } // Get parameters $mtime = $file->getMtime(); $fileId = $file->getId(); // Check if need to update $query = $this->connection->getQueryBuilder(); $query->select('fileid', 'mtime') ->from('memories') ->where($query->expr()->eq('fileid', $query->createNamedParameter($fileId, IQueryBuilder::PARAM_INT))) ; $cursor = $query->executeQuery(); $prevRow = $cursor->fetch(); $cursor->closeCursor(); // Check in live-photo table in case this is a video part of a live photo if (!$prevRow) { $query = $this->connection->getQueryBuilder(); $query->select('fileid', 'mtime') ->from('memories_livephoto') ->where($query->expr()->eq('fileid', $query->createNamedParameter($fileId, IQueryBuilder::PARAM_INT))) ; $cursor = $query->executeQuery(); $prevRow = $cursor->fetch(); $cursor->closeCursor(); } if ($prevRow && !$force && (int) $prevRow['mtime'] === $mtime) { return 1; } // Get exif data $exif = []; try { $exif = Exif::getExifFromFile($file); } catch (\Exception $e) { } // Hand off if live photo video part if ($isvideo && $this->livePhoto->isVideoPart($exif)) { $this->livePhoto->processVideoPart($file, $exif); return 2; } // Get more parameters $dateTaken = Exif::getDateTaken($file, $exif); $dayId = floor($dateTaken / 86400); $dateTaken = gmdate('Y-m-d H:i:s', $dateTaken); [$w, $h] = Exif::getDimensions($exif); $liveid = $this->livePhoto->getLivePhotoId($file, $exif); // Video parameters $videoDuration = 0; if ($isvideo) { $videoDuration = round((float) ($exif['Duration'] ?? $exif['TrackDuration'] ?? 0)); } // Clean up EXIF to keep only useful metadata $filteredExif = []; foreach ($exif as $key => $value) { // Truncate any fields > 2048 chars if (\is_string($value) && \strlen($value) > 2048) { $value = substr($value, 0, 2048); } // Only keep fields in the whitelist if (\array_key_exists($key, EXIF_FIELDS_LIST)) { $filteredExif[$key] = $value; } } // Store JSON string $exifJson = json_encode($filteredExif); // Store error if data > 64kb if (\is_string($exifJson)) { if (\strlen($exifJson) > 65535) { $exifJson = json_encode(['error' => 'Exif data too large']); } } else { $exifJson = json_encode(['error' => 'Exif data encoding error']); } // Store location data if (\array_key_exists('GPSLatitude', $exif) && \array_key_exists('GPSLongitude', $exif)) { $lat = $exif['GPSLatitude']; $lon = $exif['GPSLongitude']; try { $this->updateGeoData($file, (float) $lat, (float) $lon); } catch (\Exception $e) { $logger = \OC::$server->get(LoggerInterface::class); $logger->log(3, 'Error updating geo data: '.$e->getMessage(), ['app' => 'memories']); } } if ($prevRow) { // Update existing row // No need to set objectid again $query->update('memories') ->set('dayid', $query->createNamedParameter($dayId, IQueryBuilder::PARAM_INT)) ->set('datetaken', $query->createNamedParameter($dateTaken, IQueryBuilder::PARAM_STR)) ->set('mtime', $query->createNamedParameter($mtime, IQueryBuilder::PARAM_INT)) ->set('isvideo', $query->createNamedParameter($isvideo, IQueryBuilder::PARAM_INT)) ->set('video_duration', $query->createNamedParameter($videoDuration, IQueryBuilder::PARAM_INT)) ->set('w', $query->createNamedParameter($w, IQueryBuilder::PARAM_INT)) ->set('h', $query->createNamedParameter($h, IQueryBuilder::PARAM_INT)) ->set('exif', $query->createNamedParameter($exifJson, IQueryBuilder::PARAM_STR)) ->set('liveid', $query->createNamedParameter($liveid, IQueryBuilder::PARAM_STR)) ->where($query->expr()->eq('fileid', $query->createNamedParameter($fileId, IQueryBuilder::PARAM_INT))) ; $query->executeStatement(); } else { // Try to create new row try { $query->insert('memories') ->values([ 'fileid' => $query->createNamedParameter($fileId, IQueryBuilder::PARAM_INT), 'objectid' => $query->createNamedParameter((string) $fileId, IQueryBuilder::PARAM_STR), 'dayid' => $query->createNamedParameter($dayId, IQueryBuilder::PARAM_INT), 'datetaken' => $query->createNamedParameter($dateTaken, IQueryBuilder::PARAM_STR), 'mtime' => $query->createNamedParameter($mtime, IQueryBuilder::PARAM_INT), 'isvideo' => $query->createNamedParameter($isvideo, IQueryBuilder::PARAM_INT), 'video_duration' => $query->createNamedParameter($videoDuration, IQueryBuilder::PARAM_INT), 'w' => $query->createNamedParameter($w, IQueryBuilder::PARAM_INT), 'h' => $query->createNamedParameter($h, IQueryBuilder::PARAM_INT), 'exif' => $query->createNamedParameter($exifJson, IQueryBuilder::PARAM_STR), 'liveid' => $query->createNamedParameter($liveid, IQueryBuilder::PARAM_STR), ]) ; $query->executeStatement(); } catch (\Exception $ex) { error_log('Failed to create memories record: '.$ex->getMessage()); } } return 2; } /** * Remove a file from the exif database. */ public function deleteFile(File &$file) { foreach(DELETE_TABLES as $table) { $query = $this->connection->getQueryBuilder(); $query->delete($table) ->where($query->expr()->eq('fileid', $query->createNamedParameter($file->getId(), IQueryBuilder::PARAM_INT))) ; $query->executeStatement(); } } /** * Clear the entire index. Does not need confirmation! * * @param File $file */ public function clear() { $p = $this->connection->getDatabasePlatform(); foreach (DELETE_TABLES as $table) { $this->connection->executeStatement($p->getTruncateTableSQL('*PREFIX*'.$table, false)); } } /** * Mark a file as not orphaned. */ public function unorphan(File &$file) { $query = $this->connection->getQueryBuilder(); $query->update('memories') ->set('orphan', $query->createNamedParameter(false, IQueryBuilder::PARAM_BOOL)) ->where($query->expr()->eq('fileid', $query->createNamedParameter($file->getId(), IQueryBuilder::PARAM_INT))) ; $query->executeStatement(); } /** * Mark all files in the table as orphaned. * * @return int Number of rows affected */ public function orphanAll(): int { $query = $this->connection->getQueryBuilder(); $query->update('memories') ->set('orphan', $query->createNamedParameter(true, IQueryBuilder::PARAM_BOOL)) ; return $query->executeStatement(); } /** * Remove all entries that are orphans. * * @return int Number of rows affected */ public function removeOrphans(): int { $query = $this->connection->getQueryBuilder(); $query->delete('memories') ->where($query->expr()->eq('orphan', $query->createNamedParameter(true, IQueryBuilder::PARAM_BOOL))) ; return $query->executeStatement(); } /** * Add geolocation data for a file. */ public function updateGeoData(File &$file, float $lat, float $lon): void { // Get GIS type $config = \OC::$server->get(\OCP\IConfig::class); $gisType = $config->getSystemValue('memories.gis_type', 0); // Construct WHERE clause depending on GIS type $where = null; if (1 === $gisType) { $where = "ST_Contains(geometry, ST_GeomFromText('POINT({$lon} {$lat})'))"; } elseif (2 === $gisType) { $where = "POINT('{$lon},{$lat}') <@ geometry"; } else { return; } // Make query to memories_planet table $query = $this->connection->getQueryBuilder(); $query->select($query->createFunction('DISTINCT(osm_id)')) ->from('memories_planet_geometry') ->where($query->createFunction($where)) ; // memories_planet_geometry has no *PREFIX* $sql = str_replace('*PREFIX*memories_planet_geometry', 'memories_planet_geometry', $query->getSQL()); // Run query $result = $this->connection->executeQuery($sql); $rows = $result->fetchAll(); // Delete previous records $query = $this->connection->getQueryBuilder(); $query->delete('memories_places') ->where($query->expr()->eq('fileid', $query->createNamedParameter($file->getId(), IQueryBuilder::PARAM_INT))) ; // Insert records foreach ($rows as $row) { $query = $this->connection->getQueryBuilder(); $query->insert('memories_places') ->values([ 'fileid' => $query->createNamedParameter($file->getId(), IQueryBuilder::PARAM_INT), 'osm_id' => $query->createNamedParameter($row['osm_id'], IQueryBuilder::PARAM_INT), ]) ; $query->executeStatement(); } } }