[gnome-shell/wip/carlosg/screenshots-to-clipboard: 10/11] shell: Make screenshot API stream based
- From: Florian Müllner <fmuellner src gnome org>
- To: commits-list gnome org
- Cc:
- Subject: [gnome-shell/wip/carlosg/screenshots-to-clipboard: 10/11] shell: Make screenshot API stream based
- Date: Wed, 6 Nov 2019 22:45:58 +0000 (UTC)
commit be5f5ec9d40c9cad408dbd4be51947bb16cd0d48
Author: Carlos Garnacho <carlosg gnome org>
Date: Wed Nov 6 00:46:41 2019 +0100
shell: Make screenshot API stream based
Instead of dealing with filenames, make the low-level API use streams
so the target remains generic.
This so far means JS code now determines the appropriate filename to
use for storing the screenshot, but will be used in other ways in
future commits.
https://gitlab.gnome.org/GNOME/mutter/issues/789
https://gitlab.gnome.org/GNOME/gnome-shell/merge_requests/810
js/ui/screenshot.js | 77 ++++++++++++++++++++++----
src/shell-screenshot.c | 147 +++++++------------------------------------------
src/shell-screenshot.h | 9 +--
3 files changed, 91 insertions(+), 142 deletions(-)
---
diff --git a/js/ui/screenshot.js b/js/ui/screenshot.js
index db96343649..c50c4a3639 100644
--- a/js/ui/screenshot.js
+++ b/js/ui/screenshot.js
@@ -64,7 +64,52 @@ var ScreenshotService = class {
y + height <= global.screen_height;
}
- _onScreenshotComplete(result, area, filenameUsed, flash, invocation) {
+ *_resolveRelativeFilename(filename) {
+ if (GLib.str_has_suffix(filename, '.png'))
+ filename = filename.substr(0, -4);
+
+ let path = [
+ GLib.get_user_special_dir(GLib.UserDirectory.DIRECTORY_PICTURES),
+ GLib.get_home_dir(),
+ ].find(p => GLib.file_test(p, GLib.FileTest.EXISTS));
+
+ if (!path)
+ return null;
+
+ yield Gio.File.new_for_path(
+ GLib.build_filenamev([path, `${filename}.png`]));
+
+ for (let idx = 1; ; idx++) {
+ yield Gio.File.new_for_path(
+ GLib.build_filenamev([path, `${filename}-${idx}.png`]));
+ }
+ }
+
+ _createStream(filename) {
+ if (GLib.path_is_absolute(filename)) {
+ try {
+ let file = Gio.File.new_for_path(filename);
+ let stream = file.replace(null, false, Gio.FileCreateFlags.NONE, null);
+ return [stream, file];
+ } catch (e) {
+ return [null, null];
+ }
+ }
+
+ for (let file of this._resolveRelativeFilename(filename)) {
+ try {
+ let stream = file.create(Gio.FileCreateFlags.NONE, null);
+ return [stream, file];
+ } catch (e) {
+ if (!e.matches(Gio.IOErrorEnum, Gio.IOErrorEnum.EXISTS))
+ return [null, null];
+ }
+ }
+
+ return [null, null];
+ }
+
+ _onScreenshotComplete(result, area, stream, file, flash, invocation) {
if (result) {
if (flash) {
let flashspot = new Flashspot(area);
@@ -76,6 +121,9 @@ var ScreenshotService = class {
}
}
+ stream.close(null);
+
+ let filenameUsed = file.get_path();
let retval = GLib.Variant.new('(bs)', [result, filenameUsed]);
invocation.return_value(retval);
}
@@ -110,13 +158,16 @@ var ScreenshotService = class {
let screenshot = this._createScreenshot(invocation);
if (!screenshot)
return;
- screenshot.screenshot_area (x, y, width, height, filename,
+
+ let [stream, file] = this._createStream(filename);
+
+ screenshot.screenshot_area (x, y, width, height, stream,
(o, res) => {
try {
- let [result, area, filenameUsed] =
+ let [result, area] =
screenshot.screenshot_area_finish(res);
this._onScreenshotComplete(
- result, area, filenameUsed, flash, invocation);
+ result, area, stream, file, flash, invocation);
} catch (e) {
invocation.return_gerror (e);
}
@@ -128,13 +179,16 @@ var ScreenshotService = class {
let screenshot = this._createScreenshot(invocation);
if (!screenshot)
return;
- screenshot.screenshot_window (includeFrame, includeCursor, filename,
+
+ let [stream, file] = this._createStream(filename);
+
+ screenshot.screenshot_window (includeFrame, includeCursor, stream,
(o, res) => {
try {
- let [result, area, filenameUsed] =
+ let [result, area] =
screenshot.screenshot_window_finish(res);
this._onScreenshotComplete(
- result, area, filenameUsed, flash, invocation);
+ result, area, stream, file, flash, invocation);
} catch (e) {
invocation.return_gerror (e);
}
@@ -146,13 +200,16 @@ var ScreenshotService = class {
let screenshot = this._createScreenshot(invocation);
if (!screenshot)
return;
- screenshot.screenshot(includeCursor, filename,
+
+ let [stream, file] = this._createStream(filename);
+
+ screenshot.screenshot(includeCursor, stream,
(o, res) => {
try {
- let [result, area, filenameUsed] =
+ let [result, area] =
screenshot.screenshot_finish(res);
this._onScreenshotComplete(
- result, area, filenameUsed, flash, invocation);
+ result, area, stream, file, flash, invocation);
} catch (e) {
invocation.return_gerror (e);
}
diff --git a/src/shell-screenshot.c b/src/shell-screenshot.c
index a7296e5664..4af51c698d 100644
--- a/src/shell-screenshot.c
+++ b/src/shell-screenshot.c
@@ -25,8 +25,7 @@ struct _ShellScreenshotPrivate
{
ShellGlobal *global;
- char *filename;
- char *filename_used;
+ GOutputStream *stream;
GDateTime *datetime;
@@ -72,102 +71,12 @@ on_screenshot_written (GObject *source,
g_object_unref (result);
g_clear_pointer (&priv->image, cairo_surface_destroy);
- g_clear_pointer (&priv->filename, g_free);
- g_clear_pointer (&priv->filename_used, g_free);
+ g_clear_object (&priv->stream);
g_clear_pointer (&priv->datetime, g_date_time_unref);
meta_enable_unredirect_for_display (shell_global_get_display (priv->global));
}
-/* called in an I/O thread */
-static GOutputStream *
-get_stream_for_unique_path (const gchar *path,
- const gchar *filename,
- gchar **filename_used)
-{
- GOutputStream *stream;
- GFile *file;
- gchar *real_path, *real_filename, *name, *ptr;
- gint idx;
-
- ptr = g_strrstr (filename, ".png");
-
- if (ptr != NULL)
- real_filename = g_strndup (filename, ptr - filename);
- else
- real_filename = g_strdup (filename);
-
- idx = 0;
- real_path = NULL;
-
- do
- {
- if (idx == 0)
- name = g_strdup_printf ("%s.png", real_filename);
- else
- name = g_strdup_printf ("%s - %d.png", real_filename, idx);
-
- real_path = g_build_filename (path, name, NULL);
- g_free (name);
-
- file = g_file_new_for_path (real_path);
- stream = G_OUTPUT_STREAM (g_file_create (file, G_FILE_CREATE_NONE, NULL, NULL));
- g_object_unref (file);
-
- if (stream != NULL)
- *filename_used = real_path;
- else
- g_free (real_path);
-
- idx++;
- }
- while (stream == NULL);
-
- g_free (real_filename);
-
- return stream;
-}
-
-/* called in an I/O thread */
-static GOutputStream *
-get_stream_for_filename (const gchar *filename,
- gchar **filename_used)
-{
- const gchar *path;
-
- path = g_get_user_special_dir (G_USER_DIRECTORY_PICTURES);
- if (!g_file_test (path, G_FILE_TEST_EXISTS))
- {
- path = g_get_home_dir ();
- if (!g_file_test (path, G_FILE_TEST_EXISTS))
- return NULL;
- }
-
- return get_stream_for_unique_path (path, filename, filename_used);
-}
-
-static GOutputStream *
-prepare_write_stream (const gchar *filename,
- gchar **filename_used)
-{
- GOutputStream *stream;
- GFile *file;
-
- if (g_path_is_absolute (filename))
- {
- file = g_file_new_for_path (filename);
- *filename_used = g_strdup (filename);
- stream = G_OUTPUT_STREAM (g_file_replace (file, NULL, FALSE, G_FILE_CREATE_NONE, NULL, NULL));
- g_object_unref (file);
- }
- else
- {
- stream = get_stream_for_filename (filename, filename_used);
- }
-
- return stream;
-}
-
static void
write_screenshot_thread (GTask *result,
gpointer object,
@@ -184,8 +93,7 @@ write_screenshot_thread (GTask *result,
priv = screenshot->priv;
- stream = prepare_write_stream (priv->filename,
- &priv->filename_used);
+ stream = g_object_ref (priv->stream);
if (stream == NULL)
status = CAIRO_STATUS_FILE_NOT_FOUND;
@@ -517,7 +425,6 @@ static gboolean
finish_screenshot (ShellScreenshot *screenshot,
GAsyncResult *result,
cairo_rectangle_int_t **area,
- const char **filename_used,
GError **error)
{
ShellScreenshotPrivate *priv = screenshot->priv;
@@ -528,9 +435,6 @@ finish_screenshot (ShellScreenshot *screenshot,
if (area)
*area = &priv->screenshot_area;
- if (filename_used)
- *filename_used = priv->filename_used;
-
return TRUE;
}
@@ -538,19 +442,19 @@ finish_screenshot (ShellScreenshot *screenshot,
* shell_screenshot_screenshot:
* @screenshot: the #ShellScreenshot
* @include_cursor: Whether to include the cursor or not
- * @filename: The filename for the screenshot
+ * @stream: The stream for the screenshot
* @callback: (scope async): function to call returning success or failure
* of the async grabbing
* @user_data: the data to pass to callback function
*
* Takes a screenshot of the whole screen
- * in @filename as png image.
+ * in @stream as png image.
*
*/
void
shell_screenshot_screenshot (ShellScreenshot *screenshot,
gboolean include_cursor,
- const char *filename,
+ GOutputStream *stream,
GAsyncReadyCallback callback,
gpointer user_data)
{
@@ -559,7 +463,7 @@ shell_screenshot_screenshot (ShellScreenshot *screenshot,
const char *paint_signal;
GTask *result;
- if (priv->filename != NULL) {
+ if (priv->stream != NULL) {
if (callback)
g_task_report_new_error (screenshot,
callback,
@@ -575,7 +479,7 @@ shell_screenshot_screenshot (ShellScreenshot *screenshot,
result = g_task_new (screenshot, NULL, callback, user_data);
g_task_set_source_tag (result, shell_screenshot_screenshot);
- priv->filename = g_strdup (filename);
+ priv->stream = g_object_ref (stream);
priv->include_cursor = FALSE;
stage = CLUTTER_ACTOR (shell_global_get_stage (priv->global));
@@ -601,8 +505,6 @@ shell_screenshot_screenshot (ShellScreenshot *screenshot,
* @screenshot: the #ShellScreenshot
* @result: the #GAsyncResult that was provided to the callback
* @area: (out) (transfer none): the area that was grabbed in screen coordinates
- * @filename_used: (out) (transfer none): the name of the file the screenshot
- * was written to
* @error: #GError for error reporting
*
* Finish the asynchronous operation started by shell_screenshot_screenshot()
@@ -615,13 +517,12 @@ gboolean
shell_screenshot_screenshot_finish (ShellScreenshot *screenshot,
GAsyncResult *result,
cairo_rectangle_int_t **area,
- const char **filename_used,
GError **error)
{
g_return_val_if_fail (g_async_result_is_tagged (result,
shell_screenshot_screenshot),
FALSE);
- return finish_screenshot (screenshot, result, area, filename_used, error);
+ return finish_screenshot (screenshot, result, area, error);
}
/**
@@ -631,13 +532,13 @@ shell_screenshot_screenshot_finish (ShellScreenshot *screenshot,
* @y: The Y coordinate of the area
* @width: The width of the area
* @height: The height of the area
- * @filename: The filename for the screenshot
+ * @stream: The stream for the screenshot
* @callback: (scope async): function to call returning success or failure
* of the async grabbing
* @user_data: the data to pass to callback function
*
* Takes a screenshot of the passed in area and saves it
- * in @filename as png image.
+ * in @stream as png image.
*
*/
void
@@ -646,7 +547,7 @@ shell_screenshot_screenshot_area (ShellScreenshot *screenshot,
int y,
int width,
int height,
- const char *filename,
+ GOutputStream *stream,
GAsyncReadyCallback callback,
gpointer user_data)
{
@@ -654,7 +555,7 @@ shell_screenshot_screenshot_area (ShellScreenshot *screenshot,
ShellScreenshotPrivate *priv = screenshot->priv;
GTask *result;
- if (priv->filename != NULL) {
+ if (priv->stream != NULL) {
if (callback)
g_task_report_new_error (screenshot,
callback,
@@ -670,7 +571,7 @@ shell_screenshot_screenshot_area (ShellScreenshot *screenshot,
result = g_task_new (screenshot, NULL, callback, user_data);
g_task_set_source_tag (result, shell_screenshot_screenshot_area);
- priv->filename = g_strdup (filename);
+ priv->stream = g_object_ref (stream);
priv->screenshot_area.x = x;
priv->screenshot_area.y = y;
priv->screenshot_area.width = width;
@@ -690,8 +591,6 @@ shell_screenshot_screenshot_area (ShellScreenshot *screenshot,
* @screenshot: the #ShellScreenshot
* @result: the #GAsyncResult that was provided to the callback
* @area: (out) (transfer none): the area that was grabbed in screen coordinates
- * @filename_used: (out) (transfer none): the name of the file the screenshot
- * was written to
* @error: #GError for error reporting
*
* Finish the asynchronous operation started by shell_screenshot_screenshot_area()
@@ -704,13 +603,12 @@ gboolean
shell_screenshot_screenshot_area_finish (ShellScreenshot *screenshot,
GAsyncResult *result,
cairo_rectangle_int_t **area,
- const char **filename_used,
GError **error)
{
g_return_val_if_fail (g_async_result_is_tagged (result,
shell_screenshot_screenshot_area),
FALSE);
- return finish_screenshot (screenshot, result, area, filename_used, error);
+ return finish_screenshot (screenshot, result, area, error);
}
/**
@@ -718,20 +616,20 @@ shell_screenshot_screenshot_area_finish (ShellScreenshot *screenshot,
* @screenshot: the #ShellScreenshot
* @include_frame: Whether to include the frame or not
* @include_cursor: Whether to include the cursor or not
- * @filename: The filename for the screenshot
+ * @stream: The stream for the screenshot
* @callback: (scope async): function to call returning success or failure
* of the async grabbing
* @user_data: the data to pass to callback function
*
* Takes a screenshot of the focused window (optionally omitting the frame)
- * in @filename as png image.
+ * in @stream as png image.
*
*/
void
shell_screenshot_screenshot_window (ShellScreenshot *screenshot,
gboolean include_frame,
gboolean include_cursor,
- const char *filename,
+ GOutputStream *stream,
GAsyncReadyCallback callback,
gpointer user_data)
{
@@ -741,7 +639,7 @@ shell_screenshot_screenshot_window (ShellScreenshot *screenshot,
MetaWindow *window = meta_display_get_focus_window (display);
GTask *result;
- if (priv->filename != NULL || !window) {
+ if (priv->stream != NULL || !window) {
if (callback)
g_task_report_new_error (screenshot,
callback,
@@ -757,7 +655,7 @@ shell_screenshot_screenshot_window (ShellScreenshot *screenshot,
result = g_task_new (screenshot, NULL, callback, user_data);
g_task_set_source_tag (result, shell_screenshot_screenshot_window);
- priv->filename = g_strdup (filename);
+ priv->stream = g_object_ref (stream);
priv->include_frame = include_frame;
priv->include_cursor = include_cursor &&
should_draw_cursor_image (SHELL_SCREENSHOT_WINDOW);
@@ -776,8 +674,6 @@ shell_screenshot_screenshot_window (ShellScreenshot *screenshot,
* @screenshot: the #ShellScreenshot
* @result: the #GAsyncResult that was provided to the callback
* @area: (out) (transfer none): the area that was grabbed in screen coordinates
- * @filename_used: (out) (transfer none): the name of the file the screenshot
- * was written to
* @error: #GError for error reporting
*
* Finish the asynchronous operation started by shell_screenshot_screenshot_window()
@@ -790,13 +686,12 @@ gboolean
shell_screenshot_screenshot_window_finish (ShellScreenshot *screenshot,
GAsyncResult *result,
cairo_rectangle_int_t **area,
- const char **filename_used,
GError **error)
{
g_return_val_if_fail (g_async_result_is_tagged (result,
shell_screenshot_screenshot_window),
FALSE);
- return finish_screenshot (screenshot, result, area, filename_used, error);
+ return finish_screenshot (screenshot, result, area, error);
}
/**
diff --git a/src/shell-screenshot.h b/src/shell-screenshot.h
index 2367d518e3..18c6667e1b 100644
--- a/src/shell-screenshot.h
+++ b/src/shell-screenshot.h
@@ -21,36 +21,33 @@ void shell_screenshot_screenshot_area (ShellScreenshot *screenshot,
int y,
int width,
int height,
- const char *filename,
+ GOutputStream *stream,
GAsyncReadyCallback callback,
gpointer user_data);
gboolean shell_screenshot_screenshot_area_finish (ShellScreenshot *screenshot,
GAsyncResult *result,
cairo_rectangle_int_t **area,
- const char **filename_used,
GError **error);
void shell_screenshot_screenshot_window (ShellScreenshot *screenshot,
gboolean include_frame,
gboolean include_cursor,
- const char *filename,
+ GOutputStream *stream,
GAsyncReadyCallback callback,
gpointer user_data);
gboolean shell_screenshot_screenshot_window_finish (ShellScreenshot *screenshot,
GAsyncResult *result,
cairo_rectangle_int_t **area,
- const char **filename_used,
GError **error);
void shell_screenshot_screenshot (ShellScreenshot *screenshot,
gboolean include_cursor,
- const char *filename,
+ GOutputStream *stream,
GAsyncReadyCallback callback,
gpointer user_data);
gboolean shell_screenshot_screenshot_finish (ShellScreenshot *screenshot,
GAsyncResult *result,
cairo_rectangle_int_t **area,
- const char **filename_used,
GError **error);
void shell_screenshot_pick_color (ShellScreenshot *screenshot,
[
Date Prev][
Date Next] [
Thread Prev][
Thread Next]
[
Thread Index]
[
Date Index]
[
Author Index]
