完成quickPick的判断
This commit is contained in:
huangsimin
parent
409ab24cb3
commit
b88cbd8391
318
src/extension.ts
318
src/extension.ts
|
|
@ -1,125 +1,229 @@
|
||||||
|
"use strict";
|
||||||
|
Object.defineProperty(exports, "__esModule", { value: true });
|
||||||
// The module 'vscode' contains the VS Code extensibility API
|
// The module 'vscode' contains the VS Code extensibility API
|
||||||
// Import the module and reference it with the alias vscode in your code below
|
// Import the module and reference it with the alias vscode in your code below
|
||||||
|
|
||||||
import * as vscode from 'vscode';
|
import * as vscode from 'vscode';
|
||||||
|
|
||||||
|
class StructInfo {
|
||||||
|
Name: string;
|
||||||
|
Range: number[];
|
||||||
|
Fields: Field[];
|
||||||
|
|
||||||
|
constructor(name: string, fields: Field[], range: number[]) {
|
||||||
|
this.Name = name;
|
||||||
|
this.Range = range;
|
||||||
|
this.Fields = fields;
|
||||||
|
}
|
||||||
|
|
||||||
|
getFieldsString(): string[] {
|
||||||
|
var result: string[] = [];
|
||||||
|
|
||||||
|
this.Fields.forEach( (field, index) => {
|
||||||
|
result.push( "("+index+") " + this.Name + field.Parent + field.Name + " " + field.Type);
|
||||||
|
});
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
class Field {
|
||||||
|
Parent: string;
|
||||||
|
Type: string;
|
||||||
|
Name: string;
|
||||||
|
Range: number[];
|
||||||
|
|
||||||
|
constructor(parent: string, type: string, name: string, range: number[]) {
|
||||||
|
this.Parent = parent;
|
||||||
|
this.Type = type;
|
||||||
|
this.Name = name;
|
||||||
|
this.Range = range;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// this method is called when your extension is activated
|
// this method is called when your extension is activated
|
||||||
// your extension is activated the very first time the command is executed
|
// your extension is activated the very first time the command is executed
|
||||||
export function activate(context: vscode.ExtensionContext) {
|
function activate(context: vscode.ExtensionContext) {
|
||||||
|
// Use the console to output diagnostic information (console.log) and errors (console.error)
|
||||||
|
// This line of code will only be executed once when your extension is activated
|
||||||
|
console.log('Congratulations, your extension "go-quickly-generator" is now active!');
|
||||||
|
context.subscriptions.push(vscode.commands.registerCommand('go-quickly-generator.go-gen-set', () => {
|
||||||
|
// The code you place here will be executed every time your command is executed
|
||||||
|
// Display a message box to the user
|
||||||
|
let sinfo = GetStruct();
|
||||||
|
if(sinfo) {
|
||||||
|
console.log(sinfo);
|
||||||
|
const options = <vscode.QuickPickOptions>{canPickMany: true, placeHolder: "select the fields that would be generator get set"};
|
||||||
|
vscode.window.showQuickPick(sinfo.getFieldsString(), options).then( (input)=>{
|
||||||
|
if(typeof(input) !== "string") {
|
||||||
|
var selections: string[] = input as any;
|
||||||
|
console.log(selections); // TODO:
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}));
|
||||||
|
|
||||||
|
|
||||||
// Use the console to output diagnostic information (console.log) and errors (console.error)
|
|
||||||
// This line of code will only be executed once when your extension is activated
|
|
||||||
console.log('Congratulations, your extension "go-quickly-generator" is now active!');
|
|
||||||
|
|
||||||
context.subscriptions.push( vscode.commands.registerCommand('go-quickly-generator.go-gen-set', () => {
|
context.subscriptions.push(vscode.commands.registerCommand('go-quickly-generator.allGetterAndSetter', function () {
|
||||||
// The code you place here will be executed every time your command is executed
|
let editor = vscode.window.activeTextEditor;
|
||||||
|
if (editor !== undefined) {
|
||||||
// Display a message box to the user
|
const currentPos = editor.selection.active;
|
||||||
let editor = vscode.window.activeTextEditor;
|
const lineCount = editor.document.lineCount;
|
||||||
|
// let selection = editor.selection;
|
||||||
if (editor !== undefined) {
|
// let lineText = editor.document.lineAt(selection.active);
|
||||||
if (editor.document.languageId !== 'go') {
|
// vscode.window.showInformationMessage( lineText.text );
|
||||||
vscode.window.showInformationMessage('file in the active editor is not a go file(*.go)');
|
}
|
||||||
return;
|
}));
|
||||||
}
|
|
||||||
|
|
||||||
let selection = editor.selection;
|
|
||||||
|
|
||||||
// vscode.window.showInformationMessage( editor.document.version.toString() );
|
|
||||||
// vscode.window.showInformationMessage( text );
|
|
||||||
|
|
||||||
let selectline = selection.active.line;
|
|
||||||
let regex = "type +([^ ]+) +struct";
|
|
||||||
// lineText.text.match()
|
|
||||||
|
|
||||||
for(let i = selectline; i >= 0; i--) {
|
|
||||||
|
|
||||||
let lineText = editor.document.lineAt(i);
|
|
||||||
let matchs = lineText.text.match(regex);
|
|
||||||
if(matchs) {
|
|
||||||
console.log(matchs[0], matchs[1]);
|
|
||||||
|
|
||||||
vscode.window.showInformationMessage( lineText.text + "," + i.toString() );
|
|
||||||
let open = 0;
|
|
||||||
BREAK_OPEN:
|
|
||||||
for(let n = i ; n < editor.document.lineCount ; n++){
|
|
||||||
|
|
||||||
let lineText = editor.document.lineAt(n);
|
|
||||||
for(let c = 0; c < lineText.text.length ; c ++) {
|
|
||||||
switch(lineText.text[c]) {
|
|
||||||
case '{':
|
|
||||||
open ++;
|
|
||||||
break;
|
|
||||||
case '}':
|
|
||||||
open --;
|
|
||||||
if(open === 0) {
|
|
||||||
if(n >= selectline) {
|
|
||||||
let structName = matchs[1];
|
|
||||||
// i(start) n(end)
|
|
||||||
|
|
||||||
getStructInfo(editor, structName, i, n);
|
|
||||||
}
|
|
||||||
break BREAK_OPEN;
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
}));
|
|
||||||
|
|
||||||
context.subscriptions.push(vscode.commands.registerCommand('go-quickly-generator.allGetterAndSetter', function () {
|
|
||||||
let editor = vscode.window.activeTextEditor;
|
|
||||||
if (editor !== undefined) {
|
|
||||||
const currentPos = editor.selection.active;
|
|
||||||
const lineCount = editor.document.lineCount;
|
|
||||||
|
|
||||||
// let selection = editor.selection;
|
|
||||||
// let lineText = editor.document.lineAt(selection.active);
|
|
||||||
// vscode.window.showInformationMessage( lineText.text );
|
|
||||||
}
|
|
||||||
|
|
||||||
}));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
function GetStruct(): StructInfo | undefined {
|
||||||
function getStructInfo(editor: vscode.TextEditor, name: string, startline: number, endline: number) {
|
let editor = vscode.window.activeTextEditor;
|
||||||
let regex = "([^ ]+) +([^ \\(\\{]+)";
|
if (editor !== undefined) {
|
||||||
|
if (editor.document.languageId !== 'go') {
|
||||||
for(let i = startline; i <= endline; i ++) {
|
vscode.window.showInformationMessage('file in the active editor is not a go file(*.go)');
|
||||||
let textline = editor.document.lineAt(i);
|
return;
|
||||||
switch(textline.text[-1]) {
|
}
|
||||||
case '{':
|
let selection = editor.selection;
|
||||||
break;
|
// vscode.window.showInformationMessage( editor.document.version.toString() );
|
||||||
case '}':
|
// vscode.window.showInformationMessage( text );
|
||||||
break;
|
let selectline = selection.active.line;
|
||||||
case '(':
|
let regex = "type +([^ ]+) +struct";
|
||||||
break;
|
// lineText.text.match()
|
||||||
case '}':
|
for (let i = selectline; i >= 0; i--) {
|
||||||
break;
|
let lineText = editor.document.lineAt(i);
|
||||||
}
|
let matchs = lineText.text.match(regex);
|
||||||
}
|
if (matchs !== null) {
|
||||||
|
let open = 0;
|
||||||
|
BREAK_OPEN: for (let n = i; n < editor.document.lineCount; n++) {
|
||||||
|
let lineText = editor.document.lineAt(n);
|
||||||
|
for (let c = 0; c < lineText.text.length; c++) {
|
||||||
|
switch (lineText.text[c]) {
|
||||||
|
case '{':
|
||||||
|
open++;
|
||||||
|
break;
|
||||||
|
case '}':
|
||||||
|
open--;
|
||||||
|
if (open === 0) {
|
||||||
|
if (n >= selectline) {
|
||||||
|
let structName = matchs[1];
|
||||||
|
return new StructInfo(structName, getStructField(editor, "", i, n), [i, n]);
|
||||||
|
}
|
||||||
|
break BREAK_OPEN;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// type A struct {
|
exports.activate = activate;
|
||||||
// do int
|
function getStructField(editor: vscode.TextEditor, parent: string, startline: number, endline: number): Field[] {
|
||||||
// }
|
|
||||||
//
|
let result: Field[] = [];
|
||||||
|
|
||||||
|
if (endline - startline <= 1) {
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
parent += ".";
|
||||||
|
|
||||||
|
let regex = "([^ \t]+)[ \t]+([^ \\(\\{\t]+)";
|
||||||
|
for (let i = startline + 1; i < endline; i++) {
|
||||||
|
let textline = editor.document.lineAt(i);
|
||||||
|
let matchArray = textline.text.match(regex);
|
||||||
|
|
||||||
|
if (matchArray !== null) {
|
||||||
|
var end: number;
|
||||||
|
let fieldName = matchArray[matchArray.length - 2];
|
||||||
|
let fieldType = matchArray[matchArray.length - 1].trim();
|
||||||
|
|
||||||
|
switch (fieldType) {
|
||||||
|
case 'struct':
|
||||||
|
end = getFieldRange(editor, ['{', '}'], i, endline);
|
||||||
|
if(i === end ) {
|
||||||
|
// let matches = textline.text.match("struct {0,}\\{[^ \t]+\\}");
|
||||||
|
function getSingleStructRelationship(source: string, parent: string): Field | undefined {
|
||||||
|
let smatch = source.match("([^ \t]+)[^s]+struct {0,}\\{(.+)\\}");
|
||||||
|
if(smatch !== null) {
|
||||||
|
console.log(smatch[0],smatch[1],smatch[2]);
|
||||||
|
return getSingleStructRelationship(smatch[2], parent + "." + smatch[1]);
|
||||||
|
} else {
|
||||||
|
smatch = source.match("([^ \t]+)[ \t]+(.+)");
|
||||||
|
if(smatch !== null) {
|
||||||
|
return new Field(parent+".", smatch[2].trim(), smatch[1], [i, end]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let v = getSingleStructRelationship(textline.text, "");
|
||||||
|
if(v !== undefined) {
|
||||||
|
result.push(v);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
result = result.concat(getStructField(editor, parent + fieldName, i, end));
|
||||||
|
i = end;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case 'interface':
|
||||||
|
result.push(new Field( parent, fieldType + "{}", fieldName, [i,i]));
|
||||||
|
break;
|
||||||
|
case 'func':
|
||||||
|
end = getFieldRange(editor, ['(', ')'], i, endline);
|
||||||
|
if(i === end) {
|
||||||
|
let matches = textline.text.match("func\\(.+");
|
||||||
|
if(matches !== null) {
|
||||||
|
result.push(new Field( parent, matches[0].trim(), fieldName, [i, end]));
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
i = end;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
result.push(new Field( parent, fieldType, fieldName, [i,i]));
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
function getFieldRange(editor: vscode.TextEditor, pair: string[], startline: number, endline: number): number {
|
||||||
|
|
||||||
|
let open = 0;
|
||||||
|
let start = startline;
|
||||||
|
let end = startline;
|
||||||
|
|
||||||
|
BREAK_OPEN: for (let n = start; n < endline; n++) {
|
||||||
|
let lineText = editor.document.lineAt(n);
|
||||||
|
for (let c = 0; c < lineText.text.length; c++) {
|
||||||
|
switch (lineText.text[c]) {
|
||||||
|
case pair[0]:
|
||||||
|
open++;
|
||||||
|
break;
|
||||||
|
case pair[1]:
|
||||||
|
open--;
|
||||||
|
if (open === 0) {
|
||||||
|
end = n;
|
||||||
|
break BREAK_OPEN;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return end;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
// export function getStruct(editor: vscode.TextEditor) {
|
// export function getStruct(editor: vscode.TextEditor) {
|
||||||
|
|
||||||
// }
|
// }
|
||||||
|
|
||||||
// this method is called when your extension is deactivated
|
// this method is called when your extension is deactivated
|
||||||
export function deactivate() {}
|
function deactivate() { }
|
||||||
|
exports.deactivate = deactivate;
|
||||||
|
|
|
||||||
|
|
@ -6,16 +6,43 @@ type MyStruct struct {
|
||||||
Key string
|
Key string
|
||||||
Do func(
|
Do func(
|
||||||
a int,
|
a int,
|
||||||
b struct{ A, B int })
|
b struct {
|
||||||
|
A, B int
|
||||||
|
C string
|
||||||
|
})
|
||||||
S struct{ A int }
|
S struct{ A int }
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type MyStruct2 struct {
|
||||||
|
Value int
|
||||||
|
Key string
|
||||||
|
Do func(
|
||||||
|
a int,
|
||||||
|
b struct {
|
||||||
|
A, B int
|
||||||
|
C string
|
||||||
|
})
|
||||||
|
S struct{ A struct{ C interface{} } }
|
||||||
|
}
|
||||||
|
|
||||||
|
// DoFunc 非常丑陋
|
||||||
|
func DoFunc(a func(
|
||||||
|
a int,
|
||||||
|
b struct {
|
||||||
|
A, B int
|
||||||
|
C string
|
||||||
|
})) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
// ExStruct 升级
|
// ExStruct 升级
|
||||||
type ExStruct struct {
|
type ExStruct struct {
|
||||||
ChildStruct struct {
|
ChildStruct struct {
|
||||||
A int
|
A int
|
||||||
B interface{}
|
B interface{}
|
||||||
}
|
}
|
||||||
|
C string
|
||||||
|
D func(a int, b string)
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetChildStructA set
|
// SetChildStructA set
|
||||||
|
|
@ -25,5 +52,7 @@ func (e *ExStruct) SetChildStructA(a int) {
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
a := ExStruct{}
|
a := ExStruct{}
|
||||||
println(a)
|
b := MyStruct{}
|
||||||
|
c := MyStruct2{}
|
||||||
|
println(a, b, c)
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user