tools: Rearrange list of tools in Makefile.am.
[libguestfs.git] / fish / edit.c
1 /* guestfish - the filesystem interactive shell
2  * Copyright (C) 2009 Red Hat Inc.
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software
16  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17  */
18
19 #include <config.h>
20
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <unistd.h>
25 #include <fcntl.h>
26 #include <inttypes.h>
27
28 #include "fish.h"
29
30 /* guestfish edit command, suggested by Ján Ondrej, implemented by RWMJ */
31
32 static char *
33 load_file (const char *filename, size_t *len_r)
34 {
35   int fd, r, start;
36   char *content = NULL, *p;
37   char buf[65536];
38
39   *len_r = 0;
40
41   fd = open (filename, O_RDONLY);
42   if (fd == -1) {
43     perror (filename);
44     return NULL;
45   }
46
47   while ((r = read (fd, buf, sizeof buf)) > 0) {
48     start = *len_r;
49     *len_r += r;
50     p = realloc (content, *len_r + 1);
51     if (p == NULL) {
52       perror ("realloc");
53       free (content);
54       return NULL;
55     }
56     content = p;
57     memcpy (content + start, buf, r);
58     content[start+r] = '\0';
59   }
60
61   if (r == -1) {
62     perror (filename);
63     free (content);
64     return NULL;
65   }
66
67   if (close (fd) == -1) {
68     perror (filename);
69     free (content);
70     return NULL;
71   }
72
73   return content;
74 }
75
76 int
77 do_edit (const char *cmd, int argc, char *argv[])
78 {
79   char filename[] = "/tmp/guestfishXXXXXX";
80   char buf[256];
81   const char *editor;
82   char *content, *content_new;
83   int r, fd;
84
85   if (argc != 1) {
86     fprintf (stderr, _("use '%s filename' to edit a file\n"), cmd);
87     return -1;
88   }
89
90   /* Choose an editor. */
91   if (STRCASEEQ (cmd, "vi"))
92     editor = "vi";
93   else if (STRCASEEQ (cmd, "emacs"))
94     editor = "emacs -nw";
95   else {
96     editor = getenv ("EDITOR");
97     if (editor == NULL)
98       editor = "vi"; /* could be cruel here and choose ed(1) */
99   }
100
101   /* Download the file and write it to a temporary. */
102   fd = mkstemp (filename);
103   if (fd == -1) {
104     perror ("mkstemp");
105     return -1;
106   }
107
108   if ((content = guestfs_cat (g, argv[0])) == NULL) {
109     close (fd);
110     unlink (filename);
111     return -1;
112   }
113
114   if (xwrite (fd, content, strlen (content)) == -1) {
115     close (fd);
116     unlink (filename);
117     free (content);
118     return -1;
119   }
120
121   if (close (fd) == -1) {
122     perror (filename);
123     unlink (filename);
124     free (content);
125     return -1;
126   }
127
128   /* Edit it. */
129   /* XXX Safe? */
130   snprintf (buf, sizeof buf, "%s %s", editor, filename);
131
132   r = system (buf);
133   if (r != 0) {
134     perror (buf);
135     unlink (filename);
136     free (content);
137     return -1;
138   }
139
140   /* Reload it. */
141   size_t size;
142   content_new = load_file (filename, &size);
143   if (content_new == NULL) {
144     unlink (filename);
145     free (content);
146     return -1;
147   }
148
149   unlink (filename);
150
151   /* Changed? */
152   if (strlen (content) == size && STREQLEN (content, content_new, size)) {
153     free (content);
154     free (content_new);
155     return 0;
156   }
157
158   /* Write new content. */
159   if (guestfs_write_file (g, argv[0], content_new, size) == -1) {
160     free (content);
161     free (content_new);
162     return -1;
163   }
164
165   free (content);
166   free (content_new);
167   return 0;
168 }