1 files changed, 4 insertions, 17 deletions
diff --git a/main.go b/main.go
index 7903111..8e8aa09 100644
--- a/main.go
+++ b/main.go
@@ -159,27 +159,14 @@ func main() {
 	pb.addMutex(f)
 
 	// if foo.pb.go still doesn't exist, protoc failed
-	// exit here
-	if !shell.Exists(sortmap["protoc"]) {
-		log.Info("protoc build error.", sortmap["protoc"], "failed to be created with protoc and proto-gen-go")
+	if !shell.Exists(pbfile) {
+		log.Info("protoc build error.", pbfile)
 		badExit(errors.New("failed to be created with protoc and proto-gen-go"))
 	}
 
-	if argv.NoSort {
-		// log.Info("not making sort.pb.go file (--no-sort == true)")
-	} else {
-		if len(uniqueKeys) != 0 {
-		}
-		makeSortfile()
-	}
-
-	if argv.NoMarshal {
-		// log.Info("not making marshal.pb.go file (--no-marshal == true)")
-	} else {
-		// make the foo.marshal.pb.go file
-		marshal(sortmap)
-	}
+	pb.marshal(f)
 
+	// pb.makeSortfile(f)
 }
 
 func okExit(s string) {