contrib/git_id/git_id.cpp

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <string.h>
   4 #include <string>
   5 #include <sstream>
   6 #include <assert.h>
   7 #include "../../src/framework/Platform/CompilerDefs.h"
   8
   9 #if PLATFORM == PLATFORM_WINDOWS
  10 #include <direct.h>
  11 #define popen _popen
  12 #define pclose _pclose
  13 #define snprintf _snprintf
  14 #pragma warning (disable:4996)
  15 #else
  16 #include <unistd.h>
  17 #endif
  18
  19 // config
  20
  21 #define NUM_REMOTES 2
  22
  23 char remotes[NUM_REMOTES][256] = {
  24     "git@github.com:mangos/mangos.git",
  25     "git://github.com/mangos/mangos.git"        // used for fetch if present
  26 };
  27
  28 bool allow_replace = false;
  29 bool local = false;
  30 bool do_fetch = false;
  31
  32 // aux
  33
  34 char origins[NUM_REMOTES][256];
  35 int rev;
  36 char head_message[16384];
  37 char write_file[2048];
  38
  39 char buffer[256];
  40 FILE *cmd_pipe;
  41
  42 bool find_path()
  43 {
  44     printf("+ finding path\n");
  45     char cur_path[2048], *ptr;
  46     getcwd(cur_path, 2048);
  47     int len = strnlen(cur_path, 2048);
  48
  49     if(cur_path[len-1] == '/' || cur_path[len-1] == '\\')
  50     {
  51         // we're in root, don't bother
  52         return false;
  53     }
  54
  55     // don't count the root
  56     int count_fwd = 0, count_back = 0;
  57     for(ptr = cur_path-1; ptr = strchr(ptr+1, '/'); count_fwd++);
  58     for(ptr = cur_path-1; ptr = strchr(ptr+1, '\\'); count_back++);
  59     int count = std::max(count_fwd, count_back);
  60
  61     char prefix[2048] = "", path[2048];
  62     for(int i = 0; i < count; i++)
  63     {
  64         snprintf(path, 2048, "%s.git", prefix);
  65         if(0 == chdir(path))
  66         {
  67             chdir(cur_path);
  68             snprintf(write_file, 2048, "%ssrc/shared/revision_nr.h", prefix);
  69             return true;
  70         }
  71         strncat(prefix, "../", 2048);
  72     }
  73
  74     return false;
  75 }
  76
  77 bool find_origin()
  78 {
  79     printf("+ finding origin\n");
  80     if( (cmd_pipe = popen( "git remote -v", "r" )) == NULL )
  81         return false;
  82
  83     bool ret = false;
  84     while(fgets(buffer, 256, cmd_pipe))
  85     {
  86         char name[256], remote[256];
  87         sscanf(buffer, "%s %s", name, remote);
  88         for(int i = 0; i < NUM_REMOTES; i++)
  89         {
  90             if(strcmp(remote, remotes[i]) == 0)
  91             {
  92                 strncpy(origins[i], name, 256);
  93                 ret = true;
  94             }
  95         }
  96     }
  97     pclose(cmd_pipe);
  98     return ret;
  99 }
 100
 101 bool fetch_origin()
 102 {
 103     printf("+ fetching origin\n");
 104     char cmd[256];
 105     // use the public clone url if present because the private may require a password
 106     sprintf(cmd, "git fetch %s master", origins[1][0] ? origins[1] : origins[0]);
 107     int ret = system(cmd);
 108     return true;
 109 }
 110
 111 bool check_fwd()
 112 {
 113     printf("+ checking fast forward\n");
 114     char cmd[256];
 115     sprintf(cmd, "git log -n 1 --pretty=\"format:%%H\" %s/master", origins[1][0] ? origins[1] : origins[0]);
 116     if( (cmd_pipe = popen( cmd, "r" )) == NULL )
 117         return false;
 118
 119     char hash[256];
 120     if(!fgets(buffer, 256, cmd_pipe)) return false;
 121     strncpy(hash, buffer, 256);
 122     pclose(cmd_pipe);
 123
 124     if( (cmd_pipe = popen( "git log --pretty=\"format:%H\"", "r" )) == NULL )
 125         return false;
 126
 127     bool found = false;
 128     while(fgets(buffer, 256, cmd_pipe))
 129     {
 130         buffer[strnlen(buffer, 256) - 1] = '\0';
 131         if(strncmp(hash, buffer, 256) == 0)
 132         {
 133             found = true;
 134             break;
 135         }
 136     }
 137     pclose(cmd_pipe);
 138
 139     if(!found) printf("WARNING: non-fastforward, use rebase!\n");
 140     return true;
 141 }
 142
 143 int get_rev(const char *from_msg)
 144 {
 145     // accept only the rev number format, not the sql update format
 146     char nr_str[256];
 147     if(sscanf(from_msg, "[%[0123456789]]", nr_str) != 1) return 0;
 148     if(from_msg[strlen(nr_str)+1] != ']') return 0;
 149     return atoi(nr_str);
 150 }
 151
 152 bool find_rev()
 153 {
 154     printf("+ finding next revision number\n");
 155     // find the highest rev number on either of the remotes
 156     for(int i = 0; i < NUM_REMOTES; i++)
 157     {
 158         if(!local && !origins[i][0]) continue;
 159
 160         char cmd[512];
 161         if(local) sprintf(cmd, "git log HEAD --pretty=\"format:%%s\"");
 162         else sprintf(cmd, "git log %s/master --pretty=\"format:%%s\"", origins[i]);
 163         if( (cmd_pipe = popen( cmd, "r" )) == NULL )
 164             continue;
 165
 166         int nr;
 167         while(fgets(buffer, 256, cmd_pipe))
 168         {
 169             nr = get_rev(buffer);
 170             if(nr >= rev)
 171                 rev = nr+1;
 172         }
 173         pclose(cmd_pipe);
 174     }
 175
 176     if(rev > 0) printf("Found [%d].\n", rev);
 177
 178     return rev > 0;
 179 }
 180
 181 std::string generateHeader(char const* rev_str)
 182 {
 183     std::ostringstream newData;
 184     newData << "#ifndef __REVISION_NR_H__" << std::endl;
 185     newData << "#define __REVISION_NR_H__"  << std::endl;
 186     newData << " #define REVISION_NR \"" << rev_str << "\"" << std::endl;
 187     newData << "#endif // __REVISION_NR_H__" << std::endl;
 188     return newData.str();
 189 }
 190
 191 bool write_rev()
 192 {
 193     printf("+ writing revision_nr.h\n");
 194     char rev_str[256];
 195     sprintf(rev_str, "%d", rev);
 196     std::string header = generateHeader(rev_str);
 197
 198     if(FILE* OutputFile = fopen(write_file,"wb"))
 199     {
 200         fprintf(OutputFile,"%s", header.c_str());
 201         fclose(OutputFile);
 202         return true;
 203     }
 204
 205     return false;
 206 }
 207
 208 bool find_head_msg()
 209 {
 210     printf("+ finding last message on HEAD\n");
 211     if( (cmd_pipe = popen( "git log -n 1 --pretty=\"format:%s%n%n%b\"", "r" )) == NULL )
 212         return false;
 213
 214     int poz = 0;
 215     while(poz < 16384-1 && EOF != (head_message[poz++] = fgetc(cmd_pipe)));
 216     head_message[poz-1] = '\0';
 217
 218     pclose(cmd_pipe);
 219
 220     if(int head_rev = get_rev(head_message))
 221     {
 222         if(!allow_replace)
 223         {
 224             printf("Last commit on HEAD is [%d]. Use -r to replace it with [%d].\n", head_rev, rev);
 225             return false;
 226         }
 227
 228         // skip the rev number in the commit
 229         char *p = strchr(head_message, ']'), *q = head_message;
 230         assert(p && *(p+1));
 231         p+=2;
 232         while(*p) *q = *p, p++, q++;
 233         *q = 0;
 234         return true;
 235     }
 236
 237     return true;
 238 }
 239
 240 bool amend_commit()
 241 {
 242     printf("+ amending last commit\n");
 243     char cmd[512];
 244     sprintf(cmd, "git commit --amend -F- %s", write_file);
 245     if( (cmd_pipe = popen( cmd, "w" )) == NULL )
 246         return false;
 247
 248     fprintf(cmd_pipe, "[%d] %s", rev, head_message);
 249     pclose(cmd_pipe);
 250
 251     return true;
 252 }
 253
 254 #define DO(cmd) if(!cmd) { printf("FAILED\n"); return 1; }
 255
 256 int main(int argc, char *argv[])
 257 {
 258     for(int i = 1; i < argc; i++)
 259     {
 260         if(argv[i] == NULL) continue;
 261         if(strncmp(argv[i], "-r", 2) == 0 || strncmp(argv[i], "--replace", 2) == 0)
 262             allow_replace = true;
 263         if(strncmp(argv[i], "-l", 2) == 0 || strncmp(argv[i], "--local", 2) == 0)
 264             local = true;
 265         if(strncmp(argv[i], "-f", 2) == 0 || strncmp(argv[i], "--fetch", 2) == 0)
 266             do_fetch = true;
 267         if(strncmp(argv[i], "-h", 2) == 0 || strncmp(argv[i], "--help", 2) == 0)
 268         {
 269             printf("Usage: git_id [OPTION]\n");
 270             printf("Generates a new rev number and updates revision_nr.h and the commit message.\n");
 271             printf("Should be used just before push.\n");
 272             printf("   -h, --help     show the usage\n");
 273             printf("   -r, --replace  replace the rev number if it was already applied to the last\n");
 274             printf("                  commit\n");
 275             printf("   -l, --local    search for the highest rev number on HEAD\n");
 276             printf("   -f, --fetch    fetch from origin before searching for the new rev\n");
 277             return 0;
 278         }
 279     }
 280
 281     DO( find_path()     );
 282     if(!local)
 283     {
 284         DO( find_origin()   );
 285         if(do_fetch)
 286         {
 287             DO( fetch_origin()  );
 288             DO( check_fwd()     );
 289         }
 290     }
 291     DO( find_rev()      );
 292     DO( find_head_msg() );
 293     DO( write_rev()     );
 294     DO( amend_commit()  );
 295
 296     return 0;
 297 }