git.roojs.org Git - gitlive/blob - GitRepo.vala

   1
   2 /**
   3  * @class Scm.Git.Repo
   4  *
   5  * @extends Scm.Repo
   6  *
   7  *
   8  *
   9  */
  10 static GitRepo  _GitRepo;
  11
  12 public class GitRepo : Object
  13 {
  14
  15     public Gee.ArrayList<GitMonitorQueue> cmds;
  16
  17     public string name;
  18     public string gitdir;
  19     public string git_working_dir;
  20     public bool debug = false;
  21
  22     public Gee.HashMap<string,bool> ignore_files;
  23     public GitBranch currentBranch;
  24
  25         public RooTicket? activeTicket;
  26
  27         public static GitRepo singleton()
  28     {
  29         if (_GitRepo == null) {
  30             _GitRepo = new GitRepo.single();
  31             _GitRepo.cache = new Gee.HashMap<string,GitRepo>();
  32         }
  33         return _GitRepo;
  34     }
  35
  36     /**
  37     * index of.. matching gitpath..
  38     */
  39     public static int indexOf( Array<GitRepo> repos, string gitpath) {
  40         // make a fake object to compare against..
  41         var test_repo = GitRepo.get(gitpath);
  42
  43         for(var i =0; i < repos.length; i++) {
  44             if (repos.index(i).gitdir == test_repo.gitdir) {
  45                 return i;
  46             }
  47         }
  48         return -1;
  49
  50     }
  51
  52     public  Gee.HashMap<string,GitRepo> cache;
  53
  54
  55
  56     public static   Array<GitRepo> list()
  57     {
  58
  59         //if (GitRepo.list_cache !=  null) {
  60         //    unowned  Array<GitRepo>    ret = GitRepo.list_cache;
  61          //   return ret;
  62         //}
  63         var cache = GitRepo.singleton().cache;
  64         var list_cache = new Array<GitRepo>();
  65
  66         var dir = Environment.get_home_dir() + "/gitlive";
  67
  68         var f = File.new_for_path(dir);
  69         FileEnumerator file_enum;
  70         try {
  71             file_enum = f.enumerate_children(
  72                 FileAttribute.STANDARD_DISPLAY_NAME + ","+
  73                 FileAttribute.STANDARD_TYPE,
  74                 FileQueryInfoFlags.NONE,
  75                 null);
  76         } catch (Error e) {
  77
  78             return list_cache;
  79
  80         }
  81
  82         FileInfo next_file;
  83
  84         while (true) {
  85
  86             try {
  87                 next_file = file_enum.next_file(null);
  88                 if (next_file == null) {
  89                     break;
  90                 }
  91
  92             } catch (Error e) {
  93                 GLib.debug("Error: %s",e.message);
  94                 break;
  95             }
  96
  97             //print("got a file " + next_file.sudo () + '?=' + Gio.FileType.DIRECTORY);
  98
  99             if (next_file.get_file_type() !=  FileType.DIRECTORY) {
 100                 next_file = null;
 101                 continue;
 102             }
 103
 104             if (next_file.get_file_type() ==  FileType.SYMBOLIC_LINK) {
 105                 next_file = null;
 106                 continue;
 107             }
 108
 109             if (next_file.get_display_name()[0] == '.') {
 110                 next_file = null;
 111                 continue;
 112             }
 113             var sp = dir+"/"+next_file.get_display_name();
 114
 115             var gitdir = dir + "/" + next_file.get_display_name() + "/.git";
 116
 117             if (!FileUtils.test(gitdir, FileTest.IS_DIR)) {
 118                 continue;
 119             }
 120
 121                 var rep =  GitRepo.get(  sp );
 122                 list_cache.append_val(rep);
 123
 124         }
 125
 126         return list_cache;
 127
 128
 129
 130         }
 131
 132         public static GitRepo get(string path)
 133         {
 134                 var cache = GitRepo.singleton().cache;
 135                 if (cache.has_key(path)) {
 136                         return cache.get(path);
 137                 }
 138                 return new GitRepo(path);
 139         }
 140
 141     private GitRepo.single() {
 142                 // used to create the signleton
 143         }
 144     /**
 145      * constructor:
 146      *
 147      * @param {Object} cfg - Configuration
 148      *     (basically repopath is currently only critical one.)
 149      *
 150      */
 151
 152     private GitRepo(string path) {
 153         // cal parent?
 154         this.name =   File.new_for_path(path).get_basename();
 155         this.ignore_files = new Gee.HashMap<string,bool>();
 156
 157         this.git_working_dir = path;
 158         this.gitdir = path + "/.git";
 159         if (!FileUtils.test(this.gitdir , FileTest.IS_DIR)) {
 160             this.gitdir = path; // naked...
 161         }
 162         this.cmds = new  Gee.ArrayList<GitMonitorQueue> ();
 163
 164                 var cache = GitRepo.singleton().cache;
 165         //Repo.superclass.constructor.call(this,cfg);
 166                 if ( !cache.has_key(path) ) {
 167                         cache.set( path, this);
 168         }
 169         this.loadBranches();
 170         this.loadActiveTicket();
 171     }
 172
 173     public bool is_wip_branch()
 174     {
 175         return this.currentBranch.name.has_prefix("wip_");
 176
 177     }
 178
 179     public bool is_autocommit ()
 180     {
 181         return !FileUtils.test(this.gitdir + "/.gitlive-disable-autocommit" , FileTest.EXISTS);
 182     }
 183     public void set_autocommit(bool val)
 184     {
 185
 186                 var cur = this.is_autocommit();
 187                 GLib.debug("SET auto commit : %s <= %s", val ? "ON" : "OFF",  cur  ? "ON" : "OFF");
 188                 if (cur == val) {
 189                         return; // no change..
 190                 }
 191                 if (!val) {
 192                         FileUtils.set_contents(this.gitdir + "/.gitlive-disable-autocommit" , "x");
 193                 } else {
 194                         // it exists...
 195                         FileUtils.remove(this.gitdir + "/.gitlive-disable-autocommit" );
 196                 }
 197
 198     }
 199
 200     public bool is_auto_branch ()
 201     {
 202         return FileUtils.test(this.gitdir + "/.gitlive-enable-auto-branch" , FileTest.EXISTS);
 203     }
 204
 205     public void set_auto_branch(bool val)
 206     {
 207
 208                 var cur = this.is_auto_branch();
 209                 GLib.debug("SET auto branch : %s <= %s", val ? "ON" : "OFF",  cur  ? "ON" : "OFF");
 210
 211                 if (cur == val) {
 212                         return; // no change..
 213                 }
 214                 if (val) {
 215                         FileUtils.set_contents(this.gitdir + "/.gitlive-enable-auto-branch" , "x");
 216                 } else {
 217                         // it exists...
 218                         FileUtils.remove(this.gitdir + "/.gitlive-enable-auto-branch" );
 219                 }
 220
 221     }
 222     public bool is_autopush ()
 223     {
 224         return !FileUtils.test(this.gitdir + "/.gitlive-disable-autopush" , FileTest.EXISTS);
 225     }
 226     public void set_autopush(bool val)
 227     {
 228
 229                 var cur = this.is_autopush();
 230                 GLib.debug("SET auto push : %s <= %s", val ? "ON" : "OFF",  cur  ? "ON" : "OFF");
 231                 if (cur == val) {
 232                         return; // no change..
 233                 }
 234                 if (!val) {
 235                         FileUtils.set_contents(this.gitdir + "/.gitlive-disable-autopush" , "");
 236                 } else {
 237                         // it exists...
 238                         FileUtils.remove(this.gitdir + "/.gitlive-disable-autopush" );
 239                 }
 240
 241     }
 242
 243
 244
 245     Gee.HashMap<string,GitBranch> branches;
 246
 247     public void loadBranches()
 248     {
 249         this.branches = new Gee.HashMap<string,GitBranch>();
 250
 251         string[] cmd = { "branch",   "--no-color", "--verbose", "--no-abbrev" , "-a"  };
 252         var res = this.git( cmd );
 253         var lines = res.split("\n");
 254         for (var i = 0; i < lines.length ; i++) {
 255                 var br = new GitBranch(this);
 256                 if (!br.parseBranchListItem(lines[i])) {
 257                         continue;
 258                 }
 259                 GLib.debug("add branch %s", br.realName());
 260
 261                 branches.set(br.realName(), br);
 262                 if (br.active) {
 263                         this.currentBranch = br;
 264                 }
 265         }
 266
 267     }
 268
 269
 270
 271
 272     public string branchesToString()
 273     {
 274         var ret = "";
 275                 foreach( var br in this.branches.values) {
 276                         if (br.name == "") {
 277                                 continue;
 278                         }
 279                         ret += ret.length > 0 ? ","  : "";
 280                         ret += br.name;
 281
 282                 }
 283                 return ret;
 284
 285     }
 286      public static void doMerges(string action, string ticket_id, string commit_message)
 287     {
 288        GitMonitor.gitmonitor.stop();
 289
 290        var commitrevs = "";
 291        var sucess = true;
 292        foreach(var  repo in GitRepo.singleton().cache.values) {
 293                if (repo.activeTicket != null && repo.activeTicket.id == ticket_id) {
 294                        var res = repo.doMerge(action,ticket_id, commit_message);
 295                        if (!res) {
 296                                sucess = false;
 297                                continue;
 298                        }
 299                        commitrevs += commitrevs.length > 0 ? " " : "";
 300                        commitrevs += repo.currentBranch.lastrev;
 301                }
 302        }
 303        if (sucess && action == "CLOSE") {
 304                RooTicket.singleton().getById(ticket_id).close(commitrevs);
 305        }
 306        GitMonitor.gitmonitor.start();
 307     }
 308
 309
 310     public bool doMerge(string action, string ticket_id, string commit_message)
 311     {
 312        // in theory we should check to see if other repo's have got the same branch and merge all them at the same time.
 313        // also need to decide which branch we will merge into?
 314                    var ret = "";
 315                    if (action == "CLOSE" || action == "LEAVE") {
 316
 317                try {
 318                    var oldbranch = this.currentBranch.name;
 319                    this.setActiveTicket(null, "master");
 320                            string [] cmd = { "merge",   "--squash",  oldbranch };
 321                            this.git( cmd );
 322                            cmd = { "commit",   "-a" , "-m",  commit_message };
 323                            this.git( cmd );
 324                            this.push();
 325                            this.loadBranches(); // updates lastrev..
 326
 327                        var notification = new Notify.Notification(
 328                                "Merged branch %s to master".printf(oldbranch),
 329                                "",
 330                                 "dialog-information"
 331
 332                        );
 333
 334                        notification.set_timeout(5);
 335                        notification.show();
 336
 337                // close ticket..
 338                return true;
 339
 340            } catch (Error e) {
 341
 342                GitMonitor.gitmonitor.pauseError(e.message);
 343                return false;
 344            }
 345            // error~?? -- show the error dialog...
 346                    return false;
 347        }
 348        if (action == "MASTER") {
 349                // merge master into ours..
 350                        try {
 351                        string[] cmd = { "merge",  "master" };
 352                        this.git( cmd );
 353                        var notification = new Notify.Notification(
 354                                        "Merged code from master to %s".printf(this.currentBranch.name),
 355                                        "",
 356                                         "dialog-information"
 357
 358                                );
 359                                notification.set_timeout(5);
 360                                notification.show();
 361
 362                        return true;
 363                        } catch (Error e) {
 364                        GitMonitor.gitmonitor.pauseError(e.message);
 365                        return false;
 366                    }
 367            }
 368        if (action == "EXIT") {
 369                        try {
 370                        var oldbranch  = this.currentBranch.name;
 371                          this.setActiveTicket(null, "master");
 372                        this.loadBranches();
 373                        var notification = new Notify.Notification(
 374                                        "Left branch %s".printf(oldbranch),
 375                                        "",
 376                                         "dialog-information"
 377
 378                                );
 379                                notification.set_timeout(5);
 380                                notification.show();
 381
 382                        return true;
 383                    } catch (Error e) {
 384                        GitMonitor.gitmonitor.pauseError(e.message);
 385
 386                        return false;
 387                    }
 388                    // error~?? -- show the error dialog...
 389
 390        }
 391        return false;
 392     }
 393
 394     public void loadActiveTicket()
 395     {
 396         this.activeTicket = null;
 397         if (!FileUtils.test(this.gitdir + "/.gitlive-active-ticket" , FileTest.EXISTS)) {
 398                 return;
 399         }
 400         string ticket_id;
 401         FileUtils.get_contents(this.gitdir + "/.gitlive-active-ticket" , out ticket_id);
 402         if (ticket_id.length < 1) {
 403                 return;
 404                 }
 405                 this.activeTicket = RooTicket.singleton().getById(ticket_id.strip());
 406
 407
 408     }
 409
 410
 411
 412     public bool setActiveTicket(RooTicket? ticket, string branchname)
 413     {
 414         if (!this.createBranchNamed(branchname)) {
 415                 return false;
 416                 }
 417                 if (ticket != null) {
 418                 FileUtils.set_contents(this.gitdir + "/.gitlive-active-ticket" , ticket.id);
 419         } else {
 420                 FileUtils.remove(this.gitdir + "/.gitlive-active-ticket" );
 421         }
 422         this.activeTicket = ticket;
 423         return true;
 424     }
 425
 426     public bool createBranchNamed(string branchname)
 427     {
 428
 429                         var stash = false;
 430                      if (this.branches.has_key(branchname)) {
 431                         // this is where it get's tricky...
 432                                 try {
 433                                                 string[] cmd = { "ls-files" ,  "-m" };                   // list the modified files..
 434                                             var ret = this.git(cmd);
 435                                             stash = ret.length> 1 ;
 436
 437
 438                                             cmd = { "stash" };
 439                                             if (stash) { this.git(cmd); }
 440
 441                                             cmd = { "checkout", branchname  };
 442                                             this.git(cmd);
 443                                   } catch(Error e) {
 444                                                 GitMonitor.gitmonitor.pauseError(e.message);
 445                                                 return false;
 446                                   }
 447                                   try {
 448                                            if (branchname != "master") {
 449                                                string[] cmd = { "merge", "master"  };
 450                                                     this.git(cmd);
 451                                                     this.push();
 452                                                //cmd = { "commit",   "-a" , "-m",  "merge master changes" };
 453                                        //git chethis.git( cmd );
 454                                             }
 455
 456                                    } catch(Error e) {
 457                                             string[] cmd = { "checkout", "master"  };
 458                                             this.git(cmd);
 459                                                 GitMonitor.gitmonitor.pauseError(
 460                                                         "Use\n\na) git checkout %s\nb) git mergetool\nc) git commit\nd) git push\n d) stash pop \ne) start gitlive again\n".printf(
 461                                                                 branchname)
 462                                                          + e.message
 463                                                 );
 464                                                 return false;
 465
 466                                         }
 467                                    try {
 468                                            string[]  cmd = { "stash", "pop"  };
 469                                             if (stash) { this.git(cmd); }
 470                                         } catch(Error ee) {
 471                                                 GitMonitor.gitmonitor.pauseError(ee.message);
 472                                                 return false;
 473                                         }
 474                    this.push();
 475
 476                     } else {
 477                                     try {
 478
 479                                         string[] cmd = { "checkout", "-b" , branchname  };
 480                                         this.git(cmd);
 481
 482                        cmd = { "push", "-u" , "origin" ,"HEAD"  };
 483                                         this.git(cmd);
 484
 485                                         } catch(Error ee) {
 486                                                 GitMonitor.gitmonitor.pauseError(ee.message);
 487                                                 return false;
 488                                         }
 489
 490                     }
 491                        var notification = new Notify.Notification(
 492                        "Changed to branch %s".printf(branchname),
 493                        "",
 494                         "dialog-information"
 495
 496                );
 497
 498                notification.set_timeout(5);
 499                notification.show();
 500
 501
 502          this.loadBranches(); // update branch list...
 503          //GitMonitor.gitmonitor.runQueue(); // no point - we have hidden the queue..
 504          return true;
 505     }
 506
 507
 508     /**
 509      * add:
 510      * add files to track.
 511      *
 512      * @argument {Array} files the files to add.
 513      */
 514     public string add ( Gee.ArrayList<GitMonitorQueue> files ) throws Error, SpawnError
 515     {
 516         // should really find out if these are untracked files each..
 517         // we run multiple versions to make sure that if one failes, it does not ignore the whole lot..
 518         // not sure if that is how git works.. but just be certian.
 519         var ret = "";
 520         for (var i = 0; i < files.size;i++) {
 521             var f = files.get(i).vname;
 522             try {
 523                 string[] cmd = { "add",    f  };
 524                 this.git( cmd );
 525             } catch (Error e) {
 526                 ret += e.message  + "\n";
 527             }
 528
 529         }
 530         return ret;
 531     }
 532
 533     public bool is_ignore(string fname) throws Error, SpawnError
 534     {
 535                 if (fname == ".gitignore") {
 536                         this.ignore_files.clear();
 537                 }
 538
 539                 if (this.ignore_files.has_key(fname)) {
 540                         return this.ignore_files.get(fname);
 541                 }
 542
 543                 try {
 544                         var ret = this.git( { "check-ignore" , fname } );
 545                         this.ignore_files.set(fname, ret.length >  0);
 546                         return ret.length > 0;
 547                 } catch (SpawnError e) {
 548                         this.ignore_files.set(fname, false);
 549                         return false;
 550                 }
 551
 552     }
 553
 554
 555       /**
 556      * remove:
 557      * remove files to track.
 558      *
 559      * @argument {Array} files the files to add.
 560      */
 561     public string remove  ( Gee.ArrayList<GitMonitorQueue> files ) throws Error, SpawnError
 562     {
 563         // this may fail if files do not exist..
 564         // should really find out if these are untracked files each..
 565         // we run multiple versions to make sure that if one failes, it does not ignore the whole lot..
 566         // not sure if that is how git works.. but just be certian.
 567         var ret = "";
 568
 569         for (var i = 0; i < files.size;i++) {
 570             var f = files.get(i).vname;
 571             try {
 572                 string[] cmd = { "rm",  "-f" ,  f  };
 573                 this.git( cmd );
 574             } catch (Error e) {
 575                 ret += e.message  + "\n";
 576             }
 577         }
 578
 579         return ret;
 580
 581     }
 582
 583
 584     /**
 585      * commit:
 586      * perform a commit.
 587      *
 588      * @argument {Object} cfg commit configuration
 589      *
 590      * @property {String} name (optional)
 591      * @property {String} email (optional)
 592      * @property {String} changed (date) (optional)
 593      * @property {String} reason (optional)
 594      * @property {Array} files - the files that have changed.
 595      *
 596      */
 597
 598     public string commit ( string message, Gee.ArrayList<GitMonitorQueue> files  ) throws Error, SpawnError
 599     {
 600
 601
 602         /*
 603         var env = [];
 604
 605         if (typeof(cfg.name) != 'undefined') {
 606             args.push( {
 607                 'author' : cfg.name + ' <' + cfg.email + '>'
 608             });
 609             env.push(
 610                 "GIT_COMMITTER_NAME" + cfg.name,
 611                 "GIT_COMMITTER_EMAIL" + cfg.email
 612             );
 613         }
 614
 615         if (typeof(cfg.changed) != 'undefined') {
 616             env.push("GIT_AUTHOR_DATE= " + cfg.changed )
 617
 618         }
 619         */
 620         string[] args = { "commit", "-m" };
 621         args +=  (message.length > 0  ? message : "Changed" );
 622         for (var i = 0; i< files.size ; i++ ) {
 623             args += files.get(i).vname; // full path?
 624         }
 625
 626         return this.git(args);
 627     }
 628
 629     /**
 630      * pull:
 631      * Fetch and merge remote repo changes into current branch..
 632      *
 633      * At present we just need this to update the current working branch..
 634      * -- maybe later it will have a few options and do more stuff..
 635      *
 636      */
 637     public string pull () throws Error, SpawnError
 638     {
 639         // should probably hand error conditions better...
 640         string[] cmd = { "pull" , "--no-edit" };
 641         return this.git( cmd );
 642
 643
 644     }
 645
 646     public delegate void GitAsyncCallback (GitRepo repo, int err, string str);
 647     public void pull_async(GitAsyncCallback cb)
 648     {
 649
 650         string[] cmd = { "pull" , "--no-edit" };
 651          this.git_async( cmd , cb);
 652
 653
 654     }
 655
 656     /**
 657      * push:
 658      * Send local changes to remote repo(s)
 659      *
 660      * At present we just need this to push the current branch.
 661      * -- maybe later it will have a few options and do more stuff..
 662      *
 663      */
 664     public string push () throws Error, SpawnError
 665     {
 666         // should
 667         return this.git({ "push"  });
 668
 669     }
 670
 671
 672
 673      /**
 674      * git:
 675      * The meaty part.. run spawn.. with git..
 676      *
 677      *
 678      */
 679
 680     public string git(string[] args_in ) throws Error, SpawnError
 681     {
 682         // convert arguments.
 683
 684         string[]  args = { "git" };
 685         //args +=  "--git-dir";
 686         //args +=  this.gitdir;
 687         args +=  "--no-pager";
 688
 689
 690         //if (this.gitdir != this.repopath) {
 691         //    args +=   "--work-tree";
 692          //   args += this.repopath;
 693         //}
 694         for (var i = 0; i < args_in.length;i++) {
 695             args += args_in[i];
 696         }
 697
 698         //this.lastCmd = args.join(" ");
 699         //if(this.debug) {
 700             GLib.debug( "CWD=%s",  this.git_working_dir );
 701             GLib.debug( "cmd: %s", string.joinv (" ", args));
 702         //}
 703
 704         string[]   env = {};
 705         string  home = "HOME=" + Environment.get_home_dir() ;
 706         env +=  home ;
 707         // do not need to set gitpath..
 708         //if (File.exists(this.repo + '/.git/config')) {
 709             //env.push("GITPATH=" + this.repo );
 710         //}
 711
 712
 713         var cfg = new SpawnConfig(this.git_working_dir , args , env);
 714
 715
 716        // may throw error...
 717         var sp = new Spawn(cfg);
 718
 719
 720         GLib.debug( "GOT: %s" , sp.output);
 721         // parse output for some commands ?
 722         return sp.output;
 723     }
 724
 725    unowned GitAsyncCallback git_async_on_callback;
 726         public void  git_async( string[] args_in,   GitAsyncCallback cb ) throws Error, SpawnError
 727     {
 728         // convert arguments.
 729        this.git_async_on_callback = cb;
 730         string[]  args = { "git" };
 731         //args +=  "--git-dir";
 732         //args +=  this.gitdir;
 733         args +=  "--no-pager";
 734
 735
 736         //if (this.gitdir != this.repopath) {
 737         //    args +=   "--work-tree";
 738          //   args += this.repopath;
 739         //}
 740         for (var i = 0; i < args_in.length;i++) {
 741             args += args_in[i];
 742         }
 743
 744         //this.lastCmd = args.join(" ");
 745         //if(this.debug) {
 746             GLib.debug( "CWD=%s",  this.git_working_dir );
 747             //print( "cmd: %s\n", string.joinv (" ", args));
 748         //}
 749
 750         string[]   env = {};
 751         string  home = "HOME=" + Environment.get_home_dir() ;
 752         env +=  home ;
 753         // do not need to set gitpath..
 754         //if (File.exists(this.repo + '/.git/config')) {
 755             //env.push("GITPATH=" + this.repo );
 756         //}
 757
 758
 759         var cfg = new SpawnConfig(this.git_working_dir , args , env);
 760         cfg.async = true;
 761
 762
 763        // may throw error...
 764         var sp = new Spawn(cfg);
 765                 //sp.ref();
 766         //this.ref();
 767         sp.run(this.git_async_on_complete);
 768
 769     }
 770
 771     void git_async_on_complete(int err, string output)
 772     {
 773                 GLib.debug("GOT %d : %s", err, output);
 774                 this.git_async_on_callback(this, err, output);
 775 //              this.unref();
 776         //      sp.unref();
 777
 778
 779     }
 780
 781 }