src/pagespeed/automatic/static_rewriter_main.cc

/*
 * Copyright 2011 Google Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

// Author: jmarantz@google.com (Joshua Marantz)

#include <cstdio>

#include "net/instaweb/rewriter/public/process_context.h"
#include "net/instaweb/rewriter/public/rewrite_driver_factory.h"
#include "pagespeed/automatic/static_rewriter.h"
#include "pagespeed/kernel/base/file_system.h"
#include "pagespeed/kernel/base/string.h"
#include "pagespeed/kernel/base/string_util.h"
#include "pagespeed/kernel/base/string_writer.h"

namespace net_instaweb {

class MessageHandler;

}  // namespace net_instaweb

// The purpose of this program is to help us test that pagespeed_automatic.a
// contains all that's needed to successfully link a rewriter using standard
// g++, without using the gyp flow.
int main(int argc, char** argv) {
  net_instaweb::ProcessContext process_context;
  net_instaweb::RewriteDriverFactory::Initialize();
  net_instaweb::StaticRewriter static_rewriter(process_context, &argc, &argv);

  // Having stripped all the flags, there should be exactly 3
  // arguments remaining:
  //
  //   input_directory:   The directory where the origin web site is stored
  //   output_directory:  The directory where the rewritten web site is written
  //   URL:               The URL of HTML to rewrite.
  if (argc != 4) {
    fprintf(stderr, "Usage: [options] %s input_dir output_dir url.\n", argv[0]);
    fprintf(stderr, "Type '%s --help' to see the options\n", argv[0]);
    return 1;
  }
  const char* input_dir = argv[1];
  const char* output_dir = argv[2];
  const char* html_name = argv[3];

  GoogleString url = net_instaweb::StrCat("http://test.com/", html_name);
  GoogleString input_file_path = net_instaweb::StrCat(input_dir, "/",
                                                      html_name);
  GoogleString output_file_path = net_instaweb::StrCat(output_dir, "/",
                                                       html_name);
  GoogleString html_input_buffer, html_output_buffer;
  net_instaweb::FileSystem* file_system = static_rewriter.file_system();
  net_instaweb::MessageHandler* message_handler =
      static_rewriter.message_handler();
  net_instaweb::StringWriter writer(&html_output_buffer);

  int exit_status = 1;
  if (!file_system->ReadFile(input_file_path.c_str(), &html_input_buffer,
                             message_handler)) {
    fprintf(stderr, "failed to read file %s\n", input_file_path.c_str());
  } else if (!static_rewriter.ParseText(url, input_file_path, html_input_buffer,
                                        output_dir, &writer)) {
    fprintf(stderr, "StartParseId failed on url %s\n", url.c_str());
  } else if (!file_system->WriteFileAtomic(
      output_file_path, html_output_buffer, message_handler)) {
    fprintf(stderr, "failed to write file %s\n", output_file_path.c_str());
  } else {
    exit_status = 0;
  }

  // TODO(jmarantz): set up a file-based fetcher that will allow us to
  // rewrite resources in HTML files in this demonstration.

  net_instaweb::RewriteDriverFactory::Terminate();
  return 0;
}