各コマンド発火時にツリーデータをリフレッシュ

Author: neet

src/commands/compose.ts

@@ -3,6 +3,7 @@ import { commands, Uri, window, workspace } from 'vscode';
 import * as nls from 'vscode-nls';
 import { client } from '../client';
 import { configuration } from '../configuration';
+import { qiitaItemsProvider } from '../explorer/qiitaItems';
 import { tagQuickPickCreator, validateTagQuickPick } from '../quickpicks/tagQuickPickCreator';
 import { titleInputBoxCreator, validateTitleInputBox } from '../quickpicks/titleInputBoxCreator';
 import { privateLabel, visibilityQuickPickCreator } from '../quickpicks/visibilityQuickPickCreator';
@@ -84,6 +85,7 @@ export async function compose (resource?: { path: string }) {
 
     try {
       const item = await client.createItem(options);
+      qiitaItemsProvider.refresh();
 
       const openInBrowser = localize(
         'commands.compose.openInBrowser',

src/commands/deleteItem.ts

@@ -2,6 +2,7 @@ import { Item } from 'qiita-js-2';
 import { window } from 'vscode';
 import * as nls from 'vscode-nls';
 import { client } from '../client';
+import { qiitaItemsProvider } from '../explorer/qiitaItems';
 import { handleErrorMessage } from '../utils/errorHandler';
 
 const localize = nls.loadMessageBundle();
@@ -34,6 +35,7 @@ export async function deleteItem (resource: { item: Item }) {
 
   try {
     await client.deleteItem(resource.item.id);
+    qiitaItemsProvider.refresh();
 
     return window.showInformationMessage(localize(
       'commands.deleteItem.success',

src/commands/editTags.ts

@@ -2,6 +2,7 @@ import { Item } from 'qiita-js-2';
 import { QuickPickItem, window } from 'vscode';
 import * as nls from 'vscode-nls';
 import { client } from '../client';
+import { qiitaItemsProvider } from '../explorer/qiitaItems';
 import { makeQuickPickItemFromTag, tagQuickPickCreator, validateTagQuickPick } from '../quickpicks/tagQuickPickCreator';
 import { handleErrorMessage } from '../utils/errorHandler';
 
@@ -46,8 +47,8 @@ export async function editTags (resource: { item: Item }) {
 
     try {
       quickPick.hide();
-
       await updater(item, quickPick.selectedItems);
+      qiitaItemsProvider.refresh();
 
       return window.showInformationMessage(localize(
         'commands.editTags.success',

src/commands/editTitle.ts

@@ -2,6 +2,7 @@ import { Item } from 'qiita-js-2';
 import { window } from 'vscode';
 import * as nls from 'vscode-nls';
 import { client } from '../client';
+import { qiitaItemsProvider } from '../explorer/qiitaItems';
 import { titleInputBoxCreator, validateTitleInputBox } from '../quickpicks/titleInputBoxCreator';
 import { handleErrorMessage } from '../utils/errorHandler';
 
@@ -25,12 +26,12 @@ export function editTitle (resource: { item: Item }) {
 
     try {
       inputBox.hide();
-
       await client.updateItem(item.id, {
         body: item.body,
         tags: item.tags,
         title: inputBox.value,
       });
+      qiitaItemsProvider.refresh();
 
       return window.showInformationMessage(localize(
         'commands.editTitle.success',

src/commands/makePublic.ts

@@ -2,6 +2,7 @@ import { Item } from 'qiita-js-2';
 import { window } from 'vscode';
 import * as nls from 'vscode-nls';
 import { client } from '../client';
+import { qiitaItemsProvider } from '../explorer/qiitaItems';
 import { handleErrorMessage } from '../utils/errorHandler';
 
 const localize = nls.loadMessageBundle();
@@ -46,6 +47,7 @@ export async function makePublic (resource: { item: Item }) {
       tags: resource.item.tags,
       private: false,
     });
+    qiitaItemsProvider.refresh();
 
     return window.showInformationMessage(localize(
       'commands.makePublic.success',

src/explorer/qiitaItems.ts

@@ -1,11 +1,48 @@
 import { Item } from 'qiita-js-2';
-import { Command, TreeDataProvider, TreeItem, TreeItemCollapsibleState, Uri } from 'vscode';
+import {
+  Command,
+  Event,
+  EventEmitter,
+  TreeDataProvider,
+  TreeItem,
+  TreeItemCollapsibleState,
+  Uri,
+} from 'vscode';
 import * as nls from 'vscode-nls';
 import { client } from '../client';
 
 const localize = nls.loadMessageBundle();
 
-export class QiitaItemProvider implements TreeDataProvider<QiitaItem> {
+export class QiitaItem extends TreeItem {
+  /**
+   * "Qiitaの投稿" ビューに表示されるアイテム
+   * @param item Qiitaの投稿
+   * @param collapsibleState アイテムが折り畳まれているかの状態
+   * @param command クリック時に発火するコマンド
+   */
+  constructor (
+    public readonly item: Item,
+    public readonly collapsibleState: TreeItemCollapsibleState,
+    public readonly command: Command,
+  ) {
+    super(item.title, collapsibleState);
+  }
+
+  public resourceUri  = Uri.parse('file:///text.md'); // Hack: アイコンをMarkdownのものに
+  public contextValue = 'qiitaItems';
+}
+
+
+class QiitaItemsProvider implements TreeDataProvider<QiitaItem> {
+  private _onDidChangeTreeData: EventEmitter<QiitaItem | undefined> = new EventEmitter<QiitaItem | undefined>();
+  public readonly onDidChangeTreeData: Event<QiitaItem | undefined> = this._onDidChangeTreeData.event;
+
+  /**
+   * ツリーデータを更新
+   */
+  public refresh () {
+    this._onDidChangeTreeData.fire();
+  }
 
   /**
    * `element` に対応するツリーアイテムを取得
@@ -20,11 +57,7 @@ export class QiitaItemProvider implements TreeDataProvider<QiitaItem> {
    * 小要素を取得
    * @param element 取得するelement
    */
-  public async getChildren (/* element?: QiitaItem */): Promise<QiitaItem[]> {
-    // if (element) {
-    //   return array of item
-    // }
-
+  public async getChildren (): Promise<QiitaItem[]> {
     const { value: items } = await client.fetchMyItems({page: 1, per_page: 60}).next();
 
     return items.map((item) => {
@@ -39,22 +72,4 @@ export class QiitaItemProvider implements TreeDataProvider<QiitaItem> {
   }
 }
 
-export class QiitaItem extends TreeItem {
-  /**
-   * "Qiitaの投稿" ビューに表示されるアイテム
-   * @param item Qiitaの投稿
-   * @param collapsibleState アイテムが折り畳まれているかの状態
-   * @param command クリック時に発火するコマンド
-   */
-  constructor (
-    public readonly item: Item,
-    public readonly collapsibleState: TreeItemCollapsibleState,
-    public readonly command: Command,
-  ) {
-    super(item.title, collapsibleState);
-  }
-
-  public resourceUri = Uri.parse('file:///text.md');
-
-  public contextValue = 'qiitaItems';
-}
+export const qiitaItemsProvider = new QiitaItemsProvider();

src/extension.ts

@@ -7,13 +7,13 @@ import { editTitle } from './commands/editTitle';
 import { makePublic } from './commands/makePublic';
 import { openItem } from './commands/openItem';
 import { openItemExternal } from './commands/openItemExternal';
-import { QiitaItemProvider } from './explorer/qiitaItems';
+import { qiitaItemsProvider } from './explorer/qiitaItems';
 import './polyfills';
 
 nls.config(process.env.VSCODE_NLS_CONFIG as nls.Options)();
 
 export function activate (context: ExtensionContext) {
-  window.registerTreeDataProvider('qiitaItems', new QiitaItemProvider());
+  window.registerTreeDataProvider('qiitaItems', qiitaItemsProvider);
 
   context.subscriptions.push(
     commands.registerCommand('qiita.openItem', openItem(context.storagePath)),