7fdfab0ba90789194773f2bbd31bcc6ab161631a galt Tue Aug 5 10:57:28 2014 -0700 Fixes #12559. GenomeSpace support added to hgTables. diff --git src/lib/textOut.c src/lib/textOut.c index 5e74751..be1c500 100644 --- src/lib/textOut.c +++ src/lib/textOut.c @@ -18,31 +18,31 @@ if (format != NULL) { fflush(stdout); fprintf(stdout, "%s", hLine); vfprintf(stdout, format, args); fprintf(stdout, "\n"); fprintf(stdout, "%s", hLine); } } static void textOutAbortHandler() /* Text mode abort handler. */ { exit(-1); } -static char *getCompressSuffix(char *compressType) +char *getCompressSuffix(char *compressType) /* Return the file dot-suffix (including the dot) for compressType. */ { static char *gzipSuffix = ".gz"; static char *compressSuffix = ".Z"; static char *bz2Suffix = ".bz2"; static char *zipSuffix = ".zip"; if (sameWord(compressType, textOutCompressGzip)) return gzipSuffix; else if (sameWord(compressType, textOutCompressCompress)) return compressSuffix; else if (sameWord(compressType, textOutCompressBzip2)) return bz2Suffix; else if (sameWord(compressType, textOutCompressZip)) return zipSuffix; else @@ -91,76 +91,107 @@ unsetenv("BZIP"); unsetenv("BZIP2"); } else if (sameWord(compressType, textOutCompressZip)) { unsetenv("ZIPOPT"); } else { errAbort("cleanEnvVars: Unsupported textOutCompress type %s", compressType); } } -struct pipeline *textOutInit(char *fileName, char *compressType) +struct pipeline *textOutInit(char *fileName, char *compressType, int *saveStdout) /* Set up stdout to be HTTP text, file (if fileName is specified), or * compressed file (if both fileName and compressType are specified -- * see textOut.h for supported compression types). * Return NULL if no compression, otherwise a pipeline handle on which * textOutClose should be called when we're done writing stdout. */ { struct pipeline *compressPipeline = NULL; +// if path contains a slash, we are outputting to a local file +boolean outToFile = (strchr(fileName, '/') != NULL); +if (outToFile) + { + FILE *f; + f = fopen(fileName, "w"); + /* We want to capture stdout output to a file */ + fflush(stdout); + int tempOut = dup(STDOUT_FILENO); + if (saveStdout) + *saveStdout = tempOut; + dup2(fileno(f),STDOUT_FILENO); /* closes STDOUT before setting it again */ + fclose(f); + } + trimSpaces(fileName); if (isEmpty(fileName)) { printf("Content-Type: text/plain\n\n"); } else if (isEmpty(compressType) || sameWord(compressType, textOutCompressNone)) { + if (!outToFile) + { printf("Content-Type: application/octet-stream\n"); printf("Content-Disposition: attachment; filename=%s\n\n", fileName); } + } else { - char *suffix = getCompressSuffix(compressType); + if (!outToFile) + { + char *suffix = getCompressSuffix(compressType); printf("Content-Type: application/x-%s\n", compressType); if (endsWith(fileName, suffix)) printf("Content-Disposition: attachment; filename=%s\n\n", fileName); else printf("Content-Disposition: attachment; filename=%s%s\n\n", fileName, suffix); - /* Send the Content header uncompressed! */ fflush(stdout); + } /* Make sure no environment variables interfere with compressor. */ cleanEnvVars(compressType); /* Redirect stdout to compressor pipeline object. */ compressPipeline = pipelineOpen1(getCompressor(compressType), pipelineWrite, NULL, NULL); if (-1 == dup2(pipelineFd(compressPipeline), STDOUT_FILENO)) errnoAbort("dup2(pipelineFd %d, stdout %d) failed in textOpen()", pipelineFd(compressPipeline), STDOUT_FILENO); } pushWarnHandler(textOutWarnHandler); pushAbortHandler(textOutAbortHandler); return(compressPipeline); } -void textOutClose(struct pipeline **pCompressPipeline) +void textOutClose(struct pipeline **pCompressPipeline, int *saveStdout) /* Flush and close stdout, wait for the pipeline to finish, and then free * the pipeline object. */ { if (pCompressPipeline && *pCompressPipeline) { fflush(stdout); - fclose(stdout); + close(STDOUT_FILENO); // Do not use fclose pipelineWait(*pCompressPipeline); pipelineFree(pCompressPipeline); } +if (saveStdout) + { + if (*saveStdout != -1) + { + /* restore stdout */ + fflush(stdout); + dup2(*saveStdout,STDOUT_FILENO); /* closes STDOUT before setting it back to saved descriptor */ + close(*saveStdout); + *saveStdout = -1; + } + } }