From d8add139e1e816ff2653e28478a7b45081af7537 Mon Sep 17 00:00:00 2001
From: Rebecca Turner <rbt@sent.as>
Date: Tue, 15 Nov 2022 11:12:54 -0500
Subject: [PATCH] Add Treesitter support to Vim upgrade (#184)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* Add `:TSUpdate` to Vim upgrade

* Reformat `freebsd.rs`

Co-authored-by: Thomas Schönauer <37108907+DottoDev@users.noreply.github.com>
---
 src/steps/os/freebsd.rs | 2 ++
 src/steps/upgrade.vim   | 8 ++++++++
 2 files changed, 10 insertions(+)

diff --git a/src/steps/os/freebsd.rs b/src/steps/os/freebsd.rs
index 2c57b77c..67488f62 100644
--- a/src/steps/os/freebsd.rs
+++ b/src/steps/os/freebsd.rs
@@ -1,4 +1,6 @@
 use crate::command::CommandExt;
+use crate::config::Step;
+use crate::execution_context::ExecutionContext;
 use crate::executor::RunType;
 use crate::terminal::print_separator;
 use crate::utils::require_option;
diff --git a/src/steps/upgrade.vim b/src/steps/upgrade.vim
index 28e3df9f..5eac813e 100644
--- a/src/steps/upgrade.vim
+++ b/src/steps/upgrade.vim
@@ -38,6 +38,14 @@ if exists(":CocUpdateSync")
     CocUpdateSync
 endif
 
+" TODO: Should this be after `PackerSync`?
+" Not sure how to sequence this after Packer without doing something weird
+" with that `PackerComplete` autocommand.
+if exists(":TSUpdate")
+    echo "TreeSitter Update"
+    TSUpdate
+endif
+
 if exists(':PackerSync')
   echo "Packer"
   autocmd User PackerComplete quitall